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

2025-10-16 21:20:37 +08:00 · 2023-04-04 14:30:13 +08:00
parent 8c04c74640
commit 29fc9243d6
7 changed files with 299 additions and 102 deletions
--- a/main.go
+++ b/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不为空：提示词之后没有文本 -> 直接返回提示词所代表的内容
--- a/pkg/ops/opstools.go
+++ b/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
+}
--- a/pkg/process/db.go
+++ b/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
+}
--- a/pkg/process/image.go
+++ b/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
+}
--- a/pkg/process/opstools.go
+++ b/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
+}
--- a/pkg/process/process_request.go
+++ b/pkg/process/process_request.go
@@ -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
-}
--- a/public/tools.go
+++ b/public/tools.go
@@ -4,6 +4,7 @@ import (
 	"io/ioutil"
 	"os"
 	"strings"
+	"time"
 )

 // 将内容写入到文件，如果文件名带路径，则会判断路径是否存在，不存在则创建
@@ -63,3 +64,7 @@ func JudgeAdminUsers(s string) bool {
 	}
 	return false
 }
+
+func GetReadTime(t time.Time) string {
+	return t.Format("2006-01-02 15:04:05")
+}