Merge branch 'test'

.gitignore

@@ -1,2 +1,4 @@
 data.db
-test.json
+test.json
+config.toml
+*.exe

config example.toml

@@ -0,0 +1,10 @@
+[Server]
+APIURL = "0.0.0.0:5580"
+POSTURL = "http://0.0.0.0:5700"
+
+[Group]
+AllowGroup = []
+AllowUser = []
+AllowRole = []
+BlockGroup = []
+BlockUser = []

go.mod

@@ -1,13 +1,10 @@
 module go-bot
 
-go 1.21.0
+go 1.22.2
 
 require github.com/BurntSushi/toml v1.3.2
 
-require golang.org/x/net v0.17.0 // indirect
-
 require (
-	github.com/goccy/go-json v0.10.2 // indirect
-	github.com/gorilla/websocket v1.5.1
+	github.com/goccy/go-json v0.10.2
 	github.com/mattn/go-sqlite3 v1.14.22
 )

main.go

@@ -2,7 +2,6 @@ package main
 
 import (
 	"database/sql"
-	"encoding/json"
 	"fmt"
 	"go-bot/config"
 	"go-bot/utils"
@@ -11,6 +10,8 @@ import (
 	"net/http"
 	"time"
 
+	"github.com/goccy/go-json"
+
 	_ "github.com/mattn/go-sqlite3"
 )
 

tools/get_history_message.go

@@ -0,0 +1,141 @@
+package main
+
+import (
+	"database/sql"
+	"fmt"
+	"io"
+	"log"
+	"net/http"
+	"time"
+
+	"github.com/goccy/go-json"
+	_ "github.com/mattn/go-sqlite3"
+)
+
+func main() {
+	db, err := sql.Open("sqlite3", "./data.db")
+	if err != nil {
+		panic(err)
+	}
+	defer db.Close()
+
+	rows, err := db.Query("SELECT DISTINCT group_id FROM messages")
+	if err != nil {
+		panic(err)
+	}
+	defer rows.Close()
+	var groupIDs []int
+	for rows.Next() {
+		var groupID int
+		err := rows.Scan(&groupID)
+		if err != nil {
+			panic(err)
+		}
+		groupIDs = append(groupIDs, groupID)
+	}
+	for _, groupID := range groupIDs {
+		minSeq := get_min_seqs(db, groupID)
+		println("minSeq:", minSeq)
+		// if minSeq == 0 {
+		// 	continue
+		// }
+		for minSeq != 0 {
+			fetchHistoryMessages(db, minSeq, groupID)
+			temp := get_min_seqs(db, groupID)
+			if temp == minSeq {
+				println("已获取到最久数据！")
+				time.Sleep(time.Second * 2)
+				break
+			}
+			minSeq = temp
+			// println("minSeq_changed:", minSeq)
+
+		}
+	}
+
+}
+func get_min_seqs(db *sql.DB, groupID int) int {
+	var minMessageSeq int
+	err := db.QueryRow("SELECT MIN(message_seq) FROM messages WHERE group_id = ?", groupID).Scan(&minMessageSeq)
+	if err != nil {
+		panic(err)
+	}
+	// println("minMessageSeq:", minMessageSeq)
+	var count int
+	err = db.QueryRow("SELECT COUNT(*) FROM messages WHERE group_id = ? AND message_seq = ?", groupID, minMessageSeq-1).Scan(&count)
+	if err != nil {
+		panic(err)
+	}
+	// println("count:", count)
+	if count > 0 {
+		return 0
+	}
+	fmt.Printf("Group ID: %d, Min Message Seq: %d\n", groupID, minMessageSeq)
+	return minMessageSeq
+}
+func fetchHistoryMessages(db *sql.DB, message_seq int, group_id int) (int, error) {
+	url := fmt.Sprintf("http://127.0.0.1:5700/get_group_msg_history?message_seq=%d&group_id=%d", message_seq, group_id)
+	println("min message_seq :", message_seq, "min group_id: ", group_id)
+	println(url)
+	resp, err := http.Get(url)
+	if err != nil {
+		panic(err)
+	}
+	defer resp.Body.Close()
+	body, err := io.ReadAll(resp.Body)
+	if err != nil {
+		return 0, err
+	}
+	// println("Body:", string(body))
+	var messageResp map[string]interface{}
+	err = json.Unmarshal(body, &messageResp)
+	if err != nil {
+		return 0, err
+	}
+	data := messageResp["data"].(map[string]interface{})
+	messages := data["messages"].([]interface{})
+	for _, message := range messages {
+		messageMap := message.(map[string]interface{})
+		insertMessage(db, messageMap)
+
+	}
+	time.Sleep(time.Second * 1)
+	return 0, nil
+}
+
+func insertMessage(db *sql.DB, data map[string]interface{}) error {
+
+	if data["post_type"] == "message" {
+		// id := data["message_id"].(float64)
+		post_type := data["post_type"].(string)
+		message_type := data["message_type"].(string)
+		// 设置东八区时间
+		loc, err := time.LoadLocation("Asia/Shanghai")
+		if err != nil {
+			log.Fatal(err)
+		}
+		message_time := (time.Unix(int64(data["time"].(float64)), 0).In(loc)).Format("2006-01-02 15:04:05")
+		group_id := data["group_id"].(float64)
+		message_id := data["message_id"].(float64)
+		raw_message := data["raw_message"].(string)
+		sender := data["sender"].(map[string]interface{})
+		sender_user_id := sender["user_id"].(float64)
+		sender_nickname := sender["nickname"].(string)
+		sender_card := sender["card"].(string)
+		if sender_card == "" {
+			sender_card = sender_nickname
+		}
+		sender_role := sender["role"].(string)
+		message_seq := data["message_seq"].(float64)
+
+		fmt.Println(post_type, message_type, message_time, group_id, message_id, raw_message, sender_user_id, sender_nickname, sender_card, sender_role, message_seq)
+		// Insert data into database
+		_, err = db.Exec("INSERT INTO messages ( post_type, message_type, time, group_id, message_id, raw_message, sender_user_id, sender_nickname, sender_card, sender_role, message_seq) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)",
+			post_type, message_type, message_time, group_id, message_id, raw_message, sender_user_id, sender_nickname, sender_card, sender_role, message_seq)
+		fmt.Println("Data inserted successfully!")
+		if err != nil {
+			log.Fatal(err)
+		}
+	}
+	return nil
+}

workers/lsp.go

@@ -0,0 +1,44 @@
+package workers
+
+import (
+	"fmt"
+
+	"github.com/goccy/go-json"
+
+	"io"
+	"net/http"
+)
+
+type Lsp struct {
+	*StdAns
+}
+
+func (a *Lsp) GetMsg() string {
+	a.AllowGroup = []string{"313047773"}
+	url := "https://api.lolicon.app/setu/v2?r18=0&size=small"
+	resp, err := http.Get(url)
+	if err != nil {
+		return "获取失败"
+	}
+	defer resp.Body.Close()
+	budy, err := io.ReadAll(resp.Body)
+	var res map[string]interface{}
+	err = json.Unmarshal(budy, &res)
+	if err != nil {
+		return "解析失败"
+	}
+	code := res["error"].(string)
+	if code != "" {
+		return "获取失败"
+	}
+	data := res["data"].([]interface{})
+	uid := data[0].(map[string]interface{})["uid"].(float64)
+	urls := data[0].(map[string]interface{})["urls"].(map[string]interface{})
+	imgUrl := urls["small"].(string)
+	// title := data[0].(map[string]interface{})["title"].(string)
+	// println("标题：" + title + "\n" + imgUrl)
+	msg := fmt.Sprintf("[CQ:reply,id=%s]Pixiv ID:%d", a.MID, int64(uid))
+	a.SendMsg(msg)
+	return fmt.Sprintf("[CQ:image,file=%s]", imgUrl)
+
+}

workers/newworker.go

@@ -22,6 +22,10 @@ func NewWorker(parms []string, uid, gid, role, mid, rawMsg string) Worker {
 		return &Ip{
 			StdAns: NewStdAns(parms, uid, gid, role, mid, rawMsg),
 		}
+	case "lsp":
+		return &Lsp{
+			StdAns: NewStdAns(parms, uid, gid, role, mid, rawMsg),
+		}
 	default:
 		return &Emm{
 			StdAns: NewStdAns(parms, uid, gid, role, mid, rawMsg)}

workers/pkg.go

@@ -1,12 +1,13 @@
 package workers
 
 import (
-	"encoding/json"
 	"fmt"
 	"io"
 	"net/http"
 	"strings"
 	"time"
+
+	"github.com/goccy/go-json"
 )
 
 type Pkg struct {
@@ -20,23 +21,25 @@ func (a *Pkg) GetMsg() string {
 	}
 	// 去除换行符
 	raw_msg := strings.TrimRight(a.RawMsg, "\n")
+	raw_msg = strings.TrimRight(raw_msg, "\r")
+
 	fmt.Println("raw_msg:", raw_msg)
 	parms := strings.Split(raw_msg, " ")
 
 	url := "https://archlinux.org/packages/search/json/?name=" + parms[1]
-	if len(parms) > 2 {
+	if len(parms) > 2 && parms[1] != "" {
 		url += "&repo=" + strings.ToUpper(parms[2][:1]) + parms[2][1:]
 	}
 	// 输出请求地址
 	fmt.Println("pkg url:", url)
 	req, err := http.Get(url)
 	if err != nil {
-		return err.Error()
+		return "服务器网络错误！"
 	}
 	defer req.Body.Close()
 	body, err := io.ReadAll(req.Body)
 	if err != nil {
-		return err.Error()
+		return "数据解析错误！"
 	}
 	// fmt.Println("pkg body:", string(body))
 	// var pkg []Package
@@ -56,7 +59,7 @@ func (a *Pkg) GetMsg() string {
 		}
 		defer req.Body.Close()
 
-		fmt.Println("aur url:", url)
+		fmt.Println("aur suggest url:", url)
 		re, err := io.ReadAll(req.Body)
 		// fmt.Println(string(re))
 		if err != nil {
@@ -67,43 +70,62 @@ func (a *Pkg) GetMsg() string {
 		if err != nil {
 			fmt.Println(err)
 		}
-		if len(suggestions) > 0 {
-			searchUrl := "https://aur.archlinux.org/rpc/v5/search/" + suggestions[0] + "?by=name"
-			searchReq, err := http.Get(searchUrl)
-			if err != nil {
-				fmt.Println(err)
-			}
-			defer searchReq.Body.Close()
-			searchRe, err := io.ReadAll(searchReq.Body)
-			if err != nil {
-				fmt.Println(err)
-			}
-			var searchMap map[string]interface{}
-			err = json.Unmarshal(searchRe, &searchMap)
-			if err != nil {
-				fmt.Println(err)
-			}
-			searchResults := searchMap["results"].([]interface{})
-			searchResult := searchResults[0].(map[string]interface{})
-			maintainer, ok := searchResult["Maintainer"].(string)
-			if !ok {
-				maintainer = "孤儿包"
-			}
-			last_update := time.Unix(int64(searchResult["LastModified"].(float64)), 0).Format("2006-01-02 15:04:05")
-			var msg string
-			msg += "仓库：AUR\n"
-			msg += "包名：" + searchResult["Name"].(string) + "\n"
-			msg += "版本：" + searchResult["Version"].(string) + "\n"
-			msg += "描述：" + searchResult["Description"].(string) + "\n"
-			msg += "维护者：" + maintainer + "\n"
-			msg += "链接：" + searchResult["URL"].(string) + "\n"
-			msg += "更新时间：" + last_update + "\n"
-
-			fmt.Println(msg)
-			return msg
-		} else {
+		// println(len(suggestions))
+		if len(suggestions) == 0 {
 			return "没有找到相关软件"
 		}
+		searchUrl := "https://aur.archlinux.org/rpc/v5/search/" + suggestions[0] + "?by=name"
+		searchReq, err := http.Get(searchUrl)
+		if err != nil {
+			fmt.Println(err)
+		}
+		defer searchReq.Body.Close()
+		searchRe, err := io.ReadAll(searchReq.Body)
+		if err != nil {
+			fmt.Println(err)
+		}
+		var searchMap map[string]interface{}
+		err = json.Unmarshal(searchRe, &searchMap)
+		if err != nil {
+			fmt.Println(err)
+		}
+		searchResults := searchMap["results"].([]interface{})
+		// println("searchResults:", len(searchResults))
+		maxVotes := 0.0
+		var searchResult map[string]interface{}
+		// 增加根据投票人数排序
+		if len(searchResults) > 1 {
+			for _, result := range searchResults {
+				resultMap := result.(map[string]interface{})
+				numVotes := resultMap["NumVotes"].(float64)
+				if numVotes > maxVotes {
+					maxVotes = numVotes
+					searchResult = resultMap
+				}
+			}
+			fmt.Println(len(searchResult))
+		} else if len(searchResults) == 1 {
+			searchResult = searchResults[0].(map[string]interface{})
+		}
+		maintainer, ok := searchResult["Maintainer"].(string)
+		if !ok {
+			maintainer = "孤儿包"
+		}
+		println("maintainer:", maintainer)
+		// last_update := ''
+		last_update := time.Unix(int64(searchResult["LastModified"].(float64)), 0).Format("2006-01-02 15:04:05")
+
+		var msg string
+		msg += "仓库：AUR\n"
+		msg += "包名：" + searchResult["Name"].(string) + "\n"
+		msg += "版本：" + searchResult["Version"].(string) + "\n"
+		msg += "描述：" + searchResult["Description"].(string) + "\n"
+		msg += "维护者：" + maintainer + "\n"
+		msg += "链接：" + searchResult["URL"].(string) + "\n"
+		msg += "更新时间：" + last_update
+
+		fmt.Println(msg)
+		return msg
 
 	}
 	result := resultSlipe[0].(map[string]interface{})