settings.go

package main

import (
	"context"
	"encoding/json"
	"os"
	"path/filepath"
	"sync"

	"wails-cast/pkg/folders"
)

const (
	settingsFileName = "settings.json"
)

func getDefaultSettings() Settings {
	return Settings{
		SubtitleBurnIn:             true,
		IgnoreClosedCaptions:       false,
		DefaultTranslationLanguage: "English",
		GeminiApiKey:               "",
		GeminiModel:                "gemini-2.5-flash",
		DefaultQuality:             "5M",
		SubtitleFontSize:           24,
		MaxOutputWidth:             0,
		TranslatePromptTemplate:    "Create a subtitle translation in {{.TargetLanguage}} based on the references in other languages.\nMultiple language tracks from the same video are provided as reference to help you understand context and maintain consistent terminology.\n\nInput format:\ndelay: <seconds>\nduration: <seconds>\n<text>\n\n{{.SubtitleContent}}\n\nOutput the translation in the same format inside <llm_output></llm_output> tags.",
		MaxSubtitleSamples:         4,
		NoTranscodeCache:           false,
	}
}

type Settings struct {
	SubtitleBurnIn             bool   `json:"subtitleBurnIn"`
	IgnoreClosedCaptions       bool   `json:"ignoreClosedCaptions"`
	DefaultTranslationLanguage string `json:"defaultTranslationLanguage"`
	GeminiApiKey               string `json:"geminiApiKey"`
	GeminiModel                string `json:"geminiModel"`
	DefaultQuality             string `json:"defaultQuality"`
	SubtitleFontSize           int    `json:"subtitleFontSize"`
	MaxOutputWidth             int    `json:"maxOutputWidth"`
	TranslatePromptTemplate    string `json:"translatePromptTemplate"`
	MaxSubtitleSamples         int    `json:"maxSubtitleSamples"`
	NoTranscodeCache           bool   `json:"noTranscodeCache"`
}

type SettingsStore struct {
	settings Settings
	filePath string
	ctx      context.Context
	mu       sync.RWMutex
}

func NewSettingsStore() *SettingsStore {
	appConfigDir := folders.GetConfig()
	os.MkdirAll(appConfigDir, 0755)

	settingsPath := filepath.Join(appConfigDir, settingsFileName)

	store := &SettingsStore{
		settings: getDefaultSettings(),
		filePath: settingsPath,
	}

	store.load()
	return store
}

func (s *SettingsStore) SetContext(ctx context.Context) {
	s.ctx = ctx
}

func (s *SettingsStore) load() error {
	s.mu.Lock()
	defer s.mu.Unlock()

	data, err := os.ReadFile(s.filePath)
	if err != nil {
		if os.IsNotExist(err) {
			return nil // No settings file yet, use defaults
		}
		return err
	}

	return json.Unmarshal(data, &s.settings)
}

func (s *SettingsStore) save() error {
	data, err := json.MarshalIndent(s.settings, "", "  ")
	if err != nil {
		return err
	}

	return os.WriteFile(s.filePath, data, 0644)
}

func (s *SettingsStore) Get() *Settings {
	s.mu.RLock()
	defer s.mu.RUnlock()

	return &s.settings
}

func (s *SettingsStore) Update(settings Settings) error {
	s.mu.Lock()
	defer s.mu.Unlock()

	s.settings = settings
	return s.save()
}

func (s *SettingsStore) Reset() error {
	s.mu.Lock()
	defer s.mu.Unlock()

	s.settings = getDefaultSettings()
	return s.save()
}