Support systemd dynamic user

1 files changed, 5 insertions, 7 deletions

diff --git a/roseh.moe.go b/roseh.moe.go
index 681ab8e..033c6c1 100644
--- a/roseh.moe.go
+++ b/roseh.moe.go
@@ -21,9 +21,7 @@ import (
 )
 
 var (
-	port        = flag.Int("port", 42069, "port to listen on")
-	secretsFile = flag.String("secrets", "secrets", "path to the secrets file")
-	notepadFile = flag.String("notepad", "notepad", "path to save user notes")
+	port = flag.Int("port", 42069, "port to listen on")
 
 	serverStartTime = time.Now()
 )
@@ -34,7 +32,7 @@ var (
 )
 
 func loadSecrets() error {
-	secrets, err := os.ReadFile(*secretsFile)
+	secrets, err := os.ReadFile(filepath.Join(os.Getenv("CONFIGURATION_DIRECTORY"), "secrets"))
 	if err != nil {
 		return err
 	}
@@ -195,7 +193,7 @@ var (
 
 func notepad(w http.ResponseWriter, r *http.Request) {
 	var currentContent string
-	if currentContentBytes, err := os.ReadFile(*notepadFile); err != nil {
+	if currentContentBytes, err := os.ReadFile(filepath.Join(os.Getenv("STATE_DIRECTORY"), "notepad")); err != nil {
 		currentContent = fmt.Sprintf("Error reading notepad file: %s", err)
 	} else {
 		currentContent = string(currentContentBytes)
@@ -209,7 +207,7 @@ func notepad(w http.ResponseWriter, r *http.Request) {
 }
 
 func saveNote(content []byte) error {
-	f, err := os.CreateTemp(filepath.Dir(*notepadFile), "notepad")
+	f, err := os.CreateTemp(os.Getenv("STATE_DIRECTORY"), "notepad")
 	if err != nil {
 		return err
 	}
@@ -220,7 +218,7 @@ func saveNote(content []byte) error {
 	if err := f.Close(); err != nil {
 		return err
 	}
-	return os.Rename(f.Name(), *notepadFile)
+	return os.Rename(f.Name(), filepath.Join(os.Getenv("STATE_DIRECTORY"), "notepad"))
 }
 
 func autosave(w http.ResponseWriter, r *http.Request) {