@hupan56/wlkj 2.0.0

package/templates/qoder/scripts/build_style_index.py

@@ -0,0 +1,268 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+"""
+Build style index from Vue/HTML files.
+Extracts UI patterns: tables, forms, layouts, colors, components.
+
+Usage:
+  python build_style_index.py
+"""
+import os, json, re, sys
+from datetime import datetime
+
+# UTF-8 stdio (防御性: stdout 被捕获时不崩溃)
+try:
+    sys.stdout.reconfigure(encoding='utf-8', errors='replace')
+except (AttributeError, TypeError, OSError, IOError):
+    try:
+        sys.stdout.reconfigure(encoding='utf-8')
+    except Exception:
+        pass
+
+BASE = os.path.dirname(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
+DATA_CODE = os.path.join(BASE, 'data', 'code')
+INDEX_DIR = os.path.join(BASE, 'data', 'index')
+STYLE_INDEX_PATH = os.path.join(INDEX_DIR, 'style-index.json')
+STYLE_META_PATH = os.path.join(INDEX_DIR, 'style-meta.json')
+
+
+def extract_vue_patterns(filepath, content):
+    """Extract UI patterns from a Vue SFC file."""
+    patterns = {
+        'file': filepath.replace('\\', '/'),
+        'type': 'unknown',
+        'components': [],
+        'table_columns': [],
+        'form_fields': [],
+        'form_components': [],
+        'layout': {},
+    }
+
+    # Detect page type. Detectors cover multiple UI stacks so the index
+    # generalizes beyond this repo (Vben/VxeGrid, Ant Design, Element, Vant).
+    TABLE_MARKS = ('VxeGrid', 'vxe-grid', 'useVbenVxeGrid', '<a-table', '<el-table',
+                   '<vxe-table', '<van-list')
+    FORM_MARKS = ('useVbenForm', 'useForm', '<a-form', '<el-form', '<van-form')
+    if any(m in content for m in TABLE_MARKS):
+        patterns['type'] = 'table-page'
+    elif any(m in content for m in FORM_MARKS):
+        patterns['type'] = 'form-page'
+    elif 'detail' in filepath.lower() or 'Detail' in content:
+        patterns['type'] = 'detail-page'
+    elif 'modal' in filepath.lower() or 'drawer' in filepath.lower():
+        patterns['type'] = 'modal'
+    elif 'dashboard' in filepath.lower() or 'chart' in content.lower():
+        patterns['type'] = 'dashboard'
+
+    # Extract Ant Design component imports
+    ant_matches = re.findall(r"import\s+\{([^}]+)\}\s+from\s+['\"]ant-design-vue['\"]", content)
+    for m in ant_matches:
+        comps = [c.strip() for c in m.split(',') if c.strip()]
+        patterns['components'].extend(comps)
+
+    # Extract table columns: title + field
+    col_matches = re.findall(r"title:\s*['\"]([^'\"]+)['\"]\s*,\s*field:\s*['\"]([^'\"]+)['\"]", content)
+    if not col_matches:
+        col_matches = re.findall(r"field:\s*['\"]([^'\"]+)['\"]\s*,\s*title:\s*['\"]([^'\"]+)['\"]", content)
+        col_matches = [(t, f) for f, t in col_matches]
+    for title, field in col_matches:
+        patterns['table_columns'].append({'title': title, 'field': field})
+
+    # Extract form fields: fieldName + label
+    form_matches = re.findall(r"fieldName:\s*['\"]([^'\"]+)['\"]\s*,\s*label:\s*['\"]([^'\"]+)['\"]", content)
+    for field, label in form_matches:
+        patterns['form_fields'].append({'field': field, 'label': label})
+
+    # Extract form component types
+    comp_matches = re.findall(r"component:\s*['\"]([^'\"]+)['\"]", content)
+    patterns['form_components'] = list(set(comp_matches))
+
+    # Extract layout
+    label_w = re.findall(r'labelWidth:\s*(\d+)', content)
+    if label_w:
+        patterns['layout']['label_width'] = int(label_w[0])
+    grid_c = re.findall(r'grid-cols-(\d+)', content)
+    if grid_c:
+        patterns['layout']['grid_cols'] = [int(x) for x in grid_c]
+
+    return patterns
+
+
+def scan_project(proj_name, proj_path):
+    """Scan a single frontend project."""
+    pages = []
+    component_counter = {}
+    field_map = {}
+    form_components = {}
+    vue_count = 0
+    data_count = 0
+
+    for root, dirs, files in os.walk(proj_path):
+        dirs[:] = [d for d in dirs if d not in ['node_modules', 'dist', '.git', '__pycache__', '.svn']]
+
+        for f in files:
+            filepath = os.path.join(root, f)
+            rel_path = os.path.relpath(filepath, proj_path).replace('\\', '/')
+
+            if f.endswith('.vue'):
+                vue_count += 1
+                try:
+                    content = open(filepath, encoding='utf-8', errors='ignore').read()
+                    p = extract_vue_patterns(proj_name + '/' + rel_path, content)
+                    pages.append(p)
+
+                    for comp in p.get('components', []):
+                        component_counter[comp] = component_counter.get(comp, 0) + 1
+                    for col in p.get('table_columns', []):
+                        k = col['field']
+                        if k not in field_map:
+                            field_map[k] = {'field': k, 'titles': set(), 'files': []}
+                        field_map[k]['titles'].add(col['title'])
+                        field_map[k]['files'].append(proj_name + '/' + rel_path)
+                    for comp in p.get('form_components', []):
+                        form_components[comp] = form_components.get(comp, 0) + 1
+                except Exception as e:
+                    pass  # Skip unreadable files
+
+            elif f.endswith(('.tsx', '.ts')) and f.startswith('data'):
+                data_count += 1
+                try:
+                    content = open(filepath, encoding='utf-8', errors='ignore').read()
+                    if 'columns' in content or 'schema' in content:
+                        col_m = re.findall(r"title:\s*['\"]([^'\"]+)['\"]\s*,\s*field:\s*['\"]([^'\"]+)['\"]", content)
+                        for title, field in col_m:
+                            k = field
+                            if k not in field_map:
+                                field_map[k] = {'field': k, 'titles': set(), 'files': []}
+                            field_map[k]['titles'].add(title)
+                            field_map[k]['files'].append(proj_name + '/' + rel_path)
+
+                        comp_m = re.findall(r"component:\s*['\"]([^'\"]+)['\"]", content)
+                        for comp in comp_m:
+                            form_components[comp] = form_components.get(comp, 0) + 1
+                except:
+                    pass
+
+    return {
+        'pages': pages,
+        'component_counter': component_counter,
+        'field_map': field_map,
+        'form_components': form_components,
+        'vue_count': vue_count,
+        'data_count': data_count,
+    }
+
+
+def main():
+    print('Building style index...')
+    print('Code dir: ' + DATA_CODE)
+
+    if not os.path.isdir(DATA_CODE):
+        print('ERROR: data/code not found!')
+        return
+
+    style_index = {
+        'version': '1.0',
+        'updated': datetime.now().isoformat(),
+        'projects': {},
+        'common_components': {},
+        'form_components': {},
+        'field_map': {},
+        'page_templates': {},
+    }
+
+    total_component_counter = {}
+    total_form_components = {}
+    total_field_map = {}
+
+    for proj in sorted(os.listdir(DATA_CODE)):
+        proj_path = os.path.join(DATA_CODE, proj)
+        if not os.path.isdir(proj_path):
+            continue
+
+        print('Scanning: ' + proj + '...')
+        result = scan_project(proj, proj_path)
+        print('  Vue: ' + str(result['vue_count']) + ', Data: ' + str(result['data_count']))
+
+        style_index['projects'][proj] = {
+            'vue_files': result['vue_count'],
+            'data_files': result['data_count'],
+            'page_types': {},
+            'page_examples': {},
+        }
+
+        # Count page types and keep sample files per type so
+        # `search_index.py --style <type>` can return real reference pages.
+        # Prefer richer pages (more table columns / form fields) as examples.
+        pages_by_type = {}
+        for page in result['pages']:
+            pt = page.get('type', 'unknown')
+            style_index['projects'][proj]['page_types'][pt] = style_index['projects'][proj]['page_types'].get(pt, 0) + 1
+            pages_by_type.setdefault(pt, []).append(page)
+        for pt, pages in pages_by_type.items():
+            pages.sort(key=lambda p: len(p.get('table_columns', [])) + len(p.get('form_fields', [])), reverse=True)
+            style_index['projects'][proj]['page_examples'][pt] = sorted(
+                [p['file'] for p in pages[:15]])
+
+        # Merge component counters
+        for comp, count in result['component_counter'].items():
+            total_component_counter[comp] = total_component_counter.get(comp, 0) + count
+
+        for comp, count in result['form_components'].items():
+            total_form_components[comp] = total_form_components.get(comp, 0) + count
+
+        for k, v in result['field_map'].items():
+            if k not in total_field_map:
+                total_field_map[k] = v
+            else:
+                total_field_map[k]['titles'].update(v['titles'])
+                total_field_map[k]['files'].extend(v['files'])
+
+    # Sort and save
+    style_index['common_components'] = dict(sorted(total_component_counter.items(), key=lambda x: x[1], reverse=True)[:30])
+    style_index['form_components'] = dict(sorted(total_form_components.items(), key=lambda x: x[1], reverse=True))
+
+    # Convert sets to lists for JSON
+    field_map_serializable = {}
+    for k, v in total_field_map.items():
+        field_map_serializable[k] = {
+            'field': k,
+            'titles': sorted(v['titles']),
+            'count': len(v['files']),
+            'sample_files': v['files'][:5]
+        }
+    style_index['field_map'] = field_map_serializable
+
+    # Save
+    os.makedirs(INDEX_DIR, exist_ok=True)
+
+    with open(STYLE_INDEX_PATH, 'w', encoding='utf-8') as f:
+        json.dump(style_index, f, ensure_ascii=False, indent=2, sort_keys=True)
+
+    # Meta (small) - this is what hooks read at session start
+    meta = {
+        'version': style_index['version'],
+        'updated': style_index['updated'],
+        'projects': style_index['projects'],
+        'common_components': style_index['common_components'],
+        'form_components': style_index['form_components'],
+        'field_count': len(style_index['field_map']),
+    }
+    # Keep meta light: drop per-type example file lists
+    meta['projects'] = {
+        p: {k: v for k, v in pdata.items() if k != 'page_examples'}
+        for p, pdata in meta['projects'].items()
+    }
+    with open(STYLE_META_PATH, 'w', encoding='utf-8') as f:
+        json.dump(meta, f, ensure_ascii=False, indent=2, sort_keys=True)
+
+    print()
+    print('=== Done ===')
+    print('Common components: ' + str(len(style_index['common_components'])))
+    print('Form components: ' + str(len(style_index['form_components'])))
+    print('Field map: ' + str(len(style_index['field_map'])) + ' entries')
+    print('Saved to: ' + STYLE_META_PATH)
+
+
+if __name__ == '__main__':
+    main()

package/templates/qoder/scripts/code_index.py

@@ -0,0 +1,41 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+"""
+DEPRECATED: code_index.py used to keep its own (cruder) index builder
+whose schema conflicted with git_sync.py's - running it would overwrite
+the real indexes. It now delegates to the single implementation.
+
+Usage:
+  python code_index.py scan          # full index rebuild (= git_sync.py --index-only)
+  python code_index.py search <kw>   # use search_index.py instead
+"""
+
+import os
+import sys
+
+# UTF-8 stdio (防御性: stdout 被捕获时不崩溃)
+try:
+    sys.stdout.reconfigure(encoding='utf-8', errors='replace')
+except (AttributeError, TypeError, OSError, IOError):
+    try:
+        sys.stdout.reconfigure(encoding='utf-8')
+    except Exception:
+        pass
+
+sys.path.insert(0, os.path.dirname(os.path.abspath(__file__)))
+
+if __name__ == '__main__':
+    cmd = sys.argv[1] if len(sys.argv) > 1 else ''
+    if cmd == 'scan':
+        from git_sync import build_full_indexes, FAILURES
+        build_full_indexes()
+        sys.exit(1 if FAILURES else 0)
+    elif cmd == 'search' and len(sys.argv) > 2:
+        import subprocess
+        script = os.path.join(os.path.dirname(os.path.abspath(__file__)), 'search_index.py')
+        sys.exit(subprocess.call([sys.executable, script] + sys.argv[2:]))
+    else:
+        print('Usage:')
+        print('  python code_index.py scan          # full index rebuild')
+        print('  python code_index.py search <kw>   # -> search_index.py')
+        sys.exit(1)

package/templates/qoder/scripts/collect_prds.py

@@ -0,0 +1,31 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+"""
+Collect PRDs from all user workspaces to data/docs/prd/
+
+THIN WRAPPER: the actual logic lives in git_sync.py (single source of
+truth). This entry point is kept for backwards compatibility.
+
+Usage:
+  python collect_prds.py        # same as the collect step of: python git_sync.py --prd-only
+"""
+
+import os
+import sys
+
+# UTF-8 stdio (防御性: stdout 被捕获时不崩溃)
+try:
+    sys.stdout.reconfigure(encoding='utf-8', errors='replace')
+except (AttributeError, TypeError, OSError, IOError):
+    try:
+        sys.stdout.reconfigure(encoding='utf-8')
+    except Exception:
+        pass
+
+sys.path.insert(0, os.path.dirname(os.path.abspath(__file__)))
+
+from git_sync import collect_prds, FAILURES
+
+if __name__ == '__main__':
+    collect_prds()
+    sys.exit(1 if FAILURES else 0)

package/templates/qoder/scripts/common/active_task.py

@@ -0,0 +1,230 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+"""
+QODER Pipeline - 会话级活跃任务管理
+
+核心概念: "活跃任务" 是按 AI 会话/窗口存储的, 不是全局的。
+
+存储位置: .qoder/.runtime/sessions/<session-id>
+每个会话文件只包含一行: 活跃任务的相对路径。
+
+会话身份来源 (按优先级):
+1. 环境变量: QODER_SESSION_ID, CODEX_SESSION_ID, CODEX_THREAD_ID 等
+2. Hook 输入中的 session_id 字段
+3. TRELLIS_CONTEXT_ID 环境变量
+4. 后备: 全局 .qoder/.current-task 文件
+
+参考: Trellis 的 common/active_task.py 设计
+"""
+
+from __future__ import annotations
+
+import hashlib
+import json
+import os
+import sys
+from dataclasses import dataclass
+from pathlib import Path
+from typing import Any
+
+# 支持的 AI 平台及其会话环境变量
+_ENV_SESSION_KEYS: tuple[tuple[str, tuple[str, ...]], ...] = (
+    ("qoder", ("QODER_SESSION_ID",)),
+    ("codex", ("CODEX_SESSION_ID", "CODEX_THREAD_ID")),
+    ("claude", ("CLAUDE_SESSION_ID", "CLAUDE_CODE_SESSION_ID")),
+    ("cursor", ("CURSOR_SESSION_ID",)),
+    ("opencode", ("OPENCODE_SESSION_ID", "OPENCODE_RUN_ID")),
+    ("gemini", ("GEMINI_SESSION_ID",)),
+    ("copilot", ("COPILOT_SESSION_ID",)),
+)
+
+
+@dataclass(frozen=True)
+class ActiveTask:
+    """活跃任务状态。
+
+    Attributes:
+        task_path: 任务相对路径, None 表示无活跃任务。
+        source_type: 来源类型 (session / global-fallback / none)。
+        context_key: 会话标识键。
+        stale: 是否过期。
+    """
+    task_path: str | None
+    source_type: str  # "session" | "global-fallback" | "none"
+    context_key: str | None = None
+    stale: bool = False
+
+    @property
+    def source(self) -> str:
+        """人类可读的来源标签。"""
+        if self.source_type == "session" and self.context_key:
+            return f"session:{self.context_key}"
+        if self.source_type == "global-fallback":
+            return "global-fallback"
+        return "none"
+
+
+# =============================================================================
+# 会话身份解析
+# =============================================================================
+
+def resolve_context_key() -> str | None:
+    """解析当前会话的唯一标识键。
+
+    按平台优先级检查环境变量, 返回第一个找到的值。
+
+    Returns:
+        会话 ID 字符串, 无则返回 None。
+    """
+    # 1. 检查通用环境变量
+    for env_var in ("TRELLIS_CONTEXT_ID", "QODER_CONTEXT_ID"):
+        val = os.environ.get(env_var)
+        if val:
+            return val
+
+    # 2. 检查各平台专用环境变量
+    for _platform, env_vars in _ENV_SESSION_KEYS:
+        for env_var in env_vars:
+            val = os.environ.get(env_var)
+            if val:
+                return val
+
+    return None
+
+
+def _get_session_file(session_key: str, repo_root: Path) -> Path:
+    """获取会话状态文件路径。
+
+    Args:
+        session_key: 会话标识。
+        repo_root: 项目根目录。
+
+    Returns:
+        会话文件路径。
+    """
+    # 对 session_key 做 hash 避免特殊字符问题
+    key_hash = hashlib.sha256(session_key.encode()).hexdigest()[:16]
+    sessions_dir = repo_root / ".qoder" / ".runtime" / "sessions"
+    sessions_dir.mkdir(parents=True, exist_ok=True)
+    return sessions_dir / f"session-{key_hash}"
+
+
+# =============================================================================
+# 活跃任务读写
+# =============================================================================
+
+def resolve_active_task(repo_root: Path | None = None) -> ActiveTask:
+    """解析当前会话的活跃任务。
+
+    优先级:
+    1. 会话级状态文件
+    2. 全局 .current-task 文件 (后备)
+
+    Args:
+        repo_root: 项目根目录, 默认自动检测。
+
+    Returns:
+        ActiveTask 对象。
+    """
+    if repo_root is None:
+        from common.paths import get_repo_root
+        repo_root = get_repo_root()
+
+    # 1. 尝试会话级
+    session_key = resolve_context_key()
+    if session_key:
+        session_file = _get_session_file(session_key, repo_root)
+        if session_file.is_file():
+            try:
+                task_path = session_file.read_text(encoding="utf-8").strip()
+                if task_path:
+                    return ActiveTask(
+                        task_path=task_path,
+                        source_type="session",
+                        context_key=session_key,
+                    )
+            except (OSError, IOError):
+                pass
+
+    # 2. 后备: 全局文件
+    from common.paths import get_current_task
+    global_task = get_current_task(repo_root)
+    if global_task:
+        return ActiveTask(
+            task_path=global_task,
+            source_type="global-fallback",
+            context_key=session_key,
+        )
+
+    return ActiveTask(task_path=None, source_type="none")
+
+
+def set_active_task(task_path: str, repo_root: Path | None = None) -> ActiveTask | None:
+    """设置当前会话的活跃任务。
+
+    Args:
+        task_path: 任务相对路径。
+        repo_root: 项目根目录, 默认自动检测。
+
+    Returns:
+        ActiveTask 对象, 无会话身份则返回 None。
+    """
+    if repo_root is None:
+        from common.paths import get_repo_root
+        repo_root = get_repo_root()
+
+    session_key = resolve_context_key()
+    if not session_key:
+        # 无会话身份, 使用全局后备
+        from common.paths import set_current_task
+        set_current_task(task_path, repo_root)
+        return ActiveTask(
+            task_path=task_path,
+            source_type="global-fallback",
+            context_key=None,
+        )
+
+    session_file = _get_session_file(session_key, repo_root)
+    try:
+        session_file.parent.mkdir(parents=True, exist_ok=True)
+        session_file.write_text(task_path, encoding="utf-8")
+        return ActiveTask(
+            task_path=task_path,
+            source_type="session",
+            context_key=session_key,
+        )
+    except (OSError, IOError) as e:
+        print(f"Warning: Failed to write session file: {e}", file=sys.stderr)
+        return None
+
+
+def clear_active_task(repo_root: Path | None = None) -> ActiveTask:
+    """清除当前会话的活跃任务。
+
+    Args:
+        repo_root: 项目根目录, 默认自动检测。
+
+    Returns:
+        清除前的 ActiveTask 状态。
+    """
+    if repo_root is None:
+        from common.paths import get_repo_root
+        repo_root = get_repo_root()
+
+    # 获取清除前的状态
+    active = resolve_active_task(repo_root)
+
+    # 清除会话文件
+    session_key = resolve_context_key()
+    if session_key:
+        session_file = _get_session_file(session_key, repo_root)
+        try:
+            session_file.unlink(missing_ok=True)
+        except (OSError, IOError):
+            pass
+
+    # 也清除全局后备
+    from common.paths import set_current_task
+    set_current_task(None, repo_root)
+
+    return active