Go语言中实现操作系统特定逻辑的最佳实践

<pkgname>_<osname>.go

// startup/startup.go (这是一个示例文件，实际可能不需要这个文件，或者只包含公共接口定义)
package startup

import "errors"

// SetStartupProcessLaunch 将指定的程序设置为系统启动时自动运行。
// path: 程序的完整路径。
// name: 在系统启动项中显示的名称。
func SetStartupProcessLaunch(path, name string) error {
    // 这个文件可以为空，或者包含一个通用的“未实现”错误
    return errors.New("SetStartupProcessLaunch is not implemented for this OS")
}

// startup/startup_windows.go
package startup

import (
    "fmt"
    "golang.org/x/sys/windows/registry" // 示例：Windows特有的注册表操作库
)

// SetStartupProcessLaunch 在Windows上通过修改注册表设置程序开机启动。
func SetStartupProcessLaunch(path, name string) error {
    key, _, err := registry.CreateKey(registry.CURRENT_USER, `Software\Microsoft\Windows\CurrentVersion\Run`, registry.SET_VALUE)
    if err != nil {
        return fmt.Errorf("创建或打开注册表键失败: %w", err)
    }
    defer key.Close()

    if err := key.SetStringValue(name, path); err != nil {
        return fmt.Errorf("设置注册表值失败: %w", err)
    }
    fmt.Printf("Windows: 已将程序 '%s' (%s) 添加到开机启动项。\n", name, path)
    return nil
}

// startup/startup_darwin.go
package startup

import (
    "fmt"
    "os"
    "path/filepath"
    // 实际操作plist可能需要更复杂的库或手动生成XML
)

// SetStartupProcessLaunch 在macOS上通过创建.plist文件设置程序开机启动。
func SetStartupProcessLaunch(path, name string) error {
    // 简化示例：实际操作需要生成Propery List XML文件并放置到 ~/Library/LaunchAgents/
    plistContent := fmt.Sprintf(`<?xml version="1.0" encoding="UTF-8"?>
<!DOCTYPE plist PUBLIC "-//Apple//DTD PLIST 1.0//EN" "http://www.apple.com/DTDs/PropertyList-1.0.dtd">
<plist version="1.0">
<dict>
    <key>Label</key>
    <string>%s</string>
    <key>ProgramArguments</key>
    <array>
        <string>%s</string>
    </array>
    <key>RunAtLoad</key>
    <true/>
</dict>
</plist>`, name, path)

    homeDir, err := os.UserHomeDir()
    if err != nil {
        return fmt.Errorf("获取用户主目录失败: %w", err)
    }
    launchAgentsDir := filepath.Join(homeDir, "Library", "LaunchAgents")
    if err := os.MkdirAll(launchAgentsDir, 0755); err != nil {
        return fmt.Errorf("创建LaunchAgents目录失败: %w", err)
    }

    plistPath := filepath.Join(launchAgentsDir, fmt.Sprintf("%s.plist", name))
    if err := os.WriteFile(plistPath, []byte(plistContent), 0644); err != nil {
        return fmt.Errorf("写入plist文件失败: %w", err)
    }
    fmt.Printf("macOS: 已创建plist文件 '%s' (%s) 到开机启动项。\n", name, plistPath)
    return nil
}

// startup/startup_linux.go
package startup

import (
    "fmt"
    "os"
    "path/filepath"
)

// SetStartupProcessLaunch 在Linux上通过创建.desktop文件设置程序开机启动。
func SetStartupProcessLaunch(path, name string) error {
    // 简化示例：实际可能需要考虑systemd服务或更复杂的桌面环境集成
    desktopContent := fmt.Sprintf(`[Desktop Entry]
Type=Application
Exec=%s
Hidden=false
NoDisplay=false
X-GNOME-Autostart-enabled=true
Name=%s
Comment=Launch %s at startup`, path, name, name)

    homeDir, err := os.UserHomeDir()
    if err != nil {
        return fmt.Errorf("获取用户主目录失败: %w", err)
    }
    autostartDir := filepath.Join(homeDir, ".config", "autostart")
    if err := os.MkdirAll(autostartDir, 0755); err != nil {
        return fmt.Errorf("创建autostart目录失败: %w", err)
    }

    desktopFilePath := filepath.Join(autostartDir, fmt.Sprintf("%s.desktop", name))
    if err := os.WriteFile(desktopFilePath, []byte(desktopContent), 0644); err != nil {
        return fmt.Errorf("写入.desktop文件失败: %w", err)
    }
    fmt.Printf("Linux: 已创建.desktop文件 '%s' (%s) 到开机启动项。\n", name, desktopFilePath)
    return nil
}

// main.go
package main

import (
    "fmt"
    "os"
    "path/filepath"
    "your_module/startup" // 假设你的模块路径是 your_module
)

func main() {
    // 获取当前执行程序的路径
    exePath, err := os.Executable()
    if err != nil {
        fmt.Printf("获取可执行文件路径失败: %v\n", err)
        return
    }
    // 确保路径是绝对路径
    absPath, err := filepath.Abs(exePath)
    if err != nil {
        fmt.Printf("获取绝对路径失败: %v\n", err)
        return
    }

    appName := "MyGoApp" // 你希望在启动项中显示的名称

    err = startup.SetStartupProcessLaunch(absPath, appName)
    if err != nil {
        fmt.Printf("设置开机启动失败: %v\n", err)
        return
    }
    fmt.Println("程序已尝试设置为开机启动。")
}