Restore last buffer on start

review emersion

review taiite

review taiite

Use os.WriteFile instead of os.Create

1 files changed, 37 insertions, 2 deletions

diff --git a/cmd/senpai/main.go b/cmd/senpai/main.go
index 8d56cef..e41d8db 100644
--- a/cmd/senpai/main.go
+++ b/cmd/senpai/main.go
@@ -3,9 +3,11 @@ package main
 import (
 	"flag"
 	"fmt"
+	"io/ioutil"
 	"math/rand"
 	"os"
 	"path"
+	"strings"
 	"time"
 
 	"git.sr.ht/~taiite/senpai"
@@ -41,11 +43,44 @@ func main() {
 
 	cfg.Debug = cfg.Debug || debug
 
-	app, err := senpai.NewApp(cfg)
+	lastBuffer := getLastBuffer()
+
+	app, err := senpai.NewApp(cfg, lastBuffer)
 	if err != nil {
 		panic(err)
 	}
-	defer app.Close()
 
 	app.Run()
+	app.Close()
+
+	// Write last buffer on close
+	lastBufferPath := getLastBufferPath()
+	err = os.WriteFile(lastBufferPath, []byte(app.CurrentBuffer()), 0666)
+	if err != nil {
+		fmt.Fprintf(os.Stderr, "failed to write last buffer at %q: %s\n", lastBufferPath, err)
+	}
+}
+
+func getLastBufferPath() string {
+	cacheDir, err := os.UserCacheDir()
+	if err != nil {
+		panic(err)
+	}
+	cachePath := path.Join(cacheDir, "senpai")
+	err = os.MkdirAll(cachePath, 0755)
+	if err != nil {
+		panic(err)
+	}
+
+	lastBufferPath := path.Join(cachePath, "lastbuffer.txt")
+	return lastBufferPath
+}
+
+func getLastBuffer() string {
+	buf, err := ioutil.ReadFile(getLastBufferPath())
+	if err != nil {
+		return ""
+	}
+
+	return strings.TrimSpace(string(buf))
 }