gosentry/internal/core/store.go

package core

import (
	"errors"
	"os"
	"path/filepath"
	"runtime"
	"strings"

	"gopkg.in/yaml.v3"
)

type Store struct {
	Paths  Paths
	Config Config
}

func OpenStore() (*Store, []Job, error) {
	paths, err := ResolvePaths()
	if err != nil {
		return nil, nil, err
	}

	store := &Store{Paths: paths}
	config, err := loadOrCreateConfig(paths)
	if err != nil {
		return nil, nil, err
	}
	store.Config = config
	store.applyConfigPaths()
	if err := store.SaveConfig(); err != nil {
		return nil, nil, err
	}

	jobs, err := loadOrCreateJobs(store.Paths.JobsPath)
	if err != nil {
		return nil, nil, err
	}
	return store, jobs, nil
}

func (s *Store) SaveConfig() error {
	s.applyConfigPaths()
	if err := os.MkdirAll(s.Paths.AppDir, 0o755); err != nil {
		return err
	}
	return writeYAML(s.Paths.ConfigPath, s.Config)
}

func (s *Store) SaveJobs(jobs []Job) error {
	if err := os.MkdirAll(s.Paths.JobsDir, 0o755); err != nil {
		return err
	}
	return writeYAML(s.Paths.JobsPath, JobsFile{Jobs: jobs})
}

func loadOrCreateConfig(paths Paths) (Config, error) {
	config := Config{
		JobsDir:           ".",
		LogsDir:           "logs",
		MaxLogFiles:       100,
		MaxLogAgeDays:     30,
		KeepRunningInTray: true,
		NotifyOnFailure:   true,
	}

	if _, err := os.Stat(paths.ConfigPath); errors.Is(err, os.ErrNotExist) {
		return config, writeYAML(paths.ConfigPath, config)
	}

	data, err := os.ReadFile(paths.ConfigPath)
	if err != nil {
		return Config{}, err
	}
	if err := yaml.Unmarshal(data, &config); err != nil {
		return Config{}, err
	}
	if strings.TrimSpace(config.JobsDir) == "" {
		config.JobsDir = "."
	}
	if strings.TrimSpace(config.LogsDir) == "" {
		config.LogsDir = "logs"
	}
	if config.MaxLogFiles <= 0 {
		config.MaxLogFiles = 100
	}
	if config.MaxLogAgeDays <= 0 {
		config.MaxLogAgeDays = 30
	}
	return config, nil
}

func loadOrCreateJobs(path string) ([]Job, error) {
	if _, err := os.Stat(path); errors.Is(err, os.ErrNotExist) {
		jobs := defaultJobs()
		return jobs, writeYAML(path, JobsFile{Jobs: jobs})
	}

	data, err := os.ReadFile(path)
	if err != nil {
		return nil, err
	}
	var file JobsFile
	if err := yaml.Unmarshal(data, &file); err != nil {
		return nil, err
	}
	return file.Jobs, nil
}

func resolveJobsDir(appDir string, jobsDir string) string {
	return resolveConfiguredDir(appDir, jobsDir)
}

func resolveConfiguredDir(appDir string, dir string) string {
	if filepath.IsAbs(dir) {
		return dir
	}
	return filepath.Clean(filepath.Join(appDir, dir))
}

func (s *Store) applyConfigPaths() {
	s.Paths.JobsDir = resolveConfiguredDir(s.Paths.AppDir, s.Config.JobsDir)
	s.Paths.JobsPath = filepath.Join(s.Paths.JobsDir, JobsFileName)
	s.Paths.LogsDir = resolveConfiguredDir(s.Paths.AppDir, s.Config.LogsDir)
}

func writeYAML(path string, value any) error {
	if err := os.MkdirAll(filepath.Dir(path), 0o755); err != nil {
		return err
	}
	data, err := yaml.Marshal(value)
	if err != nil {
		return err
	}
	return os.WriteFile(path, data, 0o644)
}

func defaultJobs() []Job {
	return []Job{
		{
			ID:        1,
			Name:      "Hello scheduler",
			Folder:    "Examples",
			Schedule:  "@every 10s",
			Command:   echoCommand("PySentry test job: scheduler is alive"),
			Enabled:   true,
			LastRun:   "Never",
			NextRun:   "After start",
			LastState: "Ready",
			Output:    "No command output captured yet.",
		},
		{
			ID:        2,
			Name:      "Write timestamp",
			Folder:    "Examples",
			Schedule:  "@every 15s",
			Command:   echoCommand("PySentry test job: timestamp command ran"),
			Enabled:   true,
			LastRun:   "Never",
			NextRun:   "After start",
			LastState: "Ready",
			Output:    "No command output captured yet.",
		},
		{
			ID:        3,
			Name:      "Paused sample",
			Schedule:  "@every 1m",
			Command:   echoCommand("This paused sample should not run until enabled"),
			Enabled:   false,
			LastRun:   "Never",
			NextRun:   "Paused",
			LastState: "Paused",
			Output:    "No command output captured yet.",
		},
	}
}

func echoCommand(message string) string {
	if runtime.GOOS == "windows" {
		return "echo " + message
	}
	return "echo '" + strings.ReplaceAll(message, "'", "'\\''") + "'"
}