kcode-pi 0.1.0

package/vendor/kingdee-skills/ok-cosmic/scripts/cosmic-post-check.py

@@ -0,0 +1,262 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+"""
+cosmic-post-check.py — 苍穹代码生成后统一检查入口
+
+逻辑：
+1. 从目标文件/目录向上查找 Gradle 项目根目录（同时存在 build.gradle + settings.gradle）
+2. 若找到 → 直接执行 Gradle 编译，编译通过后再执行代码规范检查
+3. 若未找到 → 直接执行 cosmic-post-lint.py 代码规范检查
+
+用法:
+    python3 cosmic-post-check.py <file_or_directory> [--fix-hint] [--json] [--strict]
+
+示例:
+    python3 cosmic-post-check.py /path/to/project/code/module-bos/src/main/java/Foo.java
+    python3 cosmic-post-check.py ./src/main/java/ --fix-hint
+"""
+
+import os
+import re
+import subprocess
+import sys
+from pathlib import Path
+from typing import Dict, Optional
+
+from script_utils import run_cli
+
+SCRIPT_DIR = Path(__file__).resolve().parent
+
+# 全局 verbose 标志，通过 --verbose 开启
+VERBOSE = False
+
+
+def _verbose(msg: str):
+    """在 --verbose 模式下输出诊断信息到 stderr。"""
+    if VERBOSE:
+        print(f"  [verbose] {msg}", file=sys.stderr)
+
+
+# ══════════════════════════════════════════════
+#  Gradle 项目检测
+# ══════════════════════════════════════════════
+
+def find_gradle_root(target: str) -> Optional[Path]:
+    """从目标路径向上查找 Gradle 项目根目录。
+
+    判定条件：目录中同时存在 build.gradle(.kts) 和 settings.gradle(.kts)。
+    """
+    p = Path(target).resolve()
+    if p.is_file():
+        p = p.parent
+    while p != p.parent:
+        has_build = (p / "build.gradle").exists() or (p / "build.gradle.kts").exists()
+        has_settings = (p / "settings.gradle").exists() or (p / "settings.gradle.kts").exists()
+        if has_build and has_settings:
+            return p
+        p = p.parent
+    return None
+
+
+def parse_modules(gradle_root: Path) -> Dict[str, Path]:
+    """解析 settings.gradle 获取 模块名 → 目录 映射。
+
+    支持标准格式：
+        include 'moduleName'
+        project(':moduleName').projectDir = new File('relative/path')
+    """
+    modules: Dict[str, Path] = {}
+    for name in ("settings.gradle", "settings.gradle.kts"):
+        settings = gradle_root / name
+        if settings.exists():
+            break
+    else:
+        return modules
+
+    text = settings.read_text(encoding="utf-8")
+
+    # 收集所有 include 声明
+    includes = re.findall(r"include\s+'([^']+)'", text)
+
+    # 收集 projectDir 映射
+    project_dirs: Dict[str, str] = dict(re.findall(
+        r"project\(':([^']+)'\)\.projectDir\s*=\s*new\s+File\('([^']+)'\)",
+        text,
+    ))
+
+    for mod_name in includes:
+        if mod_name in project_dirs:
+            modules[mod_name] = (gradle_root / project_dirs[mod_name]).resolve()
+        else:
+            modules[mod_name] = (gradle_root / mod_name).resolve()
+
+    return modules
+
+
+def find_module(gradle_root: Path, target: str,
+                modules: Dict[str, Path]) -> Optional[str]:
+    """确定目标文件/目录属于哪个 Gradle 子模块。"""
+    target_path = Path(target).resolve()
+    # 优先匹配最长路径（最具体的模块）
+    best_match: Optional[str] = None
+    best_len = 0
+    for name, mod_dir in modules.items():
+        try:
+            target_path.relative_to(mod_dir)
+            if len(str(mod_dir)) > best_len:
+                best_len = len(str(mod_dir))
+                best_match = name
+        except ValueError:
+            continue
+    return best_match
+
+
+# ══════════════════════════════════════════════
+#  JAVA_HOME 解析（不做版本判断）
+# ══════════════════════════════════════════════
+
+def _read_properties(filepath: Path) -> Dict[str, str]:
+    """读取 .properties 文件为 key→value 字典。"""
+    props: Dict[str, str] = {}
+    if not filepath.exists():
+        return props
+    for line in filepath.read_text(encoding="utf-8").splitlines():
+        line = line.strip()
+        if not line or line.startswith("#") or "=" not in line:
+            continue
+        k, v = line.split("=", 1)
+        props[k.strip()] = v.strip()
+    return props
+
+
+def resolve_java_home(gradle_root: Path) -> Optional[str]:
+    """解析 JAVA_HOME 路径，不做版本判断。
+
+    优先级：gradle.properties 中 org.gradle.java.home → 环境变量 JAVA_HOME
+    """
+    props = _read_properties(gradle_root / "gradle.properties")
+    explicit = props.get("org.gradle.java.home", "").strip()
+    if explicit and os.path.isdir(explicit):
+        return explicit
+    env_home = os.environ.get("JAVA_HOME", "").strip()
+    if env_home and os.path.isdir(env_home):
+        return env_home
+    return None
+
+
+# ══════════════════════════════════════════════
+#  Gradle 编译
+# ══════════════════════════════════════════════
+
+def run_gradle(gradle_root: Path, module: Optional[str], java_home: Optional[str]) -> int:
+    """执行 Gradle 编译，返回退出码。不预判 JDK 版本，直接尝试构建。"""
+    # 定位 gradlew
+    wrapper_name = "gradlew.bat" if sys.platform == "win32" else "gradlew"
+    gradlew = gradle_root / wrapper_name
+    if not gradlew.exists():
+        print("[Warn] gradlew 不存在，尝试使用系统 gradle", file=sys.stderr)
+        cmd_prefix = ["gradle"]
+    else:
+        # 确保 gradlew 有可执行权限
+        if not os.access(gradlew, os.X_OK):
+            os.chmod(gradlew, 0o755)
+        cmd_prefix = [str(gradlew)]
+
+    task = f":{module}:compileJava" if module else "compileJava"
+    cmd = cmd_prefix + [task, "--console=plain"]
+
+    env = os.environ.copy()
+    if java_home:
+        env["JAVA_HOME"] = java_home
+
+    print("══════════════════════════════════════════════")
+    print("  [Build] Gradle 编译检查")
+    print("══════════════════════════════════════════════")
+    print(f"  [Project] 项目根目录: {gradle_root}")
+    if module:
+        print(f"  [Module] 目标模块:   {module}")
+    print(f"  [Java] JAVA_HOME:  {java_home or '(未设置，由 Gradle 自行解析)'}")
+    print(f"  [Command] 编译命令:   {' '.join(cmd)}")
+    print("──────────────────────────────────────────────")
+    print(flush=True)
+
+    result = subprocess.run(cmd, cwd=str(gradle_root), env=env)
+
+    print()
+    print("══════════════════════════════════════════════")
+    if result.returncode == 0:
+        print("  ✔️ Gradle 编译成功，未发现编译错误。")
+    else:
+        print(f"  ✖️ Gradle 编译失败 (exit code {result.returncode})")
+        print("  请根据上述编译错误信息修复代码。")
+    print("══════════════════════════════════════════════")
+
+    return result.returncode
+
+
+# ══════════════════════════════════════════════
+#  Post-Lint 代码规范检查
+# ══════════════════════════════════════════════
+
+def run_post_lint(args: list) -> int:
+    """执行 cosmic-post-lint.py 代码规范检查。"""
+    post_lint = SCRIPT_DIR / "cosmic-post-lint.py"
+    if not post_lint.exists():
+        print(f"✖️ 未找到 {post_lint}", file=sys.stderr)
+        return 1
+
+    print("[Check] 执行代码规范检查")
+    print(flush=True)
+
+    cmd = [sys.executable, str(post_lint)] + args
+    result = subprocess.run(cmd)
+    return result.returncode
+
+
+# ══════════════════════════════════════════════
+#  主入口
+# ══════════════════════════════════════════════
+
+def main():
+    global VERBOSE
+
+    if len(sys.argv) < 2:
+        print("用法: cosmic-post-check.py <file_or_directory> [--fix-hint] [--json] [--strict] [--verbose]")
+        print()
+        print("优先执行 Gradle 编译；编译通过后再执行代码规范检查。非 Gradle 项目直接执行规范检查。")
+        sys.exit(1)
+
+    # 解析 --verbose 并从 argv 中移除（不传给 post-lint）
+    if "--verbose" in sys.argv:
+        VERBOSE = True
+        sys.argv.remove("--verbose")
+
+    target = sys.argv[1]
+    remaining_args = sys.argv[1:]
+
+    # 1. 检测 Gradle 项目
+    gradle_root = find_gradle_root(target)
+    _verbose(f"Gradle 项目检测: {gradle_root or '未找到'}")
+
+    if not gradle_root:
+        # 非 Gradle → 直接执行代码规范检查
+        sys.exit(run_post_lint(remaining_args))
+
+    # 2. 解析 JAVA_HOME（不做版本判断）
+    java_home = resolve_java_home(gradle_root)
+    _verbose(f"JAVA_HOME: {java_home or '未设置'}")
+
+    # 3. 直接执行 Gradle 编译
+    modules = parse_modules(gradle_root)
+    module = find_module(gradle_root, target, modules)
+    exit_code = run_gradle(gradle_root, module, java_home)
+    if exit_code != 0:
+        sys.exit(exit_code)
+
+    # 4. 编译成功 → 执行代码规范检查
+    print("[Tip] Gradle 编译通过，继续执行代码规范检查...")
+    sys.exit(run_post_lint(remaining_args))
+
+
+if __name__ == "__main__":
+    sys.exit(run_cli(main))

