Add url option to login and logout

Author: abbiemery

src/blueapi/cli/cli.py

@@ -180,13 +180,24 @@ def start_application(ctx: click.Context, config: Path | None | tuple[Path, ...]
 
 
 @main.command(name="login")
+@click.option(
+    "--url",
+    type=str,
+    help="URL for the blueapi server to connect to.",
+    default=None,
+)
 @click.pass_obj
 @check_connection
-def login(obj: dict) -> None:
+def login(
+    obj: dict,
+    url: str | None,
+) -> None:
     """
     Authenticate with the blueapi using the OIDC (OpenID Connect) flow.
     """
-    config: ApplicationConfig = obj["config"]
+    config: ApplicationConfig = ConfigLoader(ApplicationConfig).load()
+    if url is not None:
+        config.api.url = HttpUrl(url)
     try:
         auth: SessionManager = SessionManager.from_cache(config.auth_token_path)
         access_token = auth.get_valid_access_token()
@@ -205,12 +216,23 @@ def login(obj: dict) -> None:
 
 
 @main.command(name="logout")
+@click.option(
+    "--url",
+    type=str,
+    help="URL for the blueapi server to connect to.",
+    default=None,
+)
 @click.pass_obj
-def logout(obj: dict) -> None:
+def logout(
+    obj: dict,
+    url: str | None,
+) -> None:
     """
     Logs out from the OIDC provider and removes the cached access token.
     """
-    config: ApplicationConfig = obj["config"]
+    config: ApplicationConfig = ConfigLoader(ApplicationConfig).load()
+    if url is not None:
+        config.api.url = HttpUrl(url)
     try:
         auth: SessionManager = SessionManager.from_cache(config.auth_token_path)
         auth.logout()