fix: allow the api key to be re-inserted after logout clear to credentials.ini file

Author: BartoszBlizniak

cloudsmith_cli/cli/config.py

@@ -222,19 +222,30 @@ def find_existing_files(cls):
         return paths
 
     @classmethod
-    def clear_api_key(cls, path):
-        """Clear api_key values in a credentials file, preserving structure."""
+    def _set_api_key(cls, path, api_key=""):
+        """Write api_key value in a credentials file, preserving structure."""
         with open(path) as f:
             content = f.read()
+        replacement = rf"\1 = {api_key}" if api_key else r"\1 ="
         content = re.sub(
             r"^(api_key)\s*=\s*.*$",
-            r"\1 =",
+            replacement,
             content,
             flags=re.MULTILINE,
         )
         with open(path, "w") as f:
             f.write(content)
 
+    @classmethod
+    def clear_api_key(cls, path):
+        """Clear api_key values in a credentials file, preserving structure."""
+        cls._set_api_key(path)
+
+    @classmethod
+    def update_api_key(cls, path, api_key):
+        """Update api_key value in an existing credentials file, preserving structure."""
+        cls._set_api_key(path, api_key)
+
 
 class Options:
     """Options object that holds config for the application."""

cloudsmith_cli/core/config.py

@@ -83,6 +83,28 @@ def create_config_files(ctx, opts, api_key, force=False):
                 )
             continue
 
+        # Update existing credentials file with new API key if provided
+        if (
+            config.present
+            and config.data.get("api_key")
+            and hasattr(config.reader, "update_api_key")
+        ):
+            try:
+                config.reader.update_api_key(
+                    config.reader.get_default_filepath(),
+                    config.data["api_key"],
+                )
+                click.secho("UPDATED", fg="green")
+            except OSError as exc:
+                has_errors = True
+                click.secho("ERROR", fg="red")
+                click.secho(
+                    "The following error occurred while trying to "
+                    "update the file: %(message)s"
+                    % {"message": click.style(exc.strerror, fg="red")}
+                )
+            continue
+
         click.secho("EXISTS" if config.present else "NOT CREATED", fg="yellow")
 
     return create, has_errors