autosnippet 3.2.7 → 3.2.9

package/lib/service/module/ModuleService.js

@@ -88,8 +88,8 @@ export class ModuleService {
   #logger;
 
   // AI pipeline deps
-  #aiFactory;
-  #chatAgent;
+  #agentFactory;
+  #container;
   #qualityScorer;
   #recipeExtractor;
   #guardCheckEngine;
@@ -98,8 +98,8 @@ export class ModuleService {
   /**
    * @param {string} projectRoot
    * @param {object} [options]
-   * @param {object} [options.aiFactory]
-   * @param {object} [options.chatAgent]
+   * @param {object} [options.agentFactory]
+   * @param {object} [options.container]
    * @param {object} [options.qualityScorer]
    * @param {object} [options.recipeExtractor]
    * @param {object} [options.guardCheckEngine]
@@ -109,8 +109,8 @@ export class ModuleService {
     this.#projectRoot = projectRoot;
     this.#registry = getDiscovererRegistry();
     this.#logger = Logger.getInstance();
-    this.#aiFactory = options.aiFactory || null;
-    this.#chatAgent = options.chatAgent || null;
+    this.#agentFactory = options.agentFactory || null;
+    this.#container = options.container || null;
     this.#qualityScorer = options.qualityScorer || null;
     this.#recipeExtractor = options.recipeExtractor || null;
     this.#guardCheckEngine = options.guardCheckEngine || null;
@@ -314,7 +314,16 @@ export class ModuleService {
     const allNodes = [];
     const allEdges = [];
 
+    // 如果有专业 Discoverer（非 generic），则跳过 GenericDiscoverer 的依赖图
+    // 避免 generic fallback 生成的冗余根节点（如项目名本身）干扰图结构
+    const hasSpecializedDiscoverer = this.#activeDiscoverers.some(
+      ({ discoverer }) => discoverer.id !== 'generic'
+    );
+
     for (const { discoverer } of this.#activeDiscoverers) {
+      if (hasSpecializedDiscoverer && discoverer.id === 'generic') {
+        continue;
+      }
       try {
         const graph = await discoverer.getDependencyGraph();
         for (const n of graph.nodes || []) {
@@ -434,7 +443,7 @@ export class ModuleService {
     this.#logger.info(`[ModuleService] scanTarget: ${targetName}, ${files.length} files`);
 
     // 3. AI 提取
-    if (!this.#chatAgent && !this.#aiFactory) {
+    if (!this.#agentFactory) {
       return {
         recipes: [],
         scannedFiles,
@@ -472,14 +481,16 @@ export class ModuleService {
 
     const result = { recipes, scannedFiles };
     if (recipes.length === 0) {
-      // 检查是否因为 API Key 缺失导致 mock / 空结果
-      const aiInfo = this.#aiFactory
-        ? (await import('../../external/ai/AiFactory.js')).getAiConfigInfo()
-        : null;
-      if (aiInfo && !aiInfo.hasKey) {
-        result.noAi = true;
-        result.message = 'AI 未配置，已跳过智能提取。请在 .env 中设置 API Key 后重试。';
-      } else {
+      // 检查是否因为 AI Provider mock 导致空结果
+      try {
+        const aiProvider = this.#container?.singletons?.aiProvider;
+        if (!aiProvider || aiProvider.name === 'mock') {
+          result.noAi = true;
+          result.message = 'AI 未配置，已跳过智能提取。请在 .env 中设置 API Key 后重试。';
+        } else {
+          result.message = `AI 提取完成，但未发现可复用的代码模式（${targetName}, ${files.length} 个文件）`;
+        }
+      } catch {
         result.message = `AI 提取完成，但未发现可复用的代码模式（${targetName}, ${files.length} 个文件）`;
       }
     }
@@ -576,7 +587,7 @@ export class ModuleService {
     const TOTAL_TIMEOUT = options.totalTimeout || 540000;
     let timedOut = false;
 
-    if (this.#chatAgent || this.#aiFactory) {
+    if (this.#agentFactory) {
       const BATCH_SIZE = options.batchSize || 20;
 
       for (let i = 0; i < allFiles.length; i += BATCH_SIZE) {
@@ -750,75 +761,40 @@ export class ModuleService {
   }
 
   /**
-   * AI 提取 Recipes — 统一入口
+   * AI 提取 Recipes — 委托 AgentFactory.scanKnowledge
+   *
+   * AgentFactory.scanKnowledge 内部创建 insight Agent，
+   * Agent(LLM) 直接分析代码 + 使用 AST 工具，输出 Recipe JSON。
    */
   async #aiExtractRecipes(targetName, files) {
-    const AI_EXTRACT_TIMEOUT = 120_000;
+    if (!this.#agentFactory) {
+      return [];
+    }
 
     try {
-      if (this.#chatAgent) {
-        const extractPromise = this.#chatAgent.executeTool('extract_recipes', {
-          targetName,
-          files,
-        });
-        const timeoutPromise = new Promise((_, reject) =>
-          setTimeout(
-            () => reject(new Error(`AI extraction timeout (${AI_EXTRACT_TIMEOUT / 1000}s)`)),
-            AI_EXTRACT_TIMEOUT
-          )
-        );
-        const result = await Promise.race([extractPromise, timeoutPromise]);
-        if (result?.error) {
-          // API Key 缺失属于配置问题，降为 info 级别避免频繁报错
-          if (/API_KEY_MISSING|API.Key.未配置|unregistered callers/i.test(result.error)) {
-            this.#logger.info(`[ModuleService] AI 未启用（未配置 API Key），跳过 AI 提取。`);
-          } else {
-            this.#logger.warn(`[ModuleService] AI extraction error: ${result.error}`);
-          }
-          return [];
-        }
-        return result?.recipes || [];
-      }
+      const result = await this.#agentFactory.scanKnowledge({
+        label: targetName, files, task: 'extract',
+      });
+      const recipes = result.recipes || [];
 
