bevor-cli 0.1.0__py3-none-any.whl

bevor_cli/commands/nodes/get.py

@@ -0,0 +1,52 @@
+"""``bevor code node get`` — fetch a single graph node."""
+
+import json
+from typing import Optional
+
+import typer
+from bevor_sdk.types import ID
+from rich.table import Table
+
+from bevor_cli.annotations.confirmations import JSON_OUT
+from bevor_cli.client import BevorInternalClient
+from bevor_cli.configuration.creds import Credentials
+from bevor_cli.configuration.state import State
+from bevor_cli.output import console
+
+
+def get_node(
+    node_id: str = typer.Argument(..., help="Node short id (or full wire id)."),
+    code_id: Optional[str] = typer.Option(
+        None, "--code", help="Code id to list for (defaults to current configuration)"
+    ),
+    json_out: JSON_OUT = False,
+) -> None:
+    """Get the details of a node. Uses the currently configured code version, or you can pass --code."""
+    creds = Credentials.load()
+    creds.require_token()
+
+    state = State.load()
+    workspace_id = state.validate_workspace()
+    code_id = state.validate_code(priority=code_id)
+
+    client = BevorInternalClient(
+        creds=creds, workspace_id=workspace_id, project_id=state.validate_project()
+    )
+
+    node = client.codes.nodes.get(code_id, ID(node_id))
+
+    if json_out:
+        typer.echo(json.dumps(node, indent=2))
+        return
+
+    table = Table(show_header=False, box=None, padding=(0, 1))
+    table.add_column(style="bold")
+    table.add_column()
+
+    table.add_row("ID", str(node.id))
+    table.add_row("Short id", str(node.short_id))
+    table.add_row("Name", node.name)
+    table.add_row("Type", str(node.node_type))
+    table.add_row("File", f"{node.path} ({node.file_id})")
+
+    console.print(table)

bevor_cli/commands/nodes/list.py

@@ -0,0 +1,75 @@
+"""``bevor code node list`` — list graph nodes for a code version."""
+
+from typing import Optional
+
+import typer
+from bevor_sdk.types import ID
+from bevor_sdk.types.requests.graph import NodeQueryParams
+from rich.table import Table
+
+from bevor_cli import output
+from bevor_cli.annotations.confirmations import JSON_OUT
+from bevor_cli.client import BevorInternalClient
+from bevor_cli.configuration.creds import Credentials
+from bevor_cli.configuration.state import State
+from bevor_cli.output import console
+
+
+def list_nodes(
+    code_id: Optional[str] = typer.Option(
+        None, "--code", help="Code id to list for (defaults to current configuration)"
+    ),
+    entrypoints_only: bool = typer.Option(
+        False,
+        "--entrypoints",
+        help="Only show auditable entry-point functions.",
+    ),
+    file_id: Optional[str] = typer.Option(None, "--file", help="Filter to nodes in this file."),
+    no_deps: Optional[bool] = typer.Option(
+        None, "--no-deps", help="Filter to nodes under this directory prefix."
+    ),
+    name: Optional[str] = typer.Option(None, "--name", help="Filter to nodes by name"),
+    node_type: Optional[str] = typer.Option(None, "--node-type", help="Filter to nodes by type"),
+    json_out: JSON_OUT = False,
+) -> None:
+    """List all nodes. Uses the currently configured code version, or you can pass --code."""
+    creds = Credentials.load()
+    creds.require_token()
+
+    state = State.load()
+    workspace_id = state.validate_workspace()
+    code_id = state.validate_code(priority=code_id)
+
+    client = BevorInternalClient(
+        creds=creds, workspace_id=workspace_id, project_id=state.validate_project()
+    )
+
+    nodes = client.codes.nodes.list(
+        code_id,
+        NodeQueryParams(
+            file_id=ID(file_id) if file_id else None,
+            entrypoint_only=entrypoints_only,
+            no_deps=no_deps,
+            name=name,
+            node_type=node_type,
+        ),
+    )
+
+    if json_out:
+        console.print_json(nodes.model_dump_json(indent=2))
+        return
+
+    if not nodes.results:
+        output.error("No nodes found")
+        raise typer.Exit(1)
+
+    table = Table(title=f"Nodes — version {state.code_id}", show_header=True, header_style="bold")
+    table.add_column("Short id", style="dim", no_wrap=True)
+    table.add_column("Name")
+    table.add_column("Type", style="cyan")
+    table.add_column("File", style="dim")
+    table.add_column("EP", justify="center")
+
+    for n in nodes.results:
+        table.add_row(n.short_id, n.name, n.node_type, n.path, "⚡" if n.is_entrypoint else "")
+    console.print(table)

