goclaw/internal/channels/telegram/factory.go

package telegram

import (
	"encoding/json"
	"fmt"

	"github.com/nextlevelbuilder/goclaw/internal/bus"
	"github.com/nextlevelbuilder/goclaw/internal/channels"
	"github.com/nextlevelbuilder/goclaw/internal/config"
	"github.com/nextlevelbuilder/goclaw/internal/store"
)

// telegramCreds maps the credentials JSON from the channel_instances table.
type telegramCreds struct {
	Token     string `json:"token"`
	Proxy     string `json:"proxy,omitempty"`
	APIServer string `json:"api_server,omitempty"`
}

// telegramInstanceConfig maps the non-secret config JSONB from the channel_instances table.
type telegramInstanceConfig struct {
	APIServer       string   `json:"api_server,omitempty"`
	Proxy           string   `json:"proxy,omitempty"`
	DMPolicy        string   `json:"dm_policy,omitempty"`
	GroupPolicy     string   `json:"group_policy,omitempty"`
	RequireMention  *bool    `json:"require_mention,omitempty"`
	HistoryLimit    int      `json:"history_limit,omitempty"`
	DMStream        *bool    `json:"dm_stream,omitempty"`
	GroupStream     *bool    `json:"group_stream,omitempty"`
	DraftTransport  *bool    `json:"draft_transport,omitempty"`   // sendMessageDraft for DM streaming (default true)
	ReasoningStream *bool    `json:"reasoning_stream,omitempty"` // show reasoning as separate message (default true)
	ReactionLevel   string   `json:"reaction_level,omitempty"`
	MediaMaxMB      int64    `json:"media_max_mb,omitempty"`
	MediaMaxBytes   int64    `json:"media_max_bytes,omitempty"` // deprecated: use media_max_mb
	LinkPreview     *bool    `json:"link_preview,omitempty"`
	BlockReply      *bool    `json:"block_reply,omitempty"`
	ForceIPv4       bool     `json:"force_ipv4,omitempty"`
	AllowFrom       []string `json:"allow_from,omitempty"`
}

// Factory creates a Telegram channel from DB instance data (no extra stores).
func Factory(name string, creds json.RawMessage, cfg json.RawMessage,
	msgBus *bus.MessageBus, pairingSvc store.PairingStore) (channels.Channel, error) {
	return buildChannel(name, creds, cfg, msgBus, pairingSvc, nil, nil, nil, nil)
}

// FactoryWithStores returns a ChannelFactory that includes agent, configPerm, team, and pending message stores.
func FactoryWithStores(agentStore store.AgentStore, configPermStore store.ConfigPermissionStore, teamStore store.TeamStore, pendingStore store.PendingMessageStore) channels.ChannelFactory {
	return func(name string, creds json.RawMessage, cfg json.RawMessage,
		msgBus *bus.MessageBus, pairingSvc store.PairingStore) (channels.Channel, error) {
		return buildChannel(name, creds, cfg, msgBus, pairingSvc, agentStore, configPermStore, teamStore, pendingStore)
	}
}

func buildChannel(name string, creds json.RawMessage, cfg json.RawMessage,
	msgBus *bus.MessageBus, pairingSvc store.PairingStore, agentStore store.AgentStore, configPermStore store.ConfigPermissionStore, teamStore store.TeamStore, pendingStore store.PendingMessageStore) (channels.Channel, error) {

	var c telegramCreds
	if len(creds) > 0 {
		if err := json.Unmarshal(creds, &c); err != nil {
			return nil, fmt.Errorf("decode telegram credentials: %w", err)
		}
	}
	if c.Token == "" {
		return nil, fmt.Errorf("telegram token is required")
	}

	var ic telegramInstanceConfig
	if len(cfg) > 0 {
		if err := json.Unmarshal(cfg, &ic); err != nil {
			return nil, fmt.Errorf("decode telegram config: %w", err)
		}
	}

	// Prefer config values; fall back to credentials for backward compat.
	proxy := ic.Proxy
	if proxy == "" {
		proxy = c.Proxy
	}
	apiServer := ic.APIServer
	if apiServer == "" {
		apiServer = c.APIServer
	}

	tgCfg := config.TelegramConfig{
		Enabled:        true,
		Token:          c.Token,
		Proxy:          proxy,
		APIServer:      apiServer,
		AllowFrom:      ic.AllowFrom,
		DMPolicy:       ic.DMPolicy,
		GroupPolicy:    ic.GroupPolicy,
		RequireMention: ic.RequireMention,
		HistoryLimit:   ic.HistoryLimit,
		DMStream:        ic.DMStream,
		GroupStream:     ic.GroupStream,
		DraftTransport:  ic.DraftTransport,
		ReasoningStream: ic.ReasoningStream,
		ReactionLevel:   ic.ReactionLevel,
		MediaMaxBytes:  resolveMediaMaxBytes(ic),
		LinkPreview:    ic.LinkPreview,
		BlockReply:     ic.BlockReply,
		ForceIPv4:      ic.ForceIPv4,
	}

	// DB instances default to "pairing" for groups (secure by default).
	// Config-based channels keep "open" default for backward compat.
	if tgCfg.GroupPolicy == "" {
		tgCfg.GroupPolicy = "pairing"
	}

	ch, err := New(tgCfg, msgBus, pairingSvc, agentStore, configPermStore, teamStore, pendingStore)
	if err != nil {
		return nil, err
	}

	// Override the channel name from DB instance.
	ch.SetName(name)
	return ch, nil
}

// resolveMediaMaxBytes converts media_max_mb (preferred) to bytes,
// falling back to the deprecated media_max_bytes for backward compat.
func resolveMediaMaxBytes(ic telegramInstanceConfig) int64 {
	if ic.MediaMaxMB > 0 {
		return ic.MediaMaxMB * 1024 * 1024
	}
	return ic.MediaMaxBytes
}