sqlseed-web 0.2.4__py3-none-any.whl

sqlseed_web/static/js/pages/heal.js

@@ -0,0 +1,395 @@
+// 配置校验与修复：L2 validate → L3 repair → L5 auto-heal，逐层可视。
+// 这是缺陷收敛的主战场：贴 YAML，看违规、看修复、看最终产物。
+// 未安装 sqlseed-ai 时展示 onboarding 引导卡（安装命令 + 能力说明）。
+
+import { h, post, get, store, msg, clear, table, fmt } from '../api.js';
+import { createDropdown } from '../dropdown.js';
+let yamlText = '';
+let healTimer = null;
+const templateTableDd = createDropdown({
+  options: [],
+  placeholder: '— 选择表 —',
+  onChange: () => {}
+});
+export function render() {
+  const root = h('div');
+  templateTableDd.setOptions(store.tables.map(t => ({
+    value: t.name,
+    label: t.name
+  })), store.tables[0]?.name || '');
+  root.append(h('h2', {}, '配置助手'), renderAiPanel(), h('div', {
+    class: 'panel'
+  }, h('div', {
+    class: 'muted',
+    style: 'margin-bottom:10px; line-height:1.7'
+  }, '校验、修复或生成测试数据配置：', h('br'), '① 校验配置 — 检查下方 YAML 中的生成规则是否符合数据库约束；', h('br'), '② 修复配置 — 修正下方 YAML 中可自动处理的问题，输出修复后的配置；', h('br'), '③ 生成新配置 — 根据当前数据库的表结构和约束重新生成配置，需要时由 AI 辅助。此操作不使用下方 YAML。'), h('div', {
+    class: 'row'
+  }, h('label', {
+    class: 'genform-label'
+  }, '从表导入模板:'), templateTableDd.el, h('button', {
+    onclick: loadTemplate
+  }, '导入'), h('span', {
+    class: 'muted',
+    style: 'font-size:12px'
+  }, '粘贴 YAML，或从所选表导入模板。处理后的配置可送到数据生成向导，也可保存后用 sqlseed fill --config config.yaml 执行。')), h('textarea', {
+    id: 'heal-yaml',
+    spellcheck: 'false',
+    oninput: e => {
+      yamlText = e.target.value;
+    }
+  }, yamlText), h('div', {
+    class: 'row end'
+  }, h('button', {
+    onclick: doValidate
+  }, '① 校验此配置'), h('button', {
+    onclick: doRepair
+  }, '② 修复此配置'), h('button', {
+    class: 'primary',
+    onclick: doAutoHeal
+  }, '③ 生成新配置'))), h('div', {
+    id: 'heal-out'
+  }));
+  const ta = document.getElementById('heal-yaml');
+  if (ta && yamlText) ta.value = yamlText;
+  return root;
+}
+export function mount() {
+  if (!store.connId) {
+    const out = document.getElementById('heal-out');
+    clear(out);
+    out.append(msg('请先在「数据库连接」页连接数据库，以便按表结构和约束校验配置。', 'warn'));
+  }
+}
+
+// ---- AI 配置面板（在线/本地大模型，会话级，免重启切换） ----------------------
+
+let aiState = null; // /api/ai/config 响应
+
+async function loadAiConfig() {
+  try {
+    aiState = await get('/api/ai/config');
+  } catch {
+    aiState = null;
+  }
+  const holder = document.getElementById('ai-panel-body');
+  if (holder) renderAiBody(holder);
+}
+function renderAiPanel() {
+  const body = h('div', {
+    id: 'ai-panel-body'
+  }, h('div', {
+    class: 'loading'
+  }, '加载 AI 配置…'));
+  loadAiConfig().then(() => renderAiBody(body));
+  return h('div', {
+    class: 'panel'
+  }, h('div', {
+    class: 'row'
+  }, h('span', {
+    class: 'pill gen'
+  }, 'AI 服务设置'), h('span', {
+    class: 'muted'
+  }, '选择辅助生成配置的模型服务，保存后立即生效。')), body);
+}
+
+// 未安装 sqlseed-ai 时的 onboarding 引导卡：说清楚缺了什么、怎么装、
+// 不装还能用什么（数据生成/浏览不依赖 AI）。
+function renderOnboarding(holder, reason) {
+  holder.append(h('div', {
+    class: 'msg warn'
+  }, reason || 'sqlseed-ai 未安装'), h('div', {
+    class: 'muted',
+    style: 'margin:8px 0; line-height:1.7'
+  }, 'sqlseed-ai 是可选插件，提供：① 配置校验与自动修复；' + '② 按数据库约束一键生成合规配置（即数据生成向导中的「AI 一键生成配置」）。' + '未安装时，连接、数据生成、数据浏览功能完全不受影响。'), h('div', {
+    class: 'row'
+  }, h('label', {}, '安装命令'), h('code', {}, 'pip install -e ./plugins/sqlseed-web[ai]')), h('div', {
+    class: 'muted',
+    style: 'margin-top:6px'
+  }, '安装后刷新本页即可。本地模型（Ollama / LM Studio）无需 API Key；在线模型需要 Key。'));
+}
+function renderAiBody(holder) {
+  clear(holder);
+  if (!aiState?.available) {
+    renderOnboarding(holder, aiState?.reason);
+    return;
+  }
+  const eff = aiState.effective;
+  const ov = aiState.override || {};
+  // 排序：当前生效的后端浮到第一（下拉默认就选它，放最前免得被埋没），
+  // 其余保持 API 顺序——通用（OpenAI 兼容 / Google AI Studio）在前，本地殿后。
+  const currentBackend = ov.backend || eff.backend;
+  const orderedBackends = [...aiState.backends.filter(b => b.id === currentBackend), ...aiState.backends.filter(b => b.id !== currentBackend)];
+  const backendDd = createDropdown({
+    value: currentBackend,
+    options: orderedBackends.map(b => ({
+      value: b.id,
+      label: b.label
+    })),
+    onChange: v => {
+      const b = aiState.backends.find(x => x.id === v);
+      const keyInput = document.getElementById('ai-key');
+      const urlInput = document.getElementById('ai-url');
+      if (b && keyInput && urlInput) {
+        keyInput.disabled = b.needs_key === '0';
+        urlInput.disabled = b.needs_url === '0';
+      }
+    }
+  });
+  function effectiveAIConfiguration() {
+    if (eff.backend === 'ollama' || eff.backend === 'lm_studio') {
+      return `当前生效：${eff.backend} · ${eff.model} · 本地后端无需 Key`;
+    } else {
+      return `当前生效：${eff.backend} · ${eff.model} · key ${eff.api_key_present ? '已配置' : '未配置'}`;
+    }
+  }
+  holder.append(h('div', {
+    class: 'row'
+  }, h('label', {
+    class: 'genform-label'
+  }, '后端'), backendDd.el), h('div', {
+    class: 'row'
+  }, h('label', {
+    class: 'genform-label'
+  }, '模型'), h('input', {
+    id: 'ai-model',
+    class: 'grow',
+    placeholder: '留空=自动（如 gemma4:e4b / gemma-4-26b-a4b-it）',
+    value: ov.model || ''
+  })), h('div', {
+    class: 'row'
+  }, h('label', {
+    class: 'genform-label'
+  }, 'API Key'), h('input', {
+    id: 'ai-key',
+    class: 'grow',
+    type: 'password',
+    placeholder: '在线后端需要；本地后端无需密钥',
+    value: ov.api_key || ''
+  })), h('div', {
+    class: 'row'
+  }, h('label', {
+    class: 'genform-label'
+  }, 'Base URL'), h('input', {
+    id: 'ai-url',
+    class: 'grow',
+    placeholder: 'OpenAI 兼容服务需要；其余留空',
+    value: ov.base_url || ''
+  })), h('div', {
+    class: 'row'
+  }, h('button', {
+    class: 'primary',
+    onclick: () => saveAiConfig(backendDd)
+  }, '应用配置'), h('button', {
+    onclick: testConnection
+  }, '测试连接'), h('span', {
+    class: 'muted',
+    id: 'ai-effective'
+  }, effectiveAIConfiguration())), h('div', {
+    id: 'ai-test-out'
+  }));
+  // 初始化输入框禁用态
+  const initB = aiState.backends.find(x => x.id === backendDd.get());
+  if (initB) {
+    document.getElementById('ai-key').disabled = initB.needs_key === '0';
+    document.getElementById('ai-url').disabled = initB.needs_url === '0';
+  }
+}
+async function testConnection() {
+  const out = document.getElementById('ai-test-out');
+  clear(out);
+  out.append(h('div', {
+    class: 'loading'
+  }, '探测后端连接…'));
+  try {
+    const res = await post('/api/ai/test-connection', {});
+    clear(out);
+    if (!res.available) {
+      out.append(msg(res.reason || 'sqlseed-ai 未安装', 'warn'));
+      return;
+    }
+    out.append(msg(res.message, res.ok ? 'ok' : 'warn'));
+  } catch (e) {
+    clear(out);
+    out.append(msg(`测试失败：${e.message}`));
+  }
+}
+async function saveAiConfig(backendDd) {
+  const payload = {
+    backend: backendDd.get(),
+    model: document.getElementById('ai-model').value.trim() || null,
+    api_key: document.getElementById('ai-key').value.trim() || null,
+    base_url: document.getElementById('ai-url').value.trim() || null
+  };
+  try {
+    aiState = await post('/api/ai/config', payload);
+    const holder = document.getElementById('ai-panel-body');
+    renderAiBody(holder);
+    holder.append(msg('AI 配置已应用（本会话生效）。', 'ok'));
+  } catch (e) {
+    const holder = document.getElementById('ai-panel-body');
+    holder.append(msg(`保存失败：${e.message}`));
+  }
+}
+async function loadTemplate() {
+  if (!store.connId || !store.tables.length) return;
+  const t = templateTableDd.get() || store.tables[0].name;
+  try {
+    const res = await get(`/api/connections/${store.connId}/tables/${encodeURIComponent(t)}/yaml-template`);
+    yamlText = res.yaml;
+    const ta = document.getElementById('heal-yaml');
+    if (ta) ta.value = yamlText;
+  } catch (e) {
+    const out = document.getElementById('heal-out');
+    out.append(msg(e.message));
+  }
+}
+function basePayload() {
+  return {
+    yaml: yamlText
+  };
+}
+async function doValidate() {
+  const out = document.getElementById('heal-out');
+  clear(out);
+  out.append(h('div', {
+    class: 'loading'
+  }, '校验中…'));
+  try {
+    const res = await post(`/api/connections/${store.connId}/heal/validate`, basePayload());
+    clear(out);
+    if (!res.ok) {
+      out.append(msg(`校验器异常：${res.error}`));
+      return;
+    }
+    // renderViolations 返回数组：必须 spread，否则 append 把数组当单个参数
+    // 字符串化成 "[object HTMLDivElement]"（P0 渲染 bug，实测发现）。
+    out.append(...renderViolations(res));
+  } catch (e) {
+    clear(out);
+    out.append(msg(e.message));
+  }
+}
+function renderViolations(res) {
+  const parts = [];
+  parts.push(h('div', {
+    class: 'panel'
+  }, h('h3', {}, `① 校验结果 — schema_hash ${res.schema_hash?.slice(0, 12)}`), res.is_clean ? msg('配置干净：无违规。', 'ok') : msg(`发现 ${res.violation_count} 处违规：`, 'warn')));
+  if (res.violations?.length) {
+    const validationSeverityClass = v => {
+      if (v.severity === 'crash') {
+        return 'err';
+      } else if (v.severity === 'semantic_error') {
+        return 'warn';
+      } else {
+        return 'ok';
+      }
+    };
+    parts.push(h('div', {
+      class: 'panel'
+    }, h('h3', {}, 'ViolationReport 列表'), h('div', {
+      class: 'table-scroll'
+    }, table(['severity', 'table', 'columns', '类型', 'fix_hint', 'message'], res.violations.map(v => [h('span', {
+      class: `pill ${validationSeverityClass(v)}`
+    }, v.severity), v.table, fmt(v.columns), v.constraint_type, v.fix_hint || '—', v.message || '']), {
+      monoCols: [1]
+    }))));
+  }
+  if (res.column_groups?.length) {
+    parts.push(h('div', {
+      class: 'panel'
+    }, h('h3', {}, '复合 FK 协同组（ColumnGroup）'), h('div', {
+      class: 'table-scroll'
+    }, table(['group_id', 'columns', 'parent_table', 'parent_columns'], res.column_groups.map(g => [g.group_id, fmt(g.columns), g.parent_table, fmt(g.parent_columns)]), {
+      monoCols: [0]
+    }))));
+  }
+  return parts;
+}
+async function doRepair() {
+  const out = document.getElementById('heal-out');
+  clear(out);
+  out.append(h('div', {
+    class: 'loading'
+  }, '修复中…'));
+  try {
+    const res = await post(`/api/connections/${store.connId}/heal/repair`, basePayload());
+    clear(out);
+    if (!res.ok) {
+      out.append(msg(`修复器异常：${res.error}`));
+      return;
+    }
+    out.append(h('div', {
+      class: 'panel'
+    }, h('h3', {}, `② 修复结果 — ${res.fix_count} 处已修`), res.applied_fixes?.length ? h('div', {
+      class: 'table-scroll'
+    }, table(['策略', 'table', 'columns', 'before', 'after'], res.applied_fixes.map(f => [h('span', {
+      class: 'pill gen'
+    }, f.fix_strategy), f.table, fmt(f.columns), h('code', {}, JSON.stringify(f.before)), h('code', {}, JSON.stringify(f.after))]), {
+      monoCols: [1]
+    })) : h('div', {
+      class: 'muted'
+    }, '无需修复。'), res.unfixable?.length ? msg(`不可自动修复 ${res.unfixable.length} 处，见 L2 视图。`, 'warn') : null));
+    if (res.repaired_yaml) {
+      const pre = h('pre', {}, res.repaired_yaml);
+      out.append(h('div', {
+        class: 'panel'
+      }, h('h3', {}, '修复后 YAML'), pre, h('div', {
+        class: 'row end'
+      }, h('button', {
+        onclick: () => {
+          yamlText = res.repaired_yaml;
+          document.getElementById('heal-yaml').value = yamlText;
+        }
+      }, '回填到编辑区'))));
+    }
+  } catch (e) {
+    clear(out);
+    out.append(msg(e.message));
+  }
+}
+async function doAutoHeal() {
+  const out = document.getElementById('heal-out');
+  clear(out);
+  const status = h('div', {
+    class: 'muted'
+  }, '提交全流程生成任务（读取 schema → 按约束校验/修复 → 必要时调用大模型 → 输出 YAML）…');
+  out.append(h('div', {
+    class: 'panel'
+  }, h('h3', {}, '③ 全流程生成'), status));
+  try {
+    const res = await post(`/api/connections/${store.connId}/heal/auto`, {
+      budget_seconds: 300
+    });
+    healTimer = setInterval(async () => {
+      try {
+        const job = await get(`/api/jobs/${res.job_id}`);
+        if (job.status === 'running') return;
+        clearInterval(healTimer);
+        if (job.status === 'error') {
+          status.replaceWith(msg(`生成失败：${job.error}`));
+        } else {
+          const n = job.result.llm_calls;
+          const llmNote = n === 0 ? '本次未调用大模型：schema 无语义违规，全部由确定性校验/修复完成（契约驱动设计：LLM 只在必要时介入）' : `大模型调用 ${n} 次`;
+          status.replaceWith(h('div', {}, msg(`完成（模型 ${job.result.model} @ ${job.result.backend}；${llmNote}）`, 'ok'), h('pre', {}, job.result.yaml), h('div', {
+            class: 'row end'
+          }, h('button', {
+            onclick: () => {
+              yamlText = job.result.yaml;
+              document.getElementById('heal-yaml').value = yamlText;
+            }
+          }, '回填到编辑区'), h('button', {
+            class: 'primary',
+            onclick: () => {
+              store.aiYaml = job.result.yaml;
+              location.hash = '#/wizard';
+            }
+          }, '送到数据生成向导 →'))));
+        }
+      } catch {
+        clearInterval(healTimer);
+      }
+    }, 1000);
+  } catch (e) {
+    status.replaceWith(msg(e.message));
+  }
+}