bevor_cli/commands/nodes/source.py

@@ -0,0 +1,40 @@
+"""``bevor code node content`` — show source for a graph node."""
+
+from typing import Optional
+
+import typer
+from bevor_sdk.types import ID
+from rich.syntax import Syntax
+
+from bevor_cli import output
+from bevor_cli.client import BevorInternalClient
+from bevor_cli.configuration.creds import Credentials
+from bevor_cli.configuration.state import State
+from bevor_cli.output import console
+
+
+def show_source(
+    node_id: str = typer.Argument(..., help="Node short id (or full wire id)."),
+    code_id: Optional[str] = typer.Option(
+        None, "--code", help="Code id to list for (defaults to current configuration)"
+    ),
+) -> None:
+    """Show the source code of a node. Uses the currently configured code version, or you can pass --code."""
+    creds = Credentials.load()
+    creds.require_token()
+
+    state = State.load()
+    workspace_id = state.validate_workspace()
+    code_id = state.validate_code(priority=code_id)
+
+    client = BevorInternalClient(
+        creds=creds, workspace_id=workspace_id, project_id=state.validate_project()
+    )
+
+    content = client.codes.nodes.content(code_id, ID(node_id))
+
+    if not content:
+        output.error("No source content available.")
+        raise typer.Exit(1)
+
+    console.print(Syntax(content.strip(), "solidity", theme="monokai", line_numbers=False, padding=1))

bevor_cli/commands/projects/create.py

@@ -0,0 +1,54 @@
+from typing import Optional, get_args
+
+import questionary
+import typer
+from bevor_sdk.types.literals import ProjectKind
+from bevor_sdk.types.requests.manage import CreateProjectBodyDict
+from bevor_sdk.types.responses.manage import ProjectSchema
+from rich.prompt import Confirm, Prompt
+
+from bevor_cli import output
+from bevor_cli.annotations.confirmations import YES
+from bevor_cli.client import BevorInternalBusinessClient
+from bevor_cli.configuration.creds import Credentials
+from bevor_cli.configuration.state import State
+from bevor_cli.select import _select
+
+
+def create_project(
+    name: Optional[str] = typer.Option(None, "--name", help="project name"),
+    description: Optional[str] = typer.Option(None, "--description", help="project description"),
+    kind: Optional[ProjectKind] = typer.Option(None, "--kind", help="project kind"),
+    yes: YES = False,
+) -> ProjectSchema:
+    """Create a project. Can pass arguments, or do so interactively."""
+    creds = Credentials.load()
+    creds.require_token()
+
+    state = State.load()
+    workspace_id = state.validate_workspace()
+
+    client = BevorInternalBusinessClient(creds=creds, workspace_id=workspace_id)
+
+    if name is None:
+        name = Prompt.ask("Project name")
+    if description is None:
+        description = Prompt.ask("Project description (optional)", default="")
+    if kind is None:
+        choices = [questionary.Choice(title=p_kind, value=p_kind) for p_kind in get_args(ProjectKind)]
+        kind = _select("Project Kind", choices=choices)
+
+    if not yes:
+        # Add this so that it's obvious which workspace they're creating the project under. Can override.
+        workspace = client.get_workspace()
+        if not Confirm.ask(f"Create project '{name}' under workspace '{workspace.name}'?"):
+            raise typer.Exit(0)
+
+    project = client.create_project(
+        CreateProjectBodyDict(name=name, description=description, kind=kind)
+    )
+
+    state.set_project(project.id)
+    state.save()
+    output.success(f"Created project: {project.id} {project.name}")
+    return project

bevor_cli/commands/projects/list.py

