Rebuild AgentCompany Electron app

2026-05-20 23:27:00 -04:00
1 parent bc5daa6eb8
commit fcf6d90731
@@ -0,0 +1,168 @@
+import type { DatabaseSync } from "node:sqlite";
+import { APPROVAL_STATE, RISK_LEVEL, TASK_STATUS } from "../../shared/domain";
+import { PROVIDER_CATALOG } from "../../shared/provider-catalog";
+import { createId } from "../../shared/ids";
+import type { CreateAgentInput, CreateOrganizationInput, CreateTaskInput } from "../../shared/commands";
+import { AuditService } from "./audit-service";
+
+export class OrganizationService {
+  private readonly audit: AuditService;
+
+  constructor(private readonly db: DatabaseSync) {
+    this.audit = new AuditService(db);
+  }
+
+  createOrganization(input: CreateOrganizationInput): string {
+    const orgId = createId("org");
+    const employeeId = createId("emp");
+    const now = new Date().toISOString();
+
+    this.db.prepare("insert into organizations(id, name, mission, goal, created_at) values (?, ?, ?, ?, ?)")
+      .run(orgId, input.organizationName.trim(), input.mission.trim(), input.goal.trim(), now);
+    this.db.prepare("insert into employees(id, org_id, display_name, role, created_at) values (?, ?, ?, ?, ?)")
+      .run(employeeId, orgId, input.employeeName.trim(), "Board Chair", now);
+
+    for (const space of [
+      ["company", "general", "Company coordination", "workspace"],
+      ["department", "product", "Product decisions", "workspace"],
+      ["department", "engineering", "Engineering delivery", "workspace"],
+      ["company", "board", "Board decisions", "restricted"],
+    ]) {
+      const spaceId = createId("spc");
+      this.db.prepare(`
+        insert into chat_spaces(id, org_id, space_type, name, topic, visibility, created_at)
+        values (?, ?, ?, ?, ?, ?, ?)
+      `).run(spaceId, orgId, space[0], space[1], space[2], space[3], now);
+      this.db.prepare(`
+        insert into chat_memberships(space_id, principal_type, principal_id, role, mute_state)
+        values (?, ?, ?, ?, ?)
+      `).run(spaceId, "employee", employeeId, space[1] === "board" ? "approver" : "member", "unmuted");
+    }
+
+    for (const provider of PROVIDER_CATALOG) {
+      this.db.prepare(`
+        insert into providers(id, org_id, provider_key, label, auth_type, model, credential_id, enabled, created_at)
+        values (?, ?, ?, ?, ?, ?, ?, ?, ?)
+      `).run(
+        createId("prv"),
+        orgId,
+        provider.key,
+        provider.label,
+        provider.authType,
+        provider.defaultModel,
+        null,
+        0,
+        now,
+      );
+    }
+
+    this.audit.record({
+      orgId,
+      entityType: "organization",
+      entityId: orgId,
+      actorType: "employee",
+      actorId: employeeId,
+      action: "created",
+      payload: { name: input.organizationName.trim() },
+    });
+    return orgId;
+  }
+
+  createAgent(input: CreateAgentInput): string {
+    const id = createId("agt");
+    const now = new Date().toISOString();
+    this.db.prepare(`
+      insert into agents(
+        id, org_id, name, role, department, model_provider_id, model_id,
+        budget_cents, manager_agent_id, lifecycle, created_at
+      ) values (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
+    `).run(
+      id,
+      input.orgId,
+      input.name.trim(),
+      input.role.trim(),
+      input.department.trim(),
+      input.modelProviderId,
+      input.modelId,
+      input.budgetCents,
+      input.managerAgentId,
+      "active",
+      now,
+    );
+    this.audit.record({
+      orgId: input.orgId,
+      entityType: "agent",
+      entityId: id,
+      actorType: "system",
+      actorId: null,
+      action: "created",
+      payload: { role: input.role, department: input.department },
+    });
+    return id;
+  }
+
+  createTask(input: CreateTaskInput): string {
+    const id = createId("tsk");
+    const now = new Date().toISOString();
+    this.db.prepare(`
+      insert into tasks(id, org_id, title, description, status, assignee_agent_id, risk, created_at, updated_at)
+      values (?, ?, ?, ?, ?, ?, ?, ?, ?)
+    `).run(
+      id,
+      input.orgId,
+      input.title.trim(),
+      input.description.trim(),
+      input.status,
+      input.assigneeAgentId,
+      input.risk,
+      now,
+      now,
+    );
+    if (input.risk === RISK_LEVEL.HIGH) {
+      this.db.prepare(`
+        insert into approvals(
+          id, org_id, kind, title, details, risk, requested_by_type,
+          requested_by_id, state, decided_by_employee_id, created_at, decided_at
+        ) values (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
+      `).run(
+        createId("app"),
+        input.orgId,
+        "task_risk",
+        input.title.trim(),
+        input.description.trim(),
+        RISK_LEVEL.HIGH,
+        "system",
+        null,
+        APPROVAL_STATE.PENDING,
+        null,
+        now,
+        null,
+      );
+    }
+    this.audit.record({
+      orgId: input.orgId,
+      entityType: "task",
+      entityId: id,
+      actorType: "system",
+      actorId: null,
+      action: "created",
+      payload: { status: TASK_STATUS.BACKLOG },
+    });
+    return id;
+  }
+
+  updateTaskStatus(orgId: string, taskId: string, status: string): void {
+    const result = this.db.prepare("update tasks set status = ?, updated_at = ? where id = ? and org_id = ?")
+      .run(status, new Date().toISOString(), taskId, orgId);
+    if (result.changes === 0) throw new Error("Task was not found.");
+    this.audit.record({
+      orgId,
+      entityType: "task",
+      entityId: taskId,
+      actorType: "system",
+      actorId: null,
+      action: "status_changed",
+      payload: { status },
+    });
+  }
+}