mautrix-discord/pkg/connector/config.go

// mautrix-discord - A Matrix-Discord puppeting bridge.
// Copyright (C) 2026 Tulir Asokan
//
// This program is free software: you can redistribute it and/or modify
// it under the terms of the GNU Affero General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.
//
// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
// GNU Affero General Public License for more details.
//
// You should have received a copy of the GNU Affero General Public License
// along with this program.  If not, see <https://www.gnu.org/licenses/>.

package connector

import (
	_ "embed"
	"strings"
	"text/template"

	"github.com/bwmarrin/discordgo"
	up "go.mau.fi/util/configupgrade"
	"gopkg.in/yaml.v3"
)

//go:embed example-config.yaml
var ExampleConfig string

const defaultChannelNameTemplate = `{{if and .IsGuildChannel (not .IsCategory)}}#{{end}}{{.Name}}`

type Config struct {
	Guilds struct {
		BridgingGuildIDs []string `yaml:"bridging_guild_ids"`
	} `yaml:"guilds"`

	// ChannelNameTemplate formats Matrix room names for Discord channels other
	// than 1:1 DMs, which intentionally use bridgev2's ghost-derived default.
	ChannelNameTemplate  string `yaml:"channel_name_template"`
	CustomEmojiReactions *bool  `yaml:"custom_emoji_reactions"`
	GuildAvatarsInRooms  *bool  `yaml:"guild_avatars_in_rooms"`

	ForbidDMingStrangers *bool `yaml:"forbid_dming_strangers"`

	LogWhenDroppingMessages bool `yaml:"log_when_dropping_messages"`

	channelNameTemplate *template.Template `yaml:"-"`
}

type umConfig Config

func (c *Config) UnmarshalYAML(node *yaml.Node) error {
	err := node.Decode((*umConfig)(c))
	if err != nil {
		return err
	}

	if c.ChannelNameTemplate == "" {
		c.ChannelNameTemplate = defaultChannelNameTemplate
	}

	c.channelNameTemplate, err = template.New("channel_name").Parse(c.ChannelNameTemplate)
	if err != nil {
		return err
	}

	return nil
}

// ChannelNameParams describes the values available to [Config.FormatChannelName].
//
// It intentionally includes both the raw Discord channel type and convenience
// booleans so templates can express v1-style naming rules without relying on
// numeric channel type constants.
type ChannelNameParams struct {
	Name           string
	ParentName     string
	GuildName      string
	Type           discordgo.ChannelType
	NSFW           bool
	IsDM           bool
	IsGroupDM      bool
	IsCategory     bool
	IsGuildChannel bool
}

// FormatChannelName renders [Config.ChannelNameTemplate] for non-guild-space
// channel portals. One-to-one DMs intentionally bypass this helper so bridgev2
// can derive the room name from the other user's ghost.
func (c *Config) FormatChannelName(params *ChannelNameParams) string {
	var buffer strings.Builder
	_ = c.channelNameTemplate.Execute(&buffer, params)
	return buffer.String()
}

func (c Config) ForbidDMingStrangersEnabled() bool {
	return c.ForbidDMingStrangers == nil || *c.ForbidDMingStrangers
}

func (c Config) CustomEmojiReactionsEnabled() bool {
	return c.CustomEmojiReactions == nil || *c.CustomEmojiReactions
}

func (c Config) GuildAvatarsInRoomsEnabled() bool {
	return c.GuildAvatarsInRooms != nil && *c.GuildAvatarsInRooms
}

func upgradeConfig(helper up.Helper) {
	helper.Copy(up.List, "guilds", "bridging_guild_ids")
	helper.Copy(up.Bool, "guilds", "guild_avatars_in_rooms")
	helper.Copy(up.Bool, "forbid_dming_strangers")
	helper.Copy(up.Str, "channel_name_template")
	helper.Copy(up.Bool, "custom_emoji_reactions")
	helper.Copy(up.Bool, "log_when_dropping_messages")
}

func (d *DiscordConnector) GetConfig() (example string, data any, upgrader up.Upgrader) {
	return ExampleConfig, &d.Config, up.SimpleUpgrader(upgradeConfig)
}