go-bot/workers/pkg.go

128 lines
3.3 KiB
Go
Raw Normal View History

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
)
type Pkg struct {
*StdAns
}
func (a *Pkg) GetMsg() string {
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-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]
if len(parms) > 2 {
url += "&repo=" + strings.ToUpper(parms[2][:1]) + parms[2][1:]
2024-04-01 09:42:29 +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 {
return err.Error()
}
defer req.Body.Close()
body, err := io.ReadAll(req.Body)
if err != nil {
return err.Error()
}
// 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-04-14 15:18:23 +08:00
fmt.Println("aur url:", url)
2024-04-01 12:33:13 +08:00
re, err := io.ReadAll(req.Body)
// 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)
}
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"
2024-04-01 12:33:13 +08:00
fmt.Println(msg)
return msg
} else {
return "没有找到相关软件"
}
2024-04-01 12:33:13 +08:00
2024-04-01 11:37:03 +08:00
}
result := resultSlipe[0].(map[string]interface{})
// 输出结果字符串
// 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)
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
return msg
2024-04-01 09:42:29 +08:00
}