重构：把剩余 12 个顶层包全部迁到 internal/

继续之前的重构：把根目录残留的非数据/资源类子目录（agents、agenttool、ai、 autoreply、completion、conversation、llm、message、mqtpp、stream、 toolmanager、toolrouter）一并搬到 internal/ 下，并把全工程引用它们的 import 路径从 "meshtastic_mqtt_server/<x>" 重写为 "meshtastic_mqtt_server/internal/<x>"。变更 - git mv 12 个顶层目录进 internal/，无函数体改动。 - 全工程 sed 把 import path 加上 internal/ 前缀，包括 main.go、 internal/bot/bot_service.go、internal/store/bot_store.go 中对 mqtpp 的引用，以及 ai 子系统内部 agents/agenttool/autoreply/conversation/ llm/toolmanager/toolrouter 之间的相互引用。验证 - go build ./... 通过；go test ./... 全部包通过。 - AI 自动回复链路（main → ai.NewService → autoreply.Service → botSender）保持不变，仅 import 路径调整。 - 根目录最终只剩 main.go / main_test.go / internal/ 加 go.mod / go.sum 与数据资源目录（dist、doc、firmware、py、win、meshmap_frontend）。 Co-Authored-By: Claude <noreply@anthropic.com>
2026-06-18 18:43:42 +08:00
parent 02b445884d
commit d57dff58f3
25 changed files with 36 additions and 36 deletions
@@ -0,0 +1,219 @@
+package conversation
+
+import (
+	"encoding/json"
+	"errors"
+	"fmt"
+	"os"
+	"path/filepath"
+	"sort"
+	"strings"
+	"sync"
+	"time"
+
+	"meshtastic_mqtt_server/internal/message"
+)
+
+// Store manages conversations stored as JSON files
+type Store struct {
+	dir string
+	mu  sync.Mutex
+}
+
+// NewStore creates a new conversation store
+func NewStore(dir string) *Store {
+	os.MkdirAll(dir, 0755)
+	return &Store{dir: dir}
+}
+
+// path returns the file path for a conversation ID
+func (s *Store) path(id string) string {
+	return filepath.Join(s.dir, id+".json")
+}
+
+// botPath returns the directory path for a specific bot
+func (s *Store) botPath(botID uint64) string {
+	return filepath.Join(s.dir, fmt.Sprintf("bot_%d", botID))
+}
+
+// Create creates a new conversation for a bot
+func (s *Store) Create(botID uint64, botNodeID string) (*message.Conversation, error) {
+	return s.CreateWithPreset(botID, botNodeID, "")
+}
+
+// CreateWithPreset creates a new conversation with a preset prompt
+func (s *Store) CreateWithPreset(botID uint64, botNodeID string, preset string) (*message.Conversation, error) {
+	conv := &message.Conversation{
+		ID:           generateConversationID(botID),
+		BotID:        botID,
+		BotNodeID:    botNodeID,
+		Title:        "新对话",
+		CreatedAt:    time.Now(),
+		UpdatedAt:    time.Now(),
+		PresetPrompt: strings.TrimSpace(preset),
+	}
+	if err := s.Save(conv); err != nil {
+		return nil, err
+	}
+	return conv, nil
+}
+
+// Save saves a conversation to disk
+func (s *Store) Save(conv *message.Conversation) error {
+	s.mu.Lock()
+	defer s.mu.Unlock()
+	conv.UpdatedAt = time.Now()
+	return atomicWriteJSON(s.path(conv.ID), conv)
+}
+
+// Get retrieves a conversation by ID
+func (s *Store) Get(id string) (*message.Conversation, error) {
+	s.mu.Lock()
+	defer s.mu.Unlock()
+	data, err := os.ReadFile(s.path(id))
+	if err != nil {
+		if os.IsNotExist(err) {
+			return nil, errors.New("对话不存在")
+		}
+		return nil, fmt.Errorf("读取对话失败: %w", err)
+	}
+	var conv message.Conversation
+	if err := json.Unmarshal(data, &conv); err != nil {
+		return nil, fmt.Errorf("解析对话失败: %w", err)
+	}
+	return &conv, nil
+}
+
+// GetOrCreateForBot gets or creates a conversation for a bot
+func (s *Store) GetOrCreateForBot(botID uint64, botNodeID string, peerNodeID string) (*message.Conversation, error) {
+	// Try to find an existing conversation with this peer
+	convs, err := s.ListForBot(botID)
+	if err == nil && len(convs) > 0 {
+		// Use the most recent conversation (List already sorts by UpdatedAt desc)
+		// Note: List returns convs with Messages = nil, so we need to reload
+		return s.Get(convs[0].ID)
+	}
+	// Create a new conversation
+	return s.Create(botID, botNodeID)
+}
+
+// List returns all conversations
+func (s *Store) List() ([]message.Conversation, error) {
+	s.mu.Lock()
+	defer s.mu.Unlock()
+
+	entries, err := os.ReadDir(s.dir)
+	if err != nil {
+		return nil, fmt.Errorf("读取对话目录失败: %w", err)
+	}
+
+	var list []message.Conversation
+	for _, e := range entries {
+		if e.IsDir() || filepath.Ext(e.Name()) != ".json" {
+			continue
+		}
+		data, err := os.ReadFile(filepath.Join(s.dir, e.Name()))
+		if err != nil {
+			continue
+		}
+		var conv message.Conversation
+		if err := json.Unmarshal(data, &conv); err != nil {
+			continue
+		}
+		conv.Messages = nil // Don't return messages in list
+		list = append(list, conv)
+	}
+
+	sort.Slice(list, func(i, j int) bool {
+		return list[i].UpdatedAt.After(list[j].UpdatedAt)
+	})
+	return list, nil
+}
+
+// ListForBot returns all conversations for a specific bot
+func (s *Store) ListForBot(botID uint64) ([]message.Conversation, error) {
+	all, err := s.List()
+	if err != nil {
+		return nil, err
+	}
+	var filtered []message.Conversation
+	for _, conv := range all {
+		if conv.BotID == botID {
+			filtered = append(filtered, conv)
+		}
+	}
+	return filtered, nil
+}
+
+// Delete deletes a conversation by ID
+func (s *Store) Delete(id string) error {
+	s.mu.Lock()
+	defer s.mu.Unlock()
+	if err := os.Remove(s.path(id)); err != nil && !os.IsNotExist(err) {
+		return fmt.Errorf("删除对话失败: %w", err)
+	}
+	return nil
+}
+
+// DeleteForBot deletes all conversations for a bot
+func (s *Store) DeleteForBot(botID uint64) error {
+	convs, err := s.ListForBot(botID)
+	if err != nil {
+		return err
+	}
+	for _, conv := range convs {
+		_ = s.Delete(conv.ID)
+	}
+	return nil
+}
+
+// AddMessage adds a message to a conversation
+func (s *Store) AddMessage(convID string, msg message.ChatMessage) error {
+	conv, err := s.Get(convID)
+	if err != nil {
+		return err
+	}
+	conv.Messages = append(conv.Messages, msg)
+	if conv.Title == "" || conv.Title == "新对话" {
+		conv.Title = GenerateTitle(conv.Messages)
+	}
+	return s.Save(conv)
+}
+
+// atomicWriteJSON writes JSON to a file atomically
+func atomicWriteJSON(path string, v any) error {
+	tmp := path + ".tmp"
+	data, err := json.MarshalIndent(v, "", "  ")
+	if err != nil {
+		return err
+	}
+	if err := os.WriteFile(tmp, data, 0644); err != nil {
+		return err
+	}
+	return os.Rename(tmp, path)
+}
+
+// GenerateTitle generates a title from conversation messages
+func GenerateTitle(messages []message.ChatMessage) string {
+	for _, m := range messages {
+		if m.Hidden {
+			continue
+		}
+		if m.Role == "user" && strings.TrimSpace(m.Content) != "" {
+			title := strings.TrimSpace(m.Content)
+			title = strings.ReplaceAll(title, "\r\n", " ")
+			title = strings.ReplaceAll(title, "\n", " ")
+			runes := []rune(title)
+			if len(runes) > 30 {
+				return string(runes[:30]) + "..."
+			}
+			return title
+		}
+	}
+	return "新对话"
+}
+
+// generateConversationID generates a unique conversation ID
+func generateConversationID(botID uint64) string {
+	return fmt.Sprintf("bot_%d_%d", botID, time.Now().UnixNano())
+}