codefission 0.1.0__py3-none-any.whl

codefission/__main__.py

@@ -0,0 +1,5 @@
+"""Allow running as `python -m codefission`."""
+
+from codefission.cli import main
+
+main()

codefission/cli.py

@@ -0,0 +1,20 @@
+"""CLI entry point for CodeFission."""
+
+import sys
+
+import uvicorn
+
+
+def main():
+    port = int(sys.argv[1]) if len(sys.argv) > 1 else 8080
+    uvicorn.run(
+        "codefission.main:app",
+        host="0.0.0.0",
+        port=port,
+        ws_ping_interval=30,
+        ws_ping_timeout=10,
+    )
+
+
+if __name__ == "__main__":
+    main()

codefission/config.py

@@ -0,0 +1,29 @@
+import json
+import os
+from pathlib import Path
+
+# Fixed location for the bootstrap config (read before DB exists)
+CONFIG_FILE = Path.home() / ".codefission.json"
+
+def _load_config() -> dict:
+    if CONFIG_FILE.exists():
+        try:
+            return json.loads(CONFIG_FILE.read_text())
+        except Exception:
+            pass
+    return {}
+
+def save_config(updates: dict):
+    """Merge updates into the config file."""
+    cfg = _load_config()
+    cfg.update(updates)
+    CONFIG_FILE.write_text(json.dumps(cfg, indent=2) + "\n")
+
+_cfg = _load_config()
+
+# Data directory: config file > env var > default
+DATA_DIR = Path(
+    _cfg.get("data_dir")
+    or os.environ.get("CODEFISSION_DATA_DIR")
+    or str(Path.home() / ".codefission")
+)

codefission/db.py

@@ -0,0 +1,116 @@
+import aiosqlite
+from pathlib import Path
+from contextlib import asynccontextmanager
+
+from models import DEFAULT_PROVIDER, DEFAULT_MODEL
+from config import DATA_DIR
+
+DB_PATH = DATA_DIR / "codefission.db"
+
+_conn: aiosqlite.Connection | None = None
+
+
+async def _open_connection() -> aiosqlite.Connection:
+    DB_PATH.parent.mkdir(parents=True, exist_ok=True)
+    db = await aiosqlite.connect(str(DB_PATH))
+    db.row_factory = aiosqlite.Row
+    await db.execute("PRAGMA journal_mode=WAL")
+    await db.execute("PRAGMA foreign_keys=ON")
+    return db
+
+
+@asynccontextmanager
+async def get_db():
+    global _conn
+    if _conn is None:
+        _conn = await _open_connection()
+    yield _conn
+
+
+async def close_db():
+    global _conn
+    if _conn is not None:
+        await _conn.close()
+        _conn = None
+
+
+async def init_db():
+    global _conn
+    # Close existing connection (e.g. tests changing DB_PATH)
+    if _conn is not None:
+        await _conn.close()
+        _conn = None
+
+    async with get_db() as db:
+        await db.executescript("""
+            CREATE TABLE IF NOT EXISTS trees (
+                id TEXT PRIMARY KEY,
+                name TEXT NOT NULL,
+                created_at TEXT NOT NULL
+            );
+            CREATE TABLE IF NOT EXISTS nodes (
+                id TEXT PRIMARY KEY,
+                tree_id TEXT NOT NULL REFERENCES trees(id),
+                parent_id TEXT REFERENCES nodes(id),
+                user_message TEXT NOT NULL DEFAULT '',
+                assistant_response TEXT NOT NULL DEFAULT '',
+                label TEXT NOT NULL DEFAULT '',
+                status TEXT NOT NULL DEFAULT 'idle',
+                created_at TEXT NOT NULL
+            );
+            CREATE TABLE IF NOT EXISTS settings (
+                key TEXT PRIMARY KEY,
+                value TEXT
+            );
+            CREATE INDEX IF NOT EXISTS idx_nodes_tree ON nodes(tree_id);
+            CREATE INDEX IF NOT EXISTS idx_nodes_parent ON nodes(parent_id);
+        """)
+
+        # Migrate: add provider/model columns to trees if missing
+        cursor = await db.execute("PRAGMA table_info(trees)")
+        columns = {row[1] for row in await cursor.fetchall()}
+        if "provider" not in columns:
+            await db.execute(
+                f"ALTER TABLE trees ADD COLUMN provider TEXT NOT NULL DEFAULT '{DEFAULT_PROVIDER}'"
+            )
+        if "model" not in columns:
+            await db.execute(
+                f"ALTER TABLE trees ADD COLUMN model TEXT NOT NULL DEFAULT '{DEFAULT_MODEL}'"
+            )
+
+        # Migrate: add repo_mode, repo_source to trees
+        if "repo_mode" not in columns:
+            await db.execute("ALTER TABLE trees ADD COLUMN repo_mode TEXT NOT NULL DEFAULT 'none'")
+        if "repo_source" not in columns:
+            await db.execute("ALTER TABLE trees ADD COLUMN repo_source TEXT")
+
+        # Migrate: add max_turns to trees
+        if "max_turns" not in columns:
+            await db.execute("ALTER TABLE trees ADD COLUMN max_turns INTEGER")
+
+        # Migrate: add skill to trees
+        if "skill" not in columns:
+            await db.execute("ALTER TABLE trees ADD COLUMN skill TEXT NOT NULL DEFAULT ''")
+
+        # Migrate: add notes to trees
+        if "notes" not in columns:
+            await db.execute("ALTER TABLE trees ADD COLUMN notes TEXT NOT NULL DEFAULT '[]'")
+
+        # Migrate: rename provider "anthropic" → "claude-code"
+        await db.execute("UPDATE trees SET provider = 'claude-code' WHERE provider = 'anthropic'")
+
+        # Migrate: add git_branch, git_commit to nodes
+        cursor2 = await db.execute("PRAGMA table_info(nodes)")
+        node_columns = {row[1] for row in await cursor2.fetchall()}
+        if "git_branch" not in node_columns:
+            await db.execute("ALTER TABLE nodes ADD COLUMN git_branch TEXT")
+        if "git_commit" not in node_columns:
+            await db.execute("ALTER TABLE nodes ADD COLUMN git_commit TEXT")
+        if "session_id" not in node_columns:
+            await db.execute("ALTER TABLE nodes ADD COLUMN session_id TEXT")
+        if "created_by" not in node_columns:
+            await db.execute("ALTER TABLE nodes ADD COLUMN created_by TEXT NOT NULL DEFAULT 'human'")
+        if "quoted_node_ids" not in node_columns:
+            await db.execute("ALTER TABLE nodes ADD COLUMN quoted_node_ids TEXT NOT NULL DEFAULT '[]'")
+
+        await db.commit()

