feat: add create-invite CLI command
This commit is contained in:
parent
0c3157ea3a
commit
bcddf5d1c8
2 changed files with 94 additions and 0 deletions
41
src/porchlight/cli.py
Normal file
41
src/porchlight/cli.py
Normal file
|
|
@ -0,0 +1,41 @@
|
||||||
|
import asyncio
|
||||||
|
from pathlib import Path
|
||||||
|
from typing import Annotated, Optional
|
||||||
|
|
||||||
|
import typer
|
||||||
|
|
||||||
|
from porchlight.config import Settings
|
||||||
|
from porchlight.invite.service import MagicLinkService
|
||||||
|
from porchlight.store.sqlite.db import open_db
|
||||||
|
from porchlight.store.sqlite.repositories import SQLiteMagicLinkRepository
|
||||||
|
|
||||||
|
PACKAGE_DIR = Path(__file__).resolve().parent
|
||||||
|
MIGRATIONS_DIR = PACKAGE_DIR / "store" / "sqlite" / "migrations"
|
||||||
|
|
||||||
|
app = typer.Typer()
|
||||||
|
|
||||||
|
|
||||||
|
async def _create_invite(settings: Settings, username: str, ttl: int, note: str | None) -> str:
|
||||||
|
"""Create an invite link and return the full registration URL."""
|
||||||
|
async with open_db(settings.sqlite_path, MIGRATIONS_DIR) as db:
|
||||||
|
repo = SQLiteMagicLinkRepository(db)
|
||||||
|
service = MagicLinkService(repo, ttl=ttl)
|
||||||
|
link = await service.create(username=username, created_by="cli", note=note)
|
||||||
|
return f"{settings.issuer}/register/{link.token}"
|
||||||
|
|
||||||
|
|
||||||
|
@app.command()
|
||||||
|
def create_invite(
|
||||||
|
username: str,
|
||||||
|
ttl: Annotated[Optional[int], typer.Option(help="Link expiration in seconds")] = None,
|
||||||
|
note: Annotated[Optional[str], typer.Option(help="Optional note stored with the link")] = None,
|
||||||
|
) -> None:
|
||||||
|
"""Generate a magic link registration URL for a new user."""
|
||||||
|
settings = Settings()
|
||||||
|
effective_ttl = ttl if ttl is not None else settings.invite_ttl
|
||||||
|
url = asyncio.run(_create_invite(settings, username, effective_ttl, note))
|
||||||
|
typer.echo(url)
|
||||||
|
|
||||||
|
|
||||||
|
def main() -> None:
|
||||||
|
app()
|
||||||
53
tests/test_cli.py
Normal file
53
tests/test_cli.py
Normal file
|
|
@ -0,0 +1,53 @@
|
||||||
|
import os
|
||||||
|
import tempfile
|
||||||
|
|
||||||
|
from typer.testing import CliRunner
|
||||||
|
|
||||||
|
from porchlight.cli import app
|
||||||
|
|
||||||
|
runner = CliRunner()
|
||||||
|
|
||||||
|
|
||||||
|
def test_create_invite_prints_registration_url() -> None:
|
||||||
|
"""create-invite should print a URL containing /register/."""
|
||||||
|
with tempfile.TemporaryDirectory() as tmpdir:
|
||||||
|
db_path = os.path.join(tmpdir, "test.db")
|
||||||
|
result = runner.invoke(
|
||||||
|
app,
|
||||||
|
["testuser"],
|
||||||
|
env={"OIDC_OP_ISSUER": "https://example.com", "OIDC_OP_SQLITE_PATH": db_path},
|
||||||
|
)
|
||||||
|
assert result.exit_code == 0, result.output
|
||||||
|
assert "https://example.com/register/" in result.output
|
||||||
|
|
||||||
|
|
||||||
|
def test_create_invite_with_note() -> None:
|
||||||
|
"""create-invite with --note should work."""
|
||||||
|
with tempfile.TemporaryDirectory() as tmpdir:
|
||||||
|
db_path = os.path.join(tmpdir, "test.db")
|
||||||
|
result = runner.invoke(
|
||||||
|
app,
|
||||||
|
["testuser", "--note", "Welcome aboard"],
|
||||||
|
env={"OIDC_OP_ISSUER": "https://example.com", "OIDC_OP_SQLITE_PATH": db_path},
|
||||||
|
)
|
||||||
|
assert result.exit_code == 0, result.output
|
||||||
|
assert "https://example.com/register/" in result.output
|
||||||
|
|
||||||
|
|
||||||
|
def test_create_invite_with_custom_ttl() -> None:
|
||||||
|
"""create-invite with --ttl should use the custom TTL."""
|
||||||
|
with tempfile.TemporaryDirectory() as tmpdir:
|
||||||
|
db_path = os.path.join(tmpdir, "test.db")
|
||||||
|
result = runner.invoke(
|
||||||
|
app,
|
||||||
|
["testuser", "--ttl", "3600"],
|
||||||
|
env={"OIDC_OP_ISSUER": "https://example.com", "OIDC_OP_SQLITE_PATH": db_path},
|
||||||
|
)
|
||||||
|
assert result.exit_code == 0, result.output
|
||||||
|
assert "https://example.com/register/" in result.output
|
||||||
|
|
||||||
|
|
||||||
|
def test_create_invite_missing_username_shows_error() -> None:
|
||||||
|
"""create-invite without a username should show an error."""
|
||||||
|
result = runner.invoke(app, [])
|
||||||
|
assert result.exit_code != 0
|
||||||
Loading…
Add table
Add a link
Reference in a new issue