67 lines
1.3 KiB
Go
67 lines
1.3 KiB
Go
package bootstrap
|
|
|
|
import (
|
|
"github.com/eatmoreapple/openwechat"
|
|
log "github.com/sirupsen/logrus"
|
|
"github.com/wechatgpt/wechatbot/config"
|
|
"github.com/wechatgpt/wechatbot/handler/wechat"
|
|
"os"
|
|
"strings"
|
|
)
|
|
|
|
func StartWebChat() {
|
|
keyword := getKeyword()
|
|
if len(keyword) == 0 {
|
|
log.Info("未配置微信关键字，不启动微信")
|
|
return
|
|
}
|
|
bot := openwechat.DefaultBot(openwechat.Desktop)
|
|
bot.MessageHandler = wechat.Handler
|
|
bot.UUIDCallback = openwechat.PrintlnQrcodeUrl
|
|
|
|
reloadStorage := openwechat.NewJsonFileHotReloadStorage("token.json")
|
|
err := bot.HotLogin(reloadStorage)
|
|
if err != nil {
|
|
if err = bot.Login(); err != nil {
|
|
log.Fatal(err)
|
|
return
|
|
}
|
|
}
|
|
|
|
// 获取登陆的用户
|
|
self, err := bot.GetCurrentUser()
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
return
|
|
}
|
|
|
|
friends, err := self.Friends()
|
|
|
|
for i, friend := range friends {
|
|
log.Println(i, friend)
|
|
}
|
|
groups, err := self.Groups()
|
|
for i, group := range groups {
|
|
log.Println(i, group)
|
|
}
|
|
err = bot.Block()
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
return
|
|
}
|
|
}
|
|
|
|
func getKeyword() string {
|
|
keyword := os.Getenv("wechat")
|
|
if len(strings.Trim(keyword, " ")) == 0 {
|
|
gptConfig := config.GetConfig()
|
|
if gptConfig != nil {
|
|
if gptConfig.ChatGpt.Wechat != nil {
|
|
keyword = *gptConfig.ChatGpt.Wechat
|
|
}
|
|
}
|
|
}
|
|
|
|
return keyword
|
|
}
|