@hupan56/wlkj 2.7.12 → 3.1.2

package/templates/qoder/scripts/{setup.py → deployment/setup/setup.py}

@@ -1,5 +1,17 @@
 #!/usr/bin/env python3
 # -*- coding: utf-8 -*-
+# v3.0 路径自举: 引导到 common/bootstrap, 统一 sys.path 逻辑
+import os as _o, sys as _s
+_f = _o.path.abspath(__file__)
+for _ in range(10):
+    _f = _o.path.dirname(_f)
+    _cp = _o.path.join(_f, 'foundation')
+    if _o.path.isfile(_o.path.join(_cp, 'bootstrap.py')):
+        break
+if _cp not in _s.path: _s.path.insert(0, _cp)
+from bootstrap import setup; setup()
+from foundation.core.paths import get_repo_root
+
 """
 setup.py - 一键就绪 (小而美)
 
@@ -23,6 +35,7 @@ setup.py - 一键就绪 (小而美)
 
 import argparse
 import os
+import shutil
 import subprocess
 import sys
 from pathlib import Path
@@ -34,7 +47,7 @@ except (AttributeError, TypeError, OSError):
     pass
 
 THIS_DIR = Path(__file__).resolve().parent
-BASE = THIS_DIR.parent.parent  # repo root
+BASE = get_repo_root()
 
 
 # ============================================================
@@ -420,10 +433,78 @@ def _write_team_remote_to_config(url):
         pass
 
 
+def _detect_local_branch(base):
+    """探测当前本地分支名。unborn 分支 (git init 后没 commit) 时,
+    git rev-parse --abbrev-ref HEAD 会报 fatal (exit 128), 必须用 symbolic-ref。
+    不同 git 版本默认分支是 master 或 main, 不能写死。"""
+    r = run(['git', 'symbolic-ref', '--short', 'HEAD'], cwd=str(base))
+    if r.returncode == 0 and r.stdout.strip():
+        return r.stdout.strip()
+    # 兜底 (理论上 symbolic-ref 不会失败, 但稳健起见)
+    r2 = run(['git', 'rev-parse', '--abbrev-ref', 'HEAD'], cwd=str(base))
+    if r2.returncode == 0 and r2.stdout.strip() and r2.stdout.strip() != 'HEAD':
+        return r2.stdout.strip()
+    return 'master'
+
+
+def _backup_local_data(base, remote_branch):
+    """reset --hard 会用远程覆盖整个工作树。开发者可能本地填了 config.yaml 的真实
+    mysql/feishu 密码但没 commit, 直接 reset 会吞掉这些数据。
+    在 reset 前, 把"团队数据文件"里被本地修改过的内容备份到 .wlkj-pre-reset/,
+    reset 后提示用户: 这些是 reset 前的本地版本, 你可能需要把密码重新填回去。
+    引擎文件 (gitignored) 不需要备份 — 反正由 update 重新生成。
+    返回被备份的原始相对路径列表 (用 / 分隔, 与 cli.js JS 版一致)。
+    """
+    backup_dir = base / '.wlkj-pre-reset'
+    backed = []
+    # 只备份团队数据文件 (git 跟踪的关键数据), 不备份引擎。
+    # 用正斜杠统一, 避免跨平台路径分隔符不一致 (Windows os.path.join 用反斜杠)。
+    DATA_FILES = ['.qoder/config.yaml', '.qoder/config.toml', '.qoder/settings.json']
+    backup_dir.mkdir(parents=True, exist_ok=True)
+    try:
+        for rel in DATA_FILES:
+            local = base / rel
+            if not local.is_file():
+                continue
+            try:
+                lc = local.read_text(encoding='utf-8')
+            except Exception:
+                continue
+            # 拿远程版本对比
+            # 注意: MSYS2/Git Bash 会把 'origin/master:path' 里的 / 转成 \ →
+            # 变成 'origin\master;path' 报 ambiguous argument。
+            # 设 MSYS_NO_PATHCONV=1 禁用转换 (Windows 原生 cmd 不受影响)。
+            rshow = run(['git', 'show', f'origin/{remote_branch}:{rel}'],
+                        cwd=str(base),
+                        env={**os.environ, 'MSYS_NO_PATHCONV': '1',
+                             'MSYS2_ARG_CONV_EXCL': '*'})
+            remote_content = rshow.stdout if rshow.returncode == 0 else None
+            # 只在本地与远程不同 (或远程没有) 时备份
+            same = (remote_content is not None and remote_content.strip() == lc.strip())
+            if not same:
+                # 扁平化文件名: .qoder/config.yaml -> .qoder__config.yaml (跨平台)
+                flat = rel.replace('/', '__').replace('\\', '__')
+                target = backup_dir / flat
+                shutil.copy2(str(local), str(target))
+                backed.append(rel)
+    except Exception:
+        pass  # 备份失败不阻塞 reset, 但会少一层保护
+    return backed
+
+
+
 def _setup_tracking(base):
     """git remote add 之后调: fetch 远程 + 设置当前分支的上游 tracking。
     没这步, git pull 会报 "no tracking information for the current branch"。
     fetch 可能因网络/认证失败 — 失败只警告, 不阻塞 (remote 至少配上了)。
