feat: add multi-account weixin channel

pkg/channels/compiled_channels.go

@@ -3,7 +3,10 @@ package channels
 import "sort"
 
 func CompiledChannelKeys() []string {
-	out := make([]string, 0, 7)
+	out := make([]string, 0, 8)
+	if weixinCompiled {
+		out = append(out, "weixin")
+	}
 	if telegramCompiled {
 		out = append(out, "telegram")
 	}

pkg/channels/manager.go

@@ -92,6 +92,28 @@ func (m *Manager) initChannels() error {
 		}
 	}
 
+	if m.config.Channels.Weixin.Enabled {
+		if len(m.config.Channels.Weixin.Accounts) == 0 && strings.TrimSpace(m.config.Channels.Weixin.BotToken) == "" {
+			logger.WarnCF("channels", 0, map[string]interface{}{
+				"channel": "weixin",
+				"error":   "missing accounts",
+			})
+		} else {
+			weixin, err := NewWeixinChannel(m.config.Channels.Weixin, m.bus)
+			if err != nil {
+				logger.ErrorCF("channels", 0, map[string]interface{}{
+					logger.FieldChannel: "weixin",
+					logger.FieldError:   err.Error(),
+				})
+			} else {
+				m.channels["weixin"] = weixin
+				logger.InfoCF("channels", 0, map[string]interface{}{
+					logger.FieldChannel: "weixin",
+				})
+			}
+		}
+	}
+
 	if m.config.Channels.WhatsApp.Enabled {
 		if m.config.Channels.WhatsApp.BridgeURL == "" {
 			logger.WarnC("channels", logger.C0009)
@@ -415,6 +437,12 @@ func (m *Manager) GetEnabledChannels() []string {
 	return names
 }
 
+func (m *Manager) GetChannel(name string) (Channel, bool) {
+	cur, _ := m.snapshot.Load().(map[string]Channel)
+	ch, ok := cur[strings.TrimSpace(name)]
+	return ch, ok
+}
+
 func (m *Manager) SendToChannel(ctx context.Context, channelName, chatID, content string) error {
 	m.mu.RLock()
 	channel, exists := m.channels[channelName]

pkg/channels/weixin_stub.go

@@ -0,0 +1,16 @@
+//go:build omit_weixin
+
+package channels
+
+import (
+	"github.com/YspCoder/clawgo/pkg/bus"
+	"github.com/YspCoder/clawgo/pkg/config"
+)
+
+type WeixinChannel struct{ disabledChannel }
+
+const weixinCompiled = false
+
+func NewWeixinChannel(cfg config.WeixinConfig, bus *bus.MessageBus) (*WeixinChannel, error) {
+	return nil, errChannelDisabled("weixin")
+}

pkg/channels/weixin_test.go

@@ -0,0 +1,142 @@
+//go:build !omit_weixin
+
+package channels
+
+import (
+	"context"
+	"testing"
+	"time"
+
+	"github.com/YspCoder/clawgo/pkg/bus"
+	"github.com/YspCoder/clawgo/pkg/config"
+)
+
+func TestBuildAndSplitWeixinChatID(t *testing.T) {
+	chatID := buildWeixinChatID("bot-a", "wx-user-1")
+	if chatID != "bot-a|wx-user-1" {
+		t.Fatalf("unexpected composite chat id: %s", chatID)
+	}
+	botID, rawChatID := splitWeixinChatID(chatID)
+	if botID != "bot-a" || rawChatID != "wx-user-1" {
+		t.Fatalf("unexpected split result: %s %s", botID, rawChatID)
+	}
+}
+
+func TestWeixinHandleInboundMessageUsesCompositeSessionChatID(t *testing.T) {
+	mb := bus.NewMessageBus()
+	ch, err := NewWeixinChannel(config.WeixinConfig{
+		BaseURL: "https://ilinkai.weixin.qq.com",
+		Accounts: []config.WeixinAccountConfig{
+			{BotID: "bot-a", BotToken: "token-a"},
+		},
+	}, mb)
+	if err != nil {
+		t.Fatalf("new weixin channel: %v", err)
+	}
+
+	ch.handleInboundMessage("bot-a", weixinInboundMessage{
+		FromUserID:   "wx-user-1",
+		ContextToken: "ctx-1",
+		ItemList: []weixinMessageItem{
+			{Type: 1, TextItem: struct {
+				Text string `json:"text"`
+			}{Text: "hello"}},
+		},
+	})
+
+	ctx, cancel := context.WithTimeout(context.Background(), 100*time.Millisecond)
+	defer cancel()
+	msg, ok := mb.ConsumeInbound(ctx)
+	if !ok {
+		t.Fatalf("expected inbound message")
+	}
+	if msg.ChatID != "bot-a|wx-user-1" {
+		t.Fatalf("expected composite chat id, got %s", msg.ChatID)
+	}
+	if msg.SessionKey != "weixin:bot-a|wx-user-1" {
+		t.Fatalf("expected composite session key, got %s", msg.SessionKey)
+	}
+	if msg.SenderID != "wx-user-1" {
+		t.Fatalf("expected raw sender id, got %s", msg.SenderID)
+	}
+}
+
+func TestWeixinResolveAccountForCompositeChatID(t *testing.T) {
+	mb := bus.NewMessageBus()
+	ch, err := NewWeixinChannel(config.WeixinConfig{
+		BaseURL:      "https://ilinkai.weixin.qq.com",
+		DefaultBotID: "bot-b",
+		Accounts: []config.WeixinAccountConfig{
+			{BotID: "bot-a", BotToken: "token-a", ContextToken: "ctx-a"},
+			{BotID: "bot-b", BotToken: "token-b", ContextToken: "ctx-b"},
+		},
+	}, mb)
+	if err != nil {
+		t.Fatalf("new weixin channel: %v", err)
+	}
+
+	account, rawChatID, contextToken, err := ch.resolveAccountForChat("bot-a|wx-user-7")
+	if err != nil {
+		t.Fatalf("resolve account: %v", err)
+	}
+	if account.cfg.BotID != "bot-a" {
+		t.Fatalf("expected bot-a, got %s", account.cfg.BotID)
+	}
+	if rawChatID != "wx-user-7" {
+		t.Fatalf("expected raw chat id wx-user-7, got %s", rawChatID)
+	}
+	if contextToken != "ctx-a" {
+		t.Fatalf("expected context token ctx-a, got %s", contextToken)
+	}
+}
+
+func TestWeixinSetDefaultAccount(t *testing.T) {
+	mb := bus.NewMessageBus()
+	ch, err := NewWeixinChannel(config.WeixinConfig{
+		BaseURL: "https://ilinkai.weixin.qq.com",
+		Accounts: []config.WeixinAccountConfig{
+			{BotID: "bot-a", BotToken: "token-a"},
+			{BotID: "bot-b", BotToken: "token-b"},
+		},
+	}, mb)
+	if err != nil {
+		t.Fatalf("new weixin channel: %v", err)
+	}
+
+	if err := ch.SetDefaultAccount("bot-b"); err != nil {
+		t.Fatalf("set default account: %v", err)
+	}
+	accounts := ch.ListAccounts()
+	if len(accounts) != 2 {
+		t.Fatalf("expected 2 accounts, got %d", len(accounts))
+	}
+	defaultCount := 0
+	for _, account := range accounts {
+		if account.Default {
+			defaultCount++
+			if account.BotID != "bot-b" {
+				t.Fatalf("expected bot-b to be default, got %s", account.BotID)
+			}
+		}
+	}
+	if defaultCount != 1 {
+		t.Fatalf("expected exactly one default account, got %d", defaultCount)
+	}
+}
+
+func TestWeixinCancelPendingLogin(t *testing.T) {
+	mb := bus.NewMessageBus()
+	ch, err := NewWeixinChannel(config.WeixinConfig{BaseURL: "https://ilinkai.weixin.qq.com"}, mb)
+	if err != nil {
+		t.Fatalf("new weixin channel: %v", err)
+	}
+	ch.pendingLogins["login-1"] = &WeixinPendingLogin{LoginID: "login-1", QRCode: "code-1", Status: "wait"}
+	ch.loginOrder = []string{"login-1"}
+
+	if ok := ch.CancelPendingLogin("login-1"); !ok {
+		t.Fatalf("expected cancel to succeed")
+	}
+	if got := ch.PendingLogins(); len(got) != 0 {
+		t.Fatalf("expected no pending logins after cancel, got %d", len(got))
+	}
+}