@@ -0,0 +1,57 @@
+"""``bevor projects list``."""
+
+import json
+from typing import Any, List
+
+from rich.table import Table
+from rich.text import Text
+
+from bevor_cli.annotations.confirmations import JSON_OUT
+from bevor_cli.client import BevorInternalBusinessClient
+from bevor_cli.configuration.creds import Credentials
+from bevor_cli.configuration.state import State
+from bevor_cli.output import console
+
+
+def list_projects(
+    json_out: JSON_OUT = False,
+) -> None:
+    """List projects in the configured workspace."""
+    creds = Credentials.load()
+    creds.require_token()
+
+    state = State.load()
+    workspace_id = state.validate_workspace()
+
+    client = BevorInternalBusinessClient(creds=creds, workspace_id=workspace_id)
+
+    projects = client.list_projects()
+
+    if json_out:
+        console.print_json(json.dumps(projects.model_dump(mode="json"), indent=2))
+        return
+
+    if not projects.results:
+        console.print("No projects found.")
+        return
+
+    table = Table("NAME", "ID", "CODES", "ANALYSES", "CREATED")
+    table.columns[2].justify = "right"
+    table.columns[3].justify = "right"
+
+    for p in projects.results:
+        is_active = p.id == state.project_id
+
+        row: List[Any] = [
+            p.name,
+            p.id,
+            str(p.n_codes),
+            str(p.n_analyses),
+            str(p.created_at.date()),
+        ]
+        if is_active:
+            row.append(Text("●", style="bold green"))
+
+        table.add_row(*row, style="bold green" if is_active else None)
+
+    console.print(table)

bevor_cli/commands/projects/select.py

@@ -0,0 +1,75 @@
+"""``bevor projects select`` — set the selected project."""
+
+from typing import Optional
+
+import questionary
+import typer
+from bevor_sdk.types.responses.manage import ProjectSchema
+
+from bevor_cli import output
+from bevor_cli.client import BevorInternalBusinessClient
+from bevor_cli.configuration.creds import Credentials
+from bevor_cli.configuration.state import State
+from bevor_cli.output import console
+from bevor_cli.select import _CANCEL, _select
+
+
+def select_project(
+    project_id: Optional[str] = typer.Argument(
+        None, help="Project id to select. Prompts with a list when omitted."
+    ),
+) -> None:
+    """Select a project by id, or pick one from a list."""
+    creds = Credentials.load()
+    creds.require_token()
+
+    state = State.load()
+    workspace_id = state.validate_workspace()
+
+    client = BevorInternalBusinessClient(creds=creds, workspace_id=workspace_id)
+
+    projects = client.list_projects()
+
+    if not projects.results:
+        console.print("No projects found.")
+        return
+
+    if project_id:
+        if not any(p.id == project_id for p in projects.results):
+            output.error("project_id does not exist")
+            raise typer.Exit(1)
+
+        if state.project_id != project_id:
+            state.set_project(project_id)
+            state.save()
+            output.info("Updated bevor configuration")
+        return
+
+    if len(projects.results) == 1:
+        if projects.results[0].id == state.project_id:
+            output.info("only 1 project exists, skipping")
+            return
+
+        state.set_project(projects.results[0].id)
+        state.save()
+        output.info("only 1 project exists, updating your bevor context automatically")
+        return
+
+    choices = [
+        questionary.Choice(
+            title=f"{p.name}   {p.id}" + ("  ●" if p.id == state.project_id else ""),
+            value=p,
+        )
+        for p in projects.results
+    ] + [questionary.Choice(title="↵ Cancel", value=_CANCEL)]
+
+    project: ProjectSchema | None = _select("Select a project:", choices)
+    if project is None:
+        raise typer.Exit(0)
+
+    if project.id != state.project_id:
+        output.info("Clearing active code version and analysis (different project).")
+        state.set_project(project.id)
+
+    state.save()
+    console.print(f"Active project: {project.name} ({project.id})")

bevor_cli/commands/projects/view.py

@@ -0,0 +1,49 @@
+"""``bevor projects view`` — show a project (the selected one by default)."""
+
+from typing import Optional
+
+import typer
+from rich.table import Table
+
+from bevor_cli.annotations.confirmations import JSON_OUT
+from bevor_cli.client import BevorInternalClient
+from bevor_cli.configuration.creds import Credentials
+from bevor_cli.configuration.state import State
+from bevor_cli.output import console
+
+
+def view_project(
+    project_id: Optional[str] = typer.Argument(
+        None, help="Project id. Defaults to the selected project."
+    ),
+    json_out: JSON_OUT = False,
+) -> None:
+    """Print details for a project in the selected workspace, or the selected project."""
+    creds = Credentials.load()
+    creds.require_token()
+
+    state = State.load()
+    workspace_id = state.validate_workspace()
+    project_id = state.validate_project(project_id)
+
+    client = BevorInternalClient(creds=creds, workspace_id=workspace_id, project_id=project_id)
+
+    project = client.projects.get()
+
+    if json_out:
+        console.print_json(project.model_dump_json(indent=2))
+        return
+
+    table = Table(show_header=False, box=None, padding=(0, 1))
+    table.add_column(style="bold")
+    table.add_column()
+
+    table.add_row("Name", project.name)
+    table.add_row("ID", project.id)
+    table.add_row("Description", project.description or "")
+    table.add_row("Codes", str(project.n_codes))
+    table.add_row("Analyses", str(project.n_analyses))
+    table.add_row("Created by", project.created_by_user.username)
+    table.add_row("Created", str(project.created_at.date()))
+
+    console.print(table)

