feat: 引用メッセージのコンテキストへの取り込み

go.mod

@@ -28,6 +28,7 @@ require (
 	github.com/kr/text v0.2.0 // indirect
 	github.com/mfridman/interpolate v0.0.2 // indirect
 	github.com/milvus-io/milvus-proto/go-api/v2 v2.4.10-0.20240819025435-512e3b98866a // indirect
+	github.com/motoki317/sc v1.8.2 // indirect
 	github.com/pkg/errors v0.9.1 // indirect
 	github.com/rogpeppe/go-internal v1.14.1 // indirect
 	github.com/sethvargo/go-retry v0.3.0 // indirect

go.sum

@@ -84,6 +84,8 @@ github.com/milvus-io/milvus-proto/go-api/v2 v2.4.10-0.20240819025435-512e3b98866
 github.com/milvus-io/milvus-proto/go-api/v2 v2.4.10-0.20240819025435-512e3b98866a/go.mod h1:1OIl0v5PQeNxIJhCvY+K55CBUOYDZevw9g9380u1Wek=
 github.com/milvus-io/milvus-sdk-go/v2 v2.4.2 h1:Xqf+S7iicElwYoS2Zly8Nf/zKHuZsNy1xQajfdtygVY=
 github.com/milvus-io/milvus-sdk-go/v2 v2.4.2/go.mod h1:ulO1YUXKH0PGg50q27grw048GDY9ayB4FPmh7D+FFTA=
+github.com/motoki317/sc v1.8.2 h1:JzhmFKl4ZS0VxuRYRBQ07o3DcGdvhn2NwqnUcPJmCjY=
+github.com/motoki317/sc v1.8.2/go.mod h1:IwywgSXTlBxHV8a6lHNiQYmTBh7Dc4f9KjzXVdl8/Bk=
 github.com/ncruces/go-strftime v0.1.9 h1:bY0MQC28UADQmHmaF5dgpLmImcShSi2kHU9XLdhx/f4=
 github.com/ncruces/go-strftime v0.1.9/go.mod h1:Fwc5htZGVVkseilnfgOVb9mKy6w1naJmn9CehxcKcls=
 github.com/openai/openai-go/v2 v2.7.1 h1:/tfvTJhfv7hTSL8mWwc5VL4WLLSDL5yn9VqVykdu9r8=

internal/bot/channel.go

@@ -0,0 +1,25 @@
+package bot
+
+import (
+	"context"
+	"time"
+
+	"github.com/motoki317/sc"
+)
+
+func getChannelPathInternal(ctx context.Context, channelID string) (string, error) {
+	bot := GetBot()
+
+	path, _, err := bot.API().ChannelAPI.GetChannelPath(ctx, channelID).Execute()
+	if err != nil {
+		return "", err
+	}
+
+	return path.Path, nil
+}
+
+var channelPathCache = sc.NewMust(getChannelPathInternal, time.Hour, time.Hour)
+
+func GetChannelPath(channelID string) (string, error) {
+	return channelPathCache.Get(context.Background(), channelID)
+}

internal/bot/user.go

@@ -0,0 +1,26 @@
+package bot
+
+import (
+	"context"
+	"time"
+
+	"github.com/motoki317/sc"
+	"github.com/traPtitech/go-traq"
+)
+
+func getUserInternal(ctx context.Context, userID string) (*traq.UserDetail, error) {
+	bot := GetBot()
+
+	user, _, err := bot.API().UserAPI.GetUser(ctx, userID).Execute()
+	if err != nil {
+		return nil, err
+	}
+
+	return user, nil
+}
+
+var userCache = sc.NewMust(getUserInternal, time.Hour, time.Hour)
+
+func GetUser(userID string) (*traq.UserDetail, error) {
+	return userCache.Get(context.Background(), userID)
+}

internal/handler/OnDirectMessageCreated.go

@@ -1,9 +1,11 @@
 package handler
 
 import (
+	"log"
+
 	"github.com/traPtitech/BOT_GPT/internal/bot"
+	"github.com/traPtitech/BOT_GPT/internal/pkg/formatter"
 	"github.com/traPtitech/traq-ws-bot/payload"
-	"log"
 )
 
 func (h *Handler) DirectMessageReceived() func(p *payload.DirectMessageCreated) {
@@ -16,12 +18,18 @@ func (h *Handler) DirectMessageReceived() func(p *payload.DirectMessageCreated)
 			return
 		}
 
-		plainTextWithoutMention := bot.RemoveFirstBotID(p.Message.PlainText)
+		textWithEmbed := formatter.FormatEmbeds(p.Message.Text)
+		textWithoutMention := bot.RemoveFirstBotID(textWithEmbed)
+		formattedMessage, err := formatter.FormatQuotedMessage(p.Message.User.ID, textWithoutMention)
+		if err != nil {
+			log.Printf("Error formatting quoted message: %v\n", err)
+			formattedMessage = textWithoutMention
+		}
 
 		if p.Message.User.Name != "pikachu" {
 			_ = bot.PostMessage(p.Message.ChannelID, "DMではあんまり沢山使わないでね。定期的な`/reset`を忘れない事。")
 		}
 
-		messageReceived(p.Message.Text, plainTextWithoutMention, p.Message.ChannelID)
+		messageReceived(p.Message.Text, formattedMessage, p.Message.ChannelID)
 	}
 }

internal/handler/OnMessageCreated.go

