@agenticmail/enterprise 0.2.1

package/dist/postgres-LN7A6MGQ.js

@@ -0,0 +1,518 @@
+import {
+  getAllCreateStatements
+} from "./chunk-7RGCCHIT.js";
+import {
+  DatabaseAdapter
+} from "./chunk-77IDQJL3.js";
+import "./chunk-PNKVD2UK.js";
+
+// src/db/postgres.ts
+import { randomUUID, createHash } from "crypto";
+var pg;
+async function getPg() {
+  if (!pg) {
+    try {
+      pg = await import("pg");
+    } catch {
+      throw new Error(
+        "PostgreSQL driver not found. Install it: npm install pg\nFor Supabase/Neon/CockroachDB, the same pg driver works."
+      );
+    }
+  }
+  return pg;
+}
+var PostgresAdapter = class extends DatabaseAdapter {
+  type = "postgres";
+  pool = null;
+  async connect(config) {
+    const { Pool } = await getPg();
+    this.pool = new Pool({
+      connectionString: config.connectionString,
+      host: config.host,
+      port: config.port,
+      database: config.database,
+      user: config.username,
+      password: config.password,
+      ssl: config.ssl ? { rejectUnauthorized: false } : void 0,
+      max: 20,
+      idleTimeoutMillis: 3e4
+    });
+    const client = await this.pool.connect();
+    client.release();
+  }
+  async disconnect() {
+    if (this.pool) await this.pool.end();
+  }
+  isConnected() {
+    return this.pool !== null;
+  }
+  async migrate() {
+    const stmts = getAllCreateStatements();
+    const client = await this.pool.connect();
+    try {
+      await client.query("BEGIN");
+      for (const stmt of stmts) {
+        await client.query(stmt);
+      }
+      await client.query(`
+        INSERT INTO retention_policy (id) VALUES ('default')
+        ON CONFLICT (id) DO NOTHING
+      `);
+      await client.query("COMMIT");
+    } catch (err) {
+      await client.query("ROLLBACK");
+      throw err;
+    } finally {
+      client.release();
+    }
+  }
+  // ─── Company ─────────────────────────────────────────────
+  async getSettings() {
+    const { rows } = await this.pool.query(
+      "SELECT * FROM company_settings WHERE id = $1",
+      ["default"]
+    );
+    return rows[0] ? this.mapSettings(rows[0]) : null;
+  }
+  async updateSettings(updates) {
+    const fields = [];
+    const values = [];
+    let i = 1;
+    const map = {
+      name: "name",
+      domain: "domain",
+      subdomain: "subdomain",
+      smtpHost: "smtp_host",
+      smtpPort: "smtp_port",
+      smtpUser: "smtp_user",
+      smtpPass: "smtp_pass",
+      dkimPrivateKey: "dkim_private_key",
+      logoUrl: "logo_url",
+      primaryColor: "primary_color",
+      plan: "plan"
+    };
+    for (const [key, col] of Object.entries(map)) {
+      if (updates[key] !== void 0) {
+        fields.push(`${col} = $${i}`);
+        values.push(updates[key]);
+        i++;
+      }
+    }
+    fields.push(`updated_at = NOW()`);
+    values.push("default");
+    const { rows } = await this.pool.query(
+      `UPDATE company_settings SET ${fields.join(", ")} WHERE id = $${i} RETURNING *`,
+      values
+    );
+    return this.mapSettings(rows[0]);
+  }
+  // ─── Agents ──────────────────────────────────────────────
+  async createAgent(input) {
+    const id = randomUUID();
+    const email = input.email || `${input.name.toLowerCase().replace(/\s+/g, "-")}@localhost`;
+    const { rows } = await this.pool.query(
+      `INSERT INTO agents (id, name, email, role, metadata, created_by)
+       VALUES ($1, $2, $3, $4, $5, $6) RETURNING *`,
+      [id, input.name, email, input.role || "assistant", JSON.stringify(input.metadata || {}), input.createdBy]
+    );
+    return this.mapAgent(rows[0]);
+  }
+  async getAgent(id) {
+    const { rows } = await this.pool.query("SELECT * FROM agents WHERE id = $1", [id]);
+    return rows[0] ? this.mapAgent(rows[0]) : null;
+  }
+  async getAgentByName(name) {
+    const { rows } = await this.pool.query("SELECT * FROM agents WHERE name = $1", [name]);
+    return rows[0] ? this.mapAgent(rows[0]) : null;
+  }
+  async listAgents(opts) {
+    let q = "SELECT * FROM agents";
+    const params = [];
+    if (opts?.status) {
+      q += " WHERE status = $1";
+      params.push(opts.status);
+    }
+    q += " ORDER BY created_at DESC";
+    if (opts?.limit) {
+      q += ` LIMIT ${opts.limit}`;
+    }
+    if (opts?.offset) {
+      q += ` OFFSET ${opts.offset}`;
+    }
+    const { rows } = await this.pool.query(q, params);
+    return rows.map((r) => this.mapAgent(r));
+  }
+  async updateAgent(id, updates) {
+    const fields = [];
+    const values = [];
+    let i = 1;
+    for (const [key, col] of Object.entries({ name: "name", email: "email", role: "role", status: "status" })) {
+      if (updates[key] !== void 0) {
+        fields.push(`${col} = $${i}`);
+        values.push(updates[key]);
+        i++;
+      }
+    }
+    if (updates.metadata) {
+      fields.push(`metadata = $${i}`);
+      values.push(JSON.stringify(updates.metadata));
+      i++;
+    }
+    fields.push("updated_at = NOW()");
+    values.push(id);
+    const { rows } = await this.pool.query(
+      `UPDATE agents SET ${fields.join(", ")} WHERE id = $${i} RETURNING *`,
+      values
+    );
+    return this.mapAgent(rows[0]);
+  }
+  async archiveAgent(id) {
+    await this.pool.query("UPDATE agents SET status = 'archived', updated_at = NOW() WHERE id = $1", [id]);
+  }
+  async deleteAgent(id) {
+    await this.pool.query("DELETE FROM agents WHERE id = $1", [id]);
+  }
+  async countAgents(status) {
+    const q = status ? await this.pool.query("SELECT COUNT(*) FROM agents WHERE status = $1", [status]) : await this.pool.query("SELECT COUNT(*) FROM agents");
+    return parseInt(q.rows[0].count, 10);
+  }
+  // ─── Users ───────────────────────────────────────────────
+  async createUser(input) {
+    const id = randomUUID();
+    let passwordHash = null;
+    if (input.password) {
+      const { default: bcrypt } = await import("bcryptjs");
+      passwordHash = await bcrypt.hash(input.password, 12);
+    }
+    const { rows } = await this.pool.query(
+      `INSERT INTO users (id, email, name, role, password_hash, sso_provider, sso_subject)
+       VALUES ($1, $2, $3, $4, $5, $6, $7) RETURNING *`,
+      [id, input.email, input.name, input.role, passwordHash, input.ssoProvider || null, input.ssoSubject || null]
+    );
+    return this.mapUser(rows[0]);
+  }
+  async getUser(id) {
+    const { rows } = await this.pool.query("SELECT * FROM users WHERE id = $1", [id]);
+    return rows[0] ? this.mapUser(rows[0]) : null;
+  }
+  async getUserByEmail(email) {
+    const { rows } = await this.pool.query("SELECT * FROM users WHERE email = $1", [email]);
+    return rows[0] ? this.mapUser(rows[0]) : null;
+  }
+  async getUserBySso(provider, subject) {
+    const { rows } = await this.pool.query(
+      "SELECT * FROM users WHERE sso_provider = $1 AND sso_subject = $2",
+      [provider, subject]
+    );
+    return rows[0] ? this.mapUser(rows[0]) : null;
+  }
+  async listUsers(opts) {
+    let q = "SELECT * FROM users ORDER BY created_at DESC";
+    if (opts?.limit) q += ` LIMIT ${opts.limit}`;
+    if (opts?.offset) q += ` OFFSET ${opts.offset}`;
+    const { rows } = await this.pool.query(q);
+    return rows.map((r) => this.mapUser(r));
+  }
+  async updateUser(id, updates) {
+    const fields = [];
+    const values = [];
+    let i = 1;
+    for (const key of ["email", "name", "role", "sso_provider", "sso_subject"]) {
+      const camelKey = key.replace(/_([a-z])/g, (_, c) => c.toUpperCase());
+      if (updates[camelKey] !== void 0) {
+        fields.push(`${key} = $${i}`);
+        values.push(updates[camelKey]);
+        i++;
+      }
+    }
+    fields.push("updated_at = NOW()");
+    values.push(id);
+    const { rows } = await this.pool.query(
+      `UPDATE users SET ${fields.join(", ")} WHERE id = $${i} RETURNING *`,
+      values
+    );
+    return this.mapUser(rows[0]);
+  }
+  async deleteUser(id) {
+    await this.pool.query("DELETE FROM users WHERE id = $1", [id]);
+  }
+  // ─── Audit ───────────────────────────────────────────────
+  async logEvent(event) {
+    await this.pool.query(
+      `INSERT INTO audit_log (id, actor, actor_type, action, resource, details, ip)
+       VALUES ($1, $2, $3, $4, $5, $6, $7)`,
+      [
+        randomUUID(),
+        event.actor,
+        event.actorType,
+        event.action,
+        event.resource,
+        JSON.stringify(event.details || {}),
+        event.ip || null
+      ]
+    );
+  }
+  async queryAudit(filters) {
+    const where = [];
+    const params = [];
+    let i = 1;
+    if (filters.actor) {
+      where.push(`actor = $${i++}`);
+      params.push(filters.actor);
+    }
+    if (filters.action) {
+      where.push(`action = $${i++}`);
+      params.push(filters.action);
+    }
+    if (filters.resource) {
+      where.push(`resource LIKE $${i++}`);
+      params.push(`%${filters.resource}%`);
+    }
+    if (filters.from) {
+      where.push(`timestamp >= $${i++}`);
+      params.push(filters.from);
+    }
+    if (filters.to) {
+      where.push(`timestamp <= $${i++}`);
+      params.push(filters.to);
+    }
+    const whereClause = where.length > 0 ? `WHERE ${where.join(" AND ")}` : "";
+    const countResult = await this.pool.query(`SELECT COUNT(*) FROM audit_log ${whereClause}`, params);
+    const total = parseInt(countResult.rows[0].count, 10);
+    let q = `SELECT * FROM audit_log ${whereClause} ORDER BY timestamp DESC`;
+    if (filters.limit) q += ` LIMIT ${filters.limit}`;
+    if (filters.offset) q += ` OFFSET ${filters.offset}`;
+    const { rows } = await this.pool.query(q, params);
+    return {
+      events: rows.map((r) => ({
+        id: r.id,
+        timestamp: r.timestamp,
+        actor: r.actor,
+        actorType: r.actor_type,
+        action: r.action,
+        resource: r.resource,
+        details: JSON.parse(r.details || "{}"),
+        ip: r.ip
+      })),
+      total
+    };
+  }
+  // ─── API Keys ────────────────────────────────────────────
+  async createApiKey(input) {
+    const id = randomUUID();
+    const plaintext = `ek_${randomUUID().replace(/-/g, "")}`;
+    const keyHash = createHash("sha256").update(plaintext).digest("hex");
+    const keyPrefix = plaintext.substring(0, 11);
+    const { rows } = await this.pool.query(
+      `INSERT INTO api_keys (id, name, key_hash, key_prefix, scopes, created_by, expires_at)
+       VALUES ($1, $2, $3, $4, $5, $6, $7) RETURNING *`,
+      [id, input.name, keyHash, keyPrefix, JSON.stringify(input.scopes), input.createdBy, input.expiresAt || null]
+    );
+    return { key: this.mapApiKey(rows[0]), plaintext };
+  }
+  async getApiKey(id) {
+    const { rows } = await this.pool.query("SELECT * FROM api_keys WHERE id = $1", [id]);
+    return rows[0] ? this.mapApiKey(rows[0]) : null;
+  }
+  async validateApiKey(plaintext) {
+    const keyHash = createHash("sha256").update(plaintext).digest("hex");
+    const { rows } = await this.pool.query(
+      "SELECT * FROM api_keys WHERE key_hash = $1 AND revoked = 0",
+      [keyHash]
+    );
+    if (!rows[0]) return null;
+    const key = this.mapApiKey(rows[0]);
+    if (key.expiresAt && /* @__PURE__ */ new Date() > key.expiresAt) return null;
+    await this.pool.query("UPDATE api_keys SET last_used_at = NOW() WHERE id = $1", [key.id]);
+    return key;
+  }
+  async listApiKeys(opts) {
+    let q = "SELECT * FROM api_keys";
+    const params = [];
+    if (opts?.createdBy) {
+      q += " WHERE created_by = $1";
+      params.push(opts.createdBy);
+    }
+    q += " ORDER BY created_at DESC";
+    const { rows } = await this.pool.query(q, params);
+    return rows.map((r) => this.mapApiKey(r));
+  }
+  async revokeApiKey(id) {
+    await this.pool.query("UPDATE api_keys SET revoked = 1 WHERE id = $1", [id]);
+  }
+  // ─── Rules ───────────────────────────────────────────────
+  async createRule(rule) {
+    const id = randomUUID();
+    const { rows } = await this.pool.query(
+      `INSERT INTO email_rules (id, name, agent_id, conditions, actions, priority, enabled)
+       VALUES ($1, $2, $3, $4, $5, $6, $7) RETURNING *`,
+      [
+        id,
+        rule.name,
+        rule.agentId || null,
+        JSON.stringify(rule.conditions),
+        JSON.stringify(rule.actions),
+        rule.priority,
+        rule.enabled ? 1 : 0
+      ]
+    );
+    return this.mapRule(rows[0]);
+  }
+  async getRules(agentId) {
+    let q = "SELECT * FROM email_rules";
+    const params = [];
+    if (agentId) {
+      q += " WHERE agent_id = $1 OR agent_id IS NULL";
+      params.push(agentId);
+    }
+    q += " ORDER BY priority DESC";
+    const { rows } = await this.pool.query(q, params);
+    return rows.map((r) => this.mapRule(r));
+  }
+  async updateRule(id, updates) {
+    const fields = [];
+    const values = [];
+    let i = 1;
+    if (updates.name !== void 0) {
+      fields.push(`name = $${i++}`);
+      values.push(updates.name);
+    }
+    if (updates.conditions) {
+      fields.push(`conditions = $${i++}`);
+      values.push(JSON.stringify(updates.conditions));
+    }
+    if (updates.actions) {
+      fields.push(`actions = $${i++}`);
+      values.push(JSON.stringify(updates.actions));
+    }
+    if (updates.priority !== void 0) {
+      fields.push(`priority = $${i++}`);
+      values.push(updates.priority);
+    }
+    if (updates.enabled !== void 0) {
+      fields.push(`enabled = $${i++}`);
+      values.push(updates.enabled ? 1 : 0);
+    }
+    fields.push("updated_at = NOW()");
+    values.push(id);
+    const { rows } = await this.pool.query(
+      `UPDATE email_rules SET ${fields.join(", ")} WHERE id = $${i} RETURNING *`,
+      values
+    );
+    return this.mapRule(rows[0]);
+  }
+  async deleteRule(id) {
+    await this.pool.query("DELETE FROM email_rules WHERE id = $1", [id]);
+  }
+  // ─── Retention ───────────────────────────────────────────
+  async getRetentionPolicy() {
+    const { rows } = await this.pool.query("SELECT * FROM retention_policy WHERE id = $1", ["default"]);
+    if (!rows[0]) return { enabled: false, retainDays: 365, archiveFirst: true };
+    return {
+      enabled: !!rows[0].enabled,
+      retainDays: rows[0].retain_days,
+      excludeTags: JSON.parse(rows[0].exclude_tags || "[]"),
+      archiveFirst: !!rows[0].archive_first
+    };
+  }
+  async setRetentionPolicy(policy) {
+    await this.pool.query(
+      `UPDATE retention_policy SET enabled = $1, retain_days = $2, exclude_tags = $3, archive_first = $4
+       WHERE id = 'default'`,
+      [policy.enabled ? 1 : 0, policy.retainDays, JSON.stringify(policy.excludeTags || []), policy.archiveFirst ? 1 : 0]
+    );
+  }
+  // ─── Stats ───────────────────────────────────────────────
+  async getStats() {
+    const [agents, active, users, audit] = await Promise.all([
+      this.pool.query("SELECT COUNT(*) FROM agents"),
+      this.pool.query("SELECT COUNT(*) FROM agents WHERE status = 'active'"),
+      this.pool.query("SELECT COUNT(*) FROM users"),
+      this.pool.query("SELECT COUNT(*) FROM audit_log")
+    ]);
+    return {
+      totalAgents: parseInt(agents.rows[0].count, 10),
+      activeAgents: parseInt(active.rows[0].count, 10),
+      totalUsers: parseInt(users.rows[0].count, 10),
+      totalEmails: 0,
+      // TODO: wire to email storage
+      totalAuditEvents: parseInt(audit.rows[0].count, 10)
+    };
+  }
+  // ─── Mappers ─────────────────────────────────────────────
+  mapAgent(r) {
+    return {
+      id: r.id,
+      name: r.name,
+      email: r.email,
+      role: r.role,
+      status: r.status,
+      metadata: typeof r.metadata === "string" ? JSON.parse(r.metadata) : r.metadata,
+      createdAt: new Date(r.created_at),
+      updatedAt: new Date(r.updated_at),
+      createdBy: r.created_by
+    };
+  }
+  mapUser(r) {
+    return {
+      id: r.id,
+      email: r.email,
+      name: r.name,
+      role: r.role,
+      passwordHash: r.password_hash,
+      ssoProvider: r.sso_provider,
+      ssoSubject: r.sso_subject,
+      createdAt: new Date(r.created_at),
+      updatedAt: new Date(r.updated_at),
+      lastLoginAt: r.last_login_at ? new Date(r.last_login_at) : void 0
+    };
+  }
+  mapApiKey(r) {
+    return {
+      id: r.id,
+      name: r.name,
+      keyHash: r.key_hash,
+      keyPrefix: r.key_prefix,
+      scopes: typeof r.scopes === "string" ? JSON.parse(r.scopes) : r.scopes,
+      createdBy: r.created_by,
+      createdAt: new Date(r.created_at),
+      lastUsedAt: r.last_used_at ? new Date(r.last_used_at) : void 0,
+      expiresAt: r.expires_at ? new Date(r.expires_at) : void 0,
+      revoked: !!r.revoked
+    };
+  }
+  mapRule(r) {
+    return {
+      id: r.id,
+      name: r.name,
+      agentId: r.agent_id,
+      conditions: typeof r.conditions === "string" ? JSON.parse(r.conditions) : r.conditions,
+      actions: typeof r.actions === "string" ? JSON.parse(r.actions) : r.actions,
+      priority: r.priority,
+      enabled: !!r.enabled,
+      createdAt: new Date(r.created_at),
+      updatedAt: new Date(r.updated_at)
+    };
+  }
+  mapSettings(r) {
+    return {
+      id: r.id,
+      name: r.name,
+      domain: r.domain,
+      subdomain: r.subdomain,
+      smtpHost: r.smtp_host,
+      smtpPort: r.smtp_port,
+      smtpUser: r.smtp_user,
+      smtpPass: r.smtp_pass,
+      dkimPrivateKey: r.dkim_private_key,
+      logoUrl: r.logo_url,
+      primaryColor: r.primary_color,
+      plan: r.plan,
+      createdAt: new Date(r.created_at),
+      updatedAt: new Date(r.updated_at)
+    };
+  }
+};
+export {
+  PostgresAdapter
+};