-      if (this.#aiFactory) {
-        const {
-          getProviderWithFallback,
-          isGeoOrProviderError,
-          getAvailableFallbacks,
-          createProvider,
-        } = this.#aiFactory;
-        let ai = await getProviderWithFallback();
-        if (!ai) {
-          return [];
-        }
-
-        try {
-          return await ai.extractRecipes(targetName, files, {});
-        } catch (primaryErr) {
-          if (isGeoOrProviderError(primaryErr)) {
-            const currentProvider = (process.env.ASD_AI_PROVIDER || 'google').toLowerCase();
-            const fallbacks = getAvailableFallbacks(currentProvider);
-            for (const fbName of fallbacks) {
-              try {
-                ai = createProvider({ provider: fbName });
-                return await ai.extractRecipes(targetName, files, {});
-              } catch (fbErr) {
-                this.#logger.warn(`[ModuleService] fallback "${fbName}" failed: ${fbErr.message}`);
-              }
-            }
-          }
-          throw primaryErr;
-        }
+      if (recipes.length === 0) {
+        this.#logger.info(
+          `[ModuleService] Agent 未产出 recipe (${targetName}, ${files.length} files)`,
+        );
+      } else {
+        this.#logger.info(
+          `[ModuleService] Agent 提取 ${recipes.length} recipes (${targetName})`,
+        );
       }
