@nasl/cli 0.1.17 → 0.1.18

package/dist/bin/naslc.mjs

@@ -3,8 +3,8 @@
 import require$$0$1, { EventEmitter } from 'events';
 import require$$1 from 'child_process';
 import * as sysPath from 'path';
-import sysPath__default from 'path';
-import require$$0$2 from 'fs';
+import sysPath__default, { join } from 'path';
+import require$$0$2, { promises } from 'fs';
 import require$$4 from 'process';
 import require$$0$5 from 'os';
 import require$$1$1 from 'tty';
@@ -15,7 +15,7 @@ import require$$5 from 'assert';
 import require$$3 from 'http';
 import require$$4$1 from 'https';
 import require$$0$6 from 'url';
-import crypto$1 from 'crypto';
+import crypto$1, { createHash } from 'crypto';
 import http2 from 'http2';
 import zlib from 'zlib';
 
@@ -8304,10 +8304,39 @@ function loadConfig(configDir) {
         const content = libExports.readFileSync(configPath, 'utf-8');
         const config = JSON.parse(content);
         // 验证必需字段
-        if (!config.serverBaseURL || !config.ideVersion || !config.srcDir || !config.outDir) {
-            defaultLogger.error('配置文件格式不正确，缺少必需字段');
+        if (!config.srcDir || !config.outDir) {
+            defaultLogger.error('配置文件格式不正确，缺少必需的 srcDir 或 outDir 字段');
             return defaultLogger.exit(1);
         }
+        if (!config.ideVersion) {
+            config.ideVersion = process.env.NASL_IDE_VERSION || '';
+            if (!config.ideVersion) {
+                defaultLogger.error('缺少配置 ideVersion，请在配置文件中添加，或在环境变量中配置 NASL_IDE_VERSION');
+                return defaultLogger.exit(1);
+            }
+        }
+        if (!config.serverBaseURL) {
+            config.serverBaseURL = process.env.NASL_SERVER_BASE_URL || '';
+            if (!config.serverBaseURL) {
+                defaultLogger.error('缺少配置 serverBaseURL，请在配置文件中添加，或在环境变量中配置 NASL_SERVER_BASE_URL');
+                return defaultLogger.exit(1);
+            }
+        }
+        if (!config.useOPENAPI)
+            config.useOPENAPI = process.env.USE_LCAP_OPENAPI === 'true';
+        if (!config.OPENAPI_AK)
+            config.OPENAPI_AK = process.env.LCAP_OPENAPI_AK;
+        if (!config.OPENAPI_SK)
+            config.OPENAPI_SK = process.env.LCAP_OPENAPI_SK;
+        if (config.useOPENAPI) {
+            if (!config.OPENAPI_AK || !config.OPENAPI_SK) {
+                defaultLogger.error(`配置了 useOPENAPI，但缺少配置 OPENAPI_AK 和 OPENAPI_SK:
+- 在 nasl.config.json 中配置 OPENAPI_AK 和 OPENAPI_SK，或者在环境变量中配置 LCAP_OPENAPI_AK 和 LCAP_OPENAPI_SK
+- 或将 useOPENAPI 配置为 false
+`);
+                return defaultLogger.exit(1);
+            }
+        }
         return config;
     }
     catch (error) {
@@ -28826,18 +28855,99 @@ function v4(options, buf, offset) {
   return unsafeStringify(rnds);
 }
 
+function generateHash(params) {
+    const hash = createHash('sha256');
+    hash.update(JSON.stringify(params));
+    return hash.digest('hex');
+}
+async function getCachePathWithTimestamp(hash) {
+    const cacheDir = join(process.cwd(), '.cache');
+    await promises.mkdir(cacheDir, { recursive: true });
+    return join(cacheDir, `${hash}_${Date.now()}.json`);
+}
 /**
- * 构建签名字符串
+ * 查找指定 hash 的有效缓存文件
+ * @param hash 缓存标识
+ * @param ttl 缓存有效期（毫秒）
+ * @returns 缓存文件路径和缓存内容，如果没有有效缓存则返回 null
  */
-function buildStringToSign(appKey, nonce, timestamp, secretKey) {
-    return `${appKey}&${nonce}&${timestamp}&${secretKey}`;
+async function findValidCache(hash, ttl) {
+    const cacheDir = join(process.cwd(), '.cache');
+    try {
+        await promises.mkdir(cacheDir, { recursive: true });
+        const files = await promises.readdir(cacheDir);
+        const now = Date.now();
+        // 查找所有匹配的缓存文件
+        const matchingFiles = files
+            .filter((file) => file.startsWith(`${hash}_`) && file.endsWith('.json'))
+            .map((file) => {
+            const match = file.match(/_(\d+)\.json$/);
+            if (!match)
+                return null;
+            const timestamp = parseInt(match[1], 10);
+            return {
+                path: join(cacheDir, file),
+                timestamp,
+                isExpired: now - timestamp > ttl,
+            };
+        })
+            .filter((item) => item !== null);
+        // 删除过期的缓存文件
+        const expiredFiles = matchingFiles.filter((item) => item.isExpired);
+        await Promise.all(expiredFiles.map((item) => promises.unlink(item.path).catch(() => { })));
+        // 查找有效的缓存文件（按时间戳降序排序，获取最新的）
+        const validFiles = matchingFiles.filter((item) => !item.isExpired).sort((a, b) => b.timestamp - a.timestamp);
+        if (validFiles.length > 0) {
+            const cacheFile = validFiles[0];
+            const cacheContent = await promises.readFile(cacheFile.path, 'utf-8');
+            const cachedResult = JSON.parse(cacheContent);
+            return { path: cacheFile.path, data: cachedResult };
+        }
+        return null;
+    }
+    catch (error) {
+        return null;
+    }
 }
 /**
- * 生成 MD5 签名
+ * 高阶函数，为异步函数添加带时间戳的缓存功能
+ * @param fn 需要添加缓存的函数
+ * @param ttl 缓存有效期（毫秒），默认 1 小时
+ * @param getCacheKey 可选的自定义缓存 key 生成函数，如果不提供则使用整个 params
+ * @returns 带缓存的函数版本
  */
-function generateSignature(plainText) {
-    return crypto$1.createHash('md5').update(plainText).digest('hex');
+function cachableWithTTL(fn, ttl = 3600 * 1000, // 默认 1 小时
+getCacheKey) {
+    return async (params, extraParams) => {
+        const functionName = fn.name;
+        const logPrefix = functionName ? `[${functionName}]` : '[cachableWithTTL]';
+        // 检查是否启用缓存
+        if (process.env.NO_AI_CACHE) {
+            return fn(params, extraParams);
+        }
+        // 使用自定义的 getCacheKey 或默认使用整个 params
+        const cacheKey = getCacheKey ? getCacheKey(params) : params;
+        const hash = generateHash(cacheKey);
+        // 查找有效的缓存
+        const validCache = await findValidCache(hash, ttl);
+        if (validCache) {
+            console.log(`${logPrefix} ✓ 使用缓存`);
+            return validCache.data;
+        }
+        // 调用原函数
+        const result = await fn(params, extraParams);
+        // 保存结果到缓存（带时间戳）
+        try {
+            const cachePath = await getCachePathWithTimestamp(hash);
+            await promises.writeFile(cachePath, JSON.stringify(result, null, 2), 'utf-8');
+        }
+        catch (error) {
+            // 缓存写入失败，但不影响返回结果
+        }
+        return result;
+    };
 }
+
 /**
  * 生成客户端签名信息
  * @param appKey AppKey
@@ -28847,8 +28957,8 @@ function generateSignature(plainText) {
 function generateClientSignature(appKey, secretKey) {
     const timestamp = Math.floor(Date.now() / 1000).toString();
     const nonce = v4();
-    const plainText = buildStringToSign(appKey, nonce, timestamp, secretKey);
-    const signature = generateSignature(plainText);
+    const plainText = `${appKey}&${nonce}&${timestamp}&${secretKey}`;
+    const signature = crypto$1.createHash('md5').update(plainText).digest('hex');
     return {
         appKey,
         timestamp,
@@ -28873,12 +28983,12 @@ function generateBaseHeaders(ak, sk) {
     };
 }
 /**
- * 获取租户的 signInfo
+ * 获取租户的 signInfo（内部实现）
  * @param options 服务器选项
  * @param baseHeaders 基础认证头
  * @returns x-signInfo 值
  */
-async function fetchSignInfo(options, baseHeaders) {
+async function _fetchSignInfo({ options, baseHeaders }) {
     const tenantName = options.tenantName || 'defaulttenant';
     const userName = tenantName === 'defaulttenant' ? 'admin' : `${tenantName}-admin`;
     const data = {
@@ -28900,6 +29010,14 @@ async function fetchSignInfo(options, baseHeaders) {
         return null;
     }
 }
+/**
+ * 获取租户的 signInfo（带缓存，1 小时过期）
+ * @param options 服务器选项
+ * @param baseHeaders 基础认证头
+ * @returns x-signInfo 值
+ */
+const fetchSignInfo = cachableWithTTL(_fetchSignInfo, 3600 * 1000, // 1 小时
+(params) => params.options);
 /**
  * 生成完整的认证头（包含 x-signInfo）
  * @param options 服务器选项
@@ -28909,19 +29027,18 @@ async function generateCompleteHeaders(options) {
     const headers = {
         'Content-Type': 'application/json',
     };
-    const OPENAPI_AK = options.OPENAPI_AK || process.env.LCAP_OPENAPI_AK;
-    const OPENAPI_SK = options.OPENAPI_SK || process.env.LCAP_OPENAPI_SK;
     // 如果没有提供 ak 和 sk，返回基础 headers
-    if (!OPENAPI_AK || !OPENAPI_SK) {
-        throw new Error(`配置了 useOPENAPI，但没有提供 OPENAPI_AK 和 OPENAPI_SK:
-- 可取消配置 useOPENAPI
-- 在 nasl.config.json 中配置 OPENAPI_AK 和 OPENAPI_SK，或者在环境变量中配置 LCAP_OPENAPI_AK 和 LCAP_OPENAPI_SK`);
+    if (!options.OPENAPI_AK || !options.OPENAPI_SK) {
+        throw new Error(`配置了 useOPENAPI，但缺少配置 OPENAPI_AK 和 OPENAPI_SK:
+- 在 nasl.config.json 中配置 OPENAPI_AK 和 OPENAPI_SK，或者在环境变量中配置 LCAP_OPENAPI_AK 和 LCAP_OPENAPI_SK
+- 或将 useOPENAPI 配置为 false
+`);
     }
     // 生成基础认证头
-    const baseHeaders = generateBaseHeaders(OPENAPI_AK, OPENAPI_SK);
+    const baseHeaders = generateBaseHeaders(options.OPENAPI_AK, options.OPENAPI_SK);
     Object.assign(headers, baseHeaders);
     // 获取 signInfo
-    const signInfo = await fetchSignInfo(options, headers);
+    const signInfo = await fetchSignInfo({ options, baseHeaders: headers });
     if (signInfo) {
         headers['x-signInfo'] = signInfo;
     }
@@ -28936,10 +29053,9 @@ async function generateCompleteHeaders(options) {
 async function createAxios(options) {
     // 如果需要鉴权，拼接 /openapi/v3/nasl；否则使用原始 URL
     const serverBaseURL = new URL(options.serverBaseURL).origin;
-    const useOPENAPI = options.useOPENAPI || process.env.USE_LCAP_OPENAPI === 'true';
-    const baseURL = useOPENAPI ? `${serverBaseURL}/openapi/v3/nasl` : `${serverBaseURL}/api/v1/nasl`;
+    const baseURL = options.useOPENAPI ? `${serverBaseURL}/openapi/v3/nasl` : `${serverBaseURL}/api/v1/nasl`;
     // 如果需要鉴权，生成完整的认证头；否则只使用基础 headers
-    const headers = useOPENAPI ? await generateCompleteHeaders(options) : { 'Content-Type': 'application/json' };
+    const headers = options.useOPENAPI ? await generateCompleteHeaders(options) : { 'Content-Type': 'application/json' };
     console.log('本次服务调用方为:', baseURL);
     const instance = axios.create({
         baseURL,
@@ -37877,7 +37993,7 @@ async function tryCompile(entry, options) {
     }
 }
 
-var version = "0.1.17";
+var version = "0.1.18";
 var pkg = {
 	version: version};