Basic provisioning api

Includes: login, logout, and ping.

config/appservice.go

@@ -13,6 +13,8 @@ type appservice struct {
 
 	Bot bot `yaml:"bot"`
 
+	Provisioning provisioning `yaml:"provisioning"`
+
 	Database database `yaml:"database"`
 
 	ASToken string `yaml:"as_token"`

config/config.go

@@ -1,6 +1,7 @@
 package config
 
 import (
+	"fmt"
 	"io/ioutil"
 
 	"gopkg.in/yaml.v2"
@@ -11,8 +12,12 @@ type Config struct {
 	Appservice appservice `yaml:"appservice"`
 	Bridge     bridge     `yaml:"bridge"`
 	Logging    logging    `yaml:"logging"`
+
+	filename string `yaml:"-"`
 }
 
+var configUpdated bool
+
 func (cfg *Config) validate() error {
 	if err := cfg.Homeserver.validate(); err != nil {
 		return err
@@ -30,13 +35,20 @@ func (cfg *Config) validate() error {
 		return err
 	}
 
+	if configUpdated {
+		return cfg.Save(cfg.filename)
+	}
+
 	return nil
 }
 
 func (cfg *Config) UnmarshalYAML(unmarshal func(interface{}) error) error {
 	type rawConfig Config
 
-	raw := rawConfig{}
+	raw := rawConfig{
+		filename: cfg.filename,
+	}
+
 	if err := unmarshal(&raw); err != nil {
 		return err
 	}
@@ -46,8 +58,10 @@ func (cfg *Config) UnmarshalYAML(unmarshal func(interface{}) error) error {
 	return cfg.validate()
 }
 
-func FromBytes(data []byte) (*Config, error) {
-	cfg := Config{}
+func FromBytes(filename string, data []byte) (*Config, error) {
+	cfg := Config{
+		filename: filename,
+	}
 
 	if err := yaml.Unmarshal(data, &cfg); err != nil {
 		return nil, err
@@ -61,7 +75,7 @@ func FromBytes(data []byte) (*Config, error) {
 }
 
 func FromString(str string) (*Config, error) {
-	return FromBytes([]byte(str))
+	return FromBytes("", []byte(str))
 }
 
 func FromFile(filename string) (*Config, error) {
@@ -70,10 +84,14 @@ func FromFile(filename string) (*Config, error) {
 		return nil, err
 	}
 
-	return FromBytes(data)
+	return FromBytes(filename, data)
 }
 
 func (cfg *Config) Save(filename string) error {
+	if filename == "" {
+		return fmt.Errorf("no filename specified yep")
+	}
+
 	data, err := yaml.Marshal(cfg)
 	if err != nil {
 		return err

config/provisioning.go

@@ -0,0 +1,43 @@
+package config
+
+import (
+	"strings"
+
+	as "maunium.net/go/mautrix/appservice"
+)
+
+type provisioning struct {
+	Prefix       string `yaml:"prefix"`
+	SharedSecret string `yaml:"shared_secret"`
+}
+
+func (p *provisioning) validate() error {
+	if p.Prefix == "" {
+		p.Prefix = "/_matrix/provision/v1"
+	}
+
+	if strings.ToLower(p.SharedSecret) == "generate" {
+		p.SharedSecret = as.RandomString(64)
+
+		configUpdated = true
+	}
+
+	return nil
+}
+
+func (p *provisioning) UnmarshalYAML(unmarshal func(interface{}) error) error {
+	type rawProvisioning provisioning
+
+	raw := rawProvisioning{}
+	if err := unmarshal(&raw); err != nil {
+		return err
+	}
+
+	*p = provisioning(raw)
+
+	return p.validate()
+}
+
+func (p *provisioning) Enabled() bool {
+	return strings.ToLower(p.SharedSecret) != "disable"
+}