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

main.go

@@ -72,6 +72,10 @@ func Start() {
  return process.ImageGenerate(&msgObj)
  case strings.HasPrefix(msgObj.Text.Content, "#查对话"):
  return process.SelectHistory(&msgObj)
+ case strings.HasPrefix(msgObj.Text.Content, "#域名"):
+ return process.DomainMsg(&msgObj)
+ case strings.HasPrefix(msgObj.Text.Content, "#证书"):
+ return process.DomainCertMsg(&msgObj)
  default:
  msgObj.Text.Content, err = process.GeneratePrompt(msgObj.Text.Content)
  // err不为空：提示词之后没有文本 -> 直接返回提示词所代表的内容

pkg/ops/opstools.go

@@ -0,0 +1,66 @@
+package ops
+
+import (
+ "crypto/tls"
+ "net"
+ "regexp"
+ "strings"
+ "time"
+)
+
+// 域名信息
+type DomainMsg struct {
+ CreateDate string `json:"create_date"`
+ ExpiryDate string `json:"expiry_date"`
+ Registrar string `json:"registrar"`
+}
+
+// GetDomainMsg 获取域名信息
+func GetDomainMsg(domain string) (dm DomainMsg, err error) {
+ var conn net.Conn
+ conn, err = net.Dial("tcp", "whois.verisign-grs.com:43")
+ if err != nil {
+ return
+ }
+ defer conn.Close()
+
+ _, err = conn.Write([]byte(domain + "\r\n"))
+ if err != nil {
+ return
+ }
+ buf := make([]byte, 1024)
+ var num int
+ num, err = conn.Read(buf)
+ if err != nil {
+ return
+ }
+ response := string(buf[:num])
+ re := regexp.MustCompile(`Creation Date: (.*)\n.*Expiry Date: (.*)\n.*Registrar: (.*)`)
+ match := re.FindStringSubmatch(response)
+ if len(match) > 3 {
+ dm.CreateDate = strings.TrimSpace(strings.Split(match[1], "Creation Date:")[0])
+ dm.ExpiryDate = strings.TrimSpace(strings.Split(match[2], "Expiry Date:")[0])
+ dm.Registrar = strings.TrimSpace(strings.Split(match[3], "Registrar:")[0])
+ }
+ return
+}
+
+// GetDomainCertMsg 获取域名证书信息
+func GetDomainCertMsg(domain string) (cm tls.ConnectionState, err error) {
+ var conn net.Conn
+ conn, err = net.DialTimeout("tcp", domain+":443", time.Second*10)
+ if err != nil {
+ return
+ }
+ defer conn.Close()
+ tlsConn := tls.Client(conn, &tls.Config{
+ ServerName: domain,
+ })
+ defer tlsConn.Close()
+ err = tlsConn.Handshake()
+ if err != nil {
+ return
+ }
+ cm = tlsConn.ConnectionState()
+ return
+}

pkg/process/db.go

@@ -0,0 +1,67 @@
+package process
+
+import (
+ "fmt"
+ "strings"
+ "time"
+
+ "github.com/eryajf/chatgpt-dingtalk/pkg/db"
+ "github.com/eryajf/chatgpt-dingtalk/pkg/dingbot"
+ "github.com/eryajf/chatgpt-dingtalk/pkg/logger"
+ "github.com/eryajf/chatgpt-dingtalk/public"
+)
+
+// 与数据库交互的请求处理在此
+
+// SelectHistory 查询会话历史
+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
+}

pkg/process/image.go

@@ -0,0 +1,62 @@
+package process
+
+import (
+ "fmt"
+ "strings"
+
+ "github.com/eryajf/chatgpt-dingtalk/pkg/db"
+ "github.com/eryajf/chatgpt-dingtalk/pkg/dingbot"
+ "github.com/eryajf/chatgpt-dingtalk/pkg/logger"
+ "github.com/solywsh/chatgpt"
+)
+
+// ImageGenerate openai生成图片
+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
+}

pkg/process/opstools.go

@@ -0,0 +1,95 @@
+package process
+
+import (
+ "fmt"
+ "strings"
+
+ "github.com/eryajf/chatgpt-dingtalk/pkg/db"
+ "github.com/eryajf/chatgpt-dingtalk/pkg/dingbot"
+ "github.com/eryajf/chatgpt-dingtalk/pkg/logger"
+ "github.com/eryajf/chatgpt-dingtalk/pkg/ops"
+ "github.com/eryajf/chatgpt-dingtalk/public"
+)
+
+// 一些运维方面的工具在此
+
+// 域名信息
+func DomainMsg(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)
+ }
+ domain := strings.TrimSpace(strings.Split(rmsg.Text.Content, " ")[1])
+ dm, err := ops.GetDomainMsg(domain)
+ if err != nil {
+ return err
+ }
+ // 回复@我的用户
+ reply := fmt.Sprintf("**创建时间:** %v\n\n**到期时间:** %v\n\n**服务商:** %v", dm.CreateDate, dm.ExpiryDate, dm.Registrar)
+ 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 DomainCertMsg(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)
+ }
+ domain := strings.TrimSpace(strings.Split(rmsg.Text.Content, " ")[1])
+ dm, err := ops.GetDomainCertMsg(domain)
+ if err != nil {
+ return err
+ }
+ cert := dm.PeerCertificates[0]
+ // 回复@我的用户
+ reply := fmt.Sprintf("**证书创建时间:** %v\n\n**证书到期时间:** %v\n\n**证书颁发机构:** %v\n\n", public.GetReadTime(cert.NotBefore), public.GetReadTime(cert.NotAfter), cert.Issuer.Organization)
+ 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
+}