+      return recipes;
     } catch (err) {
-      if (err.code === 'API_KEY_MISSING') {
+      if (err.code === 'API_KEY_MISSING' || /API_KEY_MISSING|API.Key.未配置|unregistered callers/i.test(err.message)) {
         this.#logger.info(`[ModuleService] AI 未启用（未配置 API Key），跳过 AI 提取。`);
       } else {
         this.#logger.warn(`[ModuleService] AI extraction failed: ${err.message}`);
       }
+      return [];
     }
-
-    return [];
   }
 
   /**

package/lib/service/skills/SignalCollector.js

@@ -2,7 +2,7 @@
  * SignalCollector — AI 驱动的后台行为分析与 Skill 推荐引擎
  *
  * 在 `asd ui` 运行时作为后台守护进程运行，周期性收集多维度信号并
- * 通过 ChatAgent（AI ReAct 循环）进行深度分析，生成 Skill 推荐。
+ * 通过 AgentFactory（统一 Agent 系统）进行深度分析，生成 Skill 推荐。
  *
  * 三种工作模式：
  *   - off      — 不收集，不推荐
@@ -10,8 +10,8 @@
  *   - auto     — 收集信号 → AI 分析 → 推送推荐 + AI 自动创建 Skill
  *
  * 核心架构：
- *   每次 tick → 收集 6 维度信号 → 构造分析 prompt → ChatAgent.execute()
- *   → AI ReAct 循环（可调用 suggest_skills / create_skill 等工具）
+ *   每次 tick → 收集 6 维度信号 → 构造分析 prompt → AgentFactory.createChat()
+ *   → Agent 执行（可调用 suggest_skills / create_skill 等工具）
  *   → 解析 AI 响应（suggestions + nextIntervalMinutes + summary）
  *   → 推送建议 → 动态调整下次执行间隔
  *
@@ -27,11 +27,11 @@
  *   1. 静默 — 不打断用户，后台运行，所有错误降级
  *   2. 增量 — 只分析上次快照以来的新数据
  *   3. 去重 — 同一推荐仅推送一次
- *   4. AI 驱动 — 所有分析决策由 ChatAgent 完成
+ *   4. AI 驱动 — 所有分析决策由 AgentRuntime 完成
  *   5. 自适应 — AI 根据信号密度动态调整执行频率
  *
  * 前提条件：
- *   需要可用的 AI Provider（chatAgent.hasAI === true）
+ *   需要可用的 AI Provider
  *
  * 生命周期：
  *   new SignalCollector(opts) → instance.start() → ... → instance.stop()
@@ -52,7 +52,8 @@ const SNAPSHOT_FILE = 'signal-snapshot.json';
 export class SignalCollector {
   #projectRoot;
   #db;
-  #chatAgent; // ChatAgent 实例 — AI 核心
+  #agentFactory; // AgentFactory 实例 — 统一 Agent 系统
+  #container; // ServiceContainer — 用于获取 aiProvider 等
   #mode; // 'off' | 'suggest' | 'auto'
   #intervalMs;
   #timer = null;
@@ -68,7 +69,8 @@ export class SignalCollector {
    * @param {object} opts
    * @param {string} opts.projectRoot  — 用户项目根目录
    * @param {object} [opts.database]   — better-sqlite3 实例
-   * @param {object} [opts.chatAgent]  — ChatAgent 实例
+   * @param {object} [opts.agentFactory] — AgentFactory 实例
+   * @param {object} [opts.container]  — ServiceContainer 实例
    * @param {string} [opts.mode]       — 'off' | 'suggest' | 'auto'
    * @param {number} [opts.intervalMs] — 初始收集间隔（毫秒），后续由 AI 动态调整
    * @param {function} [opts.onSuggestions] — 新建议回调 (suggestions[]) => void
@@ -76,14 +78,16 @@ export class SignalCollector {
   constructor({
     projectRoot,
     database = null,
-    chatAgent = null,
+    agentFactory = null,
+    container = null,
     mode = 'auto',
     intervalMs = DEFAULT_INTERVAL_MS,
     onSuggestions = null,
   }) {
     this.#projectRoot = projectRoot;
     this.#db = database;
-    this.#chatAgent = chatAgent;
+    this.#agentFactory = agentFactory;
+    this.#container = container;
     this.#mode = ['off', 'suggest', 'auto'].includes(mode) ? mode : 'auto';
     this.#intervalMs = Math.max(Math.min(intervalMs, MAX_INTERVAL_MS), MIN_INTERVAL_MS);
     this.#logger = Logger.getInstance();
@@ -115,7 +119,8 @@ export class SignalCollector {
       this.#logger.info('[SignalCollector] mode=off, skipping start');
       return;
     }
-    if (!this.#chatAgent?.hasAI) {
+    const aiProvider = this.#container?.get('aiProvider');
+    if (!aiProvider || aiProvider.name === 'mock') {
       this.#logger.info('[SignalCollector] no AI provider available, skipping start');
       return;
     }
@@ -217,12 +222,14 @@ export class SignalCollector {
       // 2. 构造分析 prompt
       const prompt = this.#buildAnalysisPrompt(signals);
 
-      // 3. 调用 ChatAgent AI 分析（source: 'system' 确保 Memory 隔离）
-      this.#logger.debug('[SignalCollector] invoking ChatAgent for analysis...');
-      const { reply, toolCalls } = await this.#chatAgent.execute(prompt, {
-        history: [],
-        source: 'system',
-      });
+      // 3. 调用 Agent 系统进行 AI 分析
+      this.#logger.debug('[SignalCollector] invoking Agent for analysis...');
+      const agent = this.#agentFactory.createChat({ lang: 'en' });
+      const { AgentMessage } = await import('../agent/AgentMessage.js');
+      const message = AgentMessage.internal(prompt, { source: 'signal_collector' });
+      const result = await agent.execute(message);
+      const reply = result?.reply ?? result?.text ?? '';
+      const toolCalls = result?.toolCalls ?? [];
 
       // 4. 解析 AI 响应 — 使用 AiProvider.extractJSON 统一 structured output 解析
       const parsed = this.#parseStructuredReply(reply);
@@ -514,13 +521,13 @@ ${JSON.stringify(signals.codeChanges, null, 2)}
   // ═══════════════════════════════════════════════════════
 
   /**
-   * 从 ChatAgent ReAct 回复中提取结构化 JSON
+   * 从 AgentRuntime ReAct 回复中提取结构化 JSON
    *
    * 优先级链:
    *   1. AiProvider.extractJSON (支持 markdown 清理、截断修复、trailing comma 等)
    *   2. 最后一行 JSON 回退 (兼容 prompt 要求的 "最后一行输出 JSON" 格式)
    *
-   * @param {string} reply — ChatAgent.execute() 的回复文本
+   * @param {string} reply — AgentRuntime.execute() 的回复文本
    * @returns {{ suggestions: Array, nextIntervalMinutes: number|null, summary: string }}
    */
   #parseStructuredReply(reply) {