@@ -1,9 +1,11 @@
 package handler
 
 import (
+	"log"
+
 	"github.com/traPtitech/BOT_GPT/internal/bot"
+	"github.com/traPtitech/BOT_GPT/internal/pkg/formatter"
 	"github.com/traPtitech/traq-ws-bot/payload"
-	"log"
 )
 
 func (h *Handler) MessageReceived() func(p *payload.MessageCreated) {
@@ -16,8 +18,14 @@ func (h *Handler) MessageReceived() func(p *payload.MessageCreated) {
 			return
 		}
 
-		plainTextWithoutMention := bot.RemoveFirstBotID(p.Message.PlainText)
+		textEmbedFormatted := formatter.FormatEmbeds(p.Message.Text)
+		textWithoutMention := bot.RemoveFirstBotID(textEmbedFormatted)
+		formattedMessage, err := formatter.FormatQuotedMessage(p.Message.User.ID, textWithoutMention)
+		if err != nil {
+			log.Printf("Error formatting quoted message: %v\n", err)
+			formattedMessage = textWithoutMention
+		}
 
-		messageReceived(p.Message.Text, plainTextWithoutMention, p.Message.ChannelID)
+		messageReceived(p.Message.Text, formattedMessage, p.Message.ChannelID)
 	}
 }

internal/pkg/formatter/embed.go

@@ -0,0 +1,9 @@
+package formatter
+
+import "regexp"
+
+var embedRegex = regexp.MustCompile(`!\{"type":"(\w+?)","raw":"(.+?)","id":"[a-z0-9-]+?"\}`)
+
+func FormatEmbeds(content string) string {
+	return embedRegex.ReplaceAllString(content, "$2")
+}

internal/pkg/formatter/embed_test.go

@@ -0,0 +1,36 @@
+package formatter
+
+import "testing"
+
+func TestFormatEmbeds(t *testing.T) {
+	tests := []struct {
+		name     string
+		input    string
+		expected string
+	}{
+		{
+			name:     "No embeds",
+			input:    "Hello, world!",
+			expected: "Hello, world!",
+		},
+		{
+			name:     "Single embed",
+			input:    `Hello, !{"type":"channel","raw":"#general","id":"04ad2c18-fdcb-4c43-beef-82e8ba26ac98"}, world`,
+			expected: "Hello, #general, world",
+		},
+		{
+			name:     "Multiple embeds",
+			input:    `!{"type":"user","raw":"@cp20","id":"be77174f-13c5-4464-8b15-7f45b96d5b18"}!{"type":"channel","raw":"#general","id":"04ad2c18-fdcb-4c43-beef-82e8ba26ac98"}`,
+			expected: "@cp20#general",
+		},
+	}
+
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			result := FormatEmbeds(tt.input)
+			if result != tt.expected {
+				t.Errorf("FormatEmbeds(%q) = %q; expected %q", tt.input, result, tt.expected)
+			}
+		})
+	}
+}

internal/pkg/formatter/quotes.go

@@ -0,0 +1,106 @@
+package formatter
+
+import (
+	"regexp"
+	"strings"
+	"unicode/utf8"
+
+	"github.com/traPtitech/BOT_GPT/internal/bot"
+	"github.com/traPtitech/go-traq"
+)
+
+const quoteRegexStr = `\bhttps://q\.trap\.jp/messages/([0-9a-f]{8}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{12})\b`
+
+var quoteRegex = regexp.MustCompile(quoteRegexStr)
+
+var allowingPrefixes = []string{"event", "general", "random", "services", "team/SysAd"}
+
+func isChannelAllowingQuotes(channelID string) (bool, error) {
+	channelPath, err := bot.GetChannelPath(channelID)
+	if err != nil {
+		return false, err
+	}
+
+	for _, prefix := range allowingPrefixes {
+		if strings.HasPrefix(channelPath, prefix) {
+			return true, nil
+		}
+	}
+
+	return false, nil
+}
+
+func isUserAllowingQuotes(userID string, messageUserID string) (bool, error) {
+	if userID == messageUserID {
+		return true, nil
+	}
+
+	messageUser, err := bot.GetUser(messageUserID)
+	if err != nil {
+		return false, err
+	}
+
+	if messageUser.Bot {
+		return true, nil
+	}
+
+	return false, nil
+}
+
+func getQuoteMarkdown(message *traq.Message) (string, error) {
+	user, err := bot.GetUser(message.UserId)
+	if err != nil {
+		return "", err
+	}
+
+	return "> " + user.Name + ":\n> " + strings.ReplaceAll(message.Content, "\n", "\n> "), nil
+}
+
+const maxQuoteLength = 10000
+
+func FormatQuotedMessage(userID string, content string) (string, error) {
+	matches := quoteRegex.FindAllSubmatch([]byte(content), len(content))
+	messageIDs := make([]string, 0, len(matches))
+	for _, match := range matches {
+		if len(match) < 2 {
+			continue
+		}
+		messageID := string(match[1])
+		messageIDs = append(messageIDs, messageID)
+	}
+
+	var formattedContent strings.Builder
+	formattedContent.WriteString(quoteRegex.ReplaceAllString(content, ""))
+
+	for _, messageID := range messageIDs {
+		message := bot.GetMessage(messageID)
+		if message == nil {
+			continue
+		}
+
+		if utf8.RuneCountInString(message.Content) > maxQuoteLength {
+			runes := []rune(message.Content)
+			message.Content = string(runes[:maxQuoteLength]) + "(以下略)"
+		}
+
+		channelAllowed, err := isChannelAllowingQuotes(message.ChannelId)
+		if err != nil {
+			return "", err
+		}
+		userAllowed, err := isUserAllowingQuotes(userID, message.UserId)
+		if err != nil {
+			return "", err
+		}
+		if !channelAllowed && !userAllowed {
+			continue
+		}
+
+		quote, err := getQuoteMarkdown(message)
+		if err != nil {
+			return "", err
+		}
+		formattedContent.WriteString("\n\n" + quote)
+	}
+
+	return formattedContent.String(), nil
+}