mirror of https://github.com/miguelmota/cointop
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
125 lines
2.5 KiB
Go
125 lines
2.5 KiB
Go
// +build windows,!linux,!freebsd,!netbsd,!openbsd,!darwin,!js
|
|
|
|
package beeep
|
|
|
|
import (
|
|
"bufio"
|
|
"bytes"
|
|
"errors"
|
|
"os/exec"
|
|
"strings"
|
|
"syscall"
|
|
"time"
|
|
|
|
toast "github.com/go-toast/toast"
|
|
"github.com/tadvi/systray"
|
|
"golang.org/x/sys/windows/registry"
|
|
)
|
|
|
|
var isWindows10 bool
|
|
var applicationID string
|
|
|
|
func init() {
|
|
k, err := registry.OpenKey(registry.LOCAL_MACHINE, `SOFTWARE\Microsoft\Windows NT\CurrentVersion`, registry.QUERY_VALUE)
|
|
if err != nil {
|
|
return
|
|
}
|
|
defer k.Close()
|
|
|
|
maj, _, err := k.GetIntegerValue("CurrentMajorVersionNumber")
|
|
if err != nil {
|
|
return
|
|
}
|
|
|
|
isWindows10 = maj == 10
|
|
|
|
if isWindows10 {
|
|
applicationID = appID()
|
|
}
|
|
}
|
|
|
|
// Notify sends desktop notification.
|
|
func Notify(title, message, appIcon string) error {
|
|
if isWindows10 {
|
|
return toastNotify(title, message, appIcon)
|
|
}
|
|
|
|
err := baloonNotify(title, message, appIcon, false)
|
|
if err != nil {
|
|
e := msgNotify(title, message)
|
|
if e != nil {
|
|
return errors.New("beeep: " + err.Error() + "; " + e.Error())
|
|
}
|
|
}
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
func msgNotify(title, message string) error {
|
|
msg, err := exec.LookPath("msg")
|
|
if err != nil {
|
|
return err
|
|
}
|
|
cmd := exec.Command(msg, "*", "/TIME:3", title+"\n\n"+message)
|
|
cmd.SysProcAttr = &syscall.SysProcAttr{HideWindow: true}
|
|
return cmd.Run()
|
|
}
|
|
|
|
func baloonNotify(title, message, appIcon string, bigIcon bool) error {
|
|
tray, err := systray.New()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
err = tray.ShowCustom(pathAbs(appIcon), title)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
go func() {
|
|
tray.Run()
|
|
time.Sleep(3 * time.Second)
|
|
tray.Stop()
|
|
}()
|
|
|
|
return tray.ShowMessage(title, message, bigIcon)
|
|
}
|
|
|
|
func toastNotify(title, message, appIcon string) error {
|
|
notification := toastNotification(title, message, pathAbs(appIcon))
|
|
return notification.Push()
|
|
}
|
|
|
|
func toastNotification(title, message, appIcon string) toast.Notification {
|
|
return toast.Notification{
|
|
AppID: applicationID,
|
|
Title: title,
|
|
Message: message,
|
|
Icon: appIcon,
|
|
}
|
|
}
|
|
|
|
func appID() string {
|
|
defID := "{1AC14E77-02E7-4E5D-B744-2EB1AE5198B7}\\WindowsPowerShell\\v1.0\\powershell.exe"
|
|
cmd := exec.Command("powershell", "Get-StartApps")
|
|
cmd.SysProcAttr = &syscall.SysProcAttr{HideWindow: true}
|
|
out, err := cmd.Output()
|
|
if err != nil {
|
|
return defID
|
|
}
|
|
|
|
scanner := bufio.NewScanner(bytes.NewReader(out))
|
|
for scanner.Scan() {
|
|
line := strings.TrimSpace(scanner.Text())
|
|
if strings.Contains(line, "powershell.exe") {
|
|
sp := strings.Split(line, " ")
|
|
if len(sp) > 0 {
|
|
return sp[len(sp)-1]
|
|
}
|
|
}
|
|
}
|
|
|
|
return defID
|
|
}
|