feat: (WIP) add storage

Co-authored-by: AHCorn <42889600+AHCorn@users.noreply.github.com>

bot/handler_conversation.go

@@ -0,0 +1,84 @@
+package bot
+
+import (
+	"sync"
+
+	"github.com/celestix/gotgproto/dispatcher"
+	"github.com/celestix/gotgproto/ext"
+	"github.com/krau/SaveAny-Bot/logger"
+)
+
+type ConversationType string
+
+const (
+	ConversationTypeManageStorage ConversationType = "manage_storage"
+)
+
+type ConversationState struct {
+	sync.Mutex
+	conversationType ConversationType
+	InConversation   bool
+	data             map[ConversationType]map[string]interface{}
+}
+
+func (c *ConversationState) Reset() {
+	c.Lock()
+	defer c.Unlock()
+	c.InConversation = false
+	c.conversationType = ""
+	c.data = make(map[ConversationType]map[string]interface{})
+}
+
+func (c *ConversationState) SetConversationType(t ConversationType) {
+	c.Lock()
+	defer c.Unlock()
+	c.conversationType = t
+}
+
+func (c *ConversationState) GetData(key string) interface{} {
+	if c.data == nil || c.data[c.conversationType] == nil {
+		return nil
+	}
+	return c.data[c.conversationType][key]
+}
+
+func (c *ConversationState) SetData(key string, value interface{}) {
+	c.Lock()
+	defer c.Unlock()
+	if c.data == nil {
+		c.data = make(map[ConversationType]map[string]interface{})
+	}
+	if c.data[c.conversationType] == nil {
+		c.data[c.conversationType] = make(map[string]interface{})
+	}
+	c.data[c.conversationType][key] = value
+}
+
+var userConversationState = make(map[int64]*ConversationState)
+
+func handleConversation(ctx *ext.Context, update *ext.Update) error {
+	userID := update.EffectiveUser().GetID()
+	state, ok := userConversationState[userID]
+	if !ok {
+		return dispatcher.ContinueGroups
+	}
+	if update.EffectiveMessage.Text == "/cancel" {
+		state.Reset()
+		ctx.Reply(update, ext.ReplyTextString("已取消"), nil)
+		return dispatcher.EndGroups
+	}
+	if !state.InConversation {
+		return dispatcher.ContinueGroups
+	}
+	return handleConversationState(ctx, update, state)
+}
+
+func handleConversationState(ctx *ext.Context, update *ext.Update, state *ConversationState) error {
+	switch state.conversationType {
+	case ConversationTypeManageStorage:
+		return handleManageStorageConversation(ctx, update, state)
+	default:
+		logger.L.Errorf("Unknown conversation type: %s", state.conversationType)
+	}
+	return dispatcher.EndGroups
+}