55 lines
852 B
Go
55 lines
852 B
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()
|
|
}
|