@wwkit/shared 1.0.12 → 1.0.14

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@wwkit/shared",
-  "version": "1.0.12",
+  "version": "1.0.14",
   "author": "bluesliu <langcai163@163.com>",
   "description": "Shared utilities for @wwkit packages — config loader, platform detection",
   "type": "module",

package/src/shell.js

@@ -110,6 +110,45 @@ export class Shell {
     return path.join(os.homedir(), '.profile')
   }
 
+  /**
+   * 检测当前用户所有需要写入的 shell profile 文件
+   *
+   * 返回数组以覆盖多种场景（login / non-login / 不同 shell）：
+   *   - zsh:  ~/.zshrc（交互式 zsh 总是读取）
+   *   - bash: ~/.bashrc（非登录交互式）+ ~/.bash_profile（登录式，仅当已存在）
+   *   - fish: ~/.config/fish/config.fish（fish 唯一配置文件）
+   *   - 未知: ~/.profile（POSIX 通用回退）
+   *
+   * 每项含 `path`（绝对路径）和 `syntax`（'export' | 'fish'），
+   * 调用方据此生成正确语法的变量设置行。
+   *
+   * @returns {{ path: string, syntax: 'export' | 'fish' }[]}
+   */
+  getProfiles() {
+    const home = os.homedir()
+    const shell = this.env.SHELL || ''
+    const profiles = []
+
+    if (shell.includes('fish')) {
+      profiles.push({
+        path: path.join(home, '.config', 'fish', 'config.fish'),
+        syntax: 'fish',
+      })
+    } else if (shell.includes('zsh')) {
+      profiles.push({ path: path.join(home, '.zshrc'), syntax: 'export' })
+    } else if (shell.includes('bash')) {
+      profiles.push({ path: path.join(home, '.bashrc'), syntax: 'export' })
+      const bp = path.join(home, '.bash_profile')
+      if (fs.existsSync(bp)) {
+        profiles.push({ path: bp, syntax: 'export' })
+      }
+    } else {
+      profiles.push({ path: path.join(home, '.profile'), syntax: 'export' })
+    }
+
+    return profiles
+  }
+
   /**
    * 将 shell 初始化行写入 profile（按 marker 去重）
    *
@@ -132,6 +171,82 @@ export class Shell {
     return true
   }
 
+  /**
+   * 持久化设置系统环境变量（跨平台）
+   *
+   * Unix: 将变量设置行写入所有相关 shell profile（getProfiles()），使用
+   *       begin/end 标记块管理区块，支持幂等（值相同则跳过）和值更新
+   *       （路径变化时自动替换）。覆盖 bash/zsh/fish 及 login/non-login 场景。
+   * Windows: 使用 `setx` 设置用户级环境变量（写入注册表，始终覆盖）。
+   *
+   * @param {string} name - 环境变量名
+   * @param {string} value - 环境变量值
+   * @returns {boolean} 是否写入/更新了至少一个目标（全部已存在且值相同则 false）
+   */
+  setEnvVar(name, value) {
+    if (this.isWindows) {
+      try {
+        const result = spawnSync('setx', [name, String(value)], {
+          encoding: 'utf8',
+          stdio: ['pipe', 'pipe', 'pipe'],
+        })
+        return result.status === 0
+      } catch {
+        return false
+      }
+    }
+
+    const escaped = String(value).replace(/[$`\\"]/g, '\\$&')
+    let wrote = false
+
+    for (const { path: profilePath, syntax } of this.getProfiles()) {
+      const line = syntax === 'fish'
+        ? `set -gx ${name} "${escaped}"`
+        : `export ${name}="${escaped}"`
+      if (this._writeEnvBlock(profilePath, name, line)) {
+        wrote = true
+      }
+    }
+
+    return wrote
+  }
+
+  /**
+   * 向指定 profile 文件写入/更新一个环境变量标记块
+   *
+   * @param {string} profilePath - profile 文件绝对路径
+   * @param {string} name - 环境变量名（用于生成标记）
+   * @param {string} line - 变量设置行（含正确语法）
+   * @returns {boolean} 是否写入/更新了该文件
+   */
+  _writeEnvBlock(profilePath, name, line) {
+    let existing = ''
+    try {
+      existing = fs.readFileSync(profilePath, 'utf8')
+    } catch {}
+
+    const beginMarker = `# >>> wwkit ${name} >>>`
+    const endMarker = `# <<< wwkit ${name} <<<`
+    const block = `${beginMarker}\n${line}\n${endMarker}`
+
+    const beginIdx = existing.indexOf(beginMarker)
+    if (beginIdx !== -1) {
+      const endIdx = existing.indexOf(endMarker)
+      if (endIdx !== -1 && existing.includes(line)) {
+        return false
+      }
+      const afterEnd = endIdx !== -1 ? endIdx + endMarker.length : beginIdx + beginMarker.length
+      existing = existing.slice(0, beginIdx) + block + existing.slice(afterEnd)
+      fs.writeFileSync(profilePath, existing, 'utf8')
+      return true
+    }
+
+    const dir = path.dirname(profilePath)
+    fs.mkdirSync(dir, { recursive: true })
+    fs.appendFileSync(profilePath, `\n${block}\n`, 'utf8')
+    return true
+  }
+
   /**
    * 检测系统命令是否可用
    * Windows: where <cmd>；Unix: which <cmd>