sovaria-cli 0.2.3__tar.gz → 0.2.5__tar.gz

{sovaria_cli-0.2.3 → sovaria_cli-0.2.5}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.5
 Name: sovaria-cli
-Version: 0.2.3
+Version: 0.2.5
 Summary: Sovaria BaaS command-line interface (sov)
 Project-URL: Homepage, https://sovaria.in
 Project-URL: Documentation, https://docs.sovaria.in/cli

{sovaria_cli-0.2.3 → sovaria_cli-0.2.5}/apps/cli/src/__init__.py

@@ -1,3 +1,3 @@
 """Sovaria CLI — sov command."""
 
-__version__ = "0.2.3"
+__version__ = "0.2.4"

{sovaria_cli-0.2.3 → sovaria_cli-0.2.5}/apps/cli/src/commands/db.py

@@ -18,6 +18,143 @@ from apps.cli.src.utils.display import print_json, print_sql_result
 
 app = typer.Typer(help="Database (SQL) management")
 
+_INTERNAL_SQL_TABLES = ("sov_documents",)
+_LIST_TABLES_SQL = (
+    "SELECT table_name FROM information_schema.tables "
+    "WHERE table_schema = current_schema() "
+    "AND table_type = 'BASE TABLE' "
+    f"AND table_name NOT IN ({', '.join(repr(name) for name in _INTERNAL_SQL_TABLES)}) "
+    "ORDER BY table_name;"
+)
+
+
+def _begin_tx(
+    *,
+    profile: str | None,
+    org_id: str | None,
+    project_id: str | None,
+    env_id: str | None,
+) -> int:
+    result = call_api(
+        method="POST",
+        path="/api/v1/data/rel/transaction/begin",
+        profile=profile,
+        org_id=org_id,
+        project_id=project_id,
+        env_id=env_id,
+        require_scope=True,
+    )
+    txid = result.get("txid") if isinstance(result, dict) else None
+    if not isinstance(txid, int) or txid < 1:
+        raise CLIRequestError(f"Invalid txid from transaction begin: {result!r}")
+    return txid
+
+
+def _commit_tx(
+    txid: int,
+    *,
+    profile: str | None,
+    org_id: str | None,
+    project_id: str | None,
+    env_id: str | None,
+) -> None:
+    call_api(
+        method="POST",
+        path="/api/v1/data/rel/transaction/commit",
+        profile=profile,
+        body={"txid": txid},
+        org_id=org_id,
+        project_id=project_id,
+        env_id=env_id,
+        require_scope=True,
+    )
+
+
+def _rollback_tx(
+    txid: int,
+    *,
+    profile: str | None,
+    org_id: str | None,
+    project_id: str | None,
+    env_id: str | None,
+) -> None:
+    try:
+        call_api(
+            method="POST",
+            path="/api/v1/data/rel/transaction/rollback",
+            profile=profile,
+            body={"txid": txid},
+            org_id=org_id,
+            project_id=project_id,
+            env_id=env_id,
+            require_scope=True,
+        )
+    except CLIRequestError:
+        pass
+
+
+def _run_sql_query(
+    sql: str,
+    *,
+    profile: str | None,
+    org_id: str | None,
+    project_id: str | None,
+    env_id: str | None,
+) -> dict:
+    scope = {
+        "profile": profile,
+        "org_id": org_id,
+        "project_id": project_id,
+        "env_id": env_id,
+    }
+    txid = _begin_tx(**scope)
+    try:
+        result = call_api(
+            method="POST",
+            path="/api/v1/sql/query",
+            profile=profile,
+            body={"sql": sql, "txid": txid},
+            org_id=org_id,
+            project_id=project_id,
+            env_id=env_id,
+            require_scope=True,
+        )
+        _commit_tx(txid, **scope)
+        return result if isinstance(result, dict) else {"rows": result}
+    except Exception:
+        _rollback_tx(txid, **scope)
+        raise
+
+
+def _fetch_live_tables(
+    *,
+    profile: str | None,
+    org_id: str | None,
+    project_id: str | None,
+    env_id: str | None,
+) -> list[str]:
+    result = _run_sql_query(
+        _LIST_TABLES_SQL,
+        profile=profile,
+        org_id=org_id,
+        project_id=project_id,
+        env_id=env_id,
+    )
+    rows = result.get("rows") if isinstance(result, dict) else None
+    if not isinstance(rows, list):
+        return []
+    tables: list[str] = []
+    for row in rows:
+        if isinstance(row, list) and row:
+            name = str(row[0]).strip()
+        elif isinstance(row, dict):
+            name = str(row.get("table_name") or row.get("TABLE_NAME") or "").strip()
+        else:
+            continue
+        if name and name not in _INTERNAL_SQL_TABLES:
+            tables.append(name)
+    return tables
+
 
 @app.command("tables")
 def list_tables(
@@ -25,22 +162,37 @@ def list_tables(
     project_id: str | None = typer.Option(None, "--project", help="Project ID"),
     env_id: str | None = typer.Option(None, "--env"),
     profile: str | None = typer.Option(None, "--profile"),
+    migrations: bool = typer.Option(
+        False,
+        "--migrations",
+        help="Show migration log only (legacy behaviour)",
+    ),
 ) -> None:
-    """List relational tables / migration log for the linked project."""
+    """List live relational tables in the linked project database."""
     try:
-        result = call_api(
-            method="GET",
-            path="/api/v1/data/rel/migrations",
+        if migrations:
+            result = call_api(
+                method="GET",
+                path="/api/v1/data/rel/migrations",
+                profile=profile,
+                org_id=org_id,
+                project_id=project_id,
+                env_id=env_id,
+                require_scope=True,
+            )
+            print_json(result)
+            return
+
+        tables = _fetch_live_tables(
             profile=profile,
             org_id=org_id,
             project_id=project_id,
             env_id=env_id,
-            require_scope=True,
         )
+        print_json({"tables": tables, "total": len(tables), "source": "information_schema"})
     except CLIRequestError as exc:
         typer.echo(str(exc), err=True)
         raise typer.Exit(1) from exc
-    print_json(result)
 
 
 @app.command("sql")
@@ -97,15 +249,12 @@ def _execute_sql(
     env_id: str | None,
 ) -> None:
     try:
-        result = call_api(
-            method="POST",
-            path="/api/v1/sql/query",
+        result = _run_sql_query(
+            sql,
             profile=profile,
-            body={"sql": sql, "txid": 0},
             org_id=org_id,
             project_id=project_id,
             env_id=env_id,
-            require_scope=True,
         )
     except CLIRequestError as exc:
         typer.echo(str(exc), err=True)
@@ -473,11 +622,8 @@ def migrations_push(
             continue
         typer.echo(f"Applying {path.name} …")
         try:
-            call_api(
-                method="POST",
-                path="/api/v1/sql/query",
-                body={"sql": sql, "txid": 0},
-                require_scope=True,
+            _run_sql_query(
+                sql,
                 **_scope_opts(profile, org_id, project_id, env_id),
             )
         except CLIRequestError as exc:

{sovaria_cli-0.2.3 → sovaria_cli-0.2.5}/apps/cli/tests/test_main_startup.py

@@ -26,4 +26,4 @@ def test_sov_console_script_help() -> None:
     )
 
     assert result.returncode == 0
-    assert "sov version 0.2.3" in result.stdout
+    assert "sov version 0.2.4" in result.stdout

{sovaria_cli-0.2.3 → sovaria_cli-0.2.5}/apps/cli/tests/test_project_and_commands.py

@@ -56,6 +56,36 @@ def test_resolve_scope_requires_linked_project(monkeypatch: pytest.MonkeyPatch)
         )
 
 
+def test_db_tables_uses_information_schema(monkeypatch: pytest.MonkeyPatch) -> None:
+    from apps.cli.src.commands import db
+
+    captured: dict[str, Any] = {}
+    calls: list[tuple[str, str]] = []
+
+    def _fake_call_api(**kwargs: Any) -> dict[str, Any]:
+        method = str(kwargs.get("method", ""))
+        path = str(kwargs.get("path", ""))
+        calls.append((method, path))
+        if path.endswith("/transaction/begin"):
+            return {"txid": 1, "snapshot_txid": 1}
+        if path.endswith("/sql/query"):
+            captured.update(kwargs)
+            return {"rows": [["users"], ["orders"]]}
+        if path.endswith("/transaction/commit"):
+            return {"ok": True}
+        return {}
+
+    monkeypatch.setattr(db, "call_api", _fake_call_api)
+    db.list_tables(profile=None, org_id=None, project_id=None, env_id=None, migrations=False)
+
+    assert captured["method"] == "POST"
+    assert captured["path"] == "/api/v1/sql/query"
+    assert "information_schema.tables" in captured["body"]["sql"]
+    assert captured["body"]["txid"] == 1
+    assert ("POST", "/api/v1/data/rel/transaction/begin") in calls
+    assert ("POST", "/api/v1/data/rel/transaction/commit") in calls
+
+
 def test_sim_discover_path_smoke(monkeypatch: pytest.MonkeyPatch) -> None:
     from apps.cli.src.commands import sim
 
@@ -194,3 +224,34 @@ def test_storage_buckets_uses_call_api(monkeypatch: pytest.MonkeyPatch) -> None:
     assert captured["path"] == "/api/v1/store/buckets"
     assert captured["require_scope"] is True
     assert any("bucket-a" in line for line in output)
+
+
+def test_fetch_live_tables_uses_transaction(monkeypatch: pytest.MonkeyPatch) -> None:
+    from apps.cli.src.commands import db
+
+    calls: list[tuple[str, str]] = []
+
+    def _fake_call_api(**kwargs: Any) -> dict[str, Any]:
+        method = str(kwargs.get("method", ""))
+        path = str(kwargs.get("path", ""))
+        calls.append((method, path))
+        if path.endswith("/transaction/begin"):
+            return {"txid": 7, "snapshot_txid": 7}
+        if path.endswith("/sql/query"):
+            return {"rows": [["users"], ["posts"]]}
+        if path.endswith("/transaction/commit"):
+            return {"ok": True}
+        return {}
+
+    monkeypatch.setattr(db, "call_api", _fake_call_api)
+    tables = db._fetch_live_tables(
+        profile=None,
+        org_id="org-1",
+        project_id="proj-1",
+        env_id=None,
+    )
+
+    assert tables == ["users", "posts"]
+    assert ("POST", "/api/v1/data/rel/transaction/begin") in calls
+    assert ("POST", "/api/v1/sql/query") in calls
+    assert ("POST", "/api/v1/data/rel/transaction/commit") in calls

{sovaria_cli-0.2.3 → sovaria_cli-0.2.5}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "sovaria-cli"
-version = "0.2.3"
+version = "0.2.5"
 description = "Sovaria BaaS command-line interface (sov)"
 readme = "apps/cli/README.md"
 requires-python = ">=3.11"