autostart.go

package main

import (
	"fmt"
	"os"
	"strings"

	"golang.org/x/sys/windows/registry"
)

// ---------------- 开机自启（HKCU Run 注册表项） ----------------

// 开机自启注册表位置：HKCU\Software\Microsoft\Windows\CurrentVersion\Run
// 仅当前用户生效，无需管理员权限；用户也可在任务管理器"启动"页中自行启停
const (
	autoStartRegPath  = `Software\Microsoft\Windows\CurrentVersion\Run`
	autoStartValueKey = "FileManager" // 与 settings.json 目录名、单实例互斥名保持一致
)

// applyAutoStart 写入/删除开机自启注册表项
func applyAutoStart(enabled bool) error {
	k, err := registry.OpenKey(registry.CURRENT_USER, autoStartRegPath, registry.QUERY_VALUE|registry.SET_VALUE)
	if err != nil {
		return fmt.Errorf("打开开机自启注册表键失败：%v", err)
	}
	defer k.Close()
	if enabled {
		exe, err := os.Executable()
		if err != nil {
			return fmt.Errorf("获取程序路径失败：%v", err)
		}
		// 带引号包裹路径，防止路径含空格时启动失败
		if err := k.SetStringValue(autoStartValueKey, `"`+exe+`"`); err != nil {
			return fmt.Errorf("写入开机自启注册表失败：%v", err)
		}
	} else {
		if err := k.DeleteValue(autoStartValueKey); err != nil && err != registry.ErrNotExist {
			return fmt.Errorf("移除开机自启注册表失败：%v", err)
		}
	}
	return nil
}

// isAutoStartEnabled 检查注册表中是否已注册本程序开机自启
func isAutoStartEnabled() bool {
	k, err := registry.OpenKey(registry.CURRENT_USER, autoStartRegPath, registry.QUERY_VALUE)
	if err != nil {
		return false
	}
	defer k.Close()
	val, _, err := k.GetStringValue(autoStartValueKey)
	if err != nil {
		return false
	}
	exe, err := os.Executable()
	if err != nil {
		return false
	}
	return strings.Contains(strings.ToLower(val), strings.ToLower(exe))
}