@blynkai/cli 0.1.1

package/src/cli/index.mjs

@@ -0,0 +1,554 @@
+#!/usr/bin/env node
+
+import { spawn } from 'node:child_process'
+import { openSync } from 'node:fs'
+import { mkdir } from 'node:fs/promises'
+import { dirname, join } from 'node:path'
+import { fileURLToPath } from 'node:url'
+import { ErrorCode } from '../core/errors.mjs'
+import { logFile, pidFile, runtimeDir } from '../core/runtime-paths.mjs'
+import { runBrowserLogin } from './auth-login.mjs'
+import { printHelp } from './help.mjs'
+import {
+  buildSearchBody,
+  getFlag,
+  normalizeSearchScene,
+  parseFlags,
+  parseSearchArgs,
+} from './search-options.mjs'
+
+const preferredBaseUrl = process.env.LINGRUI_AGENT_BASE_URL ?? 'http://127.0.0.1:38100'
+const basePort = Number(new URL(preferredBaseUrl).port || 38100)
+let resolvedBaseUrl = process.env.LINGRUI_AGENT_BASE_URL ?? null
+const __dirname = dirname(fileURLToPath(import.meta.url))
+const serverEntry = join(__dirname, '../server/index.mjs')
+const requestTimeoutMs = Number(process.env.LINGRUI_AGENT_REQUEST_TIMEOUT_MS ?? 3000)
+
+async function fetchJson(url, options = {}) {
+  const controller = new AbortController()
+  const timeout = setTimeout(() => controller.abort(), options.timeoutMs ?? requestTimeoutMs)
+
+  try {
+    const response = await fetch(url, {
+      method: options.method ?? 'GET',
+      headers: {
+        'content-type': 'application/json',
+      },
+      signal: controller.signal,
+      ...(options.body ? { body: JSON.stringify(options.body) } : {}),
+    })
+
+    return response.json()
+  } finally {
+    clearTimeout(timeout)
+  }
+}
+
+async function rawRequest(path, options = {}) {
+  const baseUrl = await getBaseUrl(path)
+  return fetchJson(`${baseUrl}${path}`, options)
+}
+
+async function tryRequest(baseUrl, path, options = {}) {
+  return fetchJson(`${baseUrl}${path}`, {
+    ...options,
+    timeoutMs: options.timeoutMs ?? 800,
+  })
+}
+
+async function hasEndpoint(baseUrl, path) {
+  if (path !== '/api/search') return true
+
+  try {
+    const payload = await tryRequest(baseUrl, '/api/search', {
+      method: 'POST',
+      body: {
+        scene: 'data-center',
+        output: { dry_run: true },
+      },
+    })
+    return payload?.code !== 40400
+  } catch {
+    return false
+  }
+}
+
+async function getBaseUrl(path = '/health') {
+  if (resolvedBaseUrl && (await hasEndpoint(resolvedBaseUrl, path))) return resolvedBaseUrl
+
+  if (process.env.LINGRUI_AGENT_BASE_URL) {
+    resolvedBaseUrl = preferredBaseUrl
+    return resolvedBaseUrl
+  }
+
+  for (let port = basePort; port <= basePort + 20; port += 1) {
+    const candidate = `http://127.0.0.1:${port}`
+    try {
+      const payload = await tryRequest(candidate, '/health')
+      if (payload?.code === 0 && payload?.data?.service === 'lingrui-local-agent') {
+        if (await hasEndpoint(candidate, path)) {
+          resolvedBaseUrl = candidate
+          return resolvedBaseUrl
+        }
+      }
+    } catch {
+      // Try the next local port.
+    }
+  }
+
+  resolvedBaseUrl = preferredBaseUrl
+  return resolvedBaseUrl
+}
+
+function sleep(ms) {
+  return new Promise((resolve) => setTimeout(resolve, ms))
+}
+
+async function waitForBaseUrl(path = '/health') {
+  for (let i = 0; i < 20; i += 1) {
+    resolvedBaseUrl = null
+    const baseUrl = await getBaseUrl(path)
+    if (await hasEndpoint(baseUrl, path)) return baseUrl
+    await sleep(100)
+  }
+
+  return getBaseUrl(path)
+}
+
+function normalizeStatusPayload(payload) {
+  const data = payload?.data ?? {}
+  const authState = data.authState ?? 'missing'
+
+  return {
+    code: 0,
+    msg: 'ok',
+    data: {
+      service: data.service ?? 'lingrui-cli',
+      status: data.status ?? 'online',
+      loggedIn: authState === 'active',
+      authState,
+      user: data.user ?? null,
+      ...(data.port ? { port: data.port } : {}),
+    },
+  }
+}
+
+function getHint(msg) {
+  if (msg === 'auth_required') {
+    return '请运行 blynkai login，在浏览器中完成 CLI 授权。'
+  }
+
+  if (msg === 'session_revoked') {
+    return '当前本地会话已失效。请运行 blynkai login 重新授权。'
+  }
+
+  if (msg === 'local_agent_unavailable') {
+    return `无法连接到本地助手服务。请确认服务是否正常运行，或重试。`
+  }
+
+  return undefined
+}
+
+async function request(path, options = {}) {
+  try {
+    const payload = await rawRequest(path, options)
+    const msg = payload.msg ?? payload.message
+    if (payload.code !== 0) {
+      const hint = getHint(msg)
+      console.log(`\n❌ 请求失败 [错误代码: ${payload.code}]`)
+      if (hint) {
+        console.log(`💡 提示: ${hint}`)
+      } else {
+        console.log(`📄 原因: ${msg}`)
+      }
+      console.log()
+      process.exitCode = 1
+      return null
+    }
+
+    return payload
+  } catch (error) {
+    const msg = 'local_agent_unavailable'
+    const hint = getHint(msg)
+    console.log(`\n❌ 服务未响应 [错误代码: 50010]`)
+    console.log(`💡 提示: ${hint}\n`)
+    process.exitCode = 1
+    return null
+  }
+}
+
+async function requestStatus() {
+  try {
+    const payload = await rawRequest('/status')
+    if (payload.code === 0) return normalizeStatusPayload(payload)
+
+    if (payload.code === 40400 && payload.details?.path === '/status') {
+      const fallback = await rawRequest('/health')
+      if (fallback.code === 0) return normalizeStatusPayload(fallback)
+      return fallback
+    }
+
+    return payload
+  } catch {
+    return null
+  }
+}
+
+function printJson(value) {
+  console.log(JSON.stringify(value, null, 2))
+}
+
+function printError(code, msg, details = undefined) {
+  printJson({
+    code,
+    msg,
+    ...(details ? { details } : {}),
+  })
+  process.exitCode = 1
+}
+
+function toDisplayText(value, fallback = '-') {
+  if (value === undefined || value === null || value === '') return fallback
+  return String(value)
+}
+
+function getArticleSource(item) {
+  return (
+    item.website_name ||
+    item.website ||
+    item.media_channel ||
+    item.source ||
+    item.author ||
+    '-'
+  )
+}
+
+function formatSearchList(payload) {
+  const responseData = payload?.data?.result?.data
+  const items = Array.isArray(responseData?.items) ? responseData.items : []
+  const total = responseData?.total ?? items.length
+
+  if (items.length === 0) {
+    console.log('\n📭 未查询到结果\n')
+    return true
+  }
+
+  console.log(`\n🔎 查询结果：共 ${total} 条，当前 ${items.length} 条\n`)
+
+  for (let i = 0; i < items.length; i += 1) {
+    const item = items[i]
+    const analysis = item?.analysis || {}
+    const idx = i + 1
+    const title = toDisplayText(item?.title || item?.content, '（无标题）')
+    const publishTime = toDisplayText(item?.publish_time)
+    const pushTime = toDisplayText(item?.push_time)
+    const source = toDisplayText(getArticleSource(item))
+    const risk = toDisplayText(analysis?.risk_level || analysis?.risk_level_original)
+    const sentiment = toDisplayText(analysis?.sentiment || analysis?.sentiment_original)
+    const category = toDisplayText(analysis?.category || analysis?.category_original)
+    const summary = toDisplayText(analysis?.summary)
+    const valid = toDisplayText(analysis?.is_valid)
+
+    console.log(`${idx}. ${title}`)
+    console.log(`   来源: ${source} | 发布时间: ${publishTime} | 推送时间: ${pushTime}`)
+    console.log(`   风险: ${risk} | 情感: ${sentiment} | 分类: ${category} | 相关性: ${valid}`)
+    console.log(`   摘要: ${summary}`)
+    console.log()
+  }
+
+  return true
+}
+
+async function runDoctor() {
+  const baseUrl = await getBaseUrl('/api/search')
+  const report = {
+    code: 0,
+    msg: 'ok',
+    data: {
+      now: new Date().toISOString(),
+      baseUrl,
+      env: {
+        LINGRUI_AGENT_BASE_URL: process.env.LINGRUI_AGENT_BASE_URL ?? '',
+        LINGRUI_UPSTREAM_BASE_URL: process.env.LINGRUI_UPSTREAM_BASE_URL ?? '',
+        NODE_EXTRA_CA_CERTS: process.env.NODE_EXTRA_CA_CERTS ?? '',
+        NODE_TLS_REJECT_UNAUTHORIZED: process.env.NODE_TLS_REJECT_UNAUTHORIZED ?? '',
+      },
+      checks: {
+        health: null,
+        whoami: null,
+        upstreamReachability: null,
+      },
+    },
+  }
+
+  try {
+    report.data.checks.health = await tryRequest(baseUrl, '/health', { timeoutMs: 2000 })
+  } catch (error) {
+    report.data.checks.health = {
+      code: 50010,
+      msg: 'local_agent_unavailable',
+      error: error instanceof Error ? error.message : String(error),
+    }
+  }
+
+  try {
+    report.data.checks.whoami = await tryRequest(baseUrl, '/auth/whoami', { timeoutMs: 2000 })
+  } catch (error) {
+    report.data.checks.whoami = {
+      code: 50010,
+      msg: 'local_agent_unavailable',
+      error: error instanceof Error ? error.message : String(error),
+    }
+  }
+
+  const upstreamBaseUrl = report.data.checks.health?.data?.upstreamBaseUrl
+  if (upstreamBaseUrl) {
+    try {
+      const response = await fetch(`${upstreamBaseUrl}/v1/article/search`, {
+        method: 'POST',
+        headers: { 'content-type': 'application/json' },
+        body: '{}',
+      })
+      report.data.checks.upstreamReachability = {
+        ok: true,
+        status: response.status,
+      }
+    } catch (error) {
+      report.data.checks.upstreamReachability = {
+        ok: false,
+        message: error instanceof Error ? error.message : String(error),
+        cause:
+          error && typeof error === 'object' && 'cause' in error && error.cause instanceof Error
+            ? error.cause.message
+            : '',
+      }
+    }
+  } else {
+    report.data.checks.upstreamReachability = {
+      ok: false,
+      message: 'upstream_not_configured',
+    }
+  }
+
+  printJson(report)
+}
+
+async function main() {
+  const [command, subCommand, ...restArgs] = process.argv.slice(2)
+
+  if (!command || command === '--help' || command === '-h' || command === 'help') {
+    printHelp()
+    return
+  }
+
+  if (command === 'agent' && subCommand === 'status') {
+    const payload = await request('/health')
+    if (payload) printJson(payload)
+    return
+  }
+
+  if (command === 'agent' && subCommand === 'start') {
+    await startAgent()
+    return
+  }
+
+  if (command === 'login' || (command === 'auth' && subCommand === 'login')) {
+    await runBrowserLogin()
+    return
+  }
+
+  if (command === 'status') {
+    const payload = await requestStatus()
+    if (!payload) {
+      console.log(`\n❌ 状态异常 [错误代码: 50010]`)
+      console.log(`💡 提示: ${getHint('local_agent_unavailable')}\n`)
+      process.exitCode = 1
+      return
+    }
+
+    if (payload.code !== 0) {
+      const hint = getHint(payload.msg ?? payload.message)
+      console.log(`\n❌ 状态异常 [错误代码: ${payload.code}]`)
+      if (hint) console.log(`💡 提示: ${hint}`)
+      console.log()
+      process.exitCode = 1
+      return
+    }
+
+    console.log(`\n🟢 本地助手运行状态：正常`)
+    console.log(`服务标识：${payload.data.service}`)
+    console.log(`登录状态：${payload.data.loggedIn ? '✅ 已登录' : '❌ 未登录'}`)
+    if (payload.data.loggedIn && payload.data.user) {
+      console.log(`当前账号：${payload.data.user.nickname || payload.data.user.username}`)
+    }
+    console.log()
+    return
+  }
+
+  if (command === 'health') {
+    const payload = await request('/health')
+    if (payload) printJson(payload)
+    return
+  }
+
+  if (command === 'doctor' || command === 'diag') {
+    await runDoctor()
+    return
+  }
+
+  if (command === 'whoami') {
+    const payload = await request('/auth/whoami')
+    if (payload && payload.data && payload.data.user) {
+      const user = payload.data.user
+      console.log(`\n👤 当前登录用户信息：`)
+      console.log(`-----------------------------------`)
+      console.log(`账号名：${user.username}`)
+      console.log(`昵  称：${user.nickname}`)
+      console.log(`机  构：${user.institution || '无'}`)
+      console.log(`角  色：${user.role === 'admin' ? '管理员' : '普通用户'}`)
+      if (payload.data.expiresAt) {
+        console.log(`过  期：${new Date(payload.data.expiresAt).toLocaleString('zh-CN')}`)
+      }
+      console.log(`-----------------------------------\n`)
+    }
+    return
+  }
+
+  if (command === 'list') {
+    if (!subCommand) {
+      const payload = await request('/commands/list')
+      if (payload) printJson(payload)
+      return
+    }
+
+    if (subCommand === 'datacenter' || subCommand === 'fengxian') {
+      const auth = await requestStatus()
+      if (!auth) return
+      if (!auth?.data?.loggedIn) {
+        printError(ErrorCode.AUTH_REQUIRED, 'auth_required', {
+          hint: 'Please sign in to Lingrui in your browser first.',
+        })
+        return
+      }
+
+      const flags = parseFlags(restArgs)
+      printError(ErrorCode.NOT_IMPLEMENTED, 'not_implemented', {
+        command: `list ${subCommand}`,
+        params: flags,
+        note: 'Business API is not connected yet.',
+      })
+      return
+    }
+
+    printError(ErrorCode.INVALID_PARAMS, 'invalid_params', {
+      command: 'list',
+      supported: ['datacenter', 'fengxian'],
+    })
+    return
+  }
+
+  if (command === 'auth' && subCommand === 'logout') {
+    const payload = await request('/auth/logout', { method: 'POST' })
+    if (payload) printJson(payload)
+    return
+  }
+
+  if (command === 'logout') {
+    const payload = await request('/auth/logout', { method: 'POST' })
+    if (payload) {
+      console.log(`\n👋 已成功退出登录！\n`)
+    }
+    return
+  }
+
+  if (command === 'search') {
+    const auth = await requestStatus()
+    if (!auth) return
+    if (!auth?.data?.loggedIn) {
+      printError(ErrorCode.AUTH_REQUIRED, 'auth_required', {
+        hint: 'Please sign in to Lingrui in your browser first.',
+      })
+      return
+    }
+
+    const legacyScene = normalizeSearchScene(subCommand)
+    const isLegacyScene = ['data-center', 'warning-center', 'website-stats'].includes(legacyScene)
+    const searchArgs = isLegacyScene ? restArgs : [subCommand, ...restArgs].filter(Boolean)
+    const { flags, terms } = parseSearchArgs(searchArgs)
+
+    if (isLegacyScene) {
+      if (legacyScene === 'warning-center') flags.center = flags.center ?? 'warning'
+      if (legacyScene === 'data-center') flags.center = flags.center ?? 'data'
+      if (legacyScene === 'website-stats') flags.stats = flags.stats ?? 'website'
+    }
+
+    if (!getFlag(flags, 'q', 'keywords', 'keyword', 'kw', 'k') && terms.length > 0) {
+      flags.q = terms.join(' ')
+    }
+    const body = buildSearchBody('data-center', flags)
+    const payload = await request('/api/search', {
+      method: 'POST',
+      body,
+    })
+    if (!payload) return
+
+    const wantsJson = ['1', 'true', 'yes'].includes(
+      String(process.env.LINGRUI_CLI_JSON || '').toLowerCase(),
+    )
+    if (!wantsJson && payload?.data?.result?.data?.items) {
+      formatSearchList(payload)
+    } else {
+      printJson(payload)
+    }
+    return
+  }
+
+  console.log(`\n❌ 未知命令: ${command}`)
+  console.log(`💡 输入 'blynkai help' 查看支持的命令列表。\n`)
+  process.exitCode = 1
+}
+
+async function startAgent() {
+  try {
+    const baseUrl = await getBaseUrl('/api/search')
+    if (await hasEndpoint(baseUrl, '/api/search')) {
+      const payload = await tryRequest(baseUrl, '/health')
+      printJson({
+        code: 0,
+        msg: 'ok',
+        data: {
+          status: 'already_running',
+          health: payload.data,
+          baseUrl,
+        },
+      })
+      return
+    }
+  } catch {
+    // Start a fresh local agent below.
+  }
+
+  await mkdir(runtimeDir, { recursive: true })
+  const logFd = openSync(logFile, 'a')
+  const child = spawn(process.execPath, [serverEntry], {
+    detached: true,
+    stdio: ['ignore', logFd, logFd],
+    env: process.env,
+  })
+  child.unref()
+
+  resolvedBaseUrl = null
+  printJson({
+    code: 0,
+    msg: 'ok',
+    data: {
+      status: 'started',
+      baseUrl: await waitForBaseUrl('/api/search'),
+      pid: child.pid,
+      pidFile,
+      logFile,
+    },
+  })
+}
+
+main()