feat: 增加域名以及证书信息的查询能力 (#170)

This commit is contained in:
二丫讲梵
2023-04-04 14:30:13 +08:00
committed by GitHub
parent 8c04c74640
commit 29fc9243d6
7 changed files with 299 additions and 102 deletions

View File

@@ -3,7 +3,6 @@ package process
import (
"fmt"
"strings"
"time"
"github.com/eryajf/chatgpt-dingtalk/pkg/db"
"github.com/eryajf/chatgpt-dingtalk/pkg/dingbot"
@@ -212,104 +211,3 @@ func Do(mode string, rmsg *dingbot.ReceiveMsg) error {
}
return nil
}
func ImageGenerate(rmsg *dingbot.ReceiveMsg) error {
qObj := db.Chat{
Username: rmsg.SenderNick,
Source: rmsg.GetChatTitle(),
ChatType: db.Q,
ParentContent: 0,
Content: rmsg.Text.Content,
}
qid, err := qObj.Add()
if err != nil {
logger.Error("往MySQL新增数据失败,错误信息:", err)
}
reply, err := chatgpt.ImageQa(rmsg.Text.Content, rmsg.GetSenderIdentifier())
if err != nil {
logger.Info(fmt.Errorf("gpt request error: %v", err))
_, err = rmsg.ReplyToDingtalk(string(dingbot.TEXT), fmt.Sprintf("请求openai失败了错误信息%v", err))
if err != nil {
logger.Error(fmt.Errorf("send message error: %v", err))
return err
}
}
if reply == "" {
logger.Warning(fmt.Errorf("get gpt result falied: %v", err))
return nil
} else {
reply = strings.TrimSpace(reply)
reply = strings.Trim(reply, "\n")
reply = fmt.Sprintf(">点击图片可旋转或放大。\n![](%s)", reply)
aObj := db.Chat{
Username: rmsg.SenderNick,
Source: rmsg.GetChatTitle(),
ChatType: db.A,
ParentContent: qid,
Content: reply,
}
_, err := aObj.Add()
if err != nil {
logger.Error("往MySQL新增数据失败,错误信息:", err)
}
logger.Info(fmt.Sprintf("🤖 %s得到的答案: %#v", rmsg.SenderNick, reply))
// 回复@我的用户
_, err = rmsg.ReplyToDingtalk(string(dingbot.MARKDOWN), reply)
if err != nil {
logger.Error(fmt.Errorf("send message error: %v", err))
return err
}
}
return nil
}
func SelectHistory(rmsg *dingbot.ReceiveMsg) error {
name := strings.TrimSpace(strings.Split(rmsg.Text.Content, ":")[1])
if !public.JudgeAdminUsers(rmsg.SenderNick) {
_, err := rmsg.ReplyToDingtalk(string(dingbot.MARKDOWN), "**🤷 抱歉,您没有查询对话记录的权限,只有程序管理员可以查询!**")
if err != nil {
logger.Error(fmt.Errorf("send message error: %v", err))
return err
}
return nil
}
// 获取数据列表
var chat db.Chat
if !chat.Exist(map[string]interface{}{"username": name}) {
_, err := rmsg.ReplyToDingtalk(string(dingbot.TEXT), "用户名错误,这个用户不存在,请核实之后再进行查询")
if err != nil {
logger.Error(fmt.Errorf("send message error: %v", err))
return err
}
return fmt.Errorf("用户名错误,这个用户不存在,请核实之后重新查询")
}
chats, err := chat.List(db.ChatListReq{
Username: name,
})
if err != nil {
return err
}
var rst string
for _, chatTmp := range chats {
ctime := chatTmp.CreatedAt.Format("2006-01-02 15:04:05")
if chatTmp.ChatType == 1 {
rst += fmt.Sprintf("## 🙋 %s 问\n\n**时间:** %v\n\n**问题为:** %s\n\n", chatTmp.Username, ctime, chatTmp.Content)
} else {
rst += fmt.Sprintf("## 🤖 机器人答\n\n**时间:** %v\n\n**回答如下:** \n\n%s\n\n", ctime, chatTmp.Content)
}
// TODO: 答案应该严格放在问题之后目前只根据ID排序进行的陈列当一个用户同时提出多个问题时最终展示的可能会有点问题
}
fileName := time.Now().Format("20060102-150405") + ".md"
// 写入文件
if err = public.WriteToFile("./data/chatHistory/"+fileName, []byte(rst)); err != nil {
return err
}
// 回复@我的用户
reply := fmt.Sprintf("- 在线查看: [点我](%s)\n- 下载文件: [点我](%s)\n- 在线预览请安装插件:[Markdown Preview Plus](https://chrome.google.com/webstore/detail/markdown-preview-plus/febilkbfcbhebfnokafefeacimjdckgl)", public.Config.ServiceURL+"/history/"+fileName, public.Config.ServiceURL+"/download/"+fileName)
logger.Info(fmt.Sprintf("🤖 %s得到的答案: %#v", rmsg.SenderNick, reply))
_, err = rmsg.ReplyToDingtalk(string(dingbot.MARKDOWN), reply)
if err != nil {
logger.Error(fmt.Errorf("send message error: %v", err))
return err
}
return nil
}