feat: Phase 4 多平台接入 — Platform Bridge + 6平台适配器 + 身份权限系统 (22文件, 2129行)
Co-Authored-By: Claude Opus 4.7 <noreply@anthropic.com>
This commit is contained in:
@@ -0,0 +1,204 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"context"
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
"net/http"
|
||||
"os"
|
||||
"os/signal"
|
||||
"syscall"
|
||||
"time"
|
||||
|
||||
discordstub "github.com/yourname/cyrene-ai/platform-bridge/internal/adapter/discord"
|
||||
feishustub "github.com/yourname/cyrene-ai/platform-bridge/internal/adapter/feishu"
|
||||
qqadapter "github.com/yourname/cyrene-ai/platform-bridge/internal/adapter/qq"
|
||||
telegramadapter "github.com/yourname/cyrene-ai/platform-bridge/internal/adapter/telegram"
|
||||
wechatstub "github.com/yourname/cyrene-ai/platform-bridge/internal/adapter/wechat"
|
||||
webhookadapter "github.com/yourname/cyrene-ai/platform-bridge/internal/adapter/webhook"
|
||||
"github.com/yourname/cyrene-ai/platform-bridge/internal/bridge"
|
||||
"github.com/yourname/cyrene-ai/platform-bridge/internal/config"
|
||||
"github.com/yourname/cyrene-ai/platform-bridge/internal/handler"
|
||||
"github.com/yourname/cyrene-ai/platform-bridge/internal/permissions"
|
||||
)
|
||||
|
||||
func main() {
|
||||
cfg := config.Load()
|
||||
|
||||
// Core components.
|
||||
mapper := bridge.NewIdentityMapper()
|
||||
checker := permissions.NewChecker()
|
||||
router := bridge.NewPlatformRouter(mapper, checker)
|
||||
|
||||
// Seed default identities from environment.
|
||||
seedIdentities(mapper)
|
||||
|
||||
// Register platform adapters.
|
||||
adapters := []bridge.PlatformAdapter{
|
||||
qqadapter.NewAdapter(cfg.QQBotPort),
|
||||
telegramadapter.NewAdapter(cfg.TelegramToken, cfg.TelegramWebhookURL),
|
||||
webhookadapter.NewAdapter("webhook"),
|
||||
wechatstub.NewAdapter(),
|
||||
feishustub.NewAdapter(),
|
||||
discordstub.NewAdapter(),
|
||||
}
|
||||
|
||||
for _, a := range adapters {
|
||||
router.RegisterAdapter(a)
|
||||
}
|
||||
|
||||
// Set message handler: forward to AI-Core.
|
||||
router.SetMessageHandler(func(msg *bridge.UnifiedMessage) (*bridge.UnifiedResponse, error) {
|
||||
return forwardToAICore(cfg, msg)
|
||||
})
|
||||
|
||||
// Connect all adapters.
|
||||
ctx := context.Background()
|
||||
for _, a := range adapters {
|
||||
if err := a.Connect(ctx); err != nil {
|
||||
fmt.Printf("WARN: connect %s failed: %v\n", a.PlatformName(), err)
|
||||
} else {
|
||||
fmt.Printf("Platform adapter connected: %s\n", a.PlatformName())
|
||||
}
|
||||
}
|
||||
|
||||
// Setup HTTP server.
|
||||
mux := http.NewServeMux()
|
||||
bh := handler.NewBridgeHandler(router)
|
||||
bh.RegisterRoutes(mux)
|
||||
|
||||
// Start QQ message reader loop.
|
||||
qq, _ := router.GetAdapter("qq")
|
||||
if qqa, ok := qq.(*qqadapter.Adapter); ok {
|
||||
qqMsgCh := make(chan *qqadapter.OBv11Message, 100)
|
||||
go qqa.ReadMessages(ctx, qqMsgCh)
|
||||
go func() {
|
||||
for msg := range qqMsgCh {
|
||||
response, err := router.RouteMessage("qq", msg)
|
||||
if err != nil {
|
||||
fmt.Printf("[qq] route error: %v\n", err)
|
||||
continue
|
||||
}
|
||||
// Send response back through QQ adapter.
|
||||
msgs, err := router.SendResponse(response)
|
||||
if err != nil {
|
||||
fmt.Printf("[qq] send error: %v\n", err)
|
||||
continue
|
||||
}
|
||||
_ = msgs // QQ adapter handles sending via WebSocket
|
||||
}
|
||||
}()
|
||||
}
|
||||
|
||||
addr := ":" + cfg.Port
|
||||
srv := &http.Server{Addr: addr, Handler: mux}
|
||||
go func() {
|
||||
fmt.Printf("Platform Bridge listening on port %s\n", cfg.Port)
|
||||
if err := srv.ListenAndServe(); err != nil && err != http.ErrServerClosed {
|
||||
fmt.Printf("FATAL: %v\n", err)
|
||||
os.Exit(1)
|
||||
}
|
||||
}()
|
||||
|
||||
quit := make(chan os.Signal, 1)
|
||||
signal.Notify(quit, syscall.SIGINT, syscall.SIGTERM)
|
||||
<-quit
|
||||
fmt.Println("Shutting down Platform Bridge...")
|
||||
|
||||
shutdownCtx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
|
||||
defer cancel()
|
||||
for _, a := range adapters {
|
||||
a.Disconnect(shutdownCtx)
|
||||
}
|
||||
srv.Shutdown(shutdownCtx)
|
||||
fmt.Println("Platform Bridge stopped")
|
||||
}
|
||||
|
||||
// forwardToAICore sends a unified message to AI-Core's chat endpoint and returns the response.
|
||||
func forwardToAICore(cfg *config.Config, msg *bridge.UnifiedMessage) (*bridge.UnifiedResponse, error) {
|
||||
reqBody, _ := json.Marshal(map[string]interface{}{
|
||||
"user_id": msg.SenderID,
|
||||
"session_id": fmt.Sprintf("platform_%s_%s", msg.Platform, msg.ChannelID),
|
||||
"message": msg.Content,
|
||||
"mode": "text",
|
||||
"source": map[string]string{
|
||||
"platform": msg.Platform,
|
||||
"channel_id": msg.ChannelID,
|
||||
"channel_type": msg.ChannelType,
|
||||
"sender_name": msg.SenderName,
|
||||
},
|
||||
})
|
||||
|
||||
url := cfg.AICoreURL + "/api/v1/chat"
|
||||
req, _ := http.NewRequest("POST", url, bytes.NewReader(reqBody))
|
||||
req.Header.Set("Content-Type", "application/json")
|
||||
req.Header.Set("Accept", "text/event-stream")
|
||||
if cfg.InternalToken != "" {
|
||||
req.Header.Set("X-Internal-Token", cfg.InternalToken)
|
||||
}
|
||||
|
||||
client := &http.Client{Timeout: 120 * time.Second}
|
||||
resp, err := client.Do(req)
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("forward to ai-core failed: %w", err)
|
||||
}
|
||||
defer resp.Body.Close()
|
||||
|
||||
// For simplicity, collect full text from SSE stream.
|
||||
var result struct {
|
||||
Content string `json:"content"`
|
||||
Error string `json:"error"`
|
||||
}
|
||||
if err := json.NewDecoder(resp.Body).Decode(&result); err != nil {
|
||||
// Non-JSON response — read as raw text.
|
||||
buf := new(bytes.Buffer)
|
||||
buf.ReadFrom(resp.Body)
|
||||
return &bridge.UnifiedResponse{
|
||||
Messages: []bridge.ResponseMessage{
|
||||
{DisplayType: "chat", Content: buf.String(), FormatMode: "plain"},
|
||||
},
|
||||
Platform: msg.Platform,
|
||||
}, nil
|
||||
}
|
||||
|
||||
if result.Error != "" {
|
||||
return &bridge.UnifiedResponse{
|
||||
Messages: []bridge.ResponseMessage{
|
||||
{DisplayType: "system_info", Content: result.Error, FormatMode: "plain"},
|
||||
},
|
||||
Platform: msg.Platform,
|
||||
}, nil
|
||||
}
|
||||
|
||||
return &bridge.UnifiedResponse{
|
||||
Messages: []bridge.ResponseMessage{
|
||||
{DisplayType: "chat", Content: result.Content, FormatMode: "plain"},
|
||||
},
|
||||
Platform: msg.Platform,
|
||||
}, nil
|
||||
}
|
||||
|
||||
// seedIdentities loads default identity mappings.
|
||||
func seedIdentities(m *bridge.IdentityMapper) {
|
||||
// Admin on QQ.
|
||||
if qqAdmin := os.Getenv("QQ_ADMIN_UID"); qqAdmin != "" {
|
||||
m.Register(permissions.PlatformIdentity{
|
||||
Platform: "qq",
|
||||
PlatformUID: qqAdmin,
|
||||
CyreneUser: "admin",
|
||||
Nickname: "开拓者",
|
||||
PermissionLevel: "admin",
|
||||
})
|
||||
}
|
||||
// Admin on Telegram.
|
||||
if tgAdmin := os.Getenv("TELEGRAM_ADMIN_UID"); tgAdmin != "" {
|
||||
m.Register(permissions.PlatformIdentity{
|
||||
Platform: "telegram",
|
||||
PlatformUID: tgAdmin,
|
||||
CyreneUser: "admin",
|
||||
Nickname: "开拓者",
|
||||
PermissionLevel: "admin",
|
||||
})
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user