mautrix-discord/config/bridge.go

package config

import (
	"fmt"

	"bytes"
	"text/template"
)

type bridge struct {
	UsernameTemplate string `yaml:"username_template"`

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

func (b *bridge) setDefaults() error {
	var err error

	if b.UsernameTemplate == "" {
		b.UsernameTemplate = "Discord_{{.}}"
	}

	b.usernameTemplate, err = template.New("username").Parse(b.UsernameTemplate)
	if err != nil {
		return err
	}

	return nil
}

func (b *bridge) UnmarshalYAML(unmarshal func(interface{}) error) error {
	type rawBridge bridge

	raw := rawBridge{}

	err := unmarshal(&raw)
	if err != nil {
		return err
	}

	*b = bridge(raw)

	return b.setDefaults()
}

func (b bridge) FormatUsername(userid string) string {
	var buffer bytes.Buffer

	fmt.Printf("bridge: %#v\n", b)

	b.usernameTemplate.Execute(&buffer, userid)

	return buffer.String()
}