From a4c2fc4b37849d9157bf873da38500b8c171437f Mon Sep 17 00:00:00 2001
From: Steve <stryan@saintnet.tech>
Date: Fri, 25 Jun 2021 14:05:57 -0400
Subject: [PATCH] make statefile configurable

---
 config.yaml.sample | 1 +
 main.go            | 5 ++++-
 2 files changed, 5 insertions(+), 1 deletion(-)

diff --git a/config.yaml.sample b/config.yaml.sample
index 1ee8ce2..8428238 100644
--- a/config.yaml.sample
+++ b/config.yaml.sample
@@ -3,6 +3,7 @@ username: SimpBot
 password: SimpQAhoy
 dimension: dimension.saintnet.tech
 domain: saintnet.tech
+statefile: statefile
 vtubers:
         - name: Ina
           channelid: UCMwGHR0BTZuLsmjY_NT5Pwg
diff --git a/main.go b/main.go
index 652763e..444e484 100644
--- a/main.go
+++ b/main.go
@@ -17,6 +17,7 @@ var Password string
 var DimensionServer string
 var HomeserverDomain string
 var GitCommit string
+var Statefile string
 
 func main() {
 	viper.SetConfigName("config")
@@ -28,17 +29,19 @@ func main() {
 	viper.SetConfigType("yaml")
 	Homeserver = viper.GetString("homeserver")
 	viper.SetDefault("domain", Homeserver)
+	viper.SetDefault("statefile", "simpstate")
 	Username = viper.GetString("username")
 	Password = viper.GetString("password")
 	DimensionServer = viper.GetString("dimension")
 	HomeserverDomain = viper.GetString("domain")
+	Statefile = viper.GetString("statefile")
 	var vtubers []*Vtuber
 	log.Println("Logging into", Homeserver, "as", Username)
 	client, err := mautrix.NewClient(Homeserver, "", "")
 	if err != nil {
 		panic(err)
 	}
-	client.Store = NewLazyMemStore("statedata")
+	client.Store = NewLazyMemStore(Statefile)
 	_, err = client.Login(&mautrix.ReqLogin{
 		Type:             "m.login.password",
 		Identifier:       mautrix.UserIdentifier{Type: mautrix.IdentifierTypeUser, User: Username},