aios-management-web 0.1.0

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

@@ -0,0 +1,580 @@
+import AdmZip from "adm-zip";
+
+import { jsonParse } from "../db/index.js";
+import { badRequest, conflict, notFound } from "../utils/errors.js";
+
+const APPLICATION_NAME_PATTERN = /^[a-z0-9]+(?:-[a-z0-9]+)*$/;
+const VALID_SCHEMES = new Set(["http", "https"]);
+
+function normalizeText(value) {
+  return String(value || "").trim();
+}
+
+function normalizePort(value) {
+  const port = Number(value);
+  if (!Number.isInteger(port) || port < 1 || port > 65535) {
+    throw badRequest("端口号必须是 1-65535 的整数");
+  }
+  return port;
+}
+
+function normalizeScheme(value) {
+  const scheme = normalizeText(value).toLowerCase();
+  if (!VALID_SCHEMES.has(scheme)) {
+    throw badRequest("协议仅支持 http 或 https");
+  }
+  return scheme;
+}
+
+function normalizeHost(value) {
+  const host = normalizeText(value);
+  if (!host) {
+    throw badRequest("主机名不能为空");
+  }
+  if (host.includes("/") || host.includes("\\") || host.includes("?") || host.includes("#")) {
+    throw badRequest("主机名格式不合法");
+  }
+  return host;
+}
+
+function normalizeApplicationName(value) {
+  const applicationName = normalizeText(value).toLowerCase();
+  if (!applicationName) {
+    throw badRequest("应用名不能为空");
+  }
+  if (!APPLICATION_NAME_PATTERN.test(applicationName)) {
+    throw badRequest("应用名需符合 Ubuntu 目录名规则：仅允许小写字母、数字和中划线，且不能以中划线开头或结尾");
+  }
+  return applicationName;
+}
+
+function hasOwn(payload, key) {
+  return Object.prototype.hasOwnProperty.call(payload || {}, key);
+}
+
+function buildBaseUrl(scheme, host, port, applicationName) {
+  return `${scheme}://${host}:${port}/${applicationName}`;
+}
+
+function decorateSystemRow(row) {
+  if (!row) {
+    return null;
+  }
+
+  const baseUrl = buildBaseUrl(row.scheme, row.host, row.port, row.application_name);
+  return {
+    ...row,
+    is_builtin: Boolean(row.is_builtin),
+    ontology_name: row.application_name,
+    system_id: row.application_name,
+    service_endpoint: baseUrl,
+    session_endpoint: baseUrl,
+    base_url: baseUrl,
+    last_connectivity_test_result: jsonParse(row.last_connectivity_test_result_json, {})
+  };
+}
+
+export class SystemService {
+  constructor({ db, objectStorage, rpcClient }) {
+    this.db = db;
+    this.objectStorage = objectStorage;
+    this.rpcClient = rpcClient;
+  }
+
+  listSystems() {
+    return this.db.prepare(`
+      SELECT
+        s.*,
+        a.original_name AS ontology_file_name
+      FROM business_systems s
+      LEFT JOIN artifacts a ON a.id = s.ontology_artifact_id
+      ORDER BY s.updated_at DESC
+    `).all().map(decorateSystemRow);
+  }
+
+  getSystemById(id) {
+    return decorateSystemRow(
+      this.db.prepare(`
+        SELECT
+          s.*,
+          a.original_name AS ontology_file_name
+        FROM business_systems s
+        LEFT JOIN artifacts a ON a.id = s.ontology_artifact_id
+        WHERE s.id = ?
+      `).get(id)
+    );
+  }
+
+  getInvocationBaseUrlByApplicationName(applicationName) {
+    const normalizedApplicationName = normalizeApplicationName(applicationName);
+    const row = this.db.prepare(`
+      SELECT *
+      FROM business_systems
+      WHERE application_name = ? AND status = 'active'
+    `).get(normalizedApplicationName);
+    if (!row) {
+      throw notFound(`未找到业务系统：${normalizedApplicationName}`);
+    }
+
+    return {
+      application_name: row.application_name,
+      provider: row.provider,
+      baseUrl: buildBaseUrl(row.scheme, row.host, row.port, row.application_name),
+      scheme: row.scheme,
+      host: row.host,
+      port: row.port,
+      status: row.status
+    };
+  }
+
+  findLatestActiveSystemByProvider(provider) {
+    const row = this.db.prepare(`
+      SELECT *
+      FROM business_systems
+      WHERE provider = ? AND status = 'active'
+      ORDER BY updated_at DESC, id DESC
+      LIMIT 1
+    `).get(normalizeText(provider).toLowerCase());
+    return decorateSystemRow(row);
+  }
+
+  validateUniqueApplicationName(applicationName, excludeId = null) {
+    const existing = excludeId === null
+      ? this.db.prepare(`
+        SELECT id
+        FROM business_systems
+        WHERE application_name = ?
+      `).get(applicationName)
+      : this.db.prepare(`
+        SELECT id
+        FROM business_systems
+        WHERE application_name = ? AND id <> ?
+      `).get(applicationName, excludeId);
+
+    if (existing) {
+      throw conflict(`应用名已存在：${applicationName}`);
+    }
+  }
+
+  normalizeCreatePayload(payload) {
+    const provider = normalizeText(payload.provider).toLowerCase();
+    const applicationName = normalizeApplicationName(payload.application_name);
+    if (!provider || !applicationName) {
+      throw badRequest("提供方和应用名不能为空");
+    }
+
+    return {
+      provider,
+      application_name: applicationName,
+      description: normalizeText(payload.description),
+      scheme: normalizeScheme(payload.scheme),
+      host: normalizeHost(payload.host),
+      port: normalizePort(payload.port),
+      status: normalizeText(payload.status) === "disabled" ? "disabled" : "active"
+    };
+  }
+
+  normalizeUpdatePayload(payload, current) {
+    const provider = hasOwn(payload, "provider") ? normalizeText(payload.provider).toLowerCase() : current.provider;
+    const applicationName = hasOwn(payload, "application_name")
+      ? normalizeApplicationName(payload.application_name)
+      : current.application_name;
+    if (!provider || !applicationName) {
+      throw badRequest("提供方和应用名不能为空");
+    }
+
+    return {
+      provider,
+      application_name: applicationName,
+      description: hasOwn(payload, "description") ? normalizeText(payload.description) : current.description,
+      scheme: hasOwn(payload, "scheme") ? normalizeScheme(payload.scheme) : current.scheme,
+      host: hasOwn(payload, "host") ? normalizeHost(payload.host) : current.host,
+      port: hasOwn(payload, "port") ? normalizePort(payload.port) : current.port,
+      status: hasOwn(payload, "status")
+        ? (normalizeText(payload.status) === "disabled" ? "disabled" : "active")
+        : current.status
+    };
+  }
+
+  getArtifactStorageInfo(artifactId) {
+    if (!artifactId) {
+      return null;
+    }
+
+    return this.db.prepare(`
+      SELECT bucket, object_key
+      FROM artifacts
+      WHERE id = ?
+    `).get(artifactId) || null;
+  }
+
+  async listRemoteApplicationNames() {
+    const remote = await this.rpcClient.call("apps.list", {});
+    const items = Array.isArray(remote?.items) ? remote.items : [];
+    const names = new Set();
+
+    for (const item of items) {
+      const applicationName = normalizeText(
+        item?.applicationName || item?.id || item?.name || item?.ontologyName
+      ).toLowerCase();
+      if (applicationName) {
+        names.add(applicationName);
+      }
+    }
+
+    return names;
+  }
+
+  async deleteRemoteOntologiesIfPresent(applicationNames, { ignoreDeleteErrors = false } = {}) {
+    const normalizedNames = Array.from(new Set(
+      applicationNames
+        .map((value) => normalizeText(value).toLowerCase())
+        .filter(Boolean)
+    ));
+    if (normalizedNames.length === 0) {
+      return {
+        deletedApplicationNames: []
+      };
+    }
+
+    const remoteNames = await this.listRemoteApplicationNames();
+    const deletedApplicationNames = [];
+
+    for (const applicationName of normalizedNames) {
+      if (!remoteNames.has(applicationName)) {
+        continue;
+      }
+
+      try {
+        await this.rpcClient.call("apps.delete", { systemId: applicationName });
+        deletedApplicationNames.push(applicationName);
+      } catch (error) {
+        if (!ignoreDeleteErrors) {
+          throw error;
+        }
+      }
+    }
+
+    return { deletedApplicationNames };
+  }
+
+  async createSystem({ payload, file, createdBy }) {
+    const normalized = this.normalizeCreatePayload(payload);
+    this.validateUniqueApplicationName(normalized.application_name);
+
+    let artifactId = null;
+    let bucket = null;
+    let objectKey = null;
+    if (file) {
+      const artifact = await this.persistOntologyArtifact({ file, createdBy });
+      artifactId = artifact.id;
+      bucket = artifact.bucket;
+      objectKey = artifact.objectKey;
+    }
+
+    const { deletedApplicationNames } = await this.deleteRemoteOntologiesIfPresent([
+      normalized.application_name
+    ]);
+
+    const baseUrl = buildBaseUrl(
+      normalized.scheme,
+      normalized.host,
+      normalized.port,
+      normalized.application_name
+    );
+    await this.rpcClient.call("apps.create", {
+      systemId: normalized.application_name,
+      name: normalized.application_name,
+      provider: normalized.provider,
+      ontologyName: normalized.application_name,
+      description: normalized.description,
+      applicationEndpoint: baseUrl,
+      sessionServiceEndpoint: baseUrl,
+      status: normalized.status,
+      bucket,
+      objectKey,
+      replaceOntology: Boolean(bucket && objectKey)
+    });
+
+    const now = new Date().toISOString();
+    const result = this.db.prepare(`
+      INSERT INTO business_systems (
+        provider, application_name, description, ontology_artifact_id,
+        scheme, host, port, status, last_connectivity_test_status, last_connectivity_test_result_json,
+        created_at, updated_at
+      ) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
+    `).run(
+      normalized.provider,
+      normalized.application_name,
+      normalized.description,
+      artifactId,
+      normalized.scheme,
+      normalized.host,
+      normalized.port,
+      normalized.status,
+      "unknown",
+      JSON.stringify({}),
+      now,
+      now
+    );
+
+    return {
+      system: this.getSystemById(result.lastInsertRowid),
+      kernelOntologyDeleted: deletedApplicationNames.length > 0,
+      deletedKernelOntologyNames: deletedApplicationNames
+    };
+  }
+
+  async persistOntologyArtifact({ file, createdBy }) {
+    const zip = new AdmZip(file.buffer);
+    const names = zip
+      .getEntries()
+      .filter((entry) => !entry.isDirectory)
+      .map((entry) => entry.entryName.replace(/^\/+/, ""));
+    if (!names.includes("index.md")) {
+      throw badRequest("Ontology 压缩包第一层必须包含 index.md");
+    }
+
+    const artifact = await this.objectStorage.uploadAdminArtifact({ kind: "ontology", file });
+    const result = this.db.prepare(`
+      INSERT INTO artifacts (
+        kind, bucket, object_key, original_name, mime_type, byte_size, created_by, created_at
+      ) VALUES (?, ?, ?, ?, ?, ?, ?, ?)
+    `).run(
+      "ontology",
+      artifact.bucket,
+      artifact.objectKey,
+      file.originalname,
+      file.mimetype || "application/octet-stream",
+      file.size,
+      createdBy,
+      new Date().toISOString()
+    );
+
+    return {
+      id: result.lastInsertRowid,
+      ...artifact
+    };
+  }
+
+  async updateSystem(id, { payload, file, createdBy }) {
+    const current = this.db.prepare("SELECT * FROM business_systems WHERE id = ?").get(id);
+    if (!current) {
+      throw notFound("业务系统不存在");
+    }
+
+    const normalized = this.normalizeUpdatePayload(payload, current);
+    this.validateUniqueApplicationName(normalized.application_name, id);
+
+    let artifactId = current.ontology_artifact_id;
+    let artifact;
+    if (file) {
+      artifact = await this.persistOntologyArtifact({ file, createdBy });
+      artifactId = artifact.id;
+    }
+
+    const existingArtifact = artifact || this.getArtifactStorageInfo(artifactId);
+
+    const baseUrl = buildBaseUrl(
+      normalized.scheme,
+      normalized.host,
+      normalized.port,
+      normalized.application_name
+    );
+    const previousSystemId = current.application_name;
+    const { deletedApplicationNames } = await this.deleteRemoteOntologiesIfPresent([
+      previousSystemId,
+      normalized.application_name
+    ], { ignoreDeleteErrors: true });
+
+    await this.rpcClient.call("apps.create", {
+      systemId: normalized.application_name,
+      name: normalized.application_name,
+      provider: normalized.provider,
+      ontologyName: normalized.application_name,
+      description: normalized.description,
+      applicationEndpoint: baseUrl,
+      sessionServiceEndpoint: baseUrl,
+      status: normalized.status,
+      bucket: existingArtifact?.bucket,
+      objectKey: existingArtifact?.object_key || existingArtifact?.objectKey,
+      replaceOntology: Boolean(existingArtifact?.bucket && (existingArtifact?.object_key || existingArtifact?.objectKey))
+    });
+
+    this.db.prepare(`
+      UPDATE business_systems
+      SET provider = ?, application_name = ?, description = ?, ontology_artifact_id = ?,
+          scheme = ?, host = ?, port = ?, status = ?, updated_at = ?
+      WHERE id = ?
+    `).run(
+      normalized.provider,
+      normalized.application_name,
+      normalized.description,
+      artifactId,
+      normalized.scheme,
+      normalized.host,
+      normalized.port,
+      normalized.status,
+      new Date().toISOString(),
+      id
+    );
+
+    return {
+      system: this.getSystemById(id),
+      kernelOntologyDeleted: deletedApplicationNames.length > 0,
+      deletedKernelOntologyNames: deletedApplicationNames
+    };
+  }
+
+  async setSystemStatus(id, status) {
+    const current = this.db.prepare("SELECT * FROM business_systems WHERE id = ?").get(id);
+    if (!current) {
+      throw notFound("业务系统不存在");
+    }
+
+    const normalizedStatus = normalizeText(status) === "disabled" ? "disabled" : "active";
+    await this.rpcClient.call(normalizedStatus === "active" ? "apps.enable" : "apps.disable", {
+      systemId: current.application_name
+    });
+    this.db.prepare(`
+      UPDATE business_systems
+      SET status = ?, updated_at = ?
+      WHERE id = ?
+    `).run(normalizedStatus, new Date().toISOString(), id);
+
+    return this.getSystemById(id);
+  }
+
+  async deleteSystem(id) {
+    const current = this.db.prepare("SELECT * FROM business_systems WHERE id = ?").get(id);
+    if (!current) {
+      throw notFound("业务系统不存在");
+    }
+
+    if (current.is_builtin) {
+      throw conflict("内置 ontology 不允许删除");
+    }
+
+    await this.rpcClient.call("apps.delete", { systemId: current.application_name });
+    this.db.prepare("DELETE FROM business_systems WHERE id = ?").run(id);
+    return { ok: true };
+  }
+
+  async testConnectivity(id) {
+    const current = this.db.prepare("SELECT * FROM business_systems WHERE id = ?").get(id);
+    if (!current) {
+      throw notFound("业务系统不存在");
+    }
+
+    const baseUrl = buildBaseUrl(current.scheme, current.host, current.port, current.application_name);
+    const response = await fetch(baseUrl, { method: "GET" });
+    const ok = response.ok;
+    const result = {
+      ok,
+      baseUrl,
+      service: { status: response.status, url: baseUrl },
+      tested_at: new Date().toISOString()
+    };
+
+    this.db.prepare(`
+      UPDATE business_systems
+      SET last_connectivity_test_status = ?, last_connectivity_test_result_json = ?, updated_at = ?
+      WHERE id = ?
+    `).run(ok ? "ok" : "failed", JSON.stringify(result), new Date().toISOString(), id);
+
+    return result;
+  }
+
+  findSystemForInvocation(provider, applicationName) {
+    return decorateSystemRow(this.db.prepare(`
+      SELECT *
+      FROM business_systems
+      WHERE provider = ? AND application_name = ? AND status = 'active'
+    `).get(normalizeText(provider).toLowerCase(), normalizeApplicationName(applicationName)));
+  }
+
+  async recordInvocation(log) {
+    this.db.prepare(`
+      INSERT INTO system_invocation_logs (
+        trace_id, agent_slug, session_id, provider, application_name,
+        command_name, request_payload_json, response_payload_json, response_time_ms,
+        success, error_message, created_at
+      ) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
+    `).run(
+      log.trace_id,
+      log.agent_slug || null,
+      log.session_id || null,
+      log.provider,
+      log.application_name,
+      log.command_name,
+      JSON.stringify(log.request_payload ?? {}),
+      JSON.stringify(log.response_payload ?? {}),
+      log.response_time_ms,
+      log.success ? 1 : 0,
+      log.error_message || null,
+      log.created_at || new Date().toISOString()
+    );
+
+    try {
+      await this.rpcClient.call("apps.log.record", {
+        systemId: log.application_name,
+        agentId: log.agent_slug,
+        conversationId: log.session_id,
+        provider: log.provider,
+        applicationName: log.application_name,
+        commandName: log.command_name,
+        inputParams: log.request_payload,
+        responseTimeMs: log.response_time_ms,
+        success: log.success,
+        responseContent: log.response_payload
+      });
+    } catch {}
+  }
+
+  listLogs({ applicationName = "", agentSlug = "" } = {}) {
+    return this.db.prepare(`
+      SELECT *
+      FROM system_invocation_logs
+      WHERE (? = '' OR application_name = ?)
+        AND (? = '' OR agent_slug = ?)
+      ORDER BY created_at DESC
+      LIMIT 200
+    `).all(
+      normalizeText(applicationName).toLowerCase(),
+      normalizeText(applicationName).toLowerCase(),
+      normalizeText(agentSlug),
+      normalizeText(agentSlug)
+    ).map((row) => ({
+      ...row,
+      system_id: row.application_name,
+      request_payload: jsonParse(row.request_payload_json, {}),
+      response_payload: jsonParse(row.response_payload_json, {})
+    }));
+  }
+
+  listStats({ applicationName = "", agentSlug = "" } = {}) {
+    return this.db.prepare(`
+      SELECT
+        application_name,
+        provider,
+        command_name,
+        COUNT(*) AS call_count,
+        SUM(CASE WHEN success = 1 THEN 1 ELSE 0 END) AS success_count,
+        ROUND(AVG(response_time_ms), 1) AS avg_response_ms
+      FROM system_invocation_logs
+      WHERE (? = '' OR application_name = ?)
+        AND (? = '' OR agent_slug = ?)
+      GROUP BY application_name, provider, command_name
+      ORDER BY call_count DESC, avg_response_ms ASC
+    `).all(
+      normalizeText(applicationName).toLowerCase(),
+      normalizeText(applicationName).toLowerCase(),
+      normalizeText(agentSlug),
+      normalizeText(agentSlug)
+    ).map((row) => ({
+      ...row,
+      system_id: row.application_name,
+      success_rate: row.call_count ? Number(((row.success_count / row.call_count) * 100).toFixed(1)) : 0
+    }));
+  }
+}