Merge pull request #12 from chyroc/support-reply-card

feat: support reply card message

Author: fuergaosi233

.env.example

@@ -4,4 +4,5 @@ LARK_OPEN_BASE_URL=
 LARK_WWW_BASE_URL=
 CHATGPT_API_KEY=
 CHATGPT_API_URL=
-ENABLE_SESSION_FOR_LARK_GROUP=true
+ENABLE_SESSION_FOR_LARK_GROUP=true
+ENABLE_CARD_RESP=true

larkgpt/client.go

@@ -14,6 +14,7 @@ type Client struct {
 	serverPort                string
 	maintained                bool
 	enableSessionForLarkGroup bool
+	enableCardResp            bool
 }
 
 type ClientConfig struct {
@@ -32,6 +33,7 @@ type ClientConfig struct {
 	ServerPort                string
 	Metrics                   IMetrics
 	EnableSessionForLarkGroup bool // 给群聊的消息启动 session，session id 是消息的 root id
+	EnableCardResp            bool // 以飞书卡片消息的形式回复消息
 }
 
 func New(config *ClientConfig) *Client {
@@ -56,9 +58,9 @@ func New(config *ClientConfig) *Client {
 	res.chatGPTIns = newChatGPTClient(config.ChatGPTAPIURL, config.ChatGPTAPIKey, res.metricsIns)
 
 	res.serverPort = config.ServerPort
-
 	res.maintained = config.Maintained
 	res.enableSessionForLarkGroup = config.EnableSessionForLarkGroup
+	res.enableCardResp = config.EnableCardResp
 
 	return res
 }

larkgpt/lark.go

@@ -29,3 +29,14 @@ func (r *larkClient) replyText(ctx context.Context, msgID, text string) error {
 	}
 	return err
 }
+
+func (r *larkClient) replyCard(ctx context.Context, msgID string, card *lark.MessageContentCard) error {
+	_, _, err := r.cli.Message.Reply(msgID).SendCard(ctx, card.String())
+	if err != nil {
+		r.metricsIns.EmitLarkApiFailed()
+		log.Println("LarkAPI 调用失败 请稍后重试. ", err)
+	} else {
+		r.metricsIns.EmitLarkApiSuccess()
+	}
+	return err
+}

larkgpt/larkbot.go

@@ -49,7 +49,7 @@ func (r *Client) ReceiveChatGPTMessage(ctx context.Context, msg string, event *l
 
 	log.Print("Receive message: ", msg)
 	if r.maintained {
-		return r.larkIns.replyText(context.Background(), event.Message.MessageID, "ChatGPT Bot 正在维护中 请稍后重试.请飞书搜索 ChatGPT 讨论群, 选择同款头像进群看进度.")
+		return r.replyMaintained(event.Message.MessageID)
 	}
 
 	var result string
@@ -62,10 +62,10 @@ func (r *Client) ReceiveChatGPTMessage(ctx context.Context, msg string, event *l
 	log.Println("msg: ", msg, "session", sessionID, "result: ", result)
 	if err != nil {
 		log.Println("ChatGPT 请求失败 请稍后重试. ", err)
-		return r.larkIns.replyText(context.Background(), event.Message.MessageID, "ChatGPT 请求失败 请稍后重试.")
+		return r.replyChatGPTError(event.Message.MessageID, "ChatGPT 请求失败 请稍后重试.")
 	}
 
-	return r.larkIns.replyText(context.Background(), event.Message.MessageID, result)
+	return r.larkIns.replyChatGPTMessage(event.Message.MessageID, msg, result, r.enableCardResp)
 }
 
 func (r *Client) ReceiveCommandMessage(ctx context.Context, command string, event *lark.EventV2IMMessageReceiveV1) {
@@ -77,19 +77,21 @@ func (r *Client) ReceiveCommandMessage(ctx context.Context, command string, even
 			err = r.chatGPTIns.DeleteSession(sessionID)
 		}
 		if err != nil {
-			r.larkIns.replyText(context.Background(), event.Message.MessageID, "Reset Failed.")
+			_ = r.replyChatGPTError(event.Message.MessageID, "Reset Failed.")
 			return
 		}
-		r.larkIns.replyText(context.Background(), event.Message.MessageID, "Reset Success.")
+		_ = r.replyChatGPTSuccess(event.Message.MessageID, "Reset Success.")
 	default:
-		r.larkIns.replyText(context.Background(), event.Message.MessageID, "Unknown Command.")
+		_ = r.replyChatGPTError(event.Message.MessageID, "Unknown Command.")
 	}
 }
 
 func (r *Client) larkMessageReceiverHandler(ctx context.Context, cli *lark.Lark, schema string, header *lark.EventHeaderV2, event *lark.EventV2IMMessageReceiveV1) (string, error) {
 	content, err := lark.UnwrapMessageContent(event.Message.MessageType, event.Message.Content)
 	if err != nil {
-		return "", err
+		log.Println("解析消息内容失败. ", err)
+		_ = r.replyChatGPTError(event.Message.MessageID, "解析消息内容失败.")
+		return "", nil // 无法重试
 	}
 	msg := ""
 	switch event.Message.MessageType {
@@ -99,8 +101,8 @@ func (r *Client) larkMessageReceiverHandler(ctx context.Context, cli *lark.Lark,
 		msg = wrapLarkPostMessageText(content)
 	default:
 		log.Println("暂不支持的消息类型.")
-		_ = r.larkIns.replyText(context.Background(), event.Message.MessageID, "暂不支持的消息类型.")
-		return "", nil
+		_ = r.replyChatGPTError(event.Message.MessageID, "暂不支持的消息类型.")
+		return "", nil // 无法重试
 	}
 	msg = filterMsg(msg)
 	if isNonsense(content, msg) {

larkgpt/reply.go

@@ -0,0 +1,72 @@
+package larkgpt
+
+import (
+	"context"
+
+	"github.com/chyroc/lark"
+	"github.com/chyroc/lark/card"
+)
+
+func (r *Client) replyMaintained(msgID string) error {
+	if !r.enableCardResp {
+		text := "ChatGPT Bot 正在维护中 请稍后重试.请飞书搜索 ChatGPT 讨论群, 选择同款头像进群看进度."
+		return r.larkIns.replyText(context.Background(), msgID, text)
+	}
+	cardIns := card.Card(
+		card.Div().SetFields(
+			card.FieldMarkdown("ChatGPT Bot 正在维护中 请稍后重试."),
+		),
+		card.HR(),
+		card.Note(card.Text("请飞书搜索 ChatGPT 讨论群, 选择同款头像进群看进度.")),
+	).SetHeader(
+		card.Header("ChatGPT Bot Error").SetRed(),
+	)
+	return r.larkIns.replyCard(context.Background(), msgID, cardIns)
+}
+
+func (r *Client) replyChatGPTError(msgID string, text string) error {
+	if !r.enableCardResp {
+		return r.larkIns.replyText(context.Background(), msgID, text)
+	}
+	cardIns := card.Card(
+		card.Div().SetFields(
+			card.FieldMarkdown(text),
+		),
+	).SetHeader(
+		card.Header("ChatGPT Bot Error").SetRed(),
+	)
+	return r.larkIns.replyCard(context.Background(), msgID, cardIns)
+}
+
+func (r *Client) replyChatGPTSuccess(msgID string, text string) error {
+	if !r.enableCardResp {
+		return r.larkIns.replyText(context.Background(), msgID, text)
+	}
+	cardIns := card.Card(
+		card.Div().SetFields(
+			card.FieldMarkdown(text),
+		),
+	).SetHeader(
+		card.Header("ChatGPT Bot Error").SetGreen(),
+	)
+	return r.larkIns.replyCard(context.Background(), msgID, cardIns)
+}
+
+func (r *larkClient) replyChatGPTMessage(msgID, title, text string, enableCardResp bool) error {
+	if !enableCardResp {
+		return r.replyText(context.Background(), msgID, text)
+	}
+	cardIns := card.Card(
+		card.Div().SetFields(
+			card.FieldMarkdown(text),
+		),
+		card.HR(),
+		card.Note(
+			card.I18nText(&lark.I18NText{
+				ZhCn: "HELP: 回复消息支持上下文对话; /reset 重置对话",
+				EnUs: "HELP: Reply messages with context; send /reset resets dialogue",
+			}),
+		),
+	).SetHeader(card.Header(title))
+	return r.replyCard(context.Background(), msgID, cardIns)
+}

main.go

@@ -38,5 +38,6 @@ func loadConfig() (*larkgpt.ClientConfig, error) {
 		ServerPort:                port,
 		Maintained:                os.Getenv("MAINTAINED") == "true",
 		EnableSessionForLarkGroup: os.Getenv("ENABLE_SESSION_FOR_LARK_GROUP") == "true",
+		EnableCardResp:            os.Getenv("ENABLE_CARD_RESP") == "true",
 	}, nil
 }