diff options
Diffstat (limited to 'roseh.moe.go')
| -rw-r--r-- | roseh.moe.go | 12 |
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) { |