x

internal/wechat/manager.go

@@ -0,0 +1,57 @@
+package wechat
+
+import (
+	"strings"
+
+	"github.com/shirou/gopsutil/v4/process"
+	log "github.com/sirupsen/logrus"
+)
+
+const (
+	V3ProcessName = "WeChat"
+	V4ProcessName = "Weixin"
+)
+
+var (
+	Items   []*Info
+	ItemMap map[string]*Info
+)
+
+func Load() {
+	Items = make([]*Info, 0, 2)
+	ItemMap = make(map[string]*Info)
+
+	processes, err := process.Processes()
+	if err != nil {
+		log.Println("获取进程列表失败:", err)
+		return
+	}
+
+	for _, p := range processes {
+		name, err := p.Name()
+		name = strings.TrimSuffix(name, ".exe")
+		if err != nil || name != V3ProcessName && name != V4ProcessName {
+			continue
+		}
+
+		// v4 存在同名进程，需要继续判断 cmdline
+		if name == V4ProcessName {
+			cmdline, err := p.Cmdline()
+			if err != nil {
+				log.Error(err)
+				continue
+			}
+			if strings.Contains(cmdline, "--") {
+				continue
+			}
+		}
+
+		info, err := NewInfo(p)
+		if err != nil {
+			continue
+		}
+
+		Items = append(Items, info)
+		ItemMap[info.AccountName] = info
+	}
+}