/** * Graph Service — P1 虚拟引用寻址支撑函数(新文件,P1 专属) * * 不修改 P0 已有的 graph-db.js / graph-seed-service.js(那是并行工作线), * 只通过 graph-db.js 导出的 getDb() 读写同一个 SQLite 连接。 * * 详细需求与设计见 docs/todo/system-design/27-p1-virtual-reference.md。 */ const path = require('path'); const { getDb } = require('./graph-db'); const ROOT = path.resolve(__dirname, '../..'); let indexEnsured = false; /** * 查询某个 system_agent id 对应的共享 agent.md 绝对路径。 * * 只匹配 source_kind = 'template'(不匹配 'global_agent')——这是本次设计修复的一个 * 真实 bug:system/agents/ 和 system/templates/ 下有 6 个同名 id * (agent-creator/digester/evolver/evolver-youtube-crawler/hunter/innovator) * 内容可能不同步,hire() 只从 system/templates/ 拷贝过文件,如果查表命中了 * global_agent 那一行会读错文件(详见 27-graph-agent-architecture.md §7、 * 27-p1-virtual-reference.md §4.6)。 * * @param {string} systemAgentId * @returns {string|null} 共享 agent.md 的绝对路径;查不到对应 template 行时返回 null */ function lookupSharedAgentMd(systemAgentId) { const db = getDb(); const row = db .prepare(`SELECT legacy_path FROM system_agents WHERE id = ? AND source_kind = 'template'`) .get(systemAgentId); if (!row) return null; return path.resolve(ROOT, row.legacy_path, 'agent.md'); } /** * P2(27-p2-org-graph.md §3.2)起改为**部分**唯一索引:只约束 status='active' 的行, * 这样某个角色被 fire(status='removed')之后,同一个 role_name 能被重新 hire 进来。 * 索引本身 graph-db.js 的 migrate() 已经建过了,这里保留是为了让本模块不依赖调用顺序; * 名字/条件必须与 graph-db.js 完全一致,否则会互相把对方的索引建回来。 */ function ensureOrgGraphNodeIndex(db) { if (indexEnsured) return; db.exec(` CREATE UNIQUE INDEX IF NOT EXISTS idx_org_graph_nodes_company_role_active ON org_graph_nodes(company_id, role_name) WHERE status = 'active' `); indexEnsured = true; } /** * 保证 system_agents 里有 systemAgentId 这一行(缺则按 seedCompanyAgents() * (graph-seed-service.js 约 213-295 行)对同样场景的字段约定 fallback 插入 * source_kind='company_hired' 的记录)。已存在的行一个字段都不改——绝不把 * source_kind='template' / 'global_agent' 的公共 agent 降级成公司私有。 * * P3B/W7 起从 upsertOrgGraphNode() 里拆出来单独导出:work 层本体被雇时**不写** * org_graph_nodes(判据见 company-service.decideOrgNodeRecording),但 system_agents * 这一行仍然必须补——`work_graph_nodes.system_agent_id` 是指向该表的外键, * 缺行的话 workgraph-planning 雇完人再提交图会直接外键挂(27-P3B §四 W1)。 * * 幂等,可重复调用。调用方自己起事务时也安全(better-sqlite3 支持嵌套 transaction)。 * * @param {string} companyId * @param {string} systemAgentId * @param {string} roleName - 仅用于 fallback 行的 legacy_path * @returns {boolean} true = 本次真的插了一行;false = 早就有了 */ function ensureSystemAgentRow(companyId, systemAgentId, roleName) { const db = getDb(); const existingAgent = db.prepare(`SELECT id FROM system_agents WHERE id = ?`).get(systemAgentId); if (existingAgent) return false; const now = new Date().toISOString(); db.prepare(` INSERT INTO system_agents (id, name, scope, owner_company_id, source_kind, legacy_path, status, benchmark_score, created_at, updated_at) VALUES (@id, @name, 'company', @owner_company_id, 'company_hired', @legacy_path, 'active', NULL, @created_at, @updated_at) `).run({ id: systemAgentId, name: systemAgentId, owner_company_id: companyId, legacy_path: `data/companies/${companyId}/agents/${roleName}`, created_at: now, updated_at: now, }); return true; } /** * 这家公司的 org graph 有没有成型,以及某个角色在不在图上(P3B §四 W7 的判据之一)。 * * `hasWorkflow` 看的是**边**不是节点:节点 hire() 自己就会造(P1/P2 语义),拿节点当 * 「图已存在」的信号会把「同一家公司连雇两个人,第二个不入图」这条既有语义废掉。 * 有 active 边则说明 orggraph-planning 提交过一张连起来的图(POST /org-graph → * org-graph-service.writeGraph 是唯一写边的地方),此后这张图的作者就是它, * hire() 不再往上追加节点:`work-graph-db.js:473` 起 org run 时读的正是 org_graph_nodes * 的 active 行,混进去的 work 层本体不是画歪一张图,而是真的多出一个会被执行的 org 节点。 * * @param {string} companyId * @param {string} roleName * @returns {{hasWorkflow: boolean, isNode: boolean}} */ function orgGraphMembership(companyId, roleName) { const db = getDb(); const edges = db.prepare( `SELECT COUNT(*) AS c FROM org_graph_edges WHERE company_id = ? AND status = 'active'` ).get(companyId); const node = db.prepare( `SELECT COUNT(*) AS c FROM org_graph_nodes WHERE company_id = ? AND role_name = ? AND status = 'active'` ).get(companyId, roleName); return { hasWorkflow: ((edges && edges.c) || 0) > 0, isNode: ((node && node.c) || 0) > 0 }; } /** * 为 (companyId, roleName) 建立/复用一条 org_graph_nodes 引用记录。 * * 语义是「把这个 agent 记成本公司 org graph 的一个节点」——**显式动作**, * 调用方有两类:org-graph-service.insertNode(规划 agent 提交 org graph)与 * company-service.hire 判定为 org 层招聘时。work 层招聘不走这里(P3B §四 W7)。 * * system_agents 缺行时先补(ensureSystemAgentRow),因为 org_graph_nodes.system_agent_id * 也是外键。 * * 幂等:org_graph_nodes(company_id, role_name) WHERE status='active' 部分唯一索引 * + INSERT ... ON CONFLICT(...) WHERE status='active' DO NOTHING,重复调用不产生重复行。 * (SQLite 的 conflict target 要匹配部分索引,必须原样带上索引的 WHERE 条件。) * * @param {string} companyId * @param {string} systemAgentId * @param {string} roleName * @returns {void} */ function upsertOrgGraphNode(companyId, systemAgentId, roleName) { const db = getDb(); ensureOrgGraphNodeIndex(db); const now = new Date().toISOString(); const tx = db.transaction(() => { ensureSystemAgentRow(companyId, systemAgentId, roleName); db.prepare(` INSERT INTO org_graph_nodes (company_id, system_agent_id, role_name, status, created_at) VALUES (@company_id, @system_agent_id, @role_name, 'active', @created_at) ON CONFLICT(company_id, role_name) WHERE status = 'active' DO NOTHING `).run({ company_id: companyId, system_agent_id: systemAgentId, role_name: roleName, created_at: now, }); }); tx(); } module.exports = { lookupSharedAgentMd, ensureSystemAgentRow, orgGraphMembership, upsertOrgGraphNode, };