sqlseed_web/static/js/pages/meta.js

@@ -0,0 +1,110 @@
+// 系统信息：生成器与参数、插件扩展点、生成引擎、AI 后端状态。
+// 验收驾驶舱：这里的计数必须与代码一致（doc-sync 的 UI 版）。
+
+import { h, get, msg, clear, table } from '../api.js';
+export function render() {
+  const root = h('div');
+  root.append(h('h2', {}, '系统信息'), h('div', {
+    class: 'muted',
+    style: 'margin-bottom:12px; line-height:1.6'
+  }, '查看当前版本、可用生成器、插件扩展点、数据生成引擎、AI 服务状态和最近任务。'), h('div', {
+    id: 'meta-out'
+  }, h('div', {
+    class: 'loading'
+  }, '加载中…')));
+  return root;
+}
+export function mount() {
+  load();
+}
+async function load() {
+  const out = document.getElementById('meta-out');
+  try {
+    const [generators, hooks, providers, ai, info, jobs] = await Promise.all([get('/api/meta/generators'), get('/api/meta/hooks'), get('/api/meta/providers'), get('/api/meta/ai'), get('/api/meta/info'), get('/api/jobs')]);
+    clear(out);
+    out.append(renderStats(generators, hooks, ai, info), renderGenerators(generators), renderHooks(hooks), renderProviders(providers), renderJobs(jobs));
+  } catch (e) {
+    clear(out);
+    out.append(msg(e.message));
+  }
+}
+function renderStats(gen, hooks, ai, info) {
+  return h('div', {
+    class: 'panel'
+  }, h('div', {
+    class: 'stats'
+  }, h('div', {}, h('div', {
+    class: 'stat'
+  }, gen.count), h('div', {
+    class: 'stat-label'
+  }, '生成器类型')), h('div', {}, h('div', {
+    class: 'stat'
+  }, hooks.count), h('div', {
+    class: 'stat-label'
+  }, '插件扩展点')), h('div', {}, h('div', {
+    class: 'stat'
+  }, info.sqlseed_version || '—'), h('div', {
+    class: 'stat-label'
+  }, 'sqlseed 版本'))), h('div', {
+    class: 'row'
+  }, ai.available ? h('span', {
+    class: 'pill ok'
+  }, `AI: ${ai.model} @ ${ai.backend}（协议 ${ai.tool_calling_protocol}）`) : h('span', {
+    class: 'pill warn'
+  }, `AI 不可用：${ai.reason}`)));
+}
+function renderGenerators(gen) {
+  const paramList = name => (gen.params[name] || []).join(', ');
+  return h('div', {
+    class: 'panel'
+  }, h('h3', {}, `生成器及参数（${gen.count}）`), h('div', {
+    class: 'table-scroll'
+  }, table(['生成器', '参数'], gen.names.map(n => [h('span', {
+    class: 'pill gen'
+  }, n), paramList(n) || '—']), {
+    monoCols: [1]
+  })));
+}
+function renderHooks(hooks) {
+  return h('div', {
+    class: 'panel'
+  }, h('h3', {}, `插件扩展点（${hooks.count}）`), h('div', {
+    class: 'table-scroll'
+  }, table(['扩展点', '结果处理方式'], hooks.hooks.map(hk => [hk.name, hk.firstresult ? h('span', {
+    class: 'pill ok'
+  }, '首个有效结果') : h('span', {
+    class: 'pill'
+  }, '收集所有结果')]), {
+    monoCols: [0]
+  })));
+}
+function renderProviders(providers) {
+  return h('div', {
+    class: 'panel'
+  }, h('h3', {}, '数据生成引擎优先级'), h('div', {
+    class: 'row'
+  }, ...providers.default_chain.map(p => h('span', {
+    class: `pill ${providers.available.includes(p) ? 'ok' : 'warn'}`
+  }, p)), h('span', {
+    class: 'muted'
+  }, '优先使用前面的引擎；不可用时依次尝试后续引擎。')));
+}
+function jobStatusClass(j) {
+  if (j.status === 'done') {
+    return 'ok';
+  } else if (j.status === 'error') {
+    return 'err';
+  } else {
+    return 'warn';
+  }
+}
+function renderJobs(jobs) {
+  if (!jobs.jobs.length) return h('div');
+  return h('div', {
+    class: 'panel'
+  }, h('h3', {}, '最近任务'), h('div', {
+    class: 'table-scroll'
+  }, table(['任务', '类型', '状态', '插入行数', '错误'], jobs.jobs.map(j => [j.label, j.kind, h('span', {
+    class: `pill ${jobStatusClass(j)}`
+  }, j.status), j.rows_inserted, j.error ? j.error.slice(0, 80) : '']))));
+}