codefission/events.py

@@ -0,0 +1,103 @@
+"""Async event bus for decoupled communication between services and WebSocket layer.
+
+Adapted from WhatTheBot's core/events.py — same pub/sub pattern, CodeFission-specific events.
+"""
+
+from __future__ import annotations
+
+import asyncio
+from collections import defaultdict
+from typing import Any, Callable, Coroutine
+
+Callback = Callable[..., Coroutine[Any, Any, None]]
+
+
+class EventBus:
+    def __init__(self) -> None:
+        self._listeners: dict[str, list[Callback]] = defaultdict(list)
+
+    def on(self, event: str, callback: Callback) -> None:
+        self._listeners[event].append(callback)
+
+    def off(self, event: str, callback: Callback) -> None:
+        self._listeners[event] = [
+            cb for cb in self._listeners[event] if cb is not callback
+        ]
+
+    async def emit(self, event: str, **kwargs: Any) -> None:
+        for cb in list(self._listeners.get(event, [])):
+            asyncio.create_task(cb(**kwargs))
+
+
+# Singleton bus
+bus = EventBus()
+
+
+# ── Well-known event names (internal, backend-side) ─────────────────────
+
+STREAM_START = "stream_start"    # chat streaming begins for a node
+STREAM_DELTA = "stream_delta"    # new token(s) in a streaming response
+STREAM_END = "stream_end"        # chat streaming finished
+STREAM_ERROR = "stream_error"    # chat streaming hit an error
+NODE_CREATED = "node_created"    # a new node was created
+NODE_UPDATED = "node_updated"    # a node's data changed
+TREE_CREATED = "tree_created"    # a new tree was created
+TREE_DELETED = "tree_deleted"    # a tree was deleted
+
+
+# ── WebSocket message types (wire protocol, client ↔ server) ───────────
+
+class WS:
+    """Structured constants for WebSocket JSON message types.
+
+    Inbound  = client → server requests
+    Outbound = server → client responses/pushes
+    """
+
+    # Inbound (client → server)
+    LIST_TREES = "list_trees"
+    CREATE_TREE = "create_tree"
+    LOAD_TREE = "load_tree"
+    DELETE_TREE = "delete_tree"
+    BRANCH = "branch"
+    CHAT = "chat"
+    CANCEL = "cancel"
+    DUPLICATE = "duplicate"
+    GET_NODE = "get_node"
+    SET_REPO = "set_repo"
+    GET_NODE_FILES = "get_node_files"
+    GET_NODE_DIFF = "get_node_diff"
+    GET_FILE_CONTENT = "get_file_content"
+    SELECT_TREE = "select_tree"
+    SET_EXPANDED = "set_expanded"
+    SET_SUBTREE_COLLAPSED = "set_subtree_collapsed"
+    GET_SETTINGS = "get_settings"
+    UPDATE_GLOBAL_SETTINGS = "update_global_settings"
+    UPDATE_TREE_SETTINGS = "update_tree_settings"
+    GET_NODE_PROCESSES = "get_node_processes"
+    KILL_PROCESS = "kill_process"
+    KILL_ALL_PROCESSES = "kill_all_processes"
+    DELETE_NODE = "delete_node"
+
+    # Outbound
+    SETTINGS = "settings"
+
+    # Outbound (server → client)
+    TREES = "trees"
+    TREE_CREATED = "tree_created"
+    TREE_LOADED = "tree_loaded"
+    TREE_DELETED = "tree_deleted"
+    TREE_UPDATED = "tree_updated"
+    NODE_CREATED = "node_created"
+    NODE_DATA = "node_data"
+    NODE_FILES = "node_files"
+    NODE_DIFF = "node_diff"
+    FILE_CONTENT = "file_content"
+    STATUS = "status"
+    CHUNK = "chunk"
+    TOOL_START = "tool_start"
+    TOOL_END = "tool_end"
+    DONE = "done"
+    ERROR = "error"
+    NODE_PROCESSES = "node_processes"
+    NODES_DELETED = "nodes_deleted"