aiconductor-cli 1.0.1 → 1.1.0

package/bin/aiconductor-cli.js

@@ -1,7 +1,8 @@
 #!/usr/bin/env node
 
 const { Command } = require("commander");
-const { publishCommand } = require("../src/commands/publish");
+const { publishCommand } = require("../src/commands/html_publish");
+const { searchCommand } = require("../src/commands/web_search");
 
 const program = new Command();
 
@@ -26,6 +27,39 @@ htmlCmd
     publishCommand(html, options);
   });
 
+// search 子命令组
+const searchCmd = program
+  .command("search")
+  .description("网页/图片搜索");
+
+searchCmd
+  .command("web")
+  .description("网页搜索")
+  .argument("<query>", "搜索关键词（1~100字符）")
+  .option("-k, --api-key <key>", "AIConductor API Key（也可通过 AICONDUCTOR_API_KEY 环境变量或 .env 文件提供）")
+  .option("-n, --count <n>", "返回条数（最多50）")
+  .option("--time-range <range>", "时间范围：OneDay/OneWeek/OneMonth/OneYear 或 YYYY-MM-DD..YYYY-MM-DD")
+  .option("--no-summary", "不返回摘要")
+  .option("--content-format <format>", "内容格式：Text 或 Markdown", "Text")
+  .option("--auth-level <level>", "权威来源过滤：0=不过滤，1=优先权威来源", "0")
+  .option("--query-rewrite", "开启 Query 改写")
+  .option("-f, --format <format>", "输出格式：json 或 text", "json")
+  .action((query, options) => {
+    searchCommand(query, "web", options);
+  });
+
+searchCmd
+  .command("image")
+  .description("图片搜索")
+  .argument("<query>", "搜索关键词（1~100字符）")
+  .option("-k, --api-key <key>", "AIConductor API Key（也可通过 AICONDUCTOR_API_KEY 环境变量或 .env 文件提供）")
+  .option("-n, --count <n>", "返回条数（最多5）")
+  .option("--time-range <range>", "时间范围：OneDay/OneWeek/OneMonth/OneYear 或 YYYY-MM-DD..YYYY-MM-DD")
+  .option("-f, --format <format>", "输出格式：json 或 text", "json")
+  .action((query, options) => {
+    searchCommand(query, "image", options);
+  });
+
 // 未知命令处理
 program.on("command:*", () => {
   console.error(`未知命令: ${program.args.join(" ")}`);

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "aiconductor-cli",
-  "version": "1.0.1",
+  "version": "1.1.0",
   "description": "AIConductor CLI 工具箱",
   "keywords": [
     "aiconductor",

package/src/api.js

@@ -4,6 +4,7 @@ const path = require("path");
 
 const API_HOST = "plugin.aiconductor.fun";
 const API_PATH = "/api/html_publish";
+const SEARCH_API_PATH = "/api/aic_web_search";
 
 /**
  * 解析 API Key，三级优先级：命令行参数 > 系统环境变量 > .env 文件
@@ -38,7 +39,7 @@ function resolveApiKey(optionKey) {
  * @param {object} body
  * @returns {Promise<object>}
  */
-function httpPost(host, path, body) {
+function httpPost(host, path, body, headers = {}) {
   return new Promise((resolve, reject) => {
     const data = JSON.stringify(body);
 
@@ -50,6 +51,7 @@ function httpPost(host, path, body) {
       headers: {
         "Content-Type": "application/json",
         "Content-Length": Buffer.byteLength(data),
+        ...headers,
       },
     };
 
@@ -108,4 +110,31 @@ async function publishHtml(apiKey, htmlCode, title) {
   return safe;
 }
 
-module.exports = { resolveApiKey, publishHtml };
+/**
+ * 网页/图片搜索
+ * @param {string} apiKey
+ * @param {{ query: string, searchType: string, count?: number, timeRange?: string, needSummary?: boolean, contentFormats?: string, authLevel?: number, queryRewrite?: boolean }} params
+ * @returns {Promise<object>}
+ */
+async function webSearch(apiKey, params) {
+  const reqBody = {
+    query: params.query,
+    search_type: params.searchType,
+  };
+  if (params.count !== undefined) reqBody.count = params.count;
+  if (params.timeRange) reqBody.time_range = params.timeRange;
+  if (params.needSummary !== undefined) reqBody.need_summary = params.needSummary;
+  if (params.contentFormats) reqBody.content_formats = params.contentFormats;
+  if (params.authLevel !== undefined) reqBody.auth_level = params.authLevel;
+  if (params.queryRewrite !== undefined) reqBody.query_rewrite = params.queryRewrite;
+
+  const resp = await httpPost(API_HOST, SEARCH_API_PATH, reqBody, {
+    Authorization: `Bearer ${apiKey}`,
+  });
+
+  // 过滤计费信息
+  const { billing, ...safe } = resp;
+  return safe;
+}
+
+module.exports = { resolveApiKey, publishHtml, webSearch };

package/src/commands/web_search.js

@@ -0,0 +1,120 @@
+const { resolveApiKey, webSearch } = require("../api");
+
+/**
+ * 输出成功结果
+ * @param {object} data
+ * @param {"json"|"text"} format
+ */
+function outputResult(data, format) {
+  if (format === "json") {
+    process.stdout.write(JSON.stringify({ success: true, data }, null, 2) + "\n");
+  } else {
+    // API 响应结构：{ success, message, data: { ResultCount, TimeCost, WebResults, ImageResults } }
+    const inner = data.data || {};
+    process.stdout.write(`✓ 搜索成功\n`);
+    process.stdout.write(`  结果数: ${inner.ResultCount}\n`);
+    process.stdout.write(`  耗时: ${inner.TimeCost}ms\n\n`);
+
+    if (inner.WebResults) {
+      inner.WebResults.forEach((item) => {
+        process.stdout.write(`[${item.SortId}] ${item.Title}\n`);
+        if (item.SiteName) process.stdout.write(`    站点: ${item.SiteName}\n`);
+        if (item.Url) process.stdout.write(`    URL: ${item.Url}\n`);
+        if (item.Snippet) process.stdout.write(`    摘要: ${item.Snippet}\n`);
+        if (item.Summary) process.stdout.write(`    内容摘要: ${item.Summary}\n`);
+        if (item.PublishTime) process.stdout.write(`    发布时间: ${item.PublishTime}\n`);
+        process.stdout.write(`\n`);
+      });
+    }
+
+    if (inner.ImageResults) {
+      inner.ImageResults.forEach((item) => {
+        process.stdout.write(`[${item.SortId}] ${item.Title}\n`);
+        if (item.SiteName) process.stdout.write(`    来源: ${item.SiteName}\n`);
+        if (item.Image?.Url) process.stdout.write(`    图片: ${item.Image.Url}\n`);
+        if (item.Image?.Width && item.Image?.Height) {
+          process.stdout.write(`    尺寸: ${item.Image.Width}x${item.Image.Height}\n`);
+        }
+        if (item.Url) process.stdout.write(`    来源页面: ${item.Url}\n`);
+        process.stdout.write(`\n`);
+      });
+    }
+  }
+}
+
+/**
+ * 输出错误
+ * @param {object} err
+ * @param {"json"|"text"} format
+ */
+function outputError(err, format) {
+  if (format === "json") {
+    process.stderr.write(JSON.stringify({ success: false, error: err.message || String(err) }, null, 2) + "\n");
+  } else {
+    process.stderr.write(`✗ 错误: ${err.message || err}\n`);
+  }
+}
+
+/**
+ * search 命令
+ * @param {string} query - 搜索关键词
+ * @param {"web"|"image"} searchType - 搜索类型
+ * @param {object} options - 命令行选项
+ */
+async function searchCommand(query, searchType, options = {}) {
+  const format = options.format || "json";
+  const resolvedKey = resolveApiKey(options.apiKey);
+
+  if (!resolvedKey) {
+    outputError(
+      { message: "缺少 API Key。请通过 -k 参数、AICONDUCTOR_API_KEY 环境变量或 .env 文件提供。" },
+      format
+    );
+    process.exit(1);
+  }
+
+  const params = {
+    query,
+    searchType,
+  };
+
+  if (options.count !== undefined) {
+    params.count = parseInt(options.count, 10);
+  }
+  if (options.timeRange) {
+    params.timeRange = options.timeRange;
+  }
+  // need_summary 默认 true，--no-summary 时 options.summary 为 false
+  if (searchType === "web") {
+    params.needSummary = options.summary !== false;
+  }
+  if (options.contentFormat) {
+    params.contentFormats = options.contentFormat;
+  }
+  if (options.authLevel !== undefined) {
+    params.authLevel = parseInt(options.authLevel, 10);
+  }
+  if (options.queryRewrite) {
+    params.queryRewrite = true;
+  }
+
+  try {
+    const result = await webSearch(resolvedKey, params);
+    outputResult(result, format);
+    process.exit(0);
+  } catch (err) {
+    if (err.type === "http_error") {
+      outputError(
+        { message: `API 返回错误 (HTTP ${err.statusCode}): ${err.body?.message || JSON.stringify(err.body)}` },
+        format
+      );
+    } else if (err.type === "network_error") {
+      outputError({ message: `网络错误: ${err.message}` }, format);
+    } else {
+      outputError({ message: `未知错误: ${err.message || err}` }, format);
+    }
+    process.exit(1);
+  }
+}
+
+module.exports = { searchCommand };