97 lines
2.1 KiB
Go
97 lines
2.1 KiB
Go
package store
|
|
|
|
import (
|
|
"encoding/json"
|
|
"errors"
|
|
"fmt"
|
|
"log/slog"
|
|
"os"
|
|
"path/filepath"
|
|
"sync"
|
|
|
|
"github.com/google/uuid"
|
|
)
|
|
|
|
type Store struct {
|
|
ID string `json:"id"`
|
|
FirstTimeRun bool `json:"first-time-run"`
|
|
}
|
|
|
|
var (
|
|
lock sync.Mutex
|
|
store Store
|
|
)
|
|
|
|
func GetID() string {
|
|
lock.Lock()
|
|
defer lock.Unlock()
|
|
if store.ID == "" {
|
|
initStore()
|
|
}
|
|
return store.ID
|
|
}
|
|
|
|
func GetFirstTimeRun() bool {
|
|
lock.Lock()
|
|
defer lock.Unlock()
|
|
if store.ID == "" {
|
|
initStore()
|
|
}
|
|
return store.FirstTimeRun
|
|
}
|
|
|
|
func SetFirstTimeRun(val bool) {
|
|
lock.Lock()
|
|
defer lock.Unlock()
|
|
if store.FirstTimeRun == val {
|
|
return
|
|
}
|
|
store.FirstTimeRun = val
|
|
writeStore(getStorePath())
|
|
}
|
|
|
|
// lock must be held
|
|
func initStore() {
|
|
storeFile, err := os.Open(getStorePath())
|
|
if err == nil {
|
|
defer storeFile.Close()
|
|
err = json.NewDecoder(storeFile).Decode(&store)
|
|
if err == nil {
|
|
slog.Debug(fmt.Sprintf("loaded existing store %s - ID: %s", getStorePath(), store.ID))
|
|
return
|
|
}
|
|
} else if !errors.Is(err, os.ErrNotExist) {
|
|
slog.Debug(fmt.Sprintf("unexpected error searching for store: %s", err))
|
|
}
|
|
slog.Debug("initializing new store")
|
|
store.ID = uuid.NewString()
|
|
writeStore(getStorePath())
|
|
}
|
|
|
|
func writeStore(storeFilename string) {
|
|
ollamaDir := filepath.Dir(storeFilename)
|
|
_, err := os.Stat(ollamaDir)
|
|
if errors.Is(err, os.ErrNotExist) {
|
|
if err := os.MkdirAll(ollamaDir, 0o755); err != nil {
|
|
slog.Error(fmt.Sprintf("create ollama dir %s: %v", ollamaDir, err))
|
|
return
|
|
}
|
|
}
|
|
payload, err := json.Marshal(store)
|
|
if err != nil {
|
|
slog.Error(fmt.Sprintf("failed to marshal store: %s", err))
|
|
return
|
|
}
|
|
fp, err := os.OpenFile(storeFilename, os.O_WRONLY|os.O_CREATE|os.O_TRUNC, 0o755)
|
|
if err != nil {
|
|
slog.Error(fmt.Sprintf("write store payload %s: %v", storeFilename, err))
|
|
return
|
|
}
|
|
defer fp.Close()
|
|
if n, err := fp.Write(payload); err != nil || n != len(payload) {
|
|
slog.Error(fmt.Sprintf("write store payload %s: %d vs %d -- %v", storeFilename, n, len(payload), err))
|
|
return
|
|
}
|
|
slog.Debug("Store contents: " + string(payload))
|
|
slog.Info(fmt.Sprintf("wrote store: %s", storeFilename))
|
|
}
|