aios-management-web 0.1.0

package/server/src/services/agent-service.js

@@ -0,0 +1,696 @@
+import { jsonParse, jsonStringify } from "../db/index.js";
+import { badRequest, conflict, notFound } from "../utils/errors.js";
+import {
+  findQuotaViolation,
+  isAgentStatus,
+  normalizeAgentStatus,
+  normalizeUsage
+} from "./agent-quota.js";
+
+const SLUG_PATTERN = /^[a-z0-9]+(?:-[a-z0-9]+)*$/;
+
+function normalizeStringList(value) {
+  if (Array.isArray(value)) {
+    return value
+      .map((item) => String(item || "").trim())
+      .filter(Boolean);
+  }
+
+  if (typeof value === "string") {
+    return value
+      .split(",")
+      .map((item) => item.trim())
+      .filter(Boolean);
+  }
+
+  return [];
+}
+
+function resolvePermissionUsernames(payload) {
+  return [...new Set(normalizeStringList(payload?.permission_usernames ?? payload?.users))];
+}
+
+function hasOwnValue(payload, key) {
+  return Boolean(payload) && Object.prototype.hasOwnProperty.call(payload, key);
+}
+
+function hasModelConfig(payload) {
+  return Boolean(
+    payload?.model_primary
+    || (Array.isArray(payload?.model_fallbacks) && payload.model_fallbacks.length > 0)
+  );
+}
+
+function firstText(...values) {
+  for (const value of values) {
+    if (typeof value === "string" && value.trim()) {
+      return value.trim();
+    }
+  }
+
+  return "";
+}
+
+function renderTopic(template, agentId) {
+  return String(template || "").replaceAll("{agentId}", agentId);
+}
+
+function validateSlug(value, label = "ID") {
+  const normalized = String(value || "").trim();
+  if (!normalized) {
+    throw badRequest(`${label}不能为空`);
+  }
+  if (!SLUG_PATTERN.test(normalized)) {
+    throw badRequest(`${label}需符合 slug 规则：仅允许小写字母、数字和中划线，且不能以中划线开头或结尾`);
+  }
+  return normalized;
+}
+
+function normalizeRemoteAgentStatus(value) {
+  return normalizeAgentStatus(value);
+}
+
+function isRemoteAgentMissingError(error) {
+  const message = String(error?.message || error?.details?.message || "").toLowerCase();
+  const code = String(error?.code || "").toLowerCase();
+  const status = Number(error?.status || 0);
+
+  return (
+    code === "not_found"
+    || status === 404
+    || message.includes("数字员工不存在")
+    || message.includes("agent not found")
+    || (message.includes("agent") && message.includes("不存在"))
+  );
+}
+
+function computeHealth(agent, usage) {
+  if (agent.status === "disabled") {
+    return "disabled";
+  }
+  if (findQuotaViolation(agent, usage)) {
+    return "overlimit";
+  }
+  return "normal";
+}
+
+function mergeRemoteStatus(currentStatus, remoteStatus) {
+  if (remoteStatus === "disabled") {
+    return "disabled";
+  }
+  if (currentStatus === "overlimit") {
+    return "overlimit";
+  }
+  return "normal";
+}
+
+function runtimeActionForStatusChange(currentStatus, nextStatus) {
+  if (currentStatus === "disabled" && nextStatus !== "disabled") {
+    return "agent.enable";
+  }
+  if (currentStatus !== "disabled" && nextStatus === "disabled") {
+    return "agent.disable";
+  }
+  return null;
+}
+
+function formatQuotaViolationMessage(agent, violation) {
+  return `Cannot set agent ${agent.slug} to normal: ${violation.period} quota exceeded (limit ${violation.limit}, current ${violation.current}). Resolve the quota overage before setting status to normal, or set status to disabled.`;
+}
+
+function runInTransaction(database, fn) {
+  if (typeof database?.exec !== "function") {
+    return fn();
+  }
+
+  database.exec("BEGIN");
+  try {
+    const result = fn();
+    database.exec("COMMIT");
+    return result;
+  } catch (error) {
+    try {
+      database.exec("ROLLBACK");
+    } catch {}
+    throw error;
+  }
+}
+
+export class AgentService {
+  constructor({ db, rpcClient, objectStorage, env }) {
+    this.db = db;
+    this.rpcClient = rpcClient;
+    this.objectStorage = objectStorage;
+    this.env = env;
+  }
+
+  localRows() {
+    return this.db.prepare("SELECT * FROM agents ORDER BY updated_at DESC").all();
+  }
+
+  getAgentRowById(agentId) {
+    return this.db.prepare("SELECT * FROM agents WHERE id = ?").get(agentId);
+  }
+
+  async refreshAgentFromRemote(row) {
+    if (!row || typeof this.rpcClient?.call !== "function" || typeof this.rpcClient?.isConfigured !== "function" || !this.rpcClient.isConfigured()) {
+      return false;
+    }
+
+    let remoteItem = null;
+    try {
+      remoteItem = await this.rpcClient.call("agent.get", { agentId: row.slug }, 15000);
+    } catch {
+      return false;
+    }
+
+    if (!remoteItem) {
+      return false;
+    }
+
+    const nextAgentName = firstText(remoteItem.name, row.agent_name, row.slug);
+    const nextStatus = mergeRemoteStatus(row.status, normalizeRemoteAgentStatus(remoteItem.status));
+    const nextRemoteStateJson = JSON.stringify(remoteItem ?? {});
+    const changed =
+      nextAgentName !== row.agent_name
+      || nextStatus !== row.status
+      || nextRemoteStateJson !== row.remote_state_json;
+
+    if (!changed) {
+      return false;
+    }
+
+    this.db.prepare(`
+      UPDATE agents
+      SET agent_name = ?, status = ?, remote_state_json = ?, updated_at = ?
+      WHERE id = ?
+    `).run(nextAgentName, nextStatus, nextRemoteStateJson, new Date().toISOString(), row.id);
+    return true;
+  }
+
+  getPermissions(agentId) {
+    return this.db.prepare(`
+      SELECT d.id, d.username
+      FROM agent_permissions ap
+      JOIN aios_users d ON d.id = ap.aios_user_id
+      WHERE ap.agent_id = ?
+      ORDER BY d.username
+    `).all(agentId);
+  }
+
+  getSkillSlugs(agentId) {
+    return this.db.prepare(`
+      SELECT s.slug
+      FROM agent_skill_bindings b
+      JOIN skills s ON s.id = b.skill_id
+      WHERE b.agent_id = ?
+      ORDER BY s.slug
+    `).all(agentId).map((item) => item.slug);
+  }
+
+  async listAgents() {
+    const rows = this.localRows();
+    return rows.map((row) => {
+      const usage = normalizeUsage(jsonParse(row.usage_snapshot_json, {}).usage);
+      const remote = jsonParse(row.remote_state_json, {});
+      const permissions = this.getPermissions(row.id);
+      const skills = this.getSkillSlugs(row.id);
+      return {
+        ...row,
+        tags: jsonParse(row.tags_json),
+        usage,
+        permissions,
+        skills,
+        remote_state: remote,
+        health: computeHealth(row, usage)
+      };
+    });
+  }
+
+  async getAgentBySlug(slug) {
+    const normalizedSlug = String(slug || "").trim();
+    if (!normalizedSlug) {
+      throw notFound("数字员工不存在");
+    }
+
+    const current = this.localRows().find((item) => item.slug === normalizedSlug);
+    if (!current) {
+      throw notFound("数字员工不存在");
+    }
+
+    await this.refreshAgentFromRemote(current);
+
+    const agent = (await this.listAgents()).find((item) => item.slug === normalizedSlug);
+    if (!agent) {
+      throw notFound("数字员工不存在");
+    }
+
+    return agent;
+  }
+
+  async listActiveAgentDirectory() {
+    const rows = this.db.prepare(`
+      SELECT * FROM agents
+      WHERE status = 'normal'
+      ORDER BY slug
+    `).all();
+
+    return rows.map((row) => {
+      const remote = jsonParse(row.remote_state_json, {});
+      const permissions = this.getPermissions(row.id);
+      return {
+        agent_id: row.slug,
+        agent_name: firstText(row.agent_name, row.slug),
+        status: row.status,
+        inbound_topic: firstText(remote?.inboundTopic, remote?.["inbound-topic"]),
+        outbound_topic: firstText(remote?.outboundTopic, remote?.["outbound-topic"]),
+        users: permissions.map((item) => item.username)
+      };
+    });
+  }
+
+  async listAgentDirectoryForUser(username) {
+    const normalizedUsername = String(username || "").trim();
+    const rows = this.db.prepare(`
+      SELECT ag.*
+      FROM agents ag
+      JOIN agent_permissions ap ON ap.agent_id = ag.id
+      JOIN aios_users d ON d.id = ap.aios_user_id
+      WHERE d.username = ?
+      ORDER BY ag.slug
+    `).all(normalizedUsername);
+
+    return rows.map((row) => {
+      const remote = jsonParse(row.remote_state_json, {});
+      return {
+        agent_id: row.slug,
+        agent_name: firstText(row.agent_name, row.slug),
+        status: row.status,
+        inbound_topic: firstText(remote?.inboundTopic, remote?.["inbound-topic"]),
+        outbound_topic: firstText(remote?.outboundTopic, remote?.["outbound-topic"]),
+        users: [normalizedUsername]
+      };
+    });
+  }
+
+  buildLocalRemoteState(slug, agentName) {
+    return {
+      agentId: slug,
+      name: agentName,
+      inboundTopic: renderTopic(this.env?.mqtt?.agentInboundTopicTemplate, slug),
+      outboundTopic: renderTopic(this.env?.mqtt?.agentOutboundTopicTemplate, slug)
+    };
+  }
+
+  replacePermissions(agentId, usernames) {
+    this.db.prepare("DELETE FROM agent_permissions WHERE agent_id = ?").run(agentId);
+    this.ensureAiosUsers(usernames);
+    const insert = this.db.prepare(`
+      INSERT INTO agent_permissions (agent_id, aios_user_id)
+      VALUES (?, ?)
+    `);
+    for (const username of usernames) {
+      const aiosUser = this.db.prepare(`
+        SELECT id
+        FROM aios_users
+        WHERE username = ?
+      `).get(username);
+      if (aiosUser) {
+        insert.run(agentId, aiosUser.id);
+      }
+    }
+  }
+
+  ensureAiosUsers(usernames) {
+    const insert = this.db.prepare(`
+      INSERT INTO aios_users (username, created_at, updated_at)
+      VALUES (?, ?, ?)
+      ON CONFLICT(username) DO UPDATE SET
+        updated_at = excluded.updated_at
+    `);
+    const now = new Date().toISOString();
+    for (const username of usernames) {
+      insert.run(username, now, now);
+    }
+  }
+
+  listAiosUsers(query = "", limit = 20) {
+    const normalizedQuery = String(query || "").trim();
+    const safeLimit = Math.max(1, Math.min(50, Number(limit) || 20));
+    if (!normalizedQuery) {
+      return this.db.prepare(`
+        SELECT id, username
+        FROM aios_users
+        ORDER BY updated_at DESC, username ASC
+        LIMIT ?
+      `).all(safeLimit);
+    }
+
+    return this.db.prepare(`
+      SELECT id, username
+      FROM aios_users
+      WHERE username LIKE ?
+      ORDER BY
+        CASE WHEN username = ? THEN 0 ELSE 1 END,
+        CASE WHEN username LIKE ? THEN 0 ELSE 1 END,
+        updated_at DESC,
+        username ASC
+      LIMIT ?
+    `).all(`%${normalizedQuery}%`, normalizedQuery, `${normalizedQuery}%`, safeLimit);
+  }
+
+  getAiosUserUsageCount(aiosUserId) {
+    const row = this.db.prepare(`
+      SELECT COUNT(*) AS count
+      FROM agent_permissions
+      WHERE aios_user_id = ?
+    `).get(aiosUserId);
+    return Number(row?.count || 0);
+  }
+
+  getAiosUserAssignedAgents(aiosUserId) {
+    return this.db.prepare(`
+      SELECT ag.id, ag.slug, ag.agent_name
+      FROM agent_permissions ap
+      JOIN agents ag ON ag.id = ap.agent_id
+      WHERE ap.aios_user_id = ?
+      ORDER BY ag.slug
+    `).all(aiosUserId).map((item) => ({
+      id: item.id,
+      slug: item.slug,
+      agent_name: item.agent_name
+    }));
+  }
+
+  countAiosUsers(query = "") {
+    const normalizedQuery = String(query || "").trim();
+    if (!normalizedQuery) {
+      return Number(this.db.prepare(`
+        SELECT COUNT(*) AS count
+        FROM aios_users
+      `).get()?.count || 0);
+    }
+
+    return Number(this.db.prepare(`
+      SELECT COUNT(*) AS count
+      FROM aios_users
+      WHERE username LIKE ?
+    `).get(`%${normalizedQuery}%`)?.count || 0);
+  }
+
+  listAiosUsersWithUsage(query = "", limit = 50, offset = 0) {
+    const normalizedQuery = String(query || "").trim();
+    const safeLimit = Math.max(1, Math.min(200, Number(limit) || 50));
+    const safeOffset = Math.max(0, Number(offset) || 0);
+    const rows = !normalizedQuery
+      ? this.db.prepare(`
+        SELECT id, username
+        FROM aios_users
+        ORDER BY updated_at DESC, username ASC
+        LIMIT ? OFFSET ?
+      `).all(safeLimit, safeOffset)
+      : this.db.prepare(`
+        SELECT id, username
+        FROM aios_users
+        WHERE username LIKE ?
+        ORDER BY
+          CASE WHEN username = ? THEN 0 ELSE 1 END,
+          CASE WHEN username LIKE ? THEN 0 ELSE 1 END,
+          updated_at DESC,
+          username ASC
+        LIMIT ? OFFSET ?
+      `).all(`%${normalizedQuery}%`, normalizedQuery, `${normalizedQuery}%`, safeLimit, safeOffset);
+
+    return rows.map((item) => ({
+      ...item,
+      assigned_agents: this.getAiosUserUsageCount(item.id),
+      assigned_agent_items: this.getAiosUserAssignedAgents(item.id)
+    }));
+  }
+
+  createAiosUser(username) {
+    const normalizedUsername = String(username || "").trim();
+    if (!normalizedUsername) {
+      throw badRequest("用户名不能为空");
+    }
+
+    this.ensureAiosUsers([normalizedUsername]);
+    const row = this.db.prepare(`
+      SELECT id, username
+      FROM aios_users
+      WHERE username = ?
+    `).get(normalizedUsername);
+
+    return {
+      ...row,
+      assigned_agents: row ? this.getAiosUserUsageCount(row.id) : 0,
+      assigned_agent_items: row ? this.getAiosUserAssignedAgents(row.id) : []
+    };
+  }
+
+  importAiosUsers(content) {
+    const usernames = [...new Set(
+      String(content || "")
+        .split(/\r?\n/)
+        .map((item) => item.trim())
+        .filter(Boolean)
+    )];
+
+    if (usernames.length === 0) {
+      throw badRequest("请至少输入一个用户名");
+    }
+
+    const existing = new Set(this.listAiosUsers("", 100000).map((item) => item.username));
+    this.ensureAiosUsers(usernames);
+
+    return {
+      total: usernames.length,
+      created: usernames.filter((username) => !existing.has(username)).length,
+      items: usernames.map((username) => ({
+        username,
+        existed: existing.has(username)
+      }))
+    };
+  }
+
+  deleteAiosUser(aiosUserId) {
+    const current = this.db.prepare(`
+      SELECT id, username
+      FROM aios_users
+      WHERE id = ?
+    `).get(aiosUserId);
+    if (!current) {
+      throw notFound("AIOS 用户不存在");
+    }
+
+    const usageCount = this.getAiosUserUsageCount(aiosUserId);
+    if (usageCount > 0) {
+      throw conflict("该用户名仍已分配给数字员工，无法删除");
+    }
+
+    this.db.prepare("DELETE FROM aios_users WHERE id = ?").run(aiosUserId);
+    return { ok: true, username: current.username };
+  }
+
+  replaceSkills(agentId, skillSlugs) {
+    this.db.prepare("DELETE FROM agent_skill_bindings WHERE agent_id = ?").run(agentId);
+    const insert = this.db.prepare(`
+      INSERT INTO agent_skill_bindings (agent_id, skill_id)
+      VALUES (?, ?)
+    `);
+    for (const slug of skillSlugs) {
+      const skill = this.db.prepare("SELECT id FROM skills WHERE slug = ?").get(slug);
+      if (skill) {
+        insert.run(agentId, skill.id);
+      }
+    }
+  }
+
+  async createAgent(payload) {
+    const slug = validateSlug(payload.slug || payload.id || "", "数字员工 ID");
+    const agentName = firstText(payload.agent_name, payload.name, slug);
+    if (hasModelConfig(payload)) {
+      throw badRequest("当前版本不支持在创建时配置模型");
+    }
+
+    const exists = this.db.prepare("SELECT 1 FROM agents WHERE slug = ?").get(slug);
+    if (exists) {
+      throw conflict("数字员工 ID 已存在");
+    }
+
+    const templateName = String(payload.template_name || payload.template || "default").trim() || "default";
+    await this.rpcClient.call("agent.create", {
+      agentId: slug,
+      name: agentName,
+      agentName,
+      templateName,
+      restart: payload.restart !== false
+    });
+
+    if (payload.docs_content) {
+      try {
+        await this.rpcClient.call("agent.docs.update", {
+          agentId: slug,
+          content: payload.docs_content
+        });
+      } catch (error) {
+        try {
+          await this.rpcClient.call("agent.delete", {
+            agentId: slug,
+            restart: true
+          });
+        } catch {}
+        throw error;
+      }
+    }
+
+    let agentId;
+    try {
+      agentId = runInTransaction(this.db, () => {
+        const now = new Date().toISOString();
+        const remoteState = this.buildLocalRemoteState(slug, agentName);
+        const result = this.db.prepare(`
+          INSERT INTO agents (
+            slug, agent_name, description, docs_content, template_name, status, tags_json,
+            daily_limit, remote_state_json, created_at, updated_at
+          ) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
+        `).run(
+          slug,
+          agentName,
+          payload.description || "",
+          payload.docs_content || "",
+          templateName,
+          "normal",
+          jsonStringify(payload.tags),
+          Number(payload.daily_limit ?? -1),
+          JSON.stringify(remoteState ?? {}),
+          now,
+          now
+        );
+
+        const insertedAgentId = result.lastInsertRowid;
+        if (hasOwnValue(payload, "permission_usernames") || hasOwnValue(payload, "users")) {
+          this.replacePermissions(insertedAgentId, resolvePermissionUsernames(payload));
+        }
+
+        if (hasOwnValue(payload, "skill_slugs")) {
+          this.replaceSkills(insertedAgentId, payload.skill_slugs || []);
+        }
+
+        return insertedAgentId;
+      });
+    } catch (error) {
+      try {
+        await this.rpcClient.call("agent.delete", {
+          agentId: slug,
+          restart: true
+        });
+      } catch {}
+      throw error;
+    }
+
+    return (await this.listAgents()).find((item) => item.id === agentId);
+  }
+
+  async listActiveAgentDirectoryForUser(username) {
+    const items = await this.listAgentDirectoryForUser(username);
+    return items.filter((item) => item.status === "normal");
+  }
+
+  async updateAgent(agentId, payload) {
+    const current = this.db.prepare("SELECT * FROM agents WHERE id = ?").get(agentId);
+    if (!current) {
+      throw notFound("数字员工不存在");
+    }
+    if (hasModelConfig(payload)) {
+      throw badRequest("当前版本不支持在编辑时配置模型");
+    }
+
+    const nextStatus = payload.status || current.status;
+    if (!isAgentStatus(nextStatus)) {
+      throw badRequest("状态只能是 normal、disabled 或 overlimit");
+    }
+    const nextQuotaFields = {
+      ...current,
+      daily_limit: Number(payload.daily_limit ?? current.daily_limit)
+    };
+    if (nextStatus === "normal") {
+      const usage = normalizeUsage(jsonParse(current.usage_snapshot_json, {}).usage);
+      const violation = findQuotaViolation(nextQuotaFields, usage);
+      if (violation) {
+        throw badRequest(formatQuotaViolationMessage(current, violation), {
+          agentId: current.slug,
+          requestedStatus: nextStatus,
+          quota: {
+            period: violation.period,
+            limit: violation.limit,
+            current: violation.current
+          }
+        });
+      }
+    }
+
+    const nextAgentName = firstText(payload.agent_name, payload.name, current.agent_name, current.slug);
+    const runtimeAction = runtimeActionForStatusChange(current.status, nextStatus);
+    if (runtimeAction) {
+      await this.rpcClient.call(runtimeAction, {
+        agentId: current.slug,
+        restart: payload.restart !== false
+      });
+    }
+
+    if (payload.docs_content !== undefined && payload.docs_content !== current.docs_content) {
+      await this.rpcClient.call("agent.docs.update", {
+        agentId: current.slug,
+        content: payload.docs_content
+      });
+    }
+
+    this.db.prepare(`
+      UPDATE agents
+      SET agent_name = ?, description = ?, docs_content = ?, status = ?, tags_json = ?,
+          daily_limit = ?, updated_at = ?
+      WHERE id = ?
+    `).run(
+      nextAgentName,
+      payload.description ?? current.description,
+      payload.docs_content ?? current.docs_content,
+      nextStatus,
+      jsonStringify(payload.tags ?? jsonParse(current.tags_json)),
+      Number(payload.daily_limit ?? current.daily_limit),
+      new Date().toISOString(),
+      agentId
+    );
+
+    if (hasOwnValue(payload, "permission_usernames") || hasOwnValue(payload, "users")) {
+      this.replacePermissions(agentId, resolvePermissionUsernames(payload));
+    }
+
+    if (hasOwnValue(payload, "skill_slugs")) {
+      this.replaceSkills(agentId, payload.skill_slugs || []);
+    }
+    return (await this.listAgents()).find((item) => item.id === agentId);
+  }
+
+  async deleteAgent(agentId) {
+    const current = this.db.prepare("SELECT * FROM agents WHERE id = ?").get(agentId);
+    if (!current) {
+      throw notFound("数字员工不存在");
+    }
+
+    try {
+      await this.rpcClient.call("agent.delete", {
+        agentId: current.slug,
+        restart: true
+      });
+    } catch (error) {
+      if (!isRemoteAgentMissingError(error)) {
+        throw error;
+      }
+    }
+
+    this.db.prepare("DELETE FROM agents WHERE id = ?").run(agentId);
+    return { ok: true };
+  }
+
+}