@simonyea/holysheep-cli 1.0.0

package/src/utils/config.js

@@ -0,0 +1,52 @@
+/**
+ * 本地配置管理 — 存储 API Key 和用户偏好
+ * 配置文件: ~/.holysheep/config.json
+ */
+const fs = require('fs')
+const path = require('path')
+const os = require('os')
+
+const CONFIG_DIR = path.join(os.homedir(), '.holysheep')
+const CONFIG_FILE = path.join(CONFIG_DIR, 'config.json')
+
+const BASE_URL_ANTHROPIC = 'https://api.holysheep.ai'         // 不带 /v1 (Anthropic SDK)
+const BASE_URL_OPENAI    = 'https://api.holysheep.ai/v1'      // 带 /v1 (OpenAI 兼容)
+const SHOP_URL           = 'https://shop.holysheep.ai'
+
+function ensureDir() {
+  if (!fs.existsSync(CONFIG_DIR)) {
+    fs.mkdirSync(CONFIG_DIR, { recursive: true })
+  }
+}
+
+function loadConfig() {
+  try {
+    if (fs.existsSync(CONFIG_FILE)) {
+      return JSON.parse(fs.readFileSync(CONFIG_FILE, 'utf8'))
+    }
+  } catch {}
+  return {}
+}
+
+function saveConfig(data) {
+  ensureDir()
+  const current = loadConfig()
+  const merged = { ...current, ...data }
+  fs.writeFileSync(CONFIG_FILE, JSON.stringify(merged, null, 2))
+  return merged
+}
+
+function getApiKey() {
+  return loadConfig().apiKey || process.env.HOLYSHEEP_API_KEY || ''
+}
+
+module.exports = {
+  CONFIG_DIR,
+  CONFIG_FILE,
+  BASE_URL_ANTHROPIC,
+  BASE_URL_OPENAI,
+  SHOP_URL,
+  loadConfig,
+  saveConfig,
+  getApiKey,
+}

package/src/utils/shell.js

@@ -0,0 +1,83 @@
+/**
+ * Shell RC 文件管理 — 写入/清理环境变量
+ */
+const fs = require('fs')
+const path = require('path')
+const os = require('os')
+
+const MARKER_START = '# >>> holysheep-cli managed >>>'
+const MARKER_END   = '# <<< holysheep-cli managed <<<'
+
+function getShellRcFiles() {
+  const home = os.homedir()
+  const shell = process.env.SHELL || ''
+  const candidates = []
+
+  if (shell.includes('zsh'))  candidates.push(path.join(home, '.zshrc'))
+  if (shell.includes('bash')) candidates.push(path.join(home, '.bashrc'), path.join(home, '.bash_profile'))
+  if (shell.includes('fish')) candidates.push(path.join(home, '.config', 'fish', 'config.fish'))
+
+  // 默认兜底
+  if (candidates.length === 0) {
+    const zshrc = path.join(home, '.zshrc')
+    const bashrc = path.join(home, '.bashrc')
+    if (fs.existsSync(zshrc))  candidates.push(zshrc)
+    if (fs.existsSync(bashrc)) candidates.push(bashrc)
+    if (candidates.length === 0) candidates.push(zshrc)
+  }
+
+  return candidates
+}
+
+function removeHsBlock(content) {
+  const re = new RegExp(
+    `\\n?${escapeRegex(MARKER_START)}[\\s\\S]*?${escapeRegex(MARKER_END)}\\n?`,
+    'g'
+  )
+  return content.replace(re, '')
+}
+
+function buildEnvBlock(envVars) {
+  const lines = [MARKER_START]
+  for (const [k, v] of Object.entries(envVars)) {
+    lines.push(`export ${k}="${v}"`)
+  }
+  lines.push(MARKER_END)
+  return '\n' + lines.join('\n') + '\n'
+}
+
+function writeEnvToShell(envVars) {
+  const files = getShellRcFiles()
+  const written = []
+
+  for (const file of files) {
+    let content = ''
+    try { content = fs.readFileSync(file, 'utf8') } catch {}
+    content = removeHsBlock(content)
+    content += buildEnvBlock(envVars)
+    fs.writeFileSync(file, content, 'utf8')
+    written.push(file)
+  }
+  return written
+}
+
+function removeEnvFromShell() {
+  const files = getShellRcFiles()
+  const cleaned = []
+  for (const file of files) {
+    if (!fs.existsSync(file)) continue
+    let content = fs.readFileSync(file, 'utf8')
+    const cleaned_content = removeHsBlock(content)
+    if (cleaned_content !== content) {
+      fs.writeFileSync(file, cleaned_content, 'utf8')
+      cleaned.push(file)
+    }
+  }
+  return cleaned
+}
+
+function escapeRegex(s) {
+  return s.replace(/[.*+?^${}()|[\]\\]/g, '\\$&')
+}
+
+module.exports = { getShellRcFiles, writeEnvToShell, removeEnvFromShell }