mautrix-whatsapp/config/bridge.go

// mautrix-whatsapp - A Matrix-WhatsApp puppeting bridge.
// Copyright (C) 2018 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 config

import (
	"bytes"
	"text/template"
)

type BridgeConfig struct {
	UsernameTemplate    string             `yaml:"username_template"`
	DisplaynameTemplate string             `yaml:"displayname_template"`
	usernameTemplate    *template.Template `yaml:"-"`
	displaynameTemplate *template.Template `yaml:"-"`
}

type umBridgeConfig BridgeConfig

func (bc *BridgeConfig) UnmarshalYAML(unmarshal func(interface{}) error) error {
	err := unmarshal((*umBridgeConfig)(bc))
	if err != nil {
		return err
	}

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

	bc.displaynameTemplate, err = template.New("displayname").Parse(bc.DisplaynameTemplate)
	return err
}

type DisplaynameTemplateArgs struct {
	Displayname string
}

type UsernameTemplateArgs struct {
	Receiver string
	UserID   string
}

func (bc BridgeConfig) FormatDisplayname(displayname string) string {
	var buf bytes.Buffer
	bc.displaynameTemplate.Execute(&buf, DisplaynameTemplateArgs{
		Displayname: displayname,
	})
	return buf.String()
}

func (bc BridgeConfig) FormatUsername(receiver, userID string) string {
	var buf bytes.Buffer
	bc.usernameTemplate.Execute(&buf, UsernameTemplateArgs{
		Receiver: receiver,
		UserID:   userID,
	})
	return buf.String()
}

func (bc BridgeConfig) MarshalYAML() (interface{}, error) {
	bc.DisplaynameTemplate = bc.FormatDisplayname("{{.Displayname}}")
	bc.UsernameTemplate = bc.FormatUsername("{{.Receiver}}", "{{.UserID}}")
	return bc, nil
}