@team-agent/installer 0.2.10 → 0.3.0

package/src/team_agent/message_store/schema.py

@@ -1,308 +0,0 @@
-from __future__ import annotations
-
-import sqlite3
-from datetime import datetime, timezone
-from pathlib import Path
-
-from team_agent.message_store import schema_migration as _schema_migration
-from team_agent.message_store.schema_migration import ensure_schema_indexes, ensure_table_layout, table_layout
-
-
-MESSAGE_COLUMNS = {
-    "owner_team_id",
-    "message_id",
-    "task_id",
-    "sender",
-    "recipient",
-    "reply_to",
-    "requires_ack",
-    "status",
-    "content",
-    "artifact_refs",
-    "created_at",
-    "updated_at",
-    "delivered_at",
-    "acknowledged_at",
-    "error",
-    "delivery_attempts",
-}
-RESULT_COLUMNS = {"owner_team_id", "result_id", "task_id", "agent_id", "envelope", "status", "created_at"}
-SCHEDULED_EVENT_COLUMNS = {
-    "id",
-    "owner_team_id",
-    "due_at",
-    "target",
-    "kind",
-    "payload_json",
-    "status",
-    "created_at",
-    "fired_at",
-    "result_json",
-}
-DELIVERY_TOKEN_COLUMNS = {
-    "message_id",
-    "unique_token",
-    "injected_at",
-    "visible_at",
-    "consumed_at",
-    "failed_at",
-    "failure_reason",
-}
-AGENT_HEALTH_COLUMNS = {
-    "owner_team_id",
-    "agent_id",
-    "status",
-    "last_output_at",
-    "context_usage_pct",
-    "current_task_id",
-    "updated_at",
-}
-PEER_ALLOWLIST_COLUMNS = {"a", "b", "created_at"}
-RESULT_WATCHER_COLUMNS = {
-    "owner_team_id",
-    "watcher_id",
-    "task_id",
-    "agent_id",
-    "message_id",
-    "leader_id",
-    "status",
-    "created_at",
-    "completed_at",
-    "result_id",
-    "notified_message_id",
-    "error",
-}
-LEADER_NOTIFICATION_LOG_COLUMNS = {
-    "result_id",
-    "owner_team_id",
-    "owner_epoch",
-    "leader_session_uuid",
-    "notified_message_id",
-    "notified_at",
-    "leader_pane_id_at_notify",
-    "envelope_content_hash",
-}
-
-
-def utcnow() -> str:
-    return datetime.now(timezone.utc).isoformat()
-
-SCHEMA_VERSION = 3
-SCHEMA_MIGRATIONS = _schema_migration.SCHEMA_MIGRATIONS
-
-
-def _table_columns(conn: sqlite3.Connection, table: str) -> set[str]:
-    return set(table_layout(conn, table))
-
-
-def _ensure_table_columns(
-    conn: sqlite3.Connection,
-    table: str,
-    required: set[str],
-    migrations: dict[str, str] | None = None,
-) -> None:
-    columns = _table_columns(conn, table)
-    missing = required - columns
-    migrations = migrations or {}
-    unsupported = missing - set(migrations)
-    if unsupported:
-        names = ", ".join(sorted(unsupported))
-        raise RuntimeError(f"team.db table {table} is missing required column(s): {names}")
-    for name in sorted(missing):
-        conn.execute(migrations[name])
-
-
-def initialize_schema(conn: sqlite3.Connection, db_path: Path | None = None) -> None:
-    _schema_migration.SCHEMA_MIGRATIONS = SCHEMA_MIGRATIONS
-    ensure_table_layout(conn, schema_version=SCHEMA_VERSION, db_path=db_path)
-    with conn:
-        conn.execute(
-            """
-            create table if not exists messages (
-              message_id text primary key,
-              owner_team_id text,
-              task_id text,
-              sender text,
-              recipient text,
-              reply_to text,
-              requires_ack integer,
-              status text,
-              content text,
-              artifact_refs text,
-              created_at text,
-              updated_at text,
-              delivered_at text,
-              acknowledged_at text,
-              error text,
-              delivery_attempts integer not null default 0
-            )
-            """
-        )
-        conn.execute(
-            """
-            create table if not exists results (
-              result_id text primary key,
-              owner_team_id text,
-              task_id text not null,
-              agent_id text not null,
-              envelope text not null,
-              status text not null,
-              created_at text not null
-            )
-            """
-        )
-        conn.execute(
-            """
-            create table if not exists scheduled_events (
-              id integer primary key,
-              owner_team_id text,
-              due_at text not null,
-              target text not null,
-              kind text not null,
-              payload_json text not null,
-              status text not null,
-              created_at text not null,
-              fired_at text,
-              result_json text
-            )
-            """
-        )
-        conn.execute(
-            """
-            create table if not exists delivery_tokens (
-              message_id text primary key,
-              unique_token text not null,
-              injected_at text not null,
-              visible_at text,
-              consumed_at text,
-              failed_at text,
-              failure_reason text
-            )
-            """
-        )
-        conn.execute(
-            """
-            create table if not exists agent_health (
-              owner_team_id text,
-              agent_id text not null,
-              status text not null,
-              last_output_at text,
-              context_usage_pct integer,
-              current_task_id text,
-              updated_at text not null,
-              unique(owner_team_id, agent_id)
-            )
-            """
-        )
-        conn.execute(
-            """
-            create table if not exists peer_allowlist (
-              a text not null,
-              b text not null,
-              created_at text not null,
-              primary key (a, b)
-            )
-            """
-        )
-        conn.execute(
-            """
-            create table if not exists result_watchers (
-              watcher_id text primary key,
-              owner_team_id text,
-              task_id text,
-              agent_id text,
-              message_id text,
-              leader_id text not null,
-              status text not null,
-              created_at text not null,
-              completed_at text,
-              result_id text,
-              notified_message_id text,
-              error text
-            )
-            """
-        )
-        _ensure_table_columns(
-            conn,
-            "messages",
-            MESSAGE_COLUMNS,
-            {
-                "delivery_attempts": (
-                    "alter table messages add column delivery_attempts integer not null default 0"
-                ),
-                "owner_team_id": "alter table messages add column owner_team_id text",
-            },
-        )
-        _ensure_table_columns(
-            conn,
-            "results",
-            RESULT_COLUMNS,
-            {"owner_team_id": "alter table results add column owner_team_id text"},
-        )
-        _ensure_table_columns(
-            conn,
-            "scheduled_events",
-            SCHEDULED_EVENT_COLUMNS,
-            {"owner_team_id": "alter table scheduled_events add column owner_team_id text"},
-        )
-        _ensure_table_columns(conn, "delivery_tokens", DELIVERY_TOKEN_COLUMNS)
-        _migrate_agent_health_owner_team_id(conn)
-        _ensure_table_columns(conn, "peer_allowlist", PEER_ALLOWLIST_COLUMNS)
-        _ensure_table_columns(
-            conn,
-            "result_watchers",
-            RESULT_WATCHER_COLUMNS,
-            {"owner_team_id": "alter table result_watchers add column owner_team_id text"},
-        )
-        # Dedupe leader notifications at the injection boundary by
-        # (result_id, owner_team_id, owner_epoch). leader_session_uuid remains
-        # nullable compatibility/audit metadata.
-        # UNIQUE primary key + INSERT OR IGNORE in claim_leader_notification_delivery gives
-        # atomic exactly-once without an advisory lock. Retires the bad6484 watcher-table
-        # UPSERT approach.
-        conn.execute(
-            """
-            create table if not exists leader_notification_log (
-              result_id text not null,
-              owner_team_id text not null default '',
-              owner_epoch integer not null default 0,
-              leader_session_uuid text,
-              notified_message_id text not null,
-              notified_at text not null,
-              leader_pane_id_at_notify text,
-              envelope_content_hash text,
-              primary key (result_id, owner_team_id, owner_epoch)
-            )
-            """
-        )
-        _ensure_table_columns(conn, "leader_notification_log", LEADER_NOTIFICATION_LOG_COLUMNS)
-        ensure_schema_indexes(conn)
-        conn.execute(f"pragma user_version = {SCHEMA_VERSION}")
-
-
-def _migrate_agent_health_owner_team_id(conn: sqlite3.Connection) -> None:
-    columns = _table_columns(conn, "agent_health")
-    if "owner_team_id" not in columns:
-        conn.execute(
-            """
-            create table agent_health_new (
-              owner_team_id text,
-              agent_id text not null,
-              status text not null,
-              last_output_at text,
-              context_usage_pct integer,
-              current_task_id text,
-              updated_at text not null,
-              unique(owner_team_id, agent_id)
-            )
-            """
-        )
-        conn.execute(
-            """
-            insert into agent_health_new(owner_team_id, agent_id, status, last_output_at, context_usage_pct, current_task_id, updated_at)
-            select null, agent_id, status, last_output_at, context_usage_pct, current_task_id, updated_at from agent_health
-            """
-        )
-        conn.execute("drop table agent_health")
-        conn.execute("alter table agent_health_new rename to agent_health")
-    _ensure_table_columns(conn, "agent_health", AGENT_HEALTH_COLUMNS)