Files

762 lines
32 KiB
Go

package cmd
import (
"context"
"fmt"
"io"
"log/slog"
"os"
"os/signal"
"path/filepath"
"strings"
"syscall"
"time"
"github.com/google/uuid"
"github.com/nextlevelbuilder/goclaw/internal/agent"
"github.com/nextlevelbuilder/goclaw/internal/bgalert"
"github.com/nextlevelbuilder/goclaw/internal/bootstrap"
"github.com/nextlevelbuilder/goclaw/internal/bus"
"github.com/nextlevelbuilder/goclaw/internal/cache"
"github.com/nextlevelbuilder/goclaw/internal/channelmemory"
"github.com/nextlevelbuilder/goclaw/internal/channels"
"github.com/nextlevelbuilder/goclaw/internal/channels/bitrix24"
"github.com/nextlevelbuilder/goclaw/internal/channels/discord"
"github.com/nextlevelbuilder/goclaw/internal/channels/facebook"
"github.com/nextlevelbuilder/goclaw/internal/channels/feishu"
"github.com/nextlevelbuilder/goclaw/internal/channels/pancake"
slackchannel "github.com/nextlevelbuilder/goclaw/internal/channels/slack"
"github.com/nextlevelbuilder/goclaw/internal/channels/telegram"
"github.com/nextlevelbuilder/goclaw/internal/channels/whatsapp"
"github.com/nextlevelbuilder/goclaw/internal/channels/zalo"
zalopersonal "github.com/nextlevelbuilder/goclaw/internal/channels/zalo/personal"
"github.com/nextlevelbuilder/goclaw/internal/config"
"github.com/nextlevelbuilder/goclaw/internal/consolidation"
"github.com/nextlevelbuilder/goclaw/internal/edition"
"github.com/nextlevelbuilder/goclaw/internal/eventbus"
"github.com/nextlevelbuilder/goclaw/internal/gateway"
"github.com/nextlevelbuilder/goclaw/internal/gateway/methods"
"github.com/nextlevelbuilder/goclaw/internal/hooks"
httpapi "github.com/nextlevelbuilder/goclaw/internal/http"
kg "github.com/nextlevelbuilder/goclaw/internal/knowledgegraph"
mcpbridge "github.com/nextlevelbuilder/goclaw/internal/mcp"
"github.com/nextlevelbuilder/goclaw/internal/media"
"github.com/nextlevelbuilder/goclaw/internal/providers"
"github.com/nextlevelbuilder/goclaw/internal/scheduler"
"github.com/nextlevelbuilder/goclaw/internal/skills"
"github.com/nextlevelbuilder/goclaw/internal/store"
"github.com/nextlevelbuilder/goclaw/internal/tools"
usagecaps "github.com/nextlevelbuilder/goclaw/internal/usage/caps"
"github.com/nextlevelbuilder/goclaw/internal/vault"
"github.com/nextlevelbuilder/goclaw/pkg/protocol"
// Register workstation backend factories via init().
_ "github.com/nextlevelbuilder/goclaw/internal/workstation/backends"
)
func gatewayLogOutput() io.Writer {
logFile := strings.TrimSpace(os.Getenv("GOCLAW_LOG_FILE"))
if logFile == "" {
if st, err := os.Stat("/var/log/goclaw"); err == nil && st.IsDir() {
logFile = "/var/log/goclaw/goclaw.log"
}
}
if logFile == "" {
return os.Stdout
}
f, err := os.OpenFile(logFile, os.O_CREATE|os.O_WRONLY|os.O_APPEND, 0o644)
if err != nil {
fmt.Fprintf(os.Stderr, "warning: cannot open GOCLAW_LOG_FILE=%q: %v\n", logFile, err)
return os.Stdout
}
fmt.Fprintf(os.Stderr, "logging to %s\n", logFile)
return io.MultiWriter(os.Stdout, f)
}
func runGateway() {
// Setup structured logging
logLevel := slog.LevelInfo
if verbose {
logLevel = slog.LevelDebug
}
// Env override (docker/K8s friendly, default: info): GOCLAW_LOG_LEVEL=debug|info|warn|error
if lvl := os.Getenv("GOCLAW_LOG_LEVEL"); lvl != "" {
switch strings.ToLower(lvl) {
case "debug":
logLevel = slog.LevelDebug
case "info":
logLevel = slog.LevelInfo
case "warn":
logLevel = slog.LevelWarn
case "error":
logLevel = slog.LevelError
default:
fmt.Fprintf(os.Stderr, "warning: unknown GOCLAW_LOG_LEVEL=%q, using info\n", lvl)
}
}
logOutput := gatewayLogOutput()
textHandler := slog.NewTextHandler(logOutput, &slog.HandlerOptions{Level: logLevel})
logTee := gateway.NewLogTee(textHandler)
slog.SetDefault(slog.New(logTee))
// Load config
cfgPath := resolveConfigPath()
cfg, err := config.Load(cfgPath)
if err != nil {
slog.Error("failed to load config", "error", err)
os.Exit(1)
}
if err := config.ValidateGatewayAuth(cfg.Gateway); err != nil {
slog.Error("unsafe gateway auth configuration", "error", err)
os.Exit(1)
}
// Edition override: explicit GOCLAW_EDITION takes precedence over auto-detection.
// Auto-detection happens later in setupStoresAndTracing (sqlite → lite).
if edName := os.Getenv("GOCLAW_EDITION"); edName != "" {
switch edName {
case "lite":
edition.SetCurrent(edition.Lite)
slog.Info("edition: lite (explicit)")
case "standard":
edition.SetCurrent(edition.Standard)
slog.Info("edition: standard (explicit)")
default:
slog.Warn("unknown GOCLAW_EDITION, using standard", "value", edName)
}
}
// Create core components
msgBus := bus.New()
// V3 domain event bus for consolidation pipeline (episodic → semantic → dreaming)
domainBus := eventbus.NewDomainEventBus(eventbus.Config{
QueueSize: 1000,
WorkerCount: 2,
})
domainBus.Start(context.Background())
defer func() {
if err := domainBus.Drain(10 * time.Second); err != nil {
slog.Warn("domain event bus drain timeout", "error", err)
}
}()
// Create model registry with forward-compat resolvers (shared across all providers)
modelReg := providers.NewInMemoryRegistry()
modelReg.RegisterResolver("anthropic", &providers.AnthropicForwardCompat{})
modelReg.RegisterResolver("openai", &providers.OpenAIForwardCompat{})
// Create provider registry
providerRegistry := providers.NewRegistry(store.TenantIDFromContext)
registerProviders(providerRegistry, cfg, modelReg)
// Resolve workspace (must be absolute for system prompt + file tool path resolution)
workspace := config.ExpandHome(cfg.Agents.Defaults.Workspace)
if !filepath.IsAbs(workspace) {
workspace, _ = filepath.Abs(workspace)
}
os.MkdirAll(workspace, 0755)
// Detect server IPs for output scrubbing (prevents IP leaks via web_fetch, exec, etc.)
// Skip for desktop/lite — localhost-only, no multi-tenant exposure risk
if !edition.Current().IsLimited() {
tools.DetectServerIPs(context.Background())
}
toolsReg, execApprovalMgr, mcpMgr, sandboxMgr, browserMgr, webFetchTool, ttsTool, audioMgr, permPE, toolPE, dataDir, agentCfg := setupToolRegistry(cfg, workspace, providerRegistry)
if browserMgr != nil {
defer browserMgr.Close()
}
if mcpMgr != nil {
defer mcpMgr.Stop()
}
pgStores, traceCollector, snapshotWorker := setupStoresAndTracing(cfg, dataDir, msgBus)
if browserMgr != nil && pgStores != nil && pgStores.BrowserCookies != nil && cfg.Tools.Browser.CookieSyncEnabled {
browserMgr.SetCookieProvider(newStoreBrowserCookieProvider(pgStores.BrowserCookies))
}
// Recover from crashes: flip ghost 'summoning' rows to 'summon_failed'.
// Summon goroutines don't survive process restart; stale DB rows would trap the UI.
if pgStores.Agents != nil {
if n, err := pgStores.Agents.ResetStuckSummoning(context.Background()); err != nil {
slog.Warn("agents.reset_stuck_summoning_failed", "err", err)
} else if n > 0 {
slog.Info("agents.reset_stuck_summoning", "count", n)
}
}
if traceCollector != nil {
defer traceCollector.Stop()
// OTel OTLP export: compiled via build tags. Build with 'go build -tags otel' to enable.
initOTelExporter(context.Background(), cfg, traceCollector)
}
if snapshotWorker != nil {
defer snapshotWorker.Stop()
}
// Redis cache: compiled via build tags. Build with 'go build -tags redis' to enable.
redisClient := initRedisClient(cfg)
defer shutdownRedis(redisClient)
// Register providers from DB (overrides config providers).
if pgStores.Providers != nil {
dbGatewayAddr := loopbackAddr(cfg.Gateway.Host, cfg.Gateway.Port)
registerProvidersFromDB(providerRegistry, pgStores.Providers, pgStores.ConfigSecrets, dbGatewayAddr, cfg.Gateway.Token, pgStores.MCP, cfg, modelReg)
}
slog.Info("model registry initialized", "anthropic_models", len(modelReg.Catalog("anthropic")), "openai_models", len(modelReg.Catalog("openai")))
// Warn if deprecated session scope settings are configured
if cfg.Sessions.Scope != "" && cfg.Sessions.Scope != "per-sender" {
slog.Warn("sessions.scope config is deprecated and ignored — fixed to per-sender", "configured", cfg.Sessions.Scope)
}
if cfg.Sessions.DmScope != "" && cfg.Sessions.DmScope != "per-channel-peer" {
slog.Warn("sessions.dm_scope config is deprecated and ignored — fixed to per-channel-peer", "configured", cfg.Sessions.DmScope)
}
seedSystemConfigs(pgStores.SystemConfigs, pgStores.Tenants, cfg)
// Read back system_configs from DB and overlay onto in-memory config.
if pgStores.SystemConfigs != nil {
if sysConfigs, err := pgStores.SystemConfigs.List(
store.WithTenantID(context.Background(), store.MasterTenantID),
); err == nil && len(sysConfigs) > 0 {
cfg.ApplySystemConfigs(sysConfigs)
slog.Info("system_configs applied to in-memory config", "keys", len(sysConfigs))
}
}
setupMemoryEmbeddings(pgStores, providerRegistry)
usageCapSvc := usagecaps.NewService(pgStores.UsageCaps, pgStores.Providers)
// Resolve background provider for consolidation + vault enrichment.
// Fallback: background.provider → agent.default_provider → first registered provider.
bgProvider, bgModel := resolveBackgroundProvider(cfg, providerRegistry)
// V3: Wire consolidation pipeline (episodic → semantic → KG → dreaming)
if pgStores.Episodic != nil {
if bgProvider != nil {
var kgExtractor *kg.Extractor
if pgStores.KnowledgeGraph != nil {
kgExtractor = kg.NewExtractor(bgProvider, bgModel, 0)
kgExtractor.SetUsageCapService(usageCapSvc)
}
cleanupConsolidation := consolidation.Register(consolidation.ConsolidationDeps{
EpisodicStore: pgStores.Episodic,
MemoryStore: pgStores.Memory,
KGStore: pgStores.KnowledgeGraph,
SessionStore: pgStores.Sessions,
EventBus: domainBus,
SystemConfigs: pgStores.SystemConfigs,
Registry: providerRegistry,
Extractor: kgExtractor,
AlertDeps: bgalert.AlertDeps{SystemConfigs: pgStores.SystemConfigs, MsgBus: msgBus},
UsageCaps: usageCapSvc,
AgentStore: pgStores.Agents,
})
defer cleanupConsolidation()
slog.Info("consolidation pipeline registered", "provider", bgProvider.Name(), "model", bgModel)
} else {
slog.Warn("consolidation pipeline skipped: no provider available")
}
}
if memorySvc := makeChannelMemoryService(pgStores, domainBus, providerRegistry, usageCapSvc); memorySvc != nil {
cleanupChannelMemory := (&channelmemory.Worker{Service: memorySvc}).Start(context.Background())
defer cleanupChannelMemory()
slog.Info("channel memory extraction worker registered")
}
// V3: Wire vault enrichment worker (async summary + embedding + auto-linking).
// Provider is resolved per-tenant at runtime — no static provider needed.
var enrichProgress *vault.EnrichProgress
var enrichWorker *vault.EnrichWorker
if pgStores.Vault != nil && providerRegistry != nil {
cleanupVaultEnrich, ep, ew := vault.RegisterEnrichWorker(vault.EnrichWorkerDeps{
VaultStore: pgStores.Vault,
SystemConfigs: pgStores.SystemConfigs,
Registry: providerRegistry,
EventBus: domainBus,
MsgBus: msgBus,
TeamStore: pgStores.Teams,
AlertDeps: bgalert.AlertDeps{SystemConfigs: pgStores.SystemConfigs, MsgBus: msgBus},
UsageCaps: usageCapSvc,
})
enrichProgress = ep
enrichWorker = ew
defer cleanupVaultEnrich()
slog.Info("vault enrichment worker registered (per-tenant provider resolution)")
}
loadBootstrapFiles(pgStores, workspace, agentCfg)
// Backfill CAPABILITIES.md for pre-v3 agents that don't have it yet.
if count, err := bootstrap.BackfillCapabilities(context.Background(), pgStores.DB); err != nil {
slog.Warn("bootstrap: capabilities backfill failed", "error", err)
} else if count > 0 {
slog.Info("bootstrap: capabilities backfill complete", "agents", count)
}
if readImage, ok := toolsReg.Get("read_image"); ok {
if t, ok := readImage.(*tools.ReadImageTool); ok {
t.SetUsageCapService(usageCapSvc)
}
}
// Subagent system (secureCLI store wired so subagent ExecTools enforce the gate)
subagentMgr := setupSubagents(providerRegistry, cfg, msgBus, toolsReg, workspace, sandboxMgr, pgStores.SecureCLI, usageCapSvc)
if subagentMgr != nil {
// Wire announce queue for batched subagent result delivery (matching TS debounce pattern).
announceQueue := tools.NewAnnounceQueue(1000, 20, makeDelegateAnnounceCallback(subagentMgr, msgBus))
subagentMgr.SetAnnounceQueue(announceQueue)
if pgStores.SubagentTasks != nil {
subagentMgr.SetTaskStore(pgStores.SubagentTasks)
}
toolsReg.Register(tools.NewSpawnTool(subagentMgr, "default", 0))
slog.Info("subagent system enabled", "tools", []string{"spawn"})
}
skillsLoader, skillSearchTool, globalSkillsDir, bundledSkillsDir, builtinSkillsDir := setupSkillsSystem(cfg, workspace, dataDir, pgStores, toolsReg, providerRegistry, msgBus)
_ = skillSearchTool // used via wireExtras → skillsLoader; kept for type clarity
// Register cron/heartbeat/session/message tools, aliases, allow-paths, store wiring.
heartbeatTool, hasMemory := wireExtraTools(pgStores, toolsReg, msgBus, workspace, dataDir, agentCfg, globalSkillsDir, builtinSkillsDir)
// Register workstation_exec + claude_remote tools (Standard edition only; deny-all until Phase 6).
// cleanupWorkstation stops the activity sink retention goroutine and drains the write buffer.
cleanupWorkstation := wireWorkstationTools(pgStores, toolsReg, domainBus)
defer cleanupWorkstation()
// Create all agents — resolved lazily from database by the managed resolver.
agentRouter := agent.NewRouter()
if traceCollector != nil {
agentRouter.SetTraceCollector(traceCollector)
}
slog.Info("agents will be resolved lazily from database")
// Create gateway server and wire enforcement
server := gateway.NewServer(cfg, msgBus, agentRouter, pgStores.Sessions, toolsReg)
server.SetVersion(Version)
server.SetDB(pgStores.DB)
server.SetPolicyEngine(permPE)
server.SetPairingService(pgStores.Pairing)
server.SetMessageBus(msgBus)
server.SetOAuthHandler(httpapi.NewOAuthHandler(pgStores.Providers, pgStores.ConfigSecrets, providerRegistry, msgBus))
// contextFileInterceptor is created inside wireExtras.
// Declared here so it can be passed to registerAllMethods → AgentsMethods
// for immediate cache invalidation on agents.files.set.
var contextFileInterceptor *tools.ContextFileInterceptor
// Set agent store for tools_invoke context injection + wire extras
if pgStores.Agents != nil {
server.SetAgentStore(pgStores.Agents)
}
var mcpPool *mcpbridge.Pool
var mediaStore *media.Store
var postTurn tools.PostTurnProcessor
contextFileInterceptor, mcpPool, mediaStore, postTurn = wireExtras(pgStores, agentRouter, providerRegistry, modelReg, msgBus, pgStores.Sessions, toolsReg, toolPE, skillsLoader, hasMemory, traceCollector, workspace, cfg.Gateway.InjectionAction, cfg, sandboxMgr, redisClient, domainBus, usageCapSvc)
if mcpPool != nil {
defer mcpPool.Stop()
}
// Populate shared deps struct used by extracted helper methods.
deps := &gatewayDeps{
cfg: cfg,
server: server,
msgBus: msgBus,
pgStores: pgStores,
providerRegistry: providerRegistry,
agentRouter: agentRouter,
toolsReg: toolsReg,
skillsLoader: skillsLoader,
enrichProgress: enrichProgress,
enrichWorker: enrichWorker,
workspace: workspace,
dataDir: dataDir,
domainBus: domainBus,
usageCapSvc: usageCapSvc,
audioMgr: audioMgr,
}
gatewayAddr := loopbackAddr(cfg.Gateway.Host, cfg.Gateway.Port)
var mcpToolLister httpapi.MCPToolLister
if mcpMgr != nil {
mcpToolLister = mcpMgr
}
httpapi.InitGatewayToken(cfg.Gateway.Token)
httpapi.InitGatewayNoAuthFallbackAllowed(config.GatewayNoAuthFallbackAllowed(cfg.Gateway))
exportTokenStore := httpapi.InitExportTokenStore()
defer exportTokenStore.Stop()
agentsH, skillsH, tracesH, mcpH, channelInstancesH, providersH, builtinToolsH, pendingMessagesH, teamEventsH, secureCLIH, secureCLIGrantH, mcpUserCredsH := wireHTTP(pgStores, cfg.Agents.Defaults.Workspace, dataDir, bundledSkillsDir, msgBus, domainBus, toolsReg, providerRegistry, modelReg, permPE.IsOwner, gatewayAddr, mcpToolLister, usageCapSvc, cfg, cfg.Skills)
// Wire dependencies for system prompt preview parity.
if agentsH != nil {
agentsH.SetPreviewDeps(toolsReg, skillsLoader)
var skillAccess store.SkillAccessStore
if pgStores.Skills != nil {
skillAccess, _ = pgStores.Skills.(store.SkillAccessStore)
}
agentsH.SetPreviewStores(pgStores.Teams, pgStores.AgentLinks, skillAccess)
}
// External wake/trigger API
wakeH := httpapi.NewWakeHandler(agentRouter)
if postTurn != nil {
wakeH.SetPostTurnProcessor(postTurn)
}
// Wire all server.Set*Handler() calls via extracted helper.
deps.wireHTTPHandlersOnServer(
httpHandlers{
agents: agentsH,
skills: skillsH,
traces: tracesH,
mcp: mcpH,
channelInstances: channelInstancesH,
providers: providersH,
builtinTools: builtinToolsH,
pendingMessages: pendingMessagesH,
teamEvents: teamEventsH,
secureCLI: secureCLIH,
secureCLIGrant: secureCLIGrantH,
mcpUserCreds: mcpUserCredsH,
},
wakeH,
mcpPool,
postTurn,
mediaStore,
)
// System backup API — admin + owner only, SSE progress streaming.
server.SetBackupHandler(httpapi.NewBackupHandler(cfg, cfg.Database.PostgresDSN, Version, permPE.IsOwner))
// System restore API — admin + owner only, multipart upload + SSE progress.
server.SetRestoreHandler(httpapi.NewRestoreHandler(cfg, cfg.Database.PostgresDSN, permPE.IsOwner))
// S3 backup integration — admin + owner only.
server.SetBackupS3Handler(httpapi.NewBackupS3Handler(cfg, cfg.Database.PostgresDSN, Version, pgStores.ConfigSecrets, permPE.IsOwner))
// Tenant-scoped backup/restore — owner or tenant admin.
if pgStores.Tenants != nil {
server.SetTenantBackupHandler(httpapi.NewTenantBackupHandler(pgStores.DB, cfg, pgStores.Tenants, Version, permPE.IsOwner))
}
// Register all RPC methods
server.SetLogTee(logTee)
server.SetRuntimeLogsHandler(httpapi.NewRuntimeLogsHandler(logTee))
pairingMethods, heartbeatMethods, chatMethods, cfgPermsMethods := registerAllMethods(server, agentRouter, pgStores.Sessions, pgStores.RunTimeline, pgStores.Cron, pgStores.Pairing, cfg, cfgPath, workspace, dataDir, msgBus, execApprovalMgr, pgStores.Agents, pgStores.Skills, pgStores.ConfigSecrets, pgStores.Teams, contextFileInterceptor, logTee, pgStores.Heartbeats, pgStores.ConfigPermissions, pgStores.SystemConfigs, pgStores.Tenants, pgStores.SkillTenantCfgs, audioMgr, usageCapSvc)
// Phase 3: Agent hooks RPC methods (hooks.list/create/update/delete/toggle/test/history).
if hs, ok := pgStores.Hooks.(hooks.HookStore); ok && hs != nil {
hm := methods.NewHookMethods(hs, edition.Current())
// Reuse dispatcher handlers for dry-run test runner so UI test panel
// exercises the exact code that will run in production.
if sharedHookHandlers != nil {
hm.SetTestRunner(methods.NewDispatcherTestRunner(sharedHookHandlers))
}
hm.Register(server.Router())
slog.Info("registered hooks RPC methods")
}
// Workstations WS methods — Standard edition only.
// Lite (desktop/SQLite) must NOT expose workstation RPC methods.
if edition.Current().Name != "lite" && pgStores.Workstations != nil && pgStores.WorkstationLinks != nil {
wsMethods := methods.NewWorkstationsMethods(pgStores.Workstations, pgStores.WorkstationLinks)
if pgStores.WorkstationPermissions != nil {
wsMethods.SetPermStore(pgStores.WorkstationPermissions)
}
if pgStores.WorkstationActivity != nil {
wsMethods.SetActivityStore(pgStores.WorkstationActivity)
}
wsMethods.Register(server.Router())
slog.Info("registered workstations RPC methods")
}
// Wire post-turn processor for team task dispatch (WS chat.send + HTTP API paths).
if postTurn != nil {
chatMethods.SetPostTurnProcessor(postTurn)
server.SetPostTurnProcessor(postTurn) // HTTP: /v1/chat/completions, /v1/responses
wakeH.SetPostTurnProcessor(postTurn) // HTTP: /v1/agents/{id}/wake
}
// Wire pairing event broadcasts to all WS clients.
pairingMethods.SetBroadcaster(server.BroadcastEvent)
// Wire pairing request callback — works for both PG and SQLite stores.
type pairingRequestNotifier interface {
SetOnRequest(func(code, senderID, channel, chatID string))
}
if ps, ok := pgStores.Pairing.(pairingRequestNotifier); ok {
ps.SetOnRequest(func(code, senderID, channel, chatID string) {
server.BroadcastEvent(*protocol.NewEvent(protocol.EventDevicePairReq, map[string]any{
"code": code, "sender_id": senderID, "channel": channel, "chat_id": chatID,
}))
})
}
// Channel manager
channelMgr := channels.NewManager(msgBus)
deps.channelMgr = channelMgr
// Wire channel member resolver into permission grant paths (WS + HTTP) so
// file_writer grants coming from the Web UI auto-enrich their metadata.
cfgPermsMethods.SetMemberResolver(channelMgr)
if channelInstancesH != nil {
channelInstancesH.SetMemberResolver(channelMgr)
// Setter (not constructor) because wireHTTP runs before channelMgr is
// created — required for handleDelete to invoke ChannelDestroyer on
// Bitrix24 channels (imbot.unregister bot cleanup).
channelInstancesH.SetChannelManager(channelMgr)
}
// Wire channel sender + tenant checker on message tool (now that channelMgr exists)
if t, ok := toolsReg.Get("message"); ok {
if cs, ok := t.(tools.ChannelSenderAware); ok {
cs.SetChannelSender(channelMgr.SendToChannel)
}
if tc, ok := t.(tools.ChannelTenantCheckerAware); ok {
tc.SetChannelTenantChecker(channelMgr.ChannelTenantID)
}
}
// Wire group member lister on list_group_members tool
if t, ok := toolsReg.Get("list_group_members"); ok {
if gl, ok := t.(tools.GroupMemberListerAware); ok {
gl.SetGroupMemberLister(channelMgr.ListGroupMembers)
}
}
// Load channel instances from DB.
var instanceLoader *channels.InstanceLoader
if pgStores.ChannelInstances != nil {
instanceLoader = channels.NewInstanceLoader(pgStores.ChannelInstances, pgStores.Agents, channelMgr, msgBus, pgStores.Pairing)
instanceLoader.SetProviderRegistry(providerRegistry)
instanceLoader.SetPendingCompactionConfig(cfg.Channels.PendingCompaction)
instanceLoader.SetUsageCapService(usageCapSvc)
instanceLoader.RegisterFactory(channels.TypeTelegram, telegram.FactoryWithStoresAndAudio(pgStores.Agents, pgStores.ConfigPermissions, pgStores.Teams, pgStores.SubagentTasks, pgStores.PendingMessages, audioMgr))
instanceLoader.RegisterFactory(channels.TypeDiscord, discord.FactoryWithStoresAndAudio(pgStores.Agents, pgStores.ConfigPermissions, pgStores.PendingMessages, audioMgr))
instanceLoader.RegisterFactory(channels.TypeFeishu, feishu.FactoryWithPendingStoreAndAudio(pgStores.PendingMessages, audioMgr))
instanceLoader.RegisterFactory(channels.TypeZaloOA, zalo.Factory)
instanceLoader.RegisterFactory(channels.TypeZaloPersonal, zalopersonal.FactoryWithPendingStore(pgStores.PendingMessages))
instanceLoader.RegisterFactory(channels.TypeWhatsApp, whatsapp.FactoryWithDBAudio(pgStores.DB, pgStores.PendingMessages, "pgx", audioMgr, pgStores.BuiltinTools))
instanceLoader.RegisterFactory(channels.TypeSlack, slackchannel.FactoryWithPendingStore(pgStores.PendingMessages))
instanceLoader.RegisterFactory(channels.TypeFacebook, facebook.Factory)
instanceLoader.RegisterFactory(channels.TypePancake, pancake.Factory)
// Bitrix24: factory needs the portal store + encKey injected so each
// Channel can resolve its portal on Start(). The encKey here mirrors
// the one used by pg.NewPGStores → NewPGBitrixPortalStore.
bitrixEncKey := os.Getenv("GOCLAW_ENCRYPTION_KEY")
// Use the MCP-aware factory variant so channels that opt into
// lazy per-user credential provisioning (via mcp_server_name +
// mcp_base_url in their instance config) can reach the partner's
// MCPServerStore. The MCP server authenticates each onboard call
// via the caller-supplied Bitrix access_token (Path B) — no shared
// admin secret is required. Channels with none of those set operate
// identically to before — the MCPStore arg is nil-safe inside the
// factory.
instanceLoader.RegisterFactory(channels.TypeBitrix24, bitrix24.FactoryWithPortalStoreAndMCP(pgStores.BitrixPortals, pgStores.MCP, bitrixEncKey))
if err := instanceLoader.LoadAll(context.Background()); err != nil {
slog.Error("failed to load channel instances from DB", "error", err)
}
// Bitrix24 portal management RPC (self-service onboarding).
// Registers bitrix.portals.list/create/get_install_url/delete methods
// on the WS router; install URL is built from the gateway's observed
// public URL via Server.PublicURLSnapshot().
if pgStores.BitrixPortals != nil {
methods.NewBitrixPortalsMethods(
pgStores.BitrixPortals,
pgStores.ChannelInstances,
server.PublicURLSnapshot().Get,
).Register(server.Router())
}
// Warm the shared Bitrix24 router with every portal row so inbound
// webhooks land on the right *Portal even before a channel instance
// is loaded for that portal. Idempotent; no-op on sqlite-lite.
if pgStores.BitrixPortals != nil {
if err := bitrix24.BootstrapPortals(context.Background(), pgStores.BitrixPortals, bitrixEncKey); err != nil {
// Surface the missing-table case loudly so an operator notices
// without having to grep logs — bitrix24 channels silently
// no-op until `goclaw migrate up` runs migration 000058.
if strings.Contains(err.Error(), "bitrix_portals") &&
(strings.Contains(err.Error(), "does not exist") || strings.Contains(err.Error(), "no such table")) {
slog.Warn("bitrix24 bootstrap skipped — bitrix_portals table missing; run `goclaw migrate up` (migration 000068) to enable Bitrix24 channels",
"err", err)
} else {
slog.Warn("bitrix24 bootstrap failed", "err", err)
}
}
}
}
// Register config-based channels as fallback when no DB instances loaded.
registerConfigChannels(cfg, channelMgr, msgBus, pgStores, instanceLoader, audioMgr)
// Register channels/instances/links/teams RPC methods
chInstancesM := wireChannelRPCMethods(server, pgStores, channelMgr, instanceLoader, agentRouter, msgBus, cfg, workspace)
// Bitrix24 orphan-bot cleaner. Fires from channel_instances delete handler
// when the channel is no longer loaded in the Manager (typical scenario:
// admin disabled the channel earlier so InstanceLoader.Reload removed it).
// Without this, deleting a disabled Bitrix24 channel would orphan the bot
// on the portal.
if pgStores.BitrixPortals != nil {
bitrixEncKey := os.Getenv("GOCLAW_ENCRYPTION_KEY")
orphanCleaner := func(ctx context.Context, tenantID uuid.UUID, cfg []byte) error {
return bitrix24.DestroyOrphanBot(ctx, pgStores.BitrixPortals, bitrixEncKey, tenantID, cfg)
}
if channelInstancesH != nil {
channelInstancesH.RegisterOrphanCleaner(channels.TypeBitrix24, orphanCleaner)
}
if chInstancesM != nil {
chInstancesM.RegisterOrphanCleaner(channels.TypeBitrix24, orphanCleaner)
}
}
// Wire channel event subscribers (cache invalidation, pairing, cascade disable)
wireChannelEventSubscribers(msgBus, server, pgStores, channelMgr, instanceLoader, pairingMethods, cfg)
// Audit log subscriber + team task event subscribers.
auditCh := deps.wireAuditSubscriber()
deps.wireEventSubscribers()
// Setup graceful shutdown
ctx, cancel := context.WithCancel(context.Background())
defer cancel()
server.StartUpdateChecker(ctx)
sigCh := make(chan os.Signal, 1)
signal.Notify(sigCh, syscall.SIGINT, syscall.SIGTERM)
// Skills directory watcher — auto-detect new/removed/modified skills at runtime.
if skillsWatcher, err := skills.NewWatcher(skillsLoader); err != nil {
slog.Warn("skills watcher unavailable", "error", err)
} else {
if err := skillsWatcher.Start(ctx); err != nil {
slog.Warn("skills watcher start failed", "error", err)
} else {
defer skillsWatcher.Stop()
}
}
// Start channels
if err := channelMgr.StartAll(ctx); err != nil {
slog.Error("failed to start channels", "error", err)
}
// Create lane-based scheduler (matching TS CommandLane pattern).
// Must be created before cron setup so cron jobs route through the scheduler.
sched := scheduler.NewScheduler(
scheduler.DefaultLanes(),
scheduler.DefaultQueueConfig(),
makeSchedulerRunFunc(agentRouter, cfg),
)
defer sched.Stop()
// Start cron + heartbeat ticker, wire wake functions and adaptive throttle.
heartbeatTicker := startCronAndHeartbeat(pgStores, server, sched, msgBus, providerRegistry, channelMgr, cfg, heartbeatTool, heartbeatMethods)
// Subscribe to agent events for channel streaming/reaction forwarding.
deps.wireChannelStreamingSubscriber()
// Slow tool notification subscriber — direct outbound when tool exceeds adaptive threshold.
wireSlowToolNotifySubscriber(msgBus)
// Inbound message consumer setup
consumerTeamStore := pgStores.Teams
// Quota checker: enforces per-user/group request limits.
config.MergeChannelGroupQuotas(cfg)
var quotaChecker *channels.QuotaChecker
if cfg.Gateway.Quota != nil && cfg.Gateway.Quota.Enabled {
quotaChecker = channels.NewQuotaChecker(pgStores.DB, *cfg.Gateway.Quota)
defer quotaChecker.Stop()
slog.Info("channel quota enabled",
"default_hour", cfg.Gateway.Quota.Default.Hour,
"default_day", cfg.Gateway.Quota.Default.Day,
"default_week", cfg.Gateway.Quota.Default.Week,
)
}
// Register quota usage RPC.
methods.NewQuotaMethods(quotaChecker, pgStores.DB).Register(server.Router())
// API key management RPC
if pgStores.APIKeys != nil {
methods.NewAPIKeysMethods(pgStores.APIKeys).Register(server.Router())
}
// Tenant management RPC + HTTP
if pgStores.Tenants != nil {
methods.NewTenantsMethods(pgStores.Tenants, msgBus, workspace).Register(server.Router())
server.SetTenantsHandler(httpapi.NewTenantsHandler(pgStores.Tenants, msgBus, workspace))
server.Router().SetTenantStore(pgStores.Tenants)
// Permission cache for tenant membership checks. Store on deps so
// lifecycle shutdown can call Close() to stop the sweep goroutines.
permCache := cache.NewPermissionCache()
deps.permCache = permCache
msgBus.Subscribe("permission-cache", func(e bus.Event) {
if p, ok := e.Payload.(bus.CacheInvalidatePayload); ok {
permCache.HandleInvalidation(p)
}
})
server.Router().SetPermissionCache(permCache)
httpapi.InitTenantStore(pgStores.Tenants, msgBus)
httpapi.InitOwnerIDs(cfg.Gateway.OwnerIDs)
}
// Wire lifecycle: config-reload subscribers, consumer, task recovery, shutdown, server start.
deps.runLifecycle(ctx, cancel, lifecycleDeps{
sched: sched,
heartbeatTicker: heartbeatTicker,
quotaChecker: quotaChecker,
webFetchTool: webFetchTool,
ttsTool: ttsTool,
sandboxMgr: sandboxMgr,
postTurn: postTurn,
subagentMgr: subagentMgr,
consumerTeamStore: consumerTeamStore,
auditCh: auditCh,
sigCh: sigCh,
})
}
// resolveBackgroundProvider picks the LLM provider+model for background workers
// (vault enrichment, consolidation). Fallback chain:
//
// background.provider/model → agent.default_provider/model → first registered provider.
func resolveBackgroundProvider(cfg *config.Config, reg *providers.Registry) (providers.Provider, string) {
try := func(name, model string) (providers.Provider, string, bool) {
if name == "" {
return nil, "", false
}
p, err := reg.GetForTenant(providers.MasterTenantID, name)
if err != nil || p == nil {
return nil, "", false
}
if model == "" {
model = p.DefaultModel()
}
return p, model, true
}
// 1. Explicit background config
if p, m, ok := try(cfg.Gateway.BackgroundProvider, cfg.Gateway.BackgroundModel); ok {
return p, m
}
// 2. Agent default provider
if p, m, ok := try(cfg.Agents.Defaults.Provider, cfg.Agents.Defaults.Model); ok {
return p, m
}
// 3. First registered provider (legacy fallback)
if names := reg.ListForTenant(providers.MasterTenantID); len(names) > 0 {
if p, m, ok := try(names[0], ""); ok {
return p, m
}
}
return nil, ""
}