+
+    关键修复 (Trellis 化): 如果本地是空的 git init 历史 (没有 commit) 或与远程
+    不相关 (unrelated histories), 就直接 git reset --hard origin/<branch> 采用远程历史。
+    原理: 引擎代码现在由 npm 包生成 (不在 git 里), 团队数据 (config/data/workspace) 在
+    远程 → 本地空历史直接采用远程是唯一正确姿势, 彻底消除 "refusing to merge unrelated
+    histories"。
+    数据安全: reset 前会先把本地修改过的团队数据文件 (config 等) 备份到 .wlkj-pre-reset/,
+    防止开发者未提交的真实密码/配置被远程覆盖丢失。
     """
     # fetch 远程分支
     r = run(['git', 'fetch', 'origin'], cwd=str(base))
@@ -435,9 +516,7 @@ def _setup_tracking(base):
         else:
             warn('git fetch 失败 (不阻塞, 之后 /wl-init 时重试): ' + err[:80])
         return
-    # 确定本地当前分支名
-    br_r = run(['git', 'rev-parse', '--abbrev-ref', 'HEAD'], cwd=str(base))
-    local_branch = br_r.stdout.strip() if br_r.returncode == 0 else 'master'
+
     # 远程有 master 还是 main? 优先 master (团队约定), fallback main
     remote_branch = None
     for cand in ('master', 'main'):
@@ -447,10 +526,46 @@ def _setup_tracking(base):
             break
     if not remote_branch:
         return  # 远程没分支(空仓库), 没法设 tracking, 等 push 后再说
-    # 设置上游 tracking
+
+    # 确定本地当前分支名 (unborn 分支用 symbolic-ref, 不能用 rev-parse)
+    local_branch = _detect_local_branch(base)
+
+    # === 检测本地是否有有效历史 (是否有 commit) ===
+    head_r = run(['git', 'rev-parse', '--verify', 'HEAD'], cwd=str(base))
+    has_local_commit = head_r.returncode == 0
+    # 探测 unrelated histories (即便本地有 commit, 无公共祖先也需采用远程)
+    unrelated = False
+    if has_local_commit:
+        mb_r = run(['git', 'merge-base', local_branch, f'origin/{remote_branch}'], cwd=str(base))
+        unrelated = (mb_r.returncode != 0)
+
+    if not has_local_commit or unrelated:
+        # 本地无有效历史 (git init 没提交过) 或与远程无公共祖先。
+        # 本地这点平行历史无价值 (团队真实历史都在远程) → 采用远程历史。
+        # 但 reset --hard 会覆盖工作树 → 先备份本地改过的团队数据, 防丢密码。
+        if unrelated:
+            warn(f'检测到无关历史 (本地与 origin/{remote_branch} 无公共祖先) — 采用远程历史')
+        backed = _backup_local_data(base, remote_branch)
+        rr = run(['git', 'reset', '--hard', f'origin/{remote_branch}'], cwd=str(base))
+        if rr.returncode == 0:
+            ok(f'已采用远程历史 (origin/{remote_branch})')
+            if backed:
+                warn('reset 前已备份你本地修改过的数据文件 (可能含未提交的真实密码/配置):')
+                for rel in backed:
+                    warn(f'  → .wlkj-pre-reset/{rel.replace("/", "__")}  (reset 前的本地版本)')
+                warn('请检查这些文件, 把需要的配置/密码重新填回 .qoder/config.yaml 后删除备份目录。')
+        else:
+            warn('reset --hard 失败: ' + (rr.stderr or '').strip()[:80])
+        # reset 后本地分支已指向远程 commit, 现在能设上游了
+        run(['git', 'branch', f'--set-upstream-to=origin/{remote_branch}', local_branch], cwd=str(base))
+        return
+
+    # 有公共祖先, 正常设上游
     run(['git', 'branch', f'--set-upstream-to=origin/{remote_branch}', local_branch], cwd=str(base))
 
 
+
+
 def configure_team_remote():
     """配置团队协作仓库的 git remote (workspace/ 产出往这里 push)。
 
