update

src/main/database/automation.ts

@@ -0,0 +1,133 @@
+import { randomUUID } from "node:crypto";
+import type { DatabaseSync } from "node:sqlite";
+import type { WorkflowPipelineRecord } from "@shared/types";
+import type { ReadAll } from "./helpers";
+import { ensureId, nowIso } from "./helpers";
+
+type SqliteRow = Record<string, any>;
+
+export function buildAutomationMethods(
+  getDb: () => DatabaseSync,
+  readAll: ReadAll,
+  deps: {
+    addActivity: (input: { companyId: string; actor: string; action: string; entityType: string; entityId: string; detail: string }) => void;
+  },
+) {
+  return {
+    saveAutomationRule(input: {
+      id?: string | null; companyId: string; name: string; description?: string;
+      trigger: string; conditionsJson?: string; action: string; actionConfigJson?: string;
+      sourceDepartment?: string | null; targetDepartment?: string | null;
+      priority?: number; status?: string;
+    }) {
+      const id = ensureId(input.id ?? null);
+      const ts = nowIso();
+      const existing = getDb().prepare("select id from automation_rules where id = ?").get(id) as { id: string } | undefined;
+      if (existing) {
+        getDb().prepare(
+          "update automation_rules set name = ?, description = ?, trigger = ?, conditions_json = ?, action = ?, action_config_json = ?, source_department = ?, target_department = ?, priority = ?, status = ?, updated_at = ? where id = ?",
+        ).run(input.name, input.description ?? "", input.trigger, input.conditionsJson ?? "{}", input.action, input.actionConfigJson ?? "{}", input.sourceDepartment ?? null, input.targetDepartment ?? null, input.priority ?? 50, input.status ?? "active", ts, id);
+      } else {
+        getDb().prepare(
+          "insert into automation_rules (id, company_id, name, description, trigger, conditions_json, action, action_config_json, source_department, target_department, priority, status, execution_count, created_at, updated_at) values (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, 0, ?, ?)",
+        ).run(id, input.companyId, input.name, input.description ?? "", input.trigger, input.conditionsJson ?? "{}", input.action, input.actionConfigJson ?? "{}", input.sourceDepartment ?? null, input.targetDepartment ?? null, input.priority ?? 50, input.status ?? "active", ts, ts);
+      }
+      deps.addActivity({ companyId: input.companyId, actor: "system", action: existing ? "updated" : "created", entityType: "automation_rule", entityId: id, detail: `${existing ? "Updated" : "Created"} automation rule: ${input.name}` });
+      return id;
+    },
+
+    deleteAutomationRule(id: string, companyId: string) {
+      getDb().prepare("delete from automation_rules where id = ? and company_id = ?").run(id, companyId);
+    },
+
+    listAutomationRuleNames(companyId: string): Set<string> {
+      return new Set(
+        readAll<{ name: string }>(
+          "select name from automation_rules where company_id = ?",
+          companyId,
+        ).map((row) => row.name),
+      );
+    },
+
+    toggleAutomationRule(id: string, companyId: string, status: "active" | "paused") {
+      getDb().prepare("update automation_rules set status = ?, updated_at = ? where id = ? and company_id = ?").run(status, nowIso(), id, companyId);
+    },
+
+    recordAutomationExecution(ruleId: string, companyId: string, ruleName: string, trigger: string, action: string, detail: string) {
+      const id = randomUUID();
+      getDb().prepare("insert into automation_log (id, company_id, rule_id, rule_name, trigger, action, detail, executed_at) values (?, ?, ?, ?, ?, ?, ?, ?)").run(id, companyId, ruleId, ruleName, trigger, action, detail, nowIso());
+      getDb().prepare("update automation_rules set execution_count = execution_count + 1, last_executed_at = ? where id = ?").run(nowIso(), ruleId);
+    },
+
+    getAutomationLog(companyId: string) {
+      return readAll<{ ruleId: string; ruleName: string; trigger: string; action: string; executedAt: string }>(
+        "select rule_id as ruleId, rule_name as ruleName, trigger, action, executed_at as executedAt from automation_log where company_id = ? order by executed_at desc limit 100",
+        companyId,
+      );
+    },
+
+    listActiveAutomationRules(companyId: string, trigger: string): Array<{
+      id: string; name: string; trigger: string; conditionsJson: string;
+      action: string; actionConfigJson: string; sourceDepartment: string | null;
+      targetDepartment: string | null; priority: number;
+    }> {
+      return readAll(
+        "select id, name, trigger, conditions_json as conditionsJson, action, action_config_json as actionConfigJson, source_department as sourceDepartment, target_department as targetDepartment, priority from automation_rules where company_id = ? and trigger = ? and status = 'active' order by priority desc",
+        companyId, trigger,
+      );
+    },
+
+    saveWorkflow(input: {
+      id?: string | null; companyId: string; name: string; description?: string;
+      stepsJson?: string; triggerType: string; triggerConfigJson?: string; status?: string;
+    }) {
+      const id = ensureId(input.id ?? null);
+      const ts = nowIso();
+      const existing = getDb().prepare("select id from workflow_pipelines where id = ?").get(id) as { id: string } | undefined;
+      if (existing) {
+        getDb().prepare(
+          "update workflow_pipelines set name = ?, description = ?, steps_json = ?, trigger_type = ?, trigger_config_json = ?, status = ?, updated_at = ? where id = ?",
+        ).run(input.name, input.description ?? "", input.stepsJson ?? "[]", input.triggerType, input.triggerConfigJson ?? "{}", input.status ?? "draft", ts, id);
+      } else {
+        getDb().prepare(
+          "insert into workflow_pipelines (id, company_id, name, description, steps_json, trigger_type, trigger_config_json, status, current_step_index, run_count, created_at, updated_at) values (?, ?, ?, ?, ?, ?, ?, ?, 0, 0, ?, ?)",
+        ).run(id, input.companyId, input.name, input.description ?? "", input.stepsJson ?? "[]", input.triggerType, input.triggerConfigJson ?? "{}", input.status ?? "draft", ts, ts);
+      }
+      deps.addActivity({ companyId: input.companyId, actor: "system", action: existing ? "updated" : "created", entityType: "workflow", entityId: id, detail: `${existing ? "Updated" : "Created"} workflow: ${input.name}` });
+      return id;
+    },
+
+    getWorkflowRecord(id: string, companyId: string): WorkflowPipelineRecord | null {
+      const row = getDb().prepare(
+        "select id, company_id as companyId, name, description, steps_json as stepsJson, trigger_type as triggerType, trigger_config_json as triggerConfigJson, status, current_step_index as currentStepIndex, run_count as runCount, last_run_at as lastRunAt, created_at as createdAt, updated_at as updatedAt from workflow_pipelines where id = ? and company_id = ?",
+      ).get(id, companyId) as WorkflowPipelineRecord | undefined;
+      return row ?? null;
+    },
+
+    updateWorkflowExecution(input: {
+      id: string; companyId: string; status: string;
+      currentStepIndex: number; stepsJson: string;
+      incrementRunCount?: boolean; lastRunAt?: string | null;
+    }) {
+      const ts = nowIso();
+      const runCountClause = input.incrementRunCount ? ", run_count = run_count + 1" : "";
+      const lastRunAt = input.lastRunAt ?? null;
+      getDb().prepare(
+        `update workflow_pipelines
+           set status = ?, current_step_index = ?, steps_json = ?, last_run_at = ?, updated_at = ?${runCountClause}
+         where id = ? and company_id = ?`,
+      ).run(input.status, input.currentStepIndex, input.stepsJson, lastRunAt, ts, input.id, input.companyId);
+    },
+
+    deleteWorkflow(id: string, companyId: string) {
+      getDb().prepare("delete from workflow_pipelines where id = ? and company_id = ?").run(id, companyId);
+    },
+
+    listTriggerableWorkflows(companyId: string, triggerType: string): WorkflowPipelineRecord[] {
+      return readAll<SqliteRow>(
+        "select id, company_id as companyId, name, description, steps_json as stepsJson, trigger_type as triggerType, trigger_config_json as triggerConfigJson, status, current_step_index as currentStepIndex, run_count as runCount, last_run_at as lastRunAt, created_at as createdAt, updated_at as updatedAt from workflow_pipelines where company_id = ? and trigger_type = ? and status = 'active' order by created_at asc",
+        companyId, triggerType,
+      ) as WorkflowPipelineRecord[];
+    },
+  };
+}