Merge pull request fastapi#11 from fastapilabs/08-01-_add_whoami_command

✨ Add whoami command

Author: patrick91

src/fastapi_cli/cli.py

@@ -13,6 +13,7 @@
 
 from . import __version__
 from .commands.login import login
+from .commands.whoami import whoami
 from .config import settings
 from .logging import setup_logging
 
@@ -281,6 +282,7 @@ def run(
 
 # Additional commands
 app.command()(login)
+app.command()(whoami)
 
 
 def main() -> None:

src/fastapi_cli/commands/whoami.py

@@ -0,0 +1,46 @@
+import logging
+from typing import Any
+
+import httpx
+import typer
+from rich import print
+
+from fastapi_cli.utils.api import APIClient
+from fastapi_cli.utils.auth import is_logged_in
+
+logger = logging.getLogger(__name__)
+
+
+def whoami() -> Any:
+    if not is_logged_in():
+        print("No credentials found. Use [blue]`fastapi login`[/] to login.")
+        return
+
+    with APIClient() as client:
+        try:
+            response = client.get("/users/me")
+        except httpx.ReadTimeout:
+            print(
+                "[red]Error[/]: The request to the FastAPI Cloud server timed out. Please try again later."
+            )
+            raise typer.Exit(code=1) from None
+
+        if response.status_code in (401, 403):
+            print(
+                "[red]Error[/]: The specified token is not valid. Use [blue]`fastapi login`[/] to generate a new token."
+            )
+            raise typer.Exit(code=1) from None
+
+        try:
+            response.raise_for_status()
+        except httpx.HTTPError as e:
+            logger.debug("Error: %s", e)
+
+            print(
+                "[red]Error[/]: Something went wrong while contacting the FastAPI Cloud server. Please try again later."
+            )
+            raise typer.Exit(code=1) from None
+
+        data = response.json()
+
+        print(f"⚡[bold]{data['email']}[/bold]")

src/fastapi_cli/utils/auth.py

@@ -31,3 +31,7 @@ def get_auth_token() -> Optional[str]:
         return None
 
     return auth_data.access_token
+
+
+def is_logged_in() -> bool:
+    return get_auth_token() is not None

tests/test_cli_whoami.py

@@ -0,0 +1,57 @@
+from pathlib import Path
+
+import pytest
+import respx
+from fastapi_cli.cli import app
+from fastapi_cli.config import settings
+from httpx import ReadTimeout, Response
+from typer.testing import CliRunner
+
+runner = CliRunner()
+
+assets_path = Path(__file__).parent / "assets"
+
+
+@pytest.mark.respx(base_url=settings.base_api_url)
+def test_shows_a_message_if_something_is_wrong(respx_mock: respx.MockRouter) -> None:
+    respx_mock.get("/users/me").mock(return_value=Response(500))
+
+    result = runner.invoke(app, ["whoami"])
+
+    assert result.exit_code == 1
+    assert (
+        "Something went wrong while contacting the FastAPI Cloud server."
+        in result.output
+    )
+
+
+@pytest.mark.respx(base_url=settings.base_api_url)
+def test_shows_a_message_when_not_logged_in(respx_mock: respx.MockRouter) -> None:
+    respx_mock.get("/users/me").mock(return_value=Response(401))
+
+    result = runner.invoke(app, ["whoami"])
+
+    assert result.exit_code == 1
+    assert "The specified token is not valid" in result.output
+
+
+@pytest.mark.respx(base_url=settings.base_api_url)
+def test_shows_email(respx_mock: respx.MockRouter) -> None:
+    respx_mock.get("/users/me").mock(
+        return_value=Response(200, json={"email": "[email protected]"})
+    )
+
+    result = runner.invoke(app, ["whoami"])
+
+    assert result.exit_code == 0
+    assert "[email protected]" in result.output
+
+
+@pytest.mark.respx(base_url=settings.base_api_url)
+def test_handles_read_timeout(respx_mock: respx.MockRouter) -> None:
+    respx_mock.get("/users/me").mock(side_effect=ReadTimeout)
+
+    result = runner.invoke(app, ["whoami"])
+
+    assert result.exit_code == 1
+    assert "The request to the FastAPI Cloud server timed out" in result.output