@@ -532,10 +647,12 @@ def offer_cron(skip=False):
         return
 
     if ask('注册周五 18:00 自动更新知识图谱? (团队只需一台机器注册)', 'y'):
+        # cron 脚本在 scripts/ 根 (THIS_DIR = scripts/deployment/setup/, 上两级)
+        cron_dir = THIS_DIR.parent.parent
         if sys.platform == 'win32':
-            r = run(['cmd', '/c', str(THIS_DIR / 'setup_weekly_cron.bat')], cwd=str(BASE))
+            r = run(['cmd', '/c', str(cron_dir / 'setup_weekly_cron.bat')], cwd=str(BASE))
         else:
-            r = run(['bash', str(THIS_DIR / 'setup_weekly_cron.sh')], cwd=str(BASE))
+            r = run(['bash', str(cron_dir / 'setup_weekly_cron.sh')], cwd=str(BASE))
         if r.returncode == 0:
             ok('cron 已注册 (周五 18:00)')
         else:
@@ -655,13 +772,25 @@ def print_ready(name, role):
         if projects:
             ok(f'源码仓库: {len(projects)} 个项目 ({", ".join(projects[:3])})')
 
-    # 索引
-    ki = BASE / 'data' / 'index' / 'code-keyword.json'
-    if ki.is_file():
+    # 知识图谱 (DuckDB 是唯一事实源; 旧 code-keyword.json 已废弃)
+    kg = BASE / 'data' / 'index' / 'kg.duckdb'
+    if kg.is_file():
         try:
-            import json
-            data = json.loads(ki.read_text(encoding='utf-8'))
-            ok(f'知识图谱: {len(data)} keywords')
+            size_kb = kg.stat().st_size // 1024
+            # 尝试读关键词行数作为"有内容"的证据 (读不到只报大小)
+            kw_n = None
+            try:
+                import duckdb as _ddb
+                con = _ddb.connect(str(kg), read_only=True)
+                row = con.execute('SELECT COUNT(*) FROM keywords').fetchone()
+                con.close()
+                kw_n = row[0] if row else None
+            except Exception:
+                pass
+            if kw_n:
+                ok(f'知识图谱: {kw_n} keywords (kg.duckdb {size_kb} KB)')
+            else:
+                ok(f'知识图谱: kg.duckdb {size_kb} KB')
         except Exception:
             pass
 
@@ -684,7 +813,7 @@ def print_ready(name, role):
 
     print()
     print('  现在可以开始了:')
-    print('    /wl-prd-full 写需求      /wl-search 查代码      /wl-task 建任务')
+    print('    /wl-prd 写需求        /wl-search 查代码      /wl-task 建任务')
     print('    /wl-status 看进度   /wl-report 出日报      /wl-commit 提交')
     print()
     print('  或直接用自然语言:')