mirror of
https://github.com/Mrs4s/go-cqhttp.git
synced 2025-05-05 03:23:49 +08:00
68 lines
2.0 KiB
Go
68 lines
2.0 KiB
Go
//go:build windows
|
||
// +build windows
|
||
|
||
package terminal
|
||
|
||
import (
|
||
"os"
|
||
"syscall"
|
||
"unsafe"
|
||
|
||
"github.com/pkg/errors"
|
||
)
|
||
|
||
// RunningByDoubleClick 检查是否通过双击直接运行
|
||
func RunningByDoubleClick() bool {
|
||
kernel32 := syscall.NewLazyDLL("kernel32.dll")
|
||
lp := kernel32.NewProc("GetConsoleProcessList")
|
||
if lp != nil {
|
||
var ids [2]uint32
|
||
var maxCount uint32 = 2
|
||
ret, _, _ := lp.Call(uintptr(unsafe.Pointer(&ids)), uintptr(maxCount))
|
||
if ret > 1 {
|
||
return false
|
||
}
|
||
}
|
||
return true
|
||
}
|
||
|
||
// NoMoreDoubleClick 提示用户不要双击运行，并生成安全启动脚本
|
||
func NoMoreDoubleClick() error {
|
||
r := boxW(0, "请勿通过双击直接运行本程序, 这将导致一些非预料的后果.\n请在shell中运行./go-cqhttp.exe\n点击确认将释出安全启动脚本，点击取消则关闭程序", "警告", 0x00000030|0x00000001)
|
||
if r == 2 {
|
||
return nil
|
||
}
|
||
r = boxW(0, "点击确认将覆盖go-cqhttp.bat，点击取消则关闭程序", "警告", 0x00000030|0x00000001)
|
||
if r == 2 {
|
||
return nil
|
||
}
|
||
f, err := os.OpenFile("go-cqhttp.bat", os.O_CREATE|os.O_RDWR, 0o666)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
if err != nil {
|
||
return errors.Errorf("打开go-cqhttp.bat失败: %v", err)
|
||
}
|
||
_ = f.Truncate(0)
|
||
_, err = f.WriteString("%Created by go-cqhttp. DO NOT EDIT ME!%\nstart cmd /K go-cqhttp.exe")
|
||
if err != nil {
|
||
return errors.Errorf("写入go-cqhttp.bat失败: %v", err)
|
||
}
|
||
f.Close()
|
||
boxW(0, "安全启动脚本已生成，请双击go-cqhttp.bat启动", "提示", 0x00000040|0x00000000)
|
||
return nil
|
||
}
|
||
|
||
// BoxW of Win32 API. Check https://docs.microsoft.com/en-us/windows/win32/api/winuser/nf-winuser-messageboxw for more detail.
|
||
func boxW(hwnd uintptr, caption, title string, flags uint) int {
|
||
captionPtr, _ := syscall.UTF16PtrFromString(caption)
|
||
titlePtr, _ := syscall.UTF16PtrFromString(title)
|
||
ret, _, _ := syscall.NewLazyDLL("user32.dll").NewProc("MessageBoxW").Call(
|
||
hwnd,
|
||
uintptr(unsafe.Pointer(captionPtr)),
|
||
uintptr(unsafe.Pointer(titlePtr)),
|
||
uintptr(flags))
|
||
|
||
return int(ret)
|
||
}
|