@@ -531,7 +538,7 @@ ${JSON.stringify(signals.codeChanges, null, 2)}
 
     try {
       // 策略 1: 通过 AiProvider.extractJSON 统一解析
-      const aiProvider = this.#chatAgent?.aiProvider;
+      const aiProvider = this.#container?.get('aiProvider');
       if (aiProvider && typeof aiProvider.extractJSON === 'function') {
         const obj = aiProvider.extractJSON(reply, '{', '}');
         if (obj && Array.isArray(obj.suggestions)) {

package/lib/service/snippet/codecs/VSCodeCodec.js

@@ -9,7 +9,7 @@
  */
 
 import { join } from 'node:path';
-import { PlaceholderConverter } from '../PlaceholderConverter.js';
+import { PlaceholderConverter } from '../../../platform/ios/snippet/PlaceholderConverter.js';
 import { SnippetCodec } from './SnippetCodec.js';
 
 /** AutoSnippet language → VSCode snippet scope */

package/lib/service/wiki/WikiGenerator.js

@@ -82,7 +82,7 @@ export class WikiGenerator {
    * @param {object} deps
    * @param {string} deps.projectRoot
    * @param {import('../../service/module/ModuleService.js').ModuleService} [deps.moduleService]
-   * @param {import('../../platform/ios/spm/SpmService.js').SpmService} [deps.spmService] — 向后兼容
+   * @param {import('../../platform/ios/spm/SpmHelper.js').SpmHelper} [deps.spmService] — 向后兼容
    * @param {import('../../service/knowledge/KnowledgeService.js').KnowledgeService} [deps.knowledgeService]
    * @param {import('../../core/ast/ProjectGraph.js').default} [deps.projectGraph]
    * @param {import('../../service/knowledge/CodeEntityGraph.js').CodeEntityGraph} [deps.codeEntityGraph]

package/lib/shared/FieldSpec.js

@@ -12,7 +12,7 @@
  * 消费方:
  *   - UnifiedValidator.js       → 字段完整性检查
  *   - dimension-text.js         → SUBMISSION_SCHEMA / REQUIRED_FIELDS_DESCRIPTION
- *   - ProducerAgent.js          → STYLE_GUIDE 字段列表
+ *   - bootstrap-producer.js      → STYLE_GUIDE 字段列表
  *   - MissionBriefingBuilder.js → submissionSpec 字段描述
  *   - lifecycle.js              → JSON Schema required 数组
  *   - consolidated.js           → 前置校验

package/lib/shared/PathGuard.js

@@ -141,7 +141,7 @@ class PathGuard {
 
   /**
    * Layer 1: 断言路径在允许的边界范围内
-   * 用于修改已有文件的场景（如 XcodeIntegration 插入 header、SpmService 修改 Package.swift）
+   * 用于修改已有文件的场景（如 XcodeIntegration 插入 header、SpmHelper 修改 Package.swift）
    * @param {string} targetPath - 要写入的绝对路径
    * @throws {PathGuardError}
    */

package/lib/shared/StyleGuide.js

@@ -1,7 +1,7 @@
 /**
  * StyleGuide.js — 项目特写写作指南（唯一权威来源）
  *
- * 从 ProducerAgent.js 提取，供内部 Agent (Producer) 和
+ * 供内部 Agent (bootstrap-producer.js) 和
  * 外部 Agent (MissionBriefing submissionSpec) 共享使用。
  *
  * @module shared/StyleGuide

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "autosnippet",
-  "version": "3.2.7",
+  "version": "3.2.9",
   "description": "Extract code patterns into a knowledge base for AI coding assistants",
   "type": "module",
   "main": "lib/bootstrap.js",
@@ -11,6 +11,7 @@
     "postinstall": "node scripts/postinstall-safe.js",
     "prepublishOnly": "npm run build:dashboard && node scripts/build-native-ui.js",
     "build:native-ui": "swiftc resources/native-ui/main.swift resources/native-ui/combined-window.swift -o resources/native-ui/native-ui -framework AppKit",
+    "build:screenshot": "swiftc -O -framework ScreenCaptureKit -framework AppKit resources/native-ui/screenshot.swift -o resources/native-ui/screenshot",
     "install:cursor-skill": "node scripts/install-cursor-skill.js",
     "install:cursor-skill:mcp": "node scripts/install-cursor-skill.js --mcp",
     "install:vscode-copilot": "node scripts/install-vscode-copilot.js",
@@ -116,6 +117,8 @@
     "resources/native-ui/main.swift",
     "resources/native-ui/combined-window.swift",
     "resources/native-ui/native-ui",
+    "resources/native-ui/screenshot.swift",
+    "resources/native-ui/screenshot",
     "resources/native-ui/README.md",
     "resources/openChrome.applescript",
     "resources/grammars"

package/resources/native-ui/screenshot.swift

@@ -0,0 +1,228 @@
+import Foundation
+import ScreenCaptureKit
+import AppKit
+import CoreGraphics
+
+/**
+ * AutoSnippet Screenshot Tool
+ *
+ * 使用 macOS ScreenCaptureKit 原生 API 截取窗口/屏幕画面。
+ * 息屏时可用（不依赖显示器输出）。无需 OBS。
+ *
+ * Usage:
+ *   screenshot                            # 截取整个主屏幕
+ *   screenshot --window "Visual Studio Code"  # 截取匹配标题的窗口
+ *   screenshot --list-windows             # 列出所有可截取窗口 (JSON)
+ *   screenshot --output /tmp/shot.png     # 指定输出路径
+ *   screenshot --format jpeg              # 指定格式 (png|jpeg)
+ *   screenshot --scale 0.5               # 缩放因子
+ *
+ * Requirements: macOS 12.3+, Screen Recording permission
+ */
+
+// MARK: - 输出格式
+enum ImageFormat: String {
+    case png, jpeg
+}
+
+// MARK: - 参数解析
+struct Args {
+    var windowTitle: String? = nil
+    var listWindows = false
+    var outputPath: String? = nil
+    var format: ImageFormat = .jpeg
+    var scale: CGFloat = 1.0
+
+    static func parse(_ args: [String]) -> Args {
+        var result = Args()
+        var i = 1 // skip executable name
+        while i < args.count {
+            switch args[i] {
+            case "--window", "-w":
+                i += 1
+                if i < args.count { result.windowTitle = args[i] }
+            case "--list-windows", "-l":
+                result.listWindows = true
+            case "--output", "-o":
+                i += 1
+                if i < args.count { result.outputPath = args[i] }
+            case "--format", "-f":
+                i += 1
+                if i < args.count {
+                    result.format = ImageFormat(rawValue: args[i].lowercased()) ?? .jpeg
+                }
+            case "--scale", "-s":
+                i += 1
+                if i < args.count { result.scale = CGFloat(Double(args[i]) ?? 1.0) }
+            case "--help", "-h":
+                printUsage()
+                exit(0)
+            default:
+                break
+            }
+            i += 1
+        }
+        return result
+    }
+
+    static func printUsage() {
+        let usage = """
+        AutoSnippet Screenshot Tool (ScreenCaptureKit)
+
+        Usage:
+          screenshot                               Capture main screen
+          screenshot --window "Code"               Capture window matching title
+          screenshot --list-windows                List capturable windows (JSON)
+          screenshot --output /path/to/file.png    Specify output path
+          screenshot --format jpeg|png             Image format (default: jpeg)
+          screenshot --scale 0.5                   Scale factor (default: 1.0)
+        """
+        fputs(usage + "\n", stderr)
+    }
+}
+
+// MARK: - 主逻辑
+
+@available(macOS 12.3, *)
+func run() async {
+    let args = Args.parse(CommandLine.arguments)
+
+    // 获取可共享内容（窗口/屏幕列表）
+    let content: SCShareableContent
+    do {
+        content = try await SCShareableContent.excludingDesktopWindows(false, onScreenWindowsOnly: false)
+    } catch {
+        fputs("{\"error\":\"ScreenCaptureKit access denied: \\(error.localizedDescription). Grant Screen Recording permission in System Settings.\"}\n", stderr)
+        exit(1)
+    }
+
+    // --list-windows: 输出 JSON 窗口列表
+    if args.listWindows {
+        let windows = content.windows.compactMap { w -> [String: Any]? in
+            guard let app = w.owningApplication else { return nil }
+            return [
+                "windowID": w.windowID,
+                "title": w.title ?? "",
+                "app": app.applicationName,
+                "bundleID": app.bundleIdentifier,
+                "width": w.frame.width,
+                "height": w.frame.height,
+                "onScreen": w.isOnScreen,
+            ]
+        }
+        if let data = try? JSONSerialization.data(withJSONObject: windows, options: [.prettyPrinted, .sortedKeys]),
+           let json = String(data: data, encoding: .utf8) {
+            print(json)
+        }
+        exit(0)
+    }
+
+    // 确定截取目标 — 统一使用 desktopIndependentWindow（亮屏/息屏均可用）
+    var filter: SCContentFilter
+
+    // 筛选有效窗口（面积 > 100x100，排除菜单栏等微型窗口）
+    let validWindows = content.windows.filter { $0.frame.width > 100 && $0.frame.height > 100 }
+
+    if let title = args.windowTitle {
+        // 模糊匹配窗口标题或 app 名，选面积最大的
+        let pattern = title.lowercased()
+        let candidates = validWindows.filter { w in
+            let wTitle = (w.title ?? "").lowercased()
+            let appName = (w.owningApplication?.applicationName ?? "").lowercased()
+            return wTitle.contains(pattern) || appName.contains(pattern)
+        }
+        let matched = candidates.max(by: { $0.frame.width * $0.frame.height < $1.frame.width * $1.frame.height })
+
+        guard let window = matched else {
+            let windowList: String = validWindows.prefix(20).compactMap {
+                "\($0.owningApplication?.applicationName ?? "?") - \($0.title ?? "(no title)") [\(Int($0.frame.width))x\(Int($0.frame.height))]"
+            }.joined(separator: "\n  ")
+            fputs("{\"error\":\"No window matching '\(title)'. Available:\\n  \(windowList)\"}\n", stderr)
+            exit(1)
+        }
+
+        filter = SCContentFilter(desktopIndependentWindow: window)
+    } else {
+        // 未指定窗口 — 自动选最大窗口
+        guard let largest = validWindows.max(by: { $0.frame.width * $0.frame.height < $1.frame.width * $1.frame.height }) else {
+            fputs("{\"error\":\"No capturable window available\"}\n", stderr)
+            exit(1)
+        }
+        filter = SCContentFilter(desktopIndependentWindow: largest)
+    }
+
+    // 配置截图参数
+    let config = SCStreamConfiguration()
+    let sourceRect = filter.contentRect
+    config.width = Int(sourceRect.width * args.scale)
+    config.height = Int(sourceRect.height * args.scale)
+    config.showsCursor = false
+    config.capturesAudio = false
+
+    // 截图
+    let image: CGImage
+    do {
+        image = try await SCScreenshotManager.captureImage(contentFilter: filter, configuration: config)
+    } catch {
+        fputs("{\"error\":\"Screenshot failed: \\(error.localizedDescription)\"}\n", stderr)
+        exit(1)
+    }
+
+    // 编码为图片数据
+    let bitmapRep = NSBitmapImageRep(cgImage: image)
+    let imageData: Data?
+    switch args.format {
+    case .jpeg:
+        imageData = bitmapRep.representation(using: .jpeg, properties: [.compressionFactor: 0.85])
+    case .png:
+        imageData = bitmapRep.representation(using: .png, properties: [:])
+    }
+
+    guard let data = imageData else {
+        fputs("{\"error\":\"Image encoding failed\"}\n", stderr)
+        exit(1)
+    }
+
+    // 确定输出路径
+    let ext = args.format == .png ? "png" : "jpg"
+    let outputPath = args.outputPath ?? NSTemporaryDirectory() + "asd-screenshot-\(Int(Date().timeIntervalSince1970 * 1000)).\(ext)"
+
+    do {
+        try data.write(to: URL(fileURLWithPath: outputPath))
+    } catch {
+        fputs("{\"error\":\"Write failed: \\(error.localizedDescription)\"}\n", stderr)
+        exit(1)
+    }
+
+    // 输出 JSON 结果到 stdout
+    let result: [String: Any] = [
+        "success": true,
+        "path": outputPath,
+        "width": image.width,
+        "height": image.height,
+        "format": args.format.rawValue,
+        "bytes": data.count,
+    ]
+    if let jsonData = try? JSONSerialization.data(withJSONObject: result, options: [.sortedKeys]),
+       let json = String(data: jsonData, encoding: .utf8) {
+        print(json)
+    }
+}
+
+// MARK: - Entry Point
+
+// 初始化 AppKit 连接到 WindowServer（解决 CGS_REQUIRE_INIT）
+let app = NSApplication.shared
+app.setActivationPolicy(.prohibited)
+
+if #available(macOS 12.3, *) {
+    let semaphore = DispatchSemaphore(value: 0)
+    Task {
+        await run()
+        semaphore.signal()
+    }
+    semaphore.wait()
+} else {
+    fputs("{\"error\":\"Requires macOS 12.3 or later\"}\n", stderr)
+    exit(1)
+}