2024-04-01 09:42:29 +08:00
|
|
|
|
package workers
|
|
|
|
|
|
|
|
|
|
import (
|
2024-04-01 11:05:57 +08:00
|
|
|
|
"fmt"
|
2024-04-01 09:42:29 +08:00
|
|
|
|
"io"
|
|
|
|
|
"net/http"
|
2024-04-01 12:33:13 +08:00
|
|
|
|
"strings"
|
|
|
|
|
"time"
|
2024-04-21 17:13:20 +08:00
|
|
|
|
|
|
|
|
|
"github.com/goccy/go-json"
|
2024-04-01 09:42:29 +08:00
|
|
|
|
)
|
|
|
|
|
|
2024-07-05 22:33:55 +08:00
|
|
|
|
func init() {
|
|
|
|
|
RegisterWorkerFactory("pkg", func(parms []string, uid, gid, role, mid, rawMsg string) Worker {
|
|
|
|
|
return &Pkg{
|
|
|
|
|
StdAns: NewStdAns(parms, uid, gid, role, mid, rawMsg),
|
|
|
|
|
}
|
|
|
|
|
})
|
|
|
|
|
}
|
|
|
|
|
|
2024-04-01 09:42:29 +08:00
|
|
|
|
type Pkg struct {
|
|
|
|
|
*StdAns
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
func (a *Pkg) GetMsg() string {
|
2024-04-02 20:29:25 +08:00
|
|
|
|
|
2024-04-01 09:42:29 +08:00
|
|
|
|
if len(a.Parms) < 2 {
|
|
|
|
|
return "请输入包名 如:/pkg linux 查询 linux 相关软件"
|
|
|
|
|
}
|
2024-04-14 15:18:23 +08:00
|
|
|
|
// 去除换行符
|
2024-04-14 15:24:35 +08:00
|
|
|
|
raw_msg := strings.TrimRight(a.RawMsg, "\n")
|
2024-05-11 19:38:19 +08:00
|
|
|
|
raw_msg = strings.TrimRight(raw_msg, "\r")
|
|
|
|
|
|
2024-04-14 15:18:23 +08:00
|
|
|
|
fmt.Println("raw_msg:", raw_msg)
|
|
|
|
|
parms := strings.Split(raw_msg, " ")
|
|
|
|
|
|
|
|
|
|
url := "https://archlinux.org/packages/search/json/?name=" + parms[1]
|
2024-04-23 20:38:37 +08:00
|
|
|
|
if len(parms) > 2 && parms[1] != "" {
|
2024-04-14 15:18:23 +08:00
|
|
|
|
url += "&repo=" + strings.ToUpper(parms[2][:1]) + parms[2][1:]
|
2024-04-01 09:42:29 +08:00
|
|
|
|
}
|
2024-04-02 20:29:25 +08:00
|
|
|
|
// 输出请求地址
|
2024-04-14 15:18:23 +08:00
|
|
|
|
fmt.Println("pkg url:", url)
|
2024-04-01 09:42:29 +08:00
|
|
|
|
req, err := http.Get(url)
|
|
|
|
|
if err != nil {
|
2024-05-11 19:38:19 +08:00
|
|
|
|
return "服务器网络错误!"
|
2024-04-01 09:42:29 +08:00
|
|
|
|
}
|
|
|
|
|
defer req.Body.Close()
|
|
|
|
|
body, err := io.ReadAll(req.Body)
|
|
|
|
|
if err != nil {
|
2024-05-11 19:38:19 +08:00
|
|
|
|
return "数据解析错误!"
|
2024-04-01 09:42:29 +08:00
|
|
|
|
}
|
2024-04-02 20:29:25 +08:00
|
|
|
|
// fmt.Println("pkg body:", string(body))
|
2024-04-01 11:05:57 +08:00
|
|
|
|
// var pkg []Package
|
|
|
|
|
var pkg map[string]interface{}
|
2024-04-01 09:42:29 +08:00
|
|
|
|
err = json.Unmarshal(body, &pkg)
|
|
|
|
|
if err != nil {
|
|
|
|
|
return err.Error()
|
|
|
|
|
}
|
2024-04-01 11:05:57 +08:00
|
|
|
|
// fmt.Println("pkg:", pkg)
|
2024-04-01 11:37:03 +08:00
|
|
|
|
resultSlipe := pkg["results"].([]interface{})
|
|
|
|
|
if len(resultSlipe) == 0 {
|
2024-04-01 12:33:13 +08:00
|
|
|
|
|
2024-04-14 15:18:23 +08:00
|
|
|
|
url := "https://aur.archlinux.org/rpc/v5/suggest/" + parms[1]
|
2024-04-01 12:33:13 +08:00
|
|
|
|
req, err := http.Get(url)
|
|
|
|
|
if err != nil {
|
|
|
|
|
fmt.Println(err)
|
|
|
|
|
}
|
|
|
|
|
defer req.Body.Close()
|
|
|
|
|
|
2024-06-27 13:16:14 +08:00
|
|
|
|
fmt.Println("aur suggest url:", url)
|
2024-04-01 12:33:13 +08:00
|
|
|
|
re, err := io.ReadAll(req.Body)
|
2024-04-03 21:53:57 +08:00
|
|
|
|
// fmt.Println(string(re))
|
2024-04-01 12:33:13 +08:00
|
|
|
|
if err != nil {
|
|
|
|
|
fmt.Println(err)
|
|
|
|
|
}
|
|
|
|
|
var suggestions []string
|
|
|
|
|
err = json.Unmarshal(re, &suggestions)
|
|
|
|
|
if err != nil {
|
|
|
|
|
fmt.Println(err)
|
|
|
|
|
}
|
2024-06-27 13:16:14 +08:00
|
|
|
|
// 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 {
|
2024-05-11 19:38:19 +08:00
|
|
|
|
for _, result := range searchResults {
|
|
|
|
|
resultMap := result.(map[string]interface{})
|
|
|
|
|
numVotes := resultMap["NumVotes"].(float64)
|
|
|
|
|
if numVotes > maxVotes {
|
|
|
|
|
maxVotes = numVotes
|
|
|
|
|
searchResult = resultMap
|
|
|
|
|
}
|
|
|
|
|
}
|
2024-06-27 13:16:14 +08:00
|
|
|
|
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")
|
2024-05-11 19:38:19 +08:00
|
|
|
|
|
2024-06-27 13:16:14 +08:00
|
|
|
|
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
|
2024-04-01 12:33:13 +08:00
|
|
|
|
|
2024-06-27 13:16:14 +08:00
|
|
|
|
fmt.Println(msg)
|
|
|
|
|
return msg
|
2024-04-01 12:33:13 +08:00
|
|
|
|
|
2024-04-01 11:37:03 +08:00
|
|
|
|
}
|
|
|
|
|
result := resultSlipe[0].(map[string]interface{})
|
2024-04-02 20:29:25 +08:00
|
|
|
|
// 输出结果字符串
|
|
|
|
|
// fmt.Println("result:", result)
|
|
|
|
|
// if result != nil {
|
|
|
|
|
var msg string
|
2024-04-18 21:13:15 +08:00
|
|
|
|
last_update := result["last_update"].(string)
|
|
|
|
|
last_update = strings.Replace(last_update, "T", " ", 1)
|
|
|
|
|
last_update = strings.Replace(last_update, "Z", "", 1)
|
2024-04-02 20:29:25 +08:00
|
|
|
|
msg += "仓库:" + result["repo"].(string) + "\n"
|
|
|
|
|
msg += "包名:" + result["pkgname"].(string) + "\n"
|
|
|
|
|
msg += "版本:" + result["pkgver"].(string) + "\n"
|
|
|
|
|
msg += "描述:" + result["pkgdesc"].(string) + "\n"
|
|
|
|
|
msg += "打包:" + result["packager"].(string) + "\n"
|
|
|
|
|
msg += "链接:" + result["url"].(string) + "\n"
|
2024-04-18 21:13:15 +08:00
|
|
|
|
msg += "更新日期:" + last_update
|
2024-04-02 20:29:25 +08:00
|
|
|
|
return msg
|
2024-04-01 09:42:29 +08:00
|
|
|
|
|
|
|
|
|
}
|