sqlseed-web 0.2.4__py3-none-any.whl

sqlseed_web/static/js/tree.js

@@ -0,0 +1,126 @@
+// 可勾选数据库对象树（参考工具 Step2 左栏）：表→列两级，
+// 列节点带语义标注（外键/序列/生成器语义），点击选中列驱动右侧属性面板。
+
+import { h, clear } from './api.js';
+import { colAnnotation } from './labels.js';
+
+/**
+ * @param {object} opts
+ * @param {Array} opts.tables - [{name, columns: ColumnInfo[], specs, fks}]
+ * @param {(table: string, col: string) => object|null} [opts.specResolver]
+ *   列标注用 spec 解析器（AI 一键配置回填后标注需反映 AI 选择而非零配置推断）；
+ *   缺省用 t.specs。
+ * @param {Map<string, Set<string>>} [opts.initialSelection]
+ *   初始勾选状态（跨步骤持久化；AI 选择不被步骤切换重置）。缺省全选。
+ * @param {(table: string, col: string) => void} opts.onSelectColumn
+ * @param {(sel: Map<string, Set<string>>) => void} opts.onChange
+ */
+export function createTree({
+  tables,
+  specResolver,
+  initialSelection,
+  onSelectColumn,
+  onChange
+}) {
+  const expanded = new Set(tables.map(t => t.name)); // 默认全展开
+  const checked = initialSelection ? new Map(tables.map(t => {
+    const want = initialSelection.get(t.name);
+    return [t.name, new Set(t.columns.map(c => c.name).filter(c => want?.has(c)))];
+  })) : new Map(tables.map(t => [t.name, new Set(t.columns.map(c => c.name))]));
+  const selected = {
+    table: null,
+    col: null
+  };
+  const el = h('div', {
+    class: 'tree'
+  });
+  function emit() {
+    if (onChange) onChange(checked);
+  }
+  function render() {
+    clear(el);
+    for (const t of tables) {
+      const tChecked = checked.get(t.name);
+      const allChecked = tChecked.size === t.columns.length;
+      const isOpen = expanded.has(t.name);
+      const tRow = h('div', {
+        class: 'tree-row table-row',
+        onclick: () => {
+          if (isOpen) expanded.delete(t.name);else expanded.add(t.name);
+          render();
+        }
+      }, h('span', {
+        class: 'tree-arrow'
+      }, isOpen ? '▾' : '▸'), h('input', {
+        type: 'checkbox',
+        checked: allChecked,
+        onclick: e => {
+          e.stopPropagation();
+          const next = e.target.checked ? new Set(t.columns.map(c => c.name)) : new Set();
+          checked.set(t.name, next);
+          emit();
+          render();
+        }
+      }), h('span', {
+        class: 'tree-icon'
+      }, '▦'), h('span', {
+        class: 'tree-name'
+      }, t.name), h('span', {
+        class: 'muted tree-count'
+      }, `${tChecked.size}/${t.columns.length}`));
+      el.append(tRow);
+      if (!isOpen) continue;
+      renderColumns(t, tChecked);
+    }
+    function renderColumns(t, tChecked) {
+      for (const col of t.columns) {
+        const spec = specResolver ? specResolver(t.name, col.name) : t.specs?.[col.name];
+        const anno = colAnnotation(col, spec, t.fks);
+        const isSel = selected.table === t.name && selected.col === col.name;
+        el.append(h('div', {
+          class: `tree-row col-row${isSel ? ' selected' : ''}`,
+          onclick: e => {
+            e.stopPropagation();
+            selected.table = t.name;
+            selected.col = col.name;
+            render();
+            if (onSelectColumn) onSelectColumn(t.name, col.name);
+          }
+        }, h('span', {
+          class: 'tree-arrow'
+        }, ''), h('input', {
+          type: 'checkbox',
+          checked: tChecked.has(col.name),
+          onclick: e => {
+            e.stopPropagation();
+            if (e.target.checked) tChecked.add(col.name);else tChecked.delete(col.name);
+            emit();
+            render();
+          }
+        }), h('span', {
+          class: 'tree-icon col-icon'
+        }, '▤'), h('span', {
+          class: 'tree-name'
+        }, col.name), anno ? h('span', {
+          class: 'muted tree-anno'
+        }, `(${anno})`) : null));
+      }
+    }
+  }
+  render();
+  return {
+    el,
+    getSelection: () => checked,
+    /** 外部整体替换勾选状态（AI 一键生成配置回填用）：Map<table, Set<col>> */
+    setSelection(sel) {
+      for (const t of tables) {
+        const cols = sel.get(t.name);
+        checked.set(t.name, cols ? new Set([...cols].filter(c => t.columns.some(x => x.name === c))) : new Set());
+      }
+      render();
+      emit();
+    },
+    getSelectedColumn: () => selected,
+    refresh: render
+  };
+}

