Merge branch 'test'

This commit is contained in:
liyp 2024-06-27 13:19:39 +08:00
commit 97f429fcd7
8 changed files with 268 additions and 47 deletions

2
.gitignore vendored
View file

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

10
config example.toml Normal file
View file

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

7
go.mod
View file

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

View file

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

View file

@ -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
}

44
workers/lsp.go Normal file
View file

@ -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)
}

View file

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

View file

@ -1,12 +1,13 @@
package workers package workers
import ( import (
"encoding/json"
"fmt" "fmt"
"io" "io"
"net/http" "net/http"
"strings" "strings"
"time" "time"
"github.com/goccy/go-json"
) )
type Pkg struct { type Pkg struct {
@ -20,23 +21,25 @@ func (a *Pkg) GetMsg() string {
} }
// 去除换行符 // 去除换行符
raw_msg := strings.TrimRight(a.RawMsg, "\n") raw_msg := strings.TrimRight(a.RawMsg, "\n")
raw_msg = strings.TrimRight(raw_msg, "\r")
fmt.Println("raw_msg:", raw_msg) fmt.Println("raw_msg:", raw_msg)
parms := strings.Split(raw_msg, " ") parms := strings.Split(raw_msg, " ")
url := "https://archlinux.org/packages/search/json/?name=" + parms[1] 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:] url += "&repo=" + strings.ToUpper(parms[2][:1]) + parms[2][1:]
} }
// 输出请求地址 // 输出请求地址
fmt.Println("pkg url:", url) fmt.Println("pkg url:", url)
req, err := http.Get(url) req, err := http.Get(url)
if err != nil { if err != nil {
return err.Error() return "服务器网络错误!"
} }
defer req.Body.Close() defer req.Body.Close()
body, err := io.ReadAll(req.Body) body, err := io.ReadAll(req.Body)
if err != nil { if err != nil {
return err.Error() return "数据解析错误!"
} }
// fmt.Println("pkg body:", string(body)) // fmt.Println("pkg body:", string(body))
// var pkg []Package // var pkg []Package
@ -56,7 +59,7 @@ func (a *Pkg) GetMsg() string {
} }
defer req.Body.Close() defer req.Body.Close()
fmt.Println("aur url:", url) fmt.Println("aur suggest url:", url)
re, err := io.ReadAll(req.Body) re, err := io.ReadAll(req.Body)
// fmt.Println(string(re)) // fmt.Println(string(re))
if err != nil { if err != nil {
@ -67,43 +70,62 @@ func (a *Pkg) GetMsg() string {
if err != nil { if err != nil {
fmt.Println(err) fmt.Println(err)
} }
if len(suggestions) > 0 { // println(len(suggestions))
searchUrl := "https://aur.archlinux.org/rpc/v5/search/" + suggestions[0] + "?by=name" if len(suggestions) == 0 {
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 {
return "没有找到相关软件" 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{}) result := resultSlipe[0].(map[string]interface{})