bevor_cli/commands/sessions/list.py

@@ -0,0 +1,36 @@
+"""``bevor sessions list``."""
+
+from rich.table import Table
+
+from bevor_cli.client import BevorInternalBusinessClient
+from bevor_cli.configuration.creds import Credentials
+from bevor_cli.output import console
+
+
+def sessions_list() -> None:
+    """List active CLI sessions."""
+    creds = Credentials.load()
+    creds.require_token()
+
+    client = BevorInternalBusinessClient(creds=creds)
+
+    data = client.get("/v1/cli-tokens").json()
+    sessions = data.get("results", [])
+
+    if not sessions:
+        console.print("No active CLI sessions.")
+        return
+
+    table = Table("ID", "Device", "IP", "Last Used", "Expires", "Current Device")
+
+    for s in sessions:
+        args = [
+            s["id"],
+            s["device_name"],
+            s["ip_address"],
+            s["last_used_at"],
+            s["expires_at"],
+            str(s["is_current"]),
+        ]
+        table.add_row(*args)
+    console.print(table)

bevor_cli/commands/sessions/revoke.py

@@ -0,0 +1,35 @@
+"""``bevor sessions revoke``."""
+
+import typer
+
+from bevor_cli import output
+from bevor_cli.client import BevorInternalBusinessClient
+from bevor_cli.configuration.creds import Credentials
+
+
+def sessions_revoke(
+    token_id: str = typer.Argument(
+        ...,
+        help="CLI session id to revoke (from `bevor sessions`).",
+    ),
+) -> None:
+    """Revoke a CLI session by id."""
+
+    creds = Credentials.load()
+    creds.require_token()
+
+    if token_id == creds.token:
+        output.error(
+            "Cannot revoke this session since it's your current one.",
+            hint="Run `bevor logout` instead.",
+        )
+        raise typer.Exit(1)
+
+    client = BevorInternalBusinessClient(creds=creds)
+
+    res = client.delete(f"/v1/cli-tokens/revoke/{token_id}").json()
+    output.success(f"Revoked CLI session {token_id}.")
+
+    if res["is_current"]:
+        creds.delete()
+        output.success("Logged out.")

bevor_cli/commands/workspaces/list.py

@@ -0,0 +1,50 @@
+"""``bevor workspace list`` — list workspaces."""
+
+import json
+from typing import Any, List
+
+from rich.table import Table
+from rich.text import Text
+
+from bevor_cli.annotations.confirmations import JSON_OUT
+from bevor_cli.client import BevorInternalBusinessClient
+from bevor_cli.configuration.creds import Credentials
+from bevor_cli.configuration.state import State
+from bevor_cli.output import console
+
+
+def list_workspaces(
+    json_out: JSON_OUT = False,
+) -> None:
+    """List workspaces you belong to."""
+    creds = Credentials.load()
+    creds.require_token()
+
+    state = State.load()
+
+    workspaces = BevorInternalBusinessClient(creds=creds).list_workspaces()
+
+    if json_out:
+        console.print_json(json.dumps(workspaces.model_dump(mode="json"), indent=2))
+        return
+
+    if not workspaces.results:
+        console.print("No workspaces found.")
+        return
+
+    table = Table("ID", "Name", "N Projects", "Role", "Created")
+
+    for workspace in workspaces.results:
+        args: List[Any] = [
+            workspace.id,
+            workspace.name,
+            str(workspace.n_projects),
+            workspace.role,
+            str(workspace.created_at.date()),
+        ]
+
+        if workspace.id == state.workspace_id:
+            args.append(Text("●", style="bold green"))
+        table.add_row(*args)
+
+    console.print(table)

bevor_cli/commands/workspaces/pick.py