sqlseed_web/static/js/workbench/ai-eligibility.js

@@ -0,0 +1,33 @@
+// Presentation policy mirrors workbench_ai_relations.locked_column targets.
+// Explicit/effective rules take precedence over schema mappings. The server
+// validates the actual schema and complete candidate before returning patches.
+const populated = value => {
+  if (Array.isArray(value)) {
+    return value.length > 0;
+  } else if (value && typeof value === 'object') {
+    return Object.keys(value).length > 0;
+  } else {
+    return Boolean(value);
+  }
+};
+export function fieldAIEligibility(table, column, rule) {
+  const denied = (code, reason) => ({
+    eligible: false,
+    code,
+    reason
+  });
+  if (!table || !column?.name) return denied('unavailable', '字段结构不可用，请刷新结构后再分析。');
+  if (column.is_autoincrement) return denied('database_generated', '由数据库自动分配，AI 不修改该字段。');
+  if (column.is_primary_key || table.primary_key?.includes(column.name)) return denied('primary_key', '主键由现有规则和数据库约束管理，AI 保持原样。');
+  if (table.foreign_keys?.some(key => key.columns?.includes(column.name))) return denied('foreign_key', '外键从关联表取值，AI 保持引用关系。');
+  if (column.is_computed) return denied('computed', '由数据库表达式计算，AI 不修改该字段。');
+  if (populated(rule?.derive_from) || populated(rule?.expression)) return denied('derived', '已有派生规则，AI 保持原样；可在取值规则中手动调整。');
+  if (['faker_method', 'mimesis_method', 'native_faker_method', 'native_mimesis_method', 'native_params'].some(key => populated(rule?.[key]))) return denied('native', '已有原生引擎配置，AI 保持原样；可在取值规则中手动调整。');
+  const effective = rule?.generator || rule?.generator_name || table.mapping?.[column.name]?.generator_name;
+  if (column.default !== null && column.default !== undefined && (!effective || effective === 'skip' || effective.startsWith('__'))) return denied('database_default', '当前使用数据库 DEFAULT，AI 保持数据库默认规则。');
+  return {
+    eligible: true,
+    code: 'editable',
+    reason: ''
+  };
+}

sqlseed_web/static/js/workbench/ai-handoff.js

@@ -0,0 +1,31 @@
+// One in-memory, explicitly requested trip to settings. No document edits,
+// suggestions, credentials, URL payloads or browser storage belong here.
+let pending=null;
+const valid=connId=>Boolean(pending && pending.connId===connId
+  && pending.model.epoch===pending.epoch
+  && pending.model.lifecycleVersion===pending.lifecycle
+  && pending.model.schema.schema_hash===pending.schemaHash
+  && pending.model.schema.target_key===pending.targetKey
+  && pending.model.saved?.id===pending.savedId);
+export function clearAIHandoff(){pending=null;}
+export function rememberAIHandoff({model,connId,returnTo,context}){
+  pending={model,connId,returnTo,context:structuredClone(context),armed:false,
+    epoch:model.epoch,lifecycle:model.lifecycleVersion,schemaHash:model.schema.schema_hash,
+    targetKey:model.schema.target_key,savedId:model.saved?.id};
+}
+export function peekAIHandoff(connId){
+  if(!valid(connId)){pending=null;return null;}
+  return {returnTo:pending.returnTo};
+}
+export function requestAIReturn(connId){
+  if(!peekAIHandoff(connId))return null;
+  pending.armed=true;return pending.returnTo;
+}
+export function leaveAISettings(destinationHash){
+  if(!pending?.armed || pending.returnTo!==destinationHash)pending=null;
+}
+export function consumeAIHandoff({model,connId,returnTo}){
+  const context=valid(connId)&&pending.armed&&pending.model===model&&pending.returnTo===returnTo
+    ?structuredClone(pending.context):null;
+  pending=null;return context;
+}

