Windows防锁屏工具技术实现

package main

import (
    _ "embed"
    "fmt"
    "github.com/getlantern/systray"
    "time"
)

//go:embed icon.ico  // 必须确保存在图标文件
var iconData []byte

func main() {
    PreventSleep()     // 启动防锁屏
    systray.Run(onReady, onExit)
}

// 托盘初始化
func onReady() {
    systray.SetIcon(iconData)          // 设置托盘图标
    systray.SetTitle("Anti-Lock")      // 任务栏悬停显示
    systray.SetTooltip("运行中")        // 状态提示
    
    // 创建退出菜单项
    quitMenu := systray.AddMenuItem("退出", "关闭防锁屏功能")

    // 异步监听菜单事件
    go func() {
        <-quitMenu.ClickedCh
        systray.Quit()
    }()
}

func onExit() {
    fmt.Println("程序已安全退出") 
}

package main

import (
    "syscall"
)

const (
    ES_CONTINUOUS       = 0x80000000  // 持续生效标志
    ES_SYSTEM_REQUIRED  = 0x00000001  // 阻止系统休眠
    ES_DISPLAY_REQUIRED = 0x00000002  // 阻止显示器关闭
)

// PreventSleep 调用Windows API实现防锁屏
func PreventSleep() {
    kernel32 := syscall.NewLazyDLL("kernel32.dll")
    execStateAPI := kernel32.NewProc("SetThreadExecutionState")
    
    // 组合控制参数
    flags := uintptr(ES_CONTINUOUS | ES_SYSTEM_REQUIRED | ES_DISPLAY_REQUIRED)
    ret, _, _ := execStateAPI.Call(flags)
    
    if ret == 0 {
        panic("API调用失败") 
    }
}