395 lines
11 KiB
Go
395 lines
11 KiB
Go
package setup
|
||
|
||
import (
|
||
"context"
|
||
"fmt"
|
||
"os"
|
||
"os/exec"
|
||
"path/filepath"
|
||
"runtime"
|
||
"strings"
|
||
|
||
"github.com/gogf/gf/v2/frame/g"
|
||
)
|
||
|
||
var (
|
||
envConfigured bool
|
||
|
||
// DetectedWhisperPath 自动检测到的 whisper 命令行路径（空则使用 python -m whisper）
|
||
DetectedWhisperPath string
|
||
)
|
||
|
||
// EnsureDependencies 启动时检查并安装 ffmpeg 和 whisper
|
||
func EnsureDependencies(ctx context.Context) {
|
||
g.Log().Info(ctx, "========== 检查依赖环境 ==========")
|
||
|
||
ensureFFmpeg(ctx)
|
||
ensureWhisper(ctx)
|
||
resolveWhisperPath(ctx)
|
||
|
||
if envConfigured {
|
||
g.Log().Info(ctx, "依赖检查完成，新环境变量已配置，建议重启终端")
|
||
} else {
|
||
g.Log().Info(ctx, "依赖检查完成，所有依赖已就绪")
|
||
}
|
||
g.Log().Info(ctx, "===================================")
|
||
}
|
||
|
||
// ensureFFmpeg 确保 ffmpeg 可用
|
||
func ensureFFmpeg(ctx context.Context) {
|
||
if _, err := exec.LookPath("ffmpeg"); err == nil {
|
||
g.Log().Info(ctx, "[ffmpeg] ✔ 已安装")
|
||
return
|
||
}
|
||
|
||
g.Log().Infof(ctx, "[ffmpeg] 未找到，尝试自动安装...")
|
||
|
||
switch runtime.GOOS {
|
||
case "darwin":
|
||
// 检查是否安装了 Homebrew
|
||
if _, err := exec.LookPath("brew"); err != nil {
|
||
g.Log().Warningf(ctx, "[ffmpeg] ⚠ 未检测到 Homebrew，请手动安装:\n brew install ffmpeg")
|
||
return
|
||
}
|
||
cmd := exec.CommandContext(ctx, "brew", "install", "ffmpeg")
|
||
output, err := cmd.CombinedOutput()
|
||
if err != nil {
|
||
g.Log().Errorf(ctx, "[ffmpeg] ❌ 安装失败: %v\n%s", err, string(output))
|
||
return
|
||
}
|
||
g.Log().Info(ctx, "[ffmpeg] ✔ 安装成功")
|
||
|
||
case "linux":
|
||
// 尝试 apt
|
||
if _, err := exec.LookPath("apt"); err == nil {
|
||
cmd := exec.CommandContext(ctx, "sudo", "apt", "install", "-y", "ffmpeg")
|
||
output, err := cmd.CombinedOutput()
|
||
if err != nil {
|
||
g.Log().Errorf(ctx, "[ffmpeg] ❌ apt 安装失败: %v\n%s", err, string(output))
|
||
return
|
||
}
|
||
g.Log().Info(ctx, "[ffmpeg] ✔ 安装成功")
|
||
return
|
||
}
|
||
// 尝试 yum
|
||
if _, err := exec.LookPath("yum"); err == nil {
|
||
cmd := exec.CommandContext(ctx, "sudo", "yum", "install", "-y", "ffmpeg")
|
||
output, err := cmd.CombinedOutput()
|
||
if err != nil {
|
||
g.Log().Errorf(ctx, "[ffmpeg] ❌ yum 安装失败: %v\n%s", err, string(output))
|
||
return
|
||
}
|
||
g.Log().Info(ctx, "[ffmpeg] ✔ 安装成功")
|
||
return
|
||
}
|
||
g.Log().Warningf(ctx, "[ffmpeg] ⚠ 请手动安装: sudo apt install ffmpeg")
|
||
|
||
default:
|
||
g.Log().Warningf(ctx, "[ffmpeg] ⚠ 不支持的平台(%s)，请手动安装 ffmpeg", runtime.GOOS)
|
||
}
|
||
}
|
||
|
||
// ensureWhisper 确保 whisper 可用（优先安装 C++ 版，速度更快）
|
||
func ensureWhisper(ctx context.Context) {
|
||
// 1. 检查是否已有 whisper-cpp（C++ 版，最快）
|
||
if path, err := exec.LookPath("whisper-cpp"); err == nil {
|
||
g.Log().Infof(ctx, "[whisper] ✔ C++ 版已安装: %s", path)
|
||
return
|
||
}
|
||
if path, err := exec.LookPath("whisper-cli"); err == nil {
|
||
g.Log().Infof(ctx, "[whisper] ✔ C++ 版已安装: %s", path)
|
||
return
|
||
}
|
||
|
||
// 2. 检查 Homebrew 安装目录（即使不在 PATH 也能找到）
|
||
if p := findHomebrewWhisperCpp(); p != "" {
|
||
DetectedWhisperPath = p
|
||
// 自动添加到 PATH 环境变量
|
||
addToShellPath(ctx, filepath.Dir(p))
|
||
g.Log().Infof(ctx, "[whisper] ✔ C++ 版已安装(自动检测): %s", p)
|
||
return
|
||
}
|
||
|
||
// 3. 尝试安装 whisper-cpp（C++ 版）
|
||
if runtime.GOOS == "darwin" {
|
||
if _, err := exec.LookPath("brew"); err == nil {
|
||
g.Log().Infof(ctx, "[whisper] 安装 C++ 版 (brew install whisper-cpp)...")
|
||
cmd := exec.CommandContext(ctx, "brew", "install", "whisper-cpp")
|
||
output, err := cmd.CombinedOutput()
|
||
if err == nil {
|
||
g.Log().Info(ctx, "[whisper] ✔ C++ 版安装成功")
|
||
// 装好后把 Homebrew bin 加到 PATH
|
||
addToShellPath(ctx, getHomebrewBinDir())
|
||
// 检测安装路径
|
||
if p := findHomebrewWhisperCpp(); p != "" {
|
||
DetectedWhisperPath = p
|
||
}
|
||
return
|
||
}
|
||
g.Log().Warningf(ctx, "[whisper] ⚠ brew 安装失败: %v\n%s", err, string(output))
|
||
g.Log().Infof(ctx, "[whisper] 降级安装 Python 版...")
|
||
}
|
||
}
|
||
|
||
// 4. 降级：检查 python -m whisper 是否可用
|
||
if pythonWhisperAvailable() {
|
||
g.Log().Info(ctx, "[whisper] ✔ Python 版已安装 (python3 -m whisper)")
|
||
return
|
||
}
|
||
|
||
// 5. 降级：pip 安装 Python 版
|
||
if _, err := exec.LookPath("pip3"); err != nil {
|
||
if _, err2 := exec.LookPath("pip"); err2 != nil {
|
||
g.Log().Warningf(ctx, "[whisper] ⚠ 未找到 pip，请手动安装:\n pip3 install openai-whisper")
|
||
return
|
||
}
|
||
}
|
||
|
||
g.Log().Infof(ctx, "[whisper] 安装 Python 版 (pip install openai-whisper)...")
|
||
pipCmd := "pip3"
|
||
if _, err := exec.LookPath("pip3"); err != nil {
|
||
pipCmd = "pip"
|
||
}
|
||
|
||
cmd := exec.CommandContext(ctx, pipCmd, "install", "--user", "openai-whisper")
|
||
output, err := cmd.CombinedOutput()
|
||
if err != nil {
|
||
g.Log().Errorf(ctx, "[whisper] ❌ pip 安装失败: %v\n%s", err, string(output))
|
||
return
|
||
}
|
||
g.Log().Info(ctx, "[whisper] ✔ Python 版安装成功")
|
||
|
||
// 安装后自动配置 PATH
|
||
configureWhisperPath(ctx)
|
||
}
|
||
|
||
// resolveWhisperPath 自动找到 whisper 二进制路径并存储
|
||
func resolveWhisperPath(ctx context.Context) {
|
||
// 0. 如果已经通过 ensure 检测到了路径，直接使用
|
||
if DetectedWhisperPath != "" {
|
||
if _, err := os.Stat(DetectedWhisperPath); err == nil {
|
||
g.Log().Infof(ctx, "[whisper] ✔ 路径: %s", DetectedWhisperPath)
|
||
return
|
||
}
|
||
}
|
||
|
||
// 1. 优先检测 C++ 版本（快 3-5 倍）
|
||
for _, name := range []string{"whisper-cpp", "whisper-cli"} {
|
||
if path, err := exec.LookPath(name); err == nil {
|
||
DetectedWhisperPath = path
|
||
g.Log().Infof(ctx, "[whisper] ✔ C++ 版: %s", path)
|
||
return
|
||
}
|
||
}
|
||
|
||
// 2. 在 Homebrew 目录查找 C++ 版本
|
||
if p := findHomebrewWhisperCpp(); p != "" {
|
||
DetectedWhisperPath = p
|
||
g.Log().Infof(ctx, "[whisper] ✔ C++ 版(自动检测): %s", p)
|
||
return
|
||
}
|
||
|
||
// 3. 从 PATH 查找 Python 版 whisper
|
||
if path, err := exec.LookPath("whisper"); err == nil {
|
||
DetectedWhisperPath = path
|
||
g.Log().Infof(ctx, "[whisper] ✔ Python 版: %s", path)
|
||
return
|
||
}
|
||
|
||
// 4. 尝试常见 pip user bin 路径
|
||
for _, p := range getWhisperCandidates() {
|
||
if info, err := os.Stat(p); err == nil && !info.IsDir() {
|
||
DetectedWhisperPath = p
|
||
g.Log().Infof(ctx, "[whisper] ✔ Python 版(自动检测): %s", p)
|
||
return
|
||
}
|
||
}
|
||
|
||
g.Log().Info(ctx, "[whisper] ✔ 使用 python3 -m whisper 方式")
|
||
}
|
||
|
||
// getWhisperCandidates 返回可能的 whisper 二进制路径
|
||
func getWhisperCandidates() []string {
|
||
var candidates []string
|
||
|
||
// 通过 python 探针获取 user-site bin 目录
|
||
if p := getUserPythonBin(); p != "" {
|
||
candidates = append(candidates, filepath.Join(p, "whisper"))
|
||
}
|
||
|
||
// 常见 pip user base 路径
|
||
userHome, _ := os.UserHomeDir()
|
||
|
||
switch runtime.GOOS {
|
||
case "darwin":
|
||
// macOS 常见的 Python 版本路径
|
||
pythonVersions := []string{"3.9", "3.10", "3.11", "3.12", "3.13"}
|
||
for _, ver := range pythonVersions {
|
||
candidates = append(candidates,
|
||
filepath.Join(userHome, "Library", "Python", ver, "bin", "whisper"),
|
||
)
|
||
}
|
||
case "linux":
|
||
candidates = append(candidates,
|
||
filepath.Join(userHome, ".local", "bin", "whisper"),
|
||
)
|
||
}
|
||
|
||
return candidates
|
||
}
|
||
|
||
// getUserPythonBin 通过 python 获取 user bin 目录
|
||
func getUserPythonBin() string {
|
||
pythonCandidates := []string{"python3", "python"}
|
||
for _, py := range pythonCandidates {
|
||
path, err := exec.LookPath(py)
|
||
if err != nil {
|
||
continue
|
||
}
|
||
cmd := exec.Command(path, "-m", "site", "--user-base")
|
||
output, err := cmd.Output()
|
||
if err != nil {
|
||
continue
|
||
}
|
||
base := strings.TrimSpace(string(output))
|
||
if base != "" {
|
||
return filepath.Join(base, "bin")
|
||
}
|
||
}
|
||
return ""
|
||
}
|
||
|
||
// configureWhisperPath 将 pip user bin 目录加到 shell 配置
|
||
func configureWhisperPath(ctx context.Context) {
|
||
binDir := getUserPythonBin()
|
||
if binDir == "" {
|
||
return
|
||
}
|
||
|
||
// 检查是否已经在 PATH 中
|
||
currentPath := os.Getenv("PATH")
|
||
if strings.Contains(currentPath, binDir) {
|
||
return
|
||
}
|
||
|
||
// 配置到 .zshrc 或 .bashrc
|
||
home, _ := os.UserHomeDir()
|
||
rcFiles := []string{".zshrc", ".bashrc", ".bash_profile"}
|
||
|
||
for _, rc := range rcFiles {
|
||
rcPath := filepath.Join(home, rc)
|
||
// 文件不存在则跳过
|
||
if _, err := os.Stat(rcPath); os.IsNotExist(err) {
|
||
continue
|
||
}
|
||
// 检查是否已添加
|
||
data, _ := os.ReadFile(rcPath)
|
||
if strings.Contains(string(data), binDir) {
|
||
continue
|
||
}
|
||
// 追加
|
||
line := fmt.Sprintf("\nexport PATH=\"%s:$PATH\"\n", binDir)
|
||
f, err := os.OpenFile(rcPath, os.O_APPEND|os.O_WRONLY, 0644)
|
||
if err != nil {
|
||
g.Log().Warningf(ctx, "[whisper] 写入 %s 失败: %v", rc, err)
|
||
continue
|
||
}
|
||
f.WriteString(line)
|
||
f.Close()
|
||
g.Log().Infof(ctx, "[whisper] 已将 %s 添加到 %s，请执行: source ~/%s", binDir, rc, rc)
|
||
envConfigured = true
|
||
break
|
||
}
|
||
}
|
||
|
||
// pythonWhisperAvailable 检查 python -m whisper 是否可用
|
||
func pythonWhisperAvailable() bool {
|
||
pythonCandidates := []string{"python3", "python"}
|
||
for _, py := range pythonCandidates {
|
||
if path, err := exec.LookPath(py); err == nil {
|
||
cmd := exec.Command(path, "-m", "whisper", "--help")
|
||
if cmd.Run() == nil {
|
||
return true
|
||
}
|
||
}
|
||
}
|
||
return false
|
||
}
|
||
|
||
// findHomebrewWhisperCpp 在 Homebrew 安装目录查找 whisper-cpp
|
||
func findHomebrewWhisperCpp() string {
|
||
dirs := getHomebrewBinDirs()
|
||
for _, dir := range dirs {
|
||
for _, name := range []string{"whisper-cpp", "whisper-cli"} {
|
||
p := filepath.Join(dir, name)
|
||
if info, err := os.Stat(p); err == nil && !info.IsDir() {
|
||
return p
|
||
}
|
||
}
|
||
}
|
||
return ""
|
||
}
|
||
|
||
// getHomebrewBinDirs 返回 Homebrew 可能的 bin 目录
|
||
func getHomebrewBinDirs() []string {
|
||
userHome, _ := os.UserHomeDir()
|
||
return []string{
|
||
"/opt/homebrew/bin", // Apple Silicon
|
||
"/usr/local/bin", // Intel
|
||
filepath.Join(userHome, ".homebrew", "bin"),
|
||
}
|
||
}
|
||
|
||
// getHomebrewBinDir 返回当前系统的 Homebrew bin 目录
|
||
func getHomebrewBinDir() string {
|
||
dirs := getHomebrewBinDirs()
|
||
for _, dir := range dirs {
|
||
if _, err := os.Stat(filepath.Join(dir, "brew")); err == nil {
|
||
return dir
|
||
}
|
||
// 也检查 brew 命令路径
|
||
if path, err := exec.LookPath("brew"); err == nil {
|
||
return filepath.Dir(path)
|
||
}
|
||
}
|
||
return "/opt/homebrew/bin" // 默认 Apple Silicon 路径
|
||
}
|
||
|
||
// addToShellPath 将目录添加到 shell rc 文件的 PATH 中
|
||
func addToShellPath(ctx context.Context, dir string) {
|
||
if dir == "" {
|
||
return
|
||
}
|
||
|
||
// 检查是否已在 PATH 中
|
||
currentPath := os.Getenv("PATH")
|
||
if strings.Contains(currentPath, dir) {
|
||
return
|
||
}
|
||
|
||
home, _ := os.UserHomeDir()
|
||
rcFiles := []string{".zshrc", ".bashrc", ".bash_profile"}
|
||
|
||
for _, rc := range rcFiles {
|
||
rcPath := filepath.Join(home, rc)
|
||
if _, err := os.Stat(rcPath); os.IsNotExist(err) {
|
||
continue
|
||
}
|
||
data, _ := os.ReadFile(rcPath)
|
||
if strings.Contains(string(data), dir) {
|
||
continue
|
||
}
|
||
line := fmt.Sprintf("\nexport PATH=\"%s:$PATH\"\n", dir)
|
||
f, err := os.OpenFile(rcPath, os.O_APPEND|os.O_WRONLY, 0644)
|
||
if err != nil {
|
||
g.Log().Warningf(ctx, "[setup] 写入 %s 失败: %v", rc, err)
|
||
continue
|
||
}
|
||
f.WriteString(line)
|
||
f.Close()
|
||
g.Log().Infof(ctx, "[setup] 已将 %s 添加到 %s", dir, rc)
|
||
envConfigured = true
|
||
break
|
||
}
|
||
}
|