@hupan56/wlkj 2.0.0 → 2.0.1

package/package.json

@@ -1,11 +1,11 @@
 {
   "name": "@hupan56/wlkj",
-  "version": "2.0.0",
-  "description": "AI 产品研发工作流 - PRD/原型/搜索/任务/报告 一键就绪",
+  "version": "2.0.1",
+  "description": "AI Product R&D Workflow - PRD/Prototype/Search/Task/Report",
   "bin": { "wlkj": "./bin/cli.js" },
   "files": ["bin/", "templates/"],
   "keywords": ["workflow", "ai", "prd", "pipeline", "qoder", "product", "team"],
   "license": "MIT",
   "publishConfig": { "access": "public" },
   "engines": { "node": ">=16" }
-}
+}

package/templates/qoder/scripts/context_pack.py

@@ -21,6 +21,7 @@ Usage:
 import os
 import sys
 import json
+import time
 
 # UTF-8 stdio (防御性: stdout 被捕获时不崩溃)
 try:
@@ -71,6 +72,53 @@ def plat_filter(files, target):
     return [f for f in files if target.lower() in f.lower()]
 
 
+def _cache_key(query, platform, page_type):
+    """生成缓存 key: query|platform|type|索引mtime 指纹。
+
+    索引文件没变就命中缓存, 变了自动失效。
+    """
+    import hashlib
+    # 取 3 个主要索引的 mtime 作为指纹
+    mtime_str = ''
+    for idx_file in ['keyword-index.json', 'api-index.json', 'prd-index.json']:
+        p = os.path.join(INDEX_DIR, idx_file)
+        if os.path.isfile(p):
+            mtime_str += '%d_' % os.path.getmtime(p)
+    raw = '{}|{}|{}|{}'.format(query, platform or '', page_type or '', mtime_str)
+    return hashlib.md5(raw.encode('utf-8')).hexdigest()[:16]
+
+
+def _try_cache(key):
+    """读缓存命中则返回内容, 未命中返回 None。"""
+    cache_path = os.path.join(BASE, '.qoder', '.runtime', 'ctx-cache-%s.md' % key)
+    if not os.path.isfile(cache_path):
+        return None
+    try:
+        # 缓存 24 小时有效 (防 mtime 精度问题)
+        age = time.time() - os.path.getmtime(cache_path)
+        if age > 86400:
+            return None
+        with open(cache_path, encoding='utf-8') as f:
+            return f.read()
+    except OSError:
+        return None
+
+
+def _save_cache(key, content):
+    """写缓存 (best-effort, 失败不阻塞)。"""
+    try:
+        cache_dir = os.path.join(BASE, '.qoder', '.runtime')
+        os.makedirs(cache_dir, exist_ok=True)
+        cache_path = os.path.join(cache_dir, 'ctx-cache-%s.md' % key)
+        # 原子写
+        tmp = cache_path + '.tmp'
+        with open(tmp, 'w', encoding='utf-8') as f:
+            f.write(content)
+        os.replace(tmp, cache_path)
+    except OSError:
+        pass
+
+
 def main():
     args = sys.argv[1:]
     platform = None
@@ -88,6 +136,30 @@ def main():
         return 1
     query = args[0]
 
+    # 结果缓存: 同 query+platform+索引未变 → 直接返回 (470ms → <10ms)
+    cache_key = _cache_key(query, platform, page_type)
+    cached = _try_cache(cache_key)
+    if cached is not None:
+        print(cached, end='')
+        return 0
+
+    # 缓存未命中, 正常计算, 末尾写入缓存
+    import io
+    old_stdout = sys.stdout
+    sys.stdout = buf = io.StringIO()
+    try:
+        _compute_and_print(query, platform, page_type)
+    finally:
+        sys.stdout = old_stdout
+    output = buf.getvalue()
+    print(output, end='')
+    _save_cache(cache_key, output)
+    return 0
+
+
+def _compute_and_print(query, platform, page_type):
+    """实际的 context_pack 计算 (原 main 逻辑)。"""
+
     targets = []
     if platform and platform.lower() not in ('both', '两端'):
         targets = [PLATFORM_MAP.get(platform.lower(), platform)]
@@ -189,8 +261,6 @@ def main():
         for w in wikis:
             print('- ' + w)
 
-    return 0
-
 
 if __name__ == '__main__':
     sys.exit(main())