@@ -0,0 +1,20 @@
+"""Interactive workspace picker (shared with ``bevor link``)."""
+
+from typing import List
+
+import questionary
+import typer
+from bevor_sdk.types.responses.manage import WorkspaceDetailedSchema
+
+from bevor_cli.select import _select
+
+
+def pick_workspace(workspaces: List[WorkspaceDetailedSchema]) -> WorkspaceDetailedSchema:
+    choices = [
+        questionary.Choice(title=f"{workspace.name}  {workspace.id}", value=workspace)
+        for workspace in workspaces
+    ]
+    chosen: WorkspaceDetailedSchema | None = _select("Select a workspace:", choices)
+    if chosen is None:
+        raise typer.Exit(0)
+    return chosen

bevor_cli/commands/workspaces/select.py

@@ -0,0 +1,68 @@
+"""``bevor workspaces select`` — set the selected workspace."""
+
+from typing import Optional
+
+import typer
+
+from bevor_cli import output
+from bevor_cli.client import BevorInternalBusinessClient
+from bevor_cli.configuration.creds import Credentials
+from bevor_cli.configuration.state import State
+from bevor_cli.output import console
+
+from .pick import pick_workspace
+
+
+def select_workspace(
+    workspace_id: Optional[str] = typer.Argument(
+        None, help="Workspace id to select. Prompts with a list when omitted."
+    ),
+) -> None:
+    """Select a workspace by id, or pick one from a list."""
+
+    creds = Credentials.load()
+    creds.require_token()
+
+    state = State.load()
+
+    if workspace_id and state.workspace_id == workspace_id:
+        return
+
+    workspaces = BevorInternalBusinessClient(creds=creds).list_workspaces()
+
+    if not workspaces.results:
+        console.print("No workspaces found.")
+        return
+
+    if workspace_id:
+        if not any(workspace.id == workspace_id for workspace in workspaces.results):
+            output.error(
+                "passed workspace does not exist",
+                hint="run `bevor workspaces list` to see available ids.",
+            )
+            raise typer.Exit(1)
+
+        state.set_workspace(workspace_id)
+        state.save()
+        return
+
+    if len(workspaces.results) == 1:
+        if workspaces.results[0].id == state.workspace_id:
+            output.info("only 1 workspace exists, skipping")
+            return
+
+        state.set_workspace(workspaces.results[0].id)
+        state.save()
+        output.info("only 1 workspace exists, updating your bevor context automatically")
+        return
+
+    workspace = pick_workspace(workspaces.results)
+
+    if state.workspace_id == workspace.id:
+        output.info("no changes to your current bevor context.")
+        return
+
+    state.set_workspace(workspace.id)
+    state.save()
+
+    output.success(f"Active workspace: {workspace.name} ({workspace.id})")

bevor_cli/commands/workspaces/view.py

@@ -0,0 +1,48 @@
+"""``bevor workspaces view`` — show a workspace (the selected one by default)."""
+
+from typing import Optional
+
+import typer
+from rich.table import Table
+
+from bevor_cli.annotations.confirmations import JSON_OUT
+from bevor_cli.client import BevorInternalBusinessClient
+from bevor_cli.configuration.creds import Credentials
+from bevor_cli.configuration.state import State
+from bevor_cli.output import console
+
+
+def view_workspace(
+    workspace_id: Optional[str] = typer.Argument(
+        None, help="Workspace id. Defaults to the selected workspace."
+    ),
+    json_out: JSON_OUT = False,
+) -> None:
+    """Print details for a workspace, or the selected one."""
+    creds = Credentials.load()
+    creds.require_token()
+
+    client = BevorInternalBusinessClient(
+        creds=creds, workspace_id=workspace_id or State.load().validate_workspace()
+    )
+
+    workspace = client.get_workspace()
+
+    if json_out:
+        console.print_json(workspace.model_dump_json(indent=2))
+        return
+
+    table = Table(show_header=False, box=None, padding=(0, 1))
+    table.add_column(style="bold")
+    table.add_column()
+
+    table.add_row("Name", workspace.name)
+    table.add_row("ID", workspace.id)
+    table.add_row("My Role", workspace.role)
+    table.add_row("Projects", str(workspace.n_projects))
+    table.add_row("Members", str(workspace.n_members))
+    table.add_row("Default workspace", str(workspace.is_default))
+    table.add_row("Created", str(workspace.created_at))
+    table.add_row("Created by", workspace.created_by_user.username)
+
+    console.print(table)