package/vendor/kingdee-skills/ok-cosmic/scripts/cosmic-post-lint.py

@@ -0,0 +1,293 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+"""
+cosmic-post-lint.py — 苍穹代码生成后自动校验脚本
+
+依次调用 lint/ 子模块执行各类检查，汇总输出报告。
+
+用法:
+    python3 cosmic-post-lint.py <file_or_directory> [--fix-hint] [--json] [--strict]
+
+示例:
+    python3 cosmic-post-lint.py ./src/main/java/
+    python3 cosmic-post-lint.py MyPlugin.java --fix-hint
+    python3 cosmic-post-lint.py ./src/ --json --strict
+"""
+
+import argparse
+import json
+import os
+import sys
+from concurrent.futures import ThreadPoolExecutor, as_completed
+from pathlib import Path
+from typing import List, Set
+
+from lint.base import Severity, LintIssue, LintReport
+from lint import scene_check, style_check, resource_check, verify_check
+from script_utils import FriendlyArgumentParser, run_cli
+
+SCRIPT_DIR = Path(__file__).resolve().parent
+
+# ──────────────────────────────────────────────
+# 检查器注册表
+# ──────────────────────────────────────────────
+
+# 每个检查器必须暴露 check(filepath, lines) -> List[LintIssue]
+# 新增检查器只需:  1) 在 lint/ 下新建 xxx_check.py  2) 在此列表注册
+CHECKERS = [
+    {"name": "场景错配",   "module": scene_check,    "always": True},
+    {"name": "编码偏好",   "module": style_check,    "always": True},
+    {"name": "资源管理",   "module": resource_check,  "always": True},
+    {"name": "验证注释",   "module": verify_check,    "always": False},  # 仅 --strict 模式
+]
+
+# A 层硬约束：从 rules/a-layer-rules.json 加载（单一可信源），
+# 以保证 lint 行为与 rules/constraints.md、rules/post-check.md 的口径一致。
+def _load_a_layer_rule_ids() -> Set[str]:
+    """从 rules/a-layer-rules.json 加载 A 层规则 ID 集合。"""
+    json_path = SCRIPT_DIR.parent / "rules" / "a-layer-rules.json"
+    try:
+        data = json.loads(json_path.read_text(encoding="utf-8"))
+        return set(data.get("a_layer_rule_ids", []))
+    except (FileNotFoundError, json.JSONDecodeError, OSError) as e:
+        print(f"[Warn] 无法加载 A 层规则配置 ({json_path}): {e}", file=sys.stderr)
+        return set()
+
+A_LAYER_RULE_IDS = _load_a_layer_rule_ids()
+
+
+def get_rule_layer(rule_id: str) -> str:
+    """根据规则 ID 推断当前问题属于 A/B/C 哪一层。"""
+    if rule_id.startswith("VERIFY-"):
+        return "C"
+    if rule_id in A_LAYER_RULE_IDS:
+        return "A"
+    return "B"
+
+
+# ──────────────────────────────────────────────
+# 文件扫描
+# ──────────────────────────────────────────────
+
+def collect_java_files(path: str) -> List[str]:
+    """递归收集所有 .java 文件"""
+    p = Path(path)
+    if p.is_file() and p.suffix == ".java":
+        return [str(p)]
+    elif p.is_dir():
+        return sorted(str(f) for f in p.rglob("*.java"))
+    else:
+        return []
+
+
+def read_file_lines(filepath: str) -> List[str]:
+    """读取文件内容，返回行列表"""
+    try:
+        with open(filepath, "r", encoding="utf-8") as f:
+            return [l.rstrip("\n") for l in f.readlines()]
+    except (UnicodeDecodeError, IOError):
+        return []
+
+
+# ──────────────────────────────────────────────
+# 核心编排
+# ──────────────────────────────────────────────
+
+def check_file(filepath: str, strict: bool = False) -> List[LintIssue]:
+    """对单个 Java 文件依次调用所有检查器"""
+    lines = read_file_lines(filepath)
+    if not lines:
+        return []
+
+    issues: List[LintIssue] = []
+    for checker in CHECKERS:
+        if not checker["always"] and not strict:
+            continue
+        try:
+            issues.extend(checker["module"].check(filepath, lines))
+        except Exception as e:
+            issues.append(LintIssue(
+                file=filepath,
+                line=1,
+                severity=Severity.ERROR,
+                rule_id="LINT-EXCEPTION",
+                message=f"{checker['name']} 检查执行失败: {e}",
+                fix_hint="检查脚本实现或输入文件内容；必要时单独运行该校验脚本定位。",
+            ))
+
+    for issue in issues:
+        if get_rule_layer(issue.rule_id) == "A" and issue.severity != Severity.ERROR:
+            issue.severity = Severity.ERROR
+
+    return issues
+
+
+# ──────────────────────────────────────────────
+# 报告输出
+# ──────────────────────────────────────────────
+
+RED = "\033[91m"
+YELLOW = "\033[93m"
+CYAN = "\033[96m"
+GREEN = "\033[92m"
+GRAY = "\033[90m"
+BOLD = "\033[1m"
+RESET = "\033[0m"
+
+SEVERITY_COLORS = {
+    Severity.ERROR: RED,
+    Severity.WARNING: YELLOW,
+    Severity.INFO: CYAN,
+}
+
+SEVERITY_ICONS = {
+    Severity.ERROR: "✖️",
+    Severity.WARNING: "[WARN]",
+    Severity.INFO: "[INFO]",
+}
+
+
+def print_report(report: LintReport, show_fix_hint: bool = False):
+    """终端友好格式输出"""
+    if not report.issues:
+        print(f"\n{GREEN}{BOLD}✔️ 检查通过！共扫描 {report.total_files} 个文件，未发现问题。{RESET}\n")
+        return
+
+    by_file: dict[str, List[LintIssue]] = {}
+    for issue in report.issues:
+        by_file.setdefault(issue.file, []).append(issue)
+
+    print(f"\n{BOLD}{'═' * 60}{RESET}")
+    print(f"{BOLD}  苍穹代码校验报告{RESET}")
+    print(f"{BOLD}{'═' * 60}{RESET}\n")
+
+    for filepath, file_issues in by_file.items():
+        rel = os.path.relpath(filepath)
+        print(f"  {BOLD}[File] {rel}{RESET}")
+        for issue in sorted(file_issues, key=lambda x: x.line):
+            color = SEVERITY_COLORS[issue.severity]
+            icon = SEVERITY_ICONS[issue.severity]
+            layer = get_rule_layer(issue.rule_id)
+            print(f"    {color}{icon} L{issue.line:>4d} [{layer}/{issue.rule_id}]{RESET} {issue.message}")
+            if issue.source_line:
+                print(f"         {GRAY}> {issue.source_line}{RESET}")
+            if show_fix_hint and issue.fix_hint:
+                print(f"         {GREEN}[Fix] {issue.fix_hint}{RESET}")
+        print()
+
+    print(f"  {BOLD}{'─' * 50}{RESET}")
+    print(f"  [Stats] 扫描文件: {report.total_files}   "
+          f"总问题: {report.total_issues}   "
+          f"{RED}ERROR: {report.errors}{RESET}   "
+          f"{YELLOW}WARNING: {report.warnings}{RESET}   "
+          f"{CYAN}INFO: {report.infos}{RESET}")
+
+    if report.errors > 0:
+        print(f"\n  {RED}{BOLD}[Error] 发现 {report.errors} 个错误，请修复后再提交。{RESET}\n")
+    elif report.warnings > 0:
+        print(f"\n  {YELLOW}{BOLD}[Warn] 发现 {report.warnings} 个警告，建议修复。{RESET}\n")
+    else:
+        print(f"\n  {GREEN}{BOLD}✔️ 仅有建议项，整体良好。{RESET}\n")
+
+
+def print_json_report(report: LintReport):
+    """JSON 格式输出（适合 CI/CD 集成）"""
+    output = {
+        "summary": {
+            "total_files": report.total_files,
+            "total_issues": report.total_issues,
+            "errors": report.errors,
+            "warnings": report.warnings,
+            "infos": report.infos,
+            "passed": report.errors == 0,
+        },
+        "issues": [
+            {
+                "file": os.path.relpath(i.file),
+                "line": i.line,
+                "severity": i.severity.value,
+                "layer": get_rule_layer(i.rule_id),
+                "rule_id": i.rule_id,
+                "message": i.message,
+                "fix_hint": i.fix_hint,
+                "source_line": i.source_line,
+            }
+            for i in report.issues
+        ],
+    }
+    print(json.dumps(output, ensure_ascii=False, indent=2))
+
+
+# ──────────────────────────────────────────────
+# 入口
+# ──────────────────────────────────────────────
+
+def main():
+    parser = FriendlyArgumentParser(
+        description="苍穹代码生成后自动校验脚本",
+        formatter_class=argparse.RawDescriptionHelpFormatter,
+        epilog="""
+检查模块:
+  lint/scene_check.py     场景错配 (SCENE-*)
+  lint/style_check.py     编码偏好 (STYLE-*)
+  lint/resource_check.py  资源管理 (RESOURCE-*)
+  lint/verify_check.py    验证注释治理 (VERIFY-*, C 层，仅 --strict)
+
+扩展方式:
+  1. 在 lint/ 下新建 xxx_check.py，暴露 check(filepath, lines) 函数
+  2. 在 cosmic-post-lint.py 的 CHECKERS 列表中注册
+        """,
+    )
+    parser.add_argument("path", help="要检查的 Java 文件或目录路径")
+    parser.add_argument("--fix-hint", action="store_true", help="显示修复建议")
+    parser.add_argument("--json", action="store_true", help="输出 JSON 格式（适合 CI）")
+    parser.add_argument("--strict", action="store_true",
+                        help="严格模式：额外启用 C 层验证来源治理检查")
+    parser.add_argument("--min-severity", choices=["error", "warning", "info"],
+                        default="info", help="最低报告级别 (默认: info)")
+
+    args = parser.parse_args()
+
+    severity_filter = {
+        "error": {Severity.ERROR},
+        "warning": {Severity.ERROR, Severity.WARNING},
+        "info": {Severity.ERROR, Severity.WARNING, Severity.INFO},
+    }[args.min_severity]
+
+    files = collect_java_files(args.path)
+    if not files:
+        print(f"✖️ 未找到 Java 文件: {args.path}", file=sys.stderr)
+        sys.exit(1)
+
+    report = LintReport(total_files=len(files))
+
+    def _check_one(fp):
+        return fp, check_file(fp, strict=args.strict)
+
+    if len(files) == 1:
+        all_results = [_check_one(files[0])]
+    else:
+        all_results = [None] * len(files)
+        with ThreadPoolExecutor(max_workers=min(len(files), 4)) as pool:
+            future_to_idx = {
+                pool.submit(_check_one, fp): i
+                for i, fp in enumerate(files)
+            }
+            for future in as_completed(future_to_idx):
+                all_results[future_to_idx[future]] = future.result()
+
+    for _, issues in all_results:
+        for issue in issues:
+            if issue.severity in severity_filter:
+                report.add(issue)
+
+    if args.json:
+        print_json_report(report)
+    else:
+        print_report(report, show_fix_hint=args.fix_hint)
+
+    sys.exit(1 if report.errors > 0 else 0)
+
+
+if __name__ == "__main__":
+    sys.exit(run_cli(main))

package/vendor/kingdee-skills/ok-cosmic/scripts/lint/__init__.py

@@ -0,0 +1,2 @@
+# -*- coding: utf-8 -*-
+"""ok-cosmic post-lint 检查模块包。"""