gitlab-ai-review 1.0.2 → 1.0.3

package/lib/config.js

@@ -3,6 +3,53 @@
  * 参考 review-bot/config.js 的实现
  */
 
+import fs from 'fs';
+import path from 'path';
+
+/**
+ * 读取项目根目录下的 AI Review Guard 配置文件
+ */
+function readAIReviewGuardConfig() {
+  const configFiles = ['.aireviewguard', 'aireviewguard', '.aireviewguard.json', 'reviewguard.md', '.reviewguard.md'];
+  
+  for (const filename of configFiles) {
+    try {
+      const configPath = path.join(process.cwd(), filename);
+      if (fs.existsSync(configPath)) {
+        const content = fs.readFileSync(configPath, 'utf-8');
+        
+        // 如果是 Markdown 文件，返回原始内容
+        if (filename.endsWith('.md')) {
+          return {
+            type: 'markdown',
+            filename,
+            content,
+          };
+        }
+        
+        // 尝试解析 JSON，如果失败则返回纯文本
+        try {
+          return {
+            type: 'json',
+            filename,
+            ...JSON.parse(content),
+          };
+        } catch {
+          return {
+            type: 'text',
+            filename,
+            content,
+          };
+        }
+      }
+    } catch (error) {
+      // 忽略读取错误，继续尝试下一个文件
+    }
+  }
+  
+  return null;
+}
+
 /**
  * 获取 GitLab 配置（支持自动检测 CI 环境）
  */
@@ -26,6 +73,22 @@ export function getGitLabConfig() {
   return { token, url };
 }
 
+/**
+ * 获取 AI 配置（ARK API Key）
+ */
+export function getAIConfig() {
+  // 从环境变量读取 ARK_API_KEY
+  const arkApiKey = process.env.ARK_API_KEY || '';
+  
+  // 从配置文件读取其他配置
+  const guardConfig = readAIReviewGuardConfig();
+  
+  return {
+    arkApiKey,
+    guardConfig,
+  };
+}
+
 /**
  * 获取项目配置（支持自动检测 CI 环境）
  */
@@ -55,10 +118,12 @@ export function getProjectConfig() {
 export function getConfig() {
   const gitlab = getGitLabConfig();
   const project = getProjectConfig();
+  const ai = getAIConfig();
 
   return {
     gitlab,
     project,
+    ai,
   };
 }
 

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "gitlab-ai-review",
-  "version": "1.0.2",
-  "description": "GitLab AI Review SDK - 支持 CI/CD 自动配置和手动配置",
+  "version": "1.0.3",
+  "description": "GitLab AI Review SDK - 支持 CI/CD 自动配置和 ARK API",
   "main": "index.js",
   "type": "module",
   "scripts": {
@@ -19,16 +19,10 @@
     "ai",
     "review",
     "sdk",
-    "ci-cd",
-    "code-review",
-    "automation"
+    "ci-cd"
   ],
-  "author": "dengjiang1",
+  "author": "",
   "license": "MIT",
-  "repository": {
-    "type": "git",
-    "url": "https://ksogitlab.wps.kingsoft.net/web_training/web-professional-25-tasks/dengjiang1"
-  },
   "engines": {
     "node": ">=18.0.0"
   }