sqlseed_web/static/js/workbench/ai-stream.js

@@ -0,0 +1,116 @@
+import { httpErrorMessage } from "../api.js";
+const stages = new Set(['context', 'model', 'validation', 'preview']);
+const invalid = () => new Error('AI 进度格式不正确，请重试。');
+function responseError(detail, status) {
+  const message = httpErrorMessage(detail, status);
+  const error = new Error(message);
+  error.status = status;
+  error.detail = detail;
+  error.code = detail?.code;
+  return error;
+}
+
+// A terminal result is required: an HTTP 200 or a progress event alone never
+// makes suggestions reviewable. The caller owns epoch and modal lifecycle checks.
+export async function requestAISuggestions(path, request, {
+  signal,
+  onProgress
+} = {}) {
+  const checkAbort = () => {
+    if (signal?.aborted) {
+      const error = new Error('AI 分析已取消。');
+      error.name = 'AbortError';
+      throw error;
+    }
+  };
+  checkAbort();
+  const response = await fetch(path, {
+    method: 'POST',
+    headers: {
+      'Content-Type': 'application/json',
+      Accept: 'application/x-ndjson'
+    },
+    body: JSON.stringify(request),
+    signal
+  });
+  checkAbort();
+  const streamed = response.headers?.get('Content-Type')?.includes('application/x-ndjson');
+  if (!response.ok || !streamed) {
+    const body = await response.json().catch(() => null);
+    checkAbort();
+    if (!response.ok) throw responseError(body?.detail, response.status);
+    if (!body || typeof body !== 'object' || Array.isArray(body)) throw invalid();
+    return body;
+  }
+  if (!response.body?.getReader) throw invalid();
+  const reader = response.body.getReader(),
+    decoder = new TextDecoder();
+  let buffer = '';
+  const cancel = () => {
+    reader.cancel().catch(() => {});
+  };
+  signal?.addEventListener('abort', cancel, {
+    once: true
+  });
+  function event(line) {
+    let value;
+    try {
+      value = JSON.parse(line);
+    } catch {
+      throw invalid();
+    }
+    if (value?.type === 'progress' && stages.has(value.stage) && typeof value.message === 'string') {
+      onProgress?.({
+        type: 'progress',
+        stage: value.stage,
+        message: value.message
+      });
+      return null;
+    }
+    if (value?.type === 'result' && value.result && typeof value.result === 'object' && !Array.isArray(value.result)) return value;
+    if (value?.type === 'error' && typeof value.message === 'string') {
+      const error = new Error(value.message);
+      error.code = value.code;
+      throw error;
+    }
+    throw invalid();
+  }
+  try {
+    for (;;) {
+      checkAbort();
+      const {
+        value,
+        done
+      } = await reader.read();
+      checkAbort();
+      const lines = decodedLines(done, value);
+      const terminal = consumeLines(lines);
+      if (terminal) return terminal.result;
+      if (done) throw new Error('AI 分析连接中断，未收到完整结果，请重试。');
+    }
+  } finally {
+    signal?.removeEventListener('abort', cancel);
+    await reader.cancel().catch(() => {});
+    reader.releaseLock();
+  }
+  function consumeLines(lines) {
+    for (const line of lines) {
+      checkAbort();
+      if (!line.trim()) continue;
+      const terminal = event(line);
+      if (terminal) return terminal;
+    }
+  }
+  function decodedLines(done, value) {
+    buffer += done ? decoder.decode() : decoder.decode(value, {
+      stream: true
+    });
+    const lines = buffer.split('\n');
+    buffer = lines.pop();
+    if (done && buffer.trim()) {
+      lines.push(buffer);
+      buffer = '';
+    }
+    return lines;
+  }
+}