1-6
107 lines
2.2 KiB
Go
107 lines
2.2 KiB
Go
package main
|
|
|
|
import (
|
|
"bufio"
|
|
"fmt"
|
|
"os"
|
|
"os/exec"
|
|
"runtime"
|
|
"time"
|
|
|
|
// _ "github.com/cdle/sillyplus/adapters/qq"
|
|
"github.com/cdle/sillyplus/adapters/web"
|
|
"github.com/cdle/sillyplus/core"
|
|
"github.com/cdle/sillyplus/core/common"
|
|
|
|
"github.com/cdle/sillyplus/utils"
|
|
)
|
|
|
|
var app = core.MakeBucket("app")
|
|
var ip = ""
|
|
|
|
func main() {
|
|
ip = app.GetString("ip")
|
|
go func() {
|
|
var err error
|
|
ip, err = utils.GetPublicIP()
|
|
if err == nil {
|
|
app.Set("ip", ip)
|
|
}
|
|
}()
|
|
core.Init()
|
|
if app.GetBool("anti_kasi") {
|
|
go utils.MonitorGoroutine()
|
|
}
|
|
d := false
|
|
for _, arg := range os.Args {
|
|
if arg == "-d" {
|
|
d = true
|
|
}
|
|
}
|
|
go func() { //弹出浏览器
|
|
if runtime.GOOS != "windows" {
|
|
return
|
|
}
|
|
time.Sleep(time.Second * 3)
|
|
if web.GetUserNumber() == 0 {
|
|
app := core.MakeBucket("app")
|
|
port := app.GetInt("port", 8080)
|
|
url := fmt.Sprintf("http://localhost:%d/admin", port)
|
|
cmd := exec.Command("cmd", "/c", "start", url)
|
|
stdout, err := cmd.Output()
|
|
if err != nil {
|
|
fmt.Println(err)
|
|
return
|
|
}
|
|
fmt.Println(string(stdout))
|
|
}
|
|
}()
|
|
if !d {
|
|
t := false
|
|
for _, arg := range os.Args {
|
|
if arg == "-t" {
|
|
t = true
|
|
}
|
|
}
|
|
if t {
|
|
// core.Logs.Info("Terminal机器人已连接")
|
|
scanner := bufio.NewScanner(os.Stdin)
|
|
a := &core.Factory{}
|
|
a.Init("terminal", "default", nil)
|
|
i := 0
|
|
a.SetReplyHandler(func(m map[string]interface{}) string {
|
|
i++
|
|
fmt.Printf("\x1b[%dm%s \x1b[0m\n", 31, m[core.CONETNT])
|
|
return fmt.Sprint(i)
|
|
})
|
|
a.SetActionHandler(func(m map[string]interface{}) string {
|
|
fmt.Println(`do action: ` + string(utils.JsonMarshal(m)))
|
|
return ""
|
|
})
|
|
|
|
for scanner.Scan() {
|
|
data := scanner.Text()
|
|
s := &core.CustomSender{
|
|
// Type: "terminal",
|
|
// Message: string(data),
|
|
// Admin: true,
|
|
F: a,
|
|
}
|
|
i++
|
|
s.SetFsps(&common.FakerSenderParams{
|
|
Content: data,
|
|
MessageID: fmt.Sprint(i),
|
|
})
|
|
s.SetContent(data)
|
|
core.Messages <- s
|
|
}
|
|
core.Logs.Info("Terminal机器人异常,请检查运行环境设置,如果是docker环境,请附加-it参数")
|
|
} else {
|
|
// core.Logs.Info("Terminal机器人不可用，运行带-t参数即可启用")
|
|
}
|
|
}
|
|
select {}
|
|
}
|
|
|
|
// git add . && git commit -m "x" && git push
|