@wwkit/opm 1.0.1

package/src/managers/pip.js

@@ -0,0 +1,555 @@
+/**
+ * pip 包管理器实现
+ *
+ * 通过执行 pip CLI 命令实现统一接口。
+ * 源查询使用 PyPI/Index JSON API，本地查询使用 pip show/list。
+ */
+
+import { execSync, spawnSync } from 'node:child_process'
+import path from 'node:path'
+import { URL } from 'node:url'
+import { PackageManager } from './base.js'
+import { fetchText } from './http.js'
+import { getActiveRegistry, setActiveRegistry } from '../config.js'
+
+export class PipManager extends PackageManager {
+  constructor() {
+    super()
+    this.name = 'pip'
+  }
+
+  /**
+   * 执行 pip 命令，返回 stdout
+   * execSync 默认通过 shell 执行，兼容 Windows 的 pip.exe
+   * @param {string[]} args
+   * @param {object} [opts]
+   * @returns {string}
+   * @private
+   */
+  _exec(args, opts = {}) {
+    const cmd = ['pip', ...args].join(' ')
+    try {
+      return execSync(cmd, { encoding: 'utf8', stdio: ['pipe', 'pipe', 'pipe'], ...opts })
+    } catch (err) {
+      if (opts.allowNonZero) {
+        return err.stdout ? err.stdout.toString() : ''
+      }
+      const stderr = err.stderr ? err.stderr.toString().trim() : ''
+      if (stderr) {
+        throw new Error(stderr)
+      }
+      return err.stdout ? err.stdout.toString() : ''
+    }
+  }
+
+  /**
+   * 执行 pip 命令，输出直接继承到终端
+   * @param {string[]} args
+   * @private
+   */
+  _execInherit(args) {
+    const result = spawnSync('pip', args, { stdio: 'inherit', shell: true })
+    if (result.error) {
+      throw new Error(result.error.message)
+    }
+    if (result.status !== 0) {
+      throw new Error(`pip ${args.join(' ')} exited with code ${result.status}`)
+    }
+  }
+
+  /**
+   * 获取系统级 Python 可执行文件路径（绕过 venv/conda）
+   *
+   * 通过 sys.base_prefix 定位系统 Python：
+   *   - venv 激活时: base_prefix 指向创建 venv 的系统 Python
+   *   - conda 激活时: base_prefix 指向 conda base 环境
+   *   - 未激活任何环境: base_prefix === sys.prefix，直接用当前 python
+   *
+   * 兼容 uv 创建的 venv（标准 venv 结构，sys.base_prefix 正确）
+   *
+   * @returns {string} 系统 Python 可执行文件路径
+   * @private
+   */
+  _getBasePython() {
+    const script = 'import sys; print(sys.base_prefix)'
+    let basePrefix = ''
+
+    try {
+      basePrefix = execSync(`python -c "${script}"`, {
+        encoding: 'utf8',
+        stdio: ['pipe', 'pipe', 'pipe']
+      }).trim()
+    } catch {}
+
+    if (!basePrefix) {
+      return 'python'
+    }
+
+    const currentPrefix = process.env.VIRTUAL_ENV
+      || process.env.CONDA_PREFIX
+      || ''
+
+    if (!currentPrefix) {
+      return 'python'
+    }
+
+    const isWindows = process.platform === 'win32'
+    const pythonExe = isWindows ? 'python.exe' : 'python'
+    return path.join(basePrefix, pythonExe)
+  }
+
+  /**
+   * 按作用域执行 pip 命令，返回 stdout
+   *
+   * global=true 时通过系统 Python 的 -m pip 绕过当前 venv/conda 环境
+   * global=false 时直接用当前环境 pip（venv/conda/系统）
+   *
+   * @param {string[]} args - pip 参数
+   * @param {object} [opts] - 选项
+   * @param {boolean} [opts.global] - 是否查询系统级
+   * @returns {string}
+   * @private
+   */
+  _execPip(args, opts = {}) {
+    if (opts.global) {
+      const basePython = this._getBasePython()
+      const cmd = [basePython, '-m', 'pip', ...args].join(' ')
+      try {
+        return execSync(cmd, {
+          encoding: 'utf8',
+          stdio: ['pipe', 'pipe', 'pipe'],
+          ...(opts.execOpts || {})
+        })
+      } catch (err) {
+        if (opts.allowNonZero) {
+          return err.stdout ? err.stdout.toString() : ''
+        }
+        const stderr = err.stderr ? err.stderr.toString().trim() : ''
+        if (stderr) {
+          throw new Error(stderr)
+        }
+        return err.stdout ? err.stdout.toString() : ''
+      }
+    }
+    return this._exec(args, opts)
+  }
+
+  /**
+   * 按作用域执行 pip 命令，输出直接继承到终端
+   * @param {string[]} args - pip 参数
+   * @param {object} [opts] - 选项
+   * @param {boolean} [opts.global] - 是否操作系统级
+   * @private
+   */
+  _execPipInherit(args, opts = {}) {
+    if (opts.global) {
+      const basePython = this._getBasePython()
+      const result = spawnSync(basePython, ['-m', 'pip', ...args], {
+        stdio: 'inherit',
+        shell: true
+      })
+      if (result.error) {
+        throw new Error(result.error.message)
+      }
+      if (result.status !== 0) {
+        throw new Error(`pip ${args.join(' ')} exited with code ${result.status}`)
+      }
+      return
+    }
+    this._execInherit(args)
+  }
+
+  /**
+   * 获取当前 index-url（来自 opm 配置的 active 镜像，key 已解析为 URL）
+   * @returns {string}
+   * @private
+   */
+  _getIndexUrl() {
+    return getActiveRegistry('pip') || 'https://pypi.org/simple'
+  }
+
+  /**
+   * 从 index-url 构造 JSON API URL
+   * pip 的 index-url 格式为 https://pypi.org/simple
+   * JSON API 为 https://pypi.org/pypi/<name>/json
+   * @param {string} indexUrl
+   * @returns {string}
+   * @private
+   */
+  _buildApiUrl(indexUrl) {
+    let base = indexUrl.replace(/\/+$/, '')
+    if (base.endsWith('/simple')) {
+      base = base.slice(0, -7)
+    }
+    return base
+  }
+
+  /**
+   * 构造 simple index URL
+   * 格式为 <index-url>/<name>/
+   * @param {string} indexUrl
+   * @param {string} name
+   * @returns {string}
+   * @private
+   */
+  _buildSimpleUrl(indexUrl, name) {
+    const base = indexUrl.replace(/\/+$/, '')
+    return `${base}/${encodeURIComponent(name)}/`
+  }
+
+  /**
+   * 构造 pip index-url 参数（使用配置中 active 镜像，key 已解析为 URL）
+   * @returns {string[]}
+   * @private
+   */
+  _indexUrlArgs() {
+    const url = this._getIndexUrl()
+    return url ? ['--index-url', url] : []
+  }
+
+  /**
+   * 通过代理或直连获取 URL 文本内容（复用共享隧道实现）
+   * @param {string} url - 请求 URL
+   * @param {{ proxy?: string, timeout?: number, maxRedirects?: number }} [opts] - 选项
+   * @returns {Promise<string|null>}
+   * @private
+   */
+  async _fetchText(url, opts = {}) {
+    return fetchText(url, opts)
+  }
+
+
+  /**
+   * 通过 JSON API 查询包信息
+   * @param {string} name
+   * @param {{ proxy?: string }} [opts] - 选项
+   * @returns {Promise<object|null>}
+   * @private
+   */
+  async _fetchApiData(name, opts = {}) {
+    const indexUrl = this._getIndexUrl()
+    const apiBase = this._buildApiUrl(indexUrl)
+    const url = `${apiBase}/pypi/${encodeURIComponent(name)}/json`
+
+    try {
+      const text = await this._fetchText(url, opts)
+      if (!text) return null
+      return JSON.parse(text)
+    } catch {
+      return null
+    }
+  }
+
+  /**
+   * 通过 simple index 解析包版本列表
+   * 当 JSON API 不可用时，fallback 到 simple index HTML 解析
+   * @param {string} name
+   * @param {{ proxy?: string }} [opts] - 选项
+   * @returns {Promise<string[]|null>}
+   * @private
+   */
+  async _fetchVersionsFromSimple(name, opts = {}) {
+    const indexUrl = this._getIndexUrl()
+    const url = this._buildSimpleUrl(indexUrl, name)
+
+    try {
+      const html = await this._fetchText(url, opts)
+      if (!html) return null
+      const versions = new Set()
+      const normalizedName = name.replace(/[-_.]/g, '[-_.]')
+      const sdistRegex = new RegExp(`${normalizedName}-(.+?)\\.tar\\.gz`, 'g')
+      const wheelRegex = new RegExp(`${normalizedName}-([^-]+)-[^-]+-[^-]+\\.whl`, 'g')
+      let match
+      while ((match = sdistRegex.exec(html)) !== null) {
+        versions.add(match[1])
+      }
+      while ((match = wheelRegex.exec(html)) !== null) {
+        versions.add(match[1])
+      }
+      return versions.size > 0 ? [...versions] : null
+    } catch {
+      return null
+    }
+  }
+
+  async getRegistry() {
+    const registry = this._getIndexUrl()
+    return { registry }
+  }
+
+  async getVersion() {
+    try {
+      const out = this._exec(['--version'], { allowNonZero: true }).trim()
+      const match = out.match(/pip\s+([\d.]+)/i)
+      if (match) {
+        return { name: 'pip', installed: true, version: match[1] }
+      }
+      if (out) {
+        return { name: 'pip', installed: true, version: out }
+      }
+    } catch {}
+    return { name: 'pip', version: '', installed: false, hint: 'Install pip from https://pip.pypa.io/en/stable/installation/' }
+  }
+
+  async setRegistry(url) {
+    const resolved = setActiveRegistry('pip', url)
+    return { registry: resolved }
+  }
+
+  /**
+   * 比较两个版本号，返回 -1/0/1
+   * 支持预发布版本（如 2.34.0.dev1）
+   * @param {string} a
+   * @param {string} b
+   * @returns {number}
+   * @private
+   */
+  _compareVersions(a, b) {
+    const parseVer = (v) => {
+      const parts = v.split('.')
+      return parts.map((p) => {
+        const n = parseInt(p, 10)
+        return isNaN(n) ? p : n
+      })
+    }
+    const pa = parseVer(a)
+    const pb = parseVer(b)
+    const len = Math.max(pa.length, pb.length)
+    for (let i = 0; i < len; i++) {
+      const va = pa[i] ?? 0
+      const vb = pb[i] ?? 0
+      if (typeof va === 'number' && typeof vb === 'number') {
+        if (va < vb) return -1
+        if (va > vb) return 1
+      } else if (typeof va === 'string' && typeof vb === 'number') {
+        return -1
+      } else if (typeof va === 'number' && typeof vb === 'string') {
+        return 1
+      } else {
+        const sa = String(va)
+        const sb = String(vb)
+        if (sa < sb) return -1
+        if (sa > sb) return 1
+      }
+    }
+    return 0
+  }
+
+  /**
+   * 从版本列表中找出最新的稳定版本（排除 dev/rc/alpha/beta 等预发布）
+   * @param {string[]} versions
+   * @returns {string}
+   * @private
+   */
+  _pickLatest(versions) {
+    const stable = versions.filter((v) => !/dev|rc|alpha|beta|a\d|b\d/i.test(v))
+    const pool = stable.length > 0 ? stable : versions
+    const sorted = [...pool].sort((a, b) => this._compareVersions(a, b))
+    return sorted[sorted.length - 1] || 'unknown'
+  }
+
+  async searchPackage(name, version, opts = {}) {
+    const { registry } = await this.getRegistry()
+    const data = await this._fetchApiData(name, opts)
+
+    if (data) {
+      const allVersions = Object.keys(data.releases || {})
+      const latest = data.info?.version || allVersions[allVersions.length - 1] || 'unknown'
+
+      if (version) {
+        return {
+          name,
+          version,
+          exists: true,
+          available: allVersions.includes(version),
+          registry
+        }
+      }
+
+      return {
+        name,
+        exists: true,
+        latest,
+        versions: allVersions,
+        registry
+      }
+    }
+
+    const simpleVersions = await this._fetchVersionsFromSimple(name, opts)
+    if (simpleVersions) {
+      if (version) {
+        return {
+          name,
+          version,
+          exists: true,
+          available: simpleVersions.includes(version),
+          registry
+        }
+      }
+      return {
+        name,
+        exists: true,
+        latest: this._pickLatest(simpleVersions),
+        versions: simpleVersions,
+        registry
+      }
+    }
+
+    if (version) {
+      return { name, version, exists: false, available: false, registry }
+    }
+    return { name, exists: false, registry }
+  }
+
+  /**
+   * 解析 pip show 输出为键值对
+   * @param {string} name - 包名
+   * @param {object} [opts] - 选项
+   * @param {boolean} [opts.global] - 是否查询系统级
+   * @returns {Record<string, string>|null}
+   * @private
+   */
+  _parsePipShow(name, opts = {}) {
+    try {
+      const out = this._execPip(['show', name], opts)
+      const lines = out.split('\n')
+      const info = {}
+      for (const line of lines) {
+        const idx = line.indexOf(':')
+        if (idx > 0) {
+          const key = line.slice(0, idx).trim().toLowerCase()
+          const val = line.slice(idx + 1).trim()
+          info[key] = val
+        }
+      }
+      if (info.version) {
+        return info
+      }
+    } catch {}
+    return null
+  }
+
+  async isInstalled(name, opts = {}) {
+    const info = this._parsePipShow(name, opts)
+    if (info) {
+      return {
+        name,
+        installed: true,
+        version: info.version,
+        path: info.location || '',
+        ...(opts.global ? { global: true } : {})
+      }
+    }
+
+    return { name, installed: false, ...(opts.global ? { global: true } : {}) }
+  }
+
+  async packageInfo(name, opts = {}) {
+    const data = await this._fetchApiData(name, opts)
+    if (data) {
+      const info = data.info || {}
+      return {
+        name,
+        description: info.summary || '',
+        latest: info.version || '',
+        homepage: info.home_page || info.project_url || '',
+        license: info.license || '',
+        author: info.author || '',
+        repository: info.project_urls?.Source || info.project_urls?.Repository || ''
+      }
+    }
+    const info = this._parsePipShow(name)
+    if (info) {
+      return {
+        name,
+        description: info.summary || '',
+        latest: info.version || '',
+        homepage: info['home-page'] || info['project-url'] || '',
+        license: info.license || '',
+        author: info.author || '',
+        repository: ''
+      }
+    }
+    return { name, description: '(not found)' }
+  }
+
+  async packageVersions(name, opts = {}) {
+    const { registry } = await this.getRegistry()
+    const data = await this._fetchApiData(name, opts)
+    if (data) {
+      const versions = Object.keys(data.releases || {})
+      return { name, versions, registry }
+    }
+    const simpleVersions = await this._fetchVersionsFromSimple(name, opts)
+    if (simpleVersions) {
+      return { name, versions: simpleVersions, registry }
+    }
+    return { name, versions: [], registry }
+  }
+
+  async listInstalled(opts = {}) {
+    try {
+      const out = this._execPip(['list', '--format=freeze'], opts)
+      const items = []
+      for (const line of out.split('\n')) {
+        const trimmed = line.trim()
+        if (!trimmed) continue
+        const match = trimmed.match(/^([a-zA-Z0-9_-]+)==(.+)$/)
+        if (match) {
+          items.push({ name: match[1], version: match[2], ...(opts.global ? { global: true } : {}) })
+        }
+      }
+      return items
+    } catch (err) {
+      throw new Error(`pip list failed: ${err.message}`)
+    }
+  }
+
+  async listOutdated(opts = {}) {
+    try {
+      const args = ['list', '--outdated', '--format=json']
+      if (opts.proxy) args.push('--proxy', opts.proxy)
+      const jsonOut = this._execPip(args, { ...opts, allowNonZero: true })
+      const parsed = JSON.parse(jsonOut)
+      if (!Array.isArray(parsed)) return []
+      return parsed.map((item) => ({
+        name: item.name || '',
+        current: item.version || '',
+        latest: item.latest_version || '',
+        ...(opts.global ? { global: true } : {})
+      }))
+    } catch (err) {
+      if (err.message.includes('No packages') || err.message.includes('No broken')) {
+        return []
+      }
+      throw new Error(`pip list --outdated failed: ${err.message}`)
+    }
+  }
+
+  async install(name, version, opts = {}) {
+    const spec = version ? `${name}==${version}` : name
+    const args = ['install', spec]
+    args.push(...this._indexUrlArgs())
+    if (opts.proxy) args.push('--proxy', opts.proxy)
+    this._execPipInherit(args, opts)
+    return { name, version: version || 'latest', action: 'installed' }
+  }
+
+  async uninstall(name, opts = {}) {
+    this._execPipInherit(['uninstall', '-y', name], opts)
+    return { name, action: 'uninstalled' }
+  }
+
+  async upgrade(name, opts = {}) {
+    const args = ['install', '--upgrade', name]
+    args.push(...this._indexUrlArgs())
+    if (opts.proxy) args.push('--proxy', opts.proxy)
+    this._execPipInherit(args, opts)
+    return { name, version: 'latest', action: 'upgraded' }
+  }
+
+  async cleanCache() {
+    this._execInherit(['cache', 'purge'])
+    return { action: 'cache cleaned' }
+  }
+}