gickup/webui/webui.go

package webui

import (
	"bytes"
	_ "embed"
	"encoding/json"
	"errors"
	"io"
	"net/http"
	"os"
	"path/filepath"
	"reflect"
	"strconv"
	"sync"
	"sync/atomic"
	"time"

	"github.com/cooperspencer/gickup/types"
	"github.com/goccy/go-yaml"
	"github.com/rs/zerolog/log"
)

// BackupStatus is the result of a single repo-to-destination backup attempt.
type BackupStatus string

const (
	StatusSuccess BackupStatus = "success"
	StatusFailed  BackupStatus = "failed"
)

// StatusFromInt converts the integer convention used by backup() (1=success, 0=failed).
func StatusFromInt(i int) BackupStatus {
	if i == 1 {
		return StatusSuccess
	}
	return StatusFailed
}

// BackupEntry records one repo-to-destination backup attempt.
type BackupEntry struct {
	Timestamp  time.Time    `json:"timestamp"`
	RepoName   string       `json:"repo_name"`
	RepoURL    string       `json:"repo_url"`
	Owner      string       `json:"owner"`
	Hoster     string       `json:"hoster"`
	DestType   string       `json:"dest_type"`
	DestAddr   string       `json:"dest_addr"`
	Status     BackupStatus `json:"status"`
	DurationMs int64        `json:"duration_ms"`
}

// ConfigInfo describes one loaded configuration block for the UI.
type ConfigInfo struct {
	Index    int    `json:"index"`
	Name     string `json:"name"`
	Sources  int    `json:"sources"`
	Dests    int    `json:"dests"`
	CronSpec string `json:"cron_spec,omitempty"`
	NextRun  string `json:"next_run,omitempty"` // RFC3339
}

// store holds backup entries, config file paths, and the run callback.
type store struct {
	mu          sync.RWMutex
	entries     []BackupEntry
	configFiles []string
	configs     []ConfigInfo
	runFunc     func(int)
	running     int32 // atomic count of active runs and API reservations
}

// Global is the shared store used across the application.
var Global = &store{}

// BeginRun tracks startup, scheduled, and manually triggered backup work.
func (s *store) BeginRun() { atomic.AddInt32(&s.running, 1) }

// EndRun releases one active run or API reservation.
func (s *store) EndRun() { atomic.AddInt32(&s.running, -1) }

// SetConfigFiles registers the config file paths with the store.
func (s *store) SetConfigFiles(files []string) {
	s.mu.Lock()
	defer s.mu.Unlock()
	s.configFiles = make([]string, len(files))
	copy(s.configFiles, files)
}

// SetConfigs registers the list of config summaries shown in the UI.
func (s *store) SetConfigs(infos []ConfigInfo) {
	s.mu.Lock()
	defer s.mu.Unlock()
	s.configs = make([]ConfigInfo, len(infos))
	copy(s.configs, infos)
}

// SetRunFunc registers the callback that main.go uses to trigger a backup run.
// index == -1 means run all configs; otherwise run the config at that index.
func (s *store) SetRunFunc(fn func(int)) {
	s.mu.Lock()
	defer s.mu.Unlock()
	s.runFunc = fn
}

// Record appends a backup entry to the store.
func (s *store) Record(e BackupEntry) {
	s.mu.Lock()
	defer s.mu.Unlock()
	s.entries = append(s.entries, e)
}

// Clear removes all recorded backup entries.
func (s *store) Clear() {
	s.mu.Lock()
	defer s.mu.Unlock()
	s.entries = nil
}

//go:embed ui/index.html
var indexHTML []byte

//go:embed ui/logo.png
var logoPNG []byte

// Serve starts the web UI HTTP server on addr.
func Serve(addr string) {
	mux := http.NewServeMux()
	mux.HandleFunc("/", handleIndex)
	mux.HandleFunc("/logo.png", handleLogo)
	mux.HandleFunc("/api/status", handleStatus)
	mux.HandleFunc("/api/config", handleConfig)
	mux.HandleFunc("/api/configs", handleConfigs)
	mux.HandleFunc("/api/run", handleRun)
	mux.HandleFunc("/api/running", handleRunning)
	log.Info().Str("addr", addr).Msg("Web UI listening")
	if err := http.ListenAndServe(addr, mux); err != nil {
		log.Error().Err(err).Msg("Web UI server error")
	}
}

func handleIndex(w http.ResponseWriter, _ *http.Request) {
	w.Header().Set("Content-Type", "text/html; charset=utf-8")
	_, _ = w.Write(indexHTML)
}

func handleLogo(w http.ResponseWriter, _ *http.Request) {
	w.Header().Set("Content-Type", "image/png")
	_, _ = w.Write(logoPNG)
}

func handleStatus(w http.ResponseWriter, r *http.Request) {
	if r.Method == http.MethodDelete {
		Global.Clear()
		w.WriteHeader(http.StatusNoContent)
		return
	}
	Global.mu.RLock()
	entries := make([]BackupEntry, len(Global.entries))
	copy(entries, Global.entries)
	Global.mu.RUnlock()
	w.Header().Set("Content-Type", "application/json")
	if err := json.NewEncoder(w).Encode(entries); err != nil {
		log.Error().Err(err).Msg("Failed to write JSON response")
	}
}

// handleConfigs returns the list of loaded configuration blocks.
func handleConfigs(w http.ResponseWriter, _ *http.Request) {
	Global.mu.RLock()
	cfgs := make([]ConfigInfo, len(Global.configs))
	copy(cfgs, Global.configs)
	Global.mu.RUnlock()

	now := time.Now()
	for i := range cfgs {
		if cfgs[i].CronSpec == "" {
			continue
		}
		sched, err := types.ParseCronSpec(cfgs[i].CronSpec)
		if err != nil {
			cfgs[i].NextRun = ""
			continue
		}
		cfgs[i].NextRun = sched.Next(now).Format(time.RFC3339)
	}

	w.Header().Set("Content-Type", "application/json")
	if err := json.NewEncoder(w).Encode(cfgs); err != nil {
		log.Error().Err(err).Msg("Failed to write JSON response")
	}
}

// handleRunning reports whether a backup run is currently in progress.
func handleRunning(w http.ResponseWriter, _ *http.Request) {
	running := atomic.LoadInt32(&Global.running) != 0
	w.Header().Set("Content-Type", "application/json")
	if err := json.NewEncoder(w).Encode(map[string]bool{"running": running}); err != nil {
		log.Error().Err(err).Msg("Failed to write JSON response")
	}
}

// handleRun triggers an immediate backup run.
// POST /api/run  body: {"index": -1}   (-1 = all, ≥0 = specific config)
func handleRun(w http.ResponseWriter, r *http.Request) {
	if r.Method != http.MethodPost {
		http.Error(w, "method not allowed", http.StatusMethodNotAllowed)
		return
	}
	if !atomic.CompareAndSwapInt32(&Global.running, 0, 1) {
		http.Error(w, "backup already running", http.StatusConflict)
		return
	}

	var req struct {
		Index int `json:"index"`
	}
	req.Index = -1
	_ = json.NewDecoder(r.Body).Decode(&req)

	Global.mu.RLock()
	fn := Global.runFunc
	Global.mu.RUnlock()

	if fn == nil {
		Global.EndRun()
		http.Error(w, "not ready", http.StatusServiceUnavailable)
		return
	}

	go func() {
		defer Global.EndRun()
		fn(req.Index)
	}()

	w.Header().Set("Content-Type", "application/json")
	w.WriteHeader(http.StatusAccepted)
	if err := json.NewEncoder(w).Encode(map[string]bool{"running": true}); err != nil {
		log.Error().Err(err).Msg("Failed to write JSON response")
	}
}

func handleConfig(w http.ResponseWriter, r *http.Request) {
	Global.mu.RLock()
	files := make([]string, len(Global.configFiles))
	copy(files, Global.configFiles)
	Global.mu.RUnlock()

	if len(files) == 0 {
		http.Error(w, "no config files registered", http.StatusNotFound)
		return
	}

	q := r.URL.Query()

	// List mode: GET /api/config?list=1
	if r.Method == http.MethodGet && q.Get("list") == "1" {
		type fileInfo struct {
			Index int    `json:"index"`
			Name  string `json:"name"`
			Path  string `json:"path"`
		}
		list := make([]fileInfo, len(files))
		for i, f := range files {
			list[i] = fileInfo{Index: i, Name: filepath.Base(f), Path: f}
		}
		w.Header().Set("Content-Type", "application/json")
		if err := json.NewEncoder(w).Encode(list); err != nil {
			log.Error().Err(err).Msg("Failed to write JSON response")
		}
		return
	}

	// Parse file index (default 0).
	idx := 0
	if s := q.Get("file"); s != "" {
		n, err := strconv.Atoi(s)
		if err != nil || n < 0 || n >= len(files) {
			http.Error(w, "invalid file index", http.StatusBadRequest)
			return
		}
		idx = n
	}

	switch r.Method {
	case http.MethodGet:
		data, err := os.ReadFile(files[idx])
		if errors.Is(err, os.ErrNotExist) {
			data = []byte{}
			err = nil
		}
		if err != nil {
			http.Error(w, err.Error(), http.StatusInternalServerError)
			return
		}
		w.Header().Set("Content-Type", "text/plain; charset=utf-8")
		_, _ = w.Write(data)

	case http.MethodPost:
		body, err := io.ReadAll(http.MaxBytesReader(w, r.Body, 1<<20))
		if err != nil {
			http.Error(w, err.Error(), http.StatusBadRequest)
			return
		}
		// Check every YAML document before replacing the file.
		dec := yaml.NewDecoder(bytes.NewReader(body))
		count := 0
		for {
			var conf types.Conf
			err := dec.Decode(&conf)
			if errors.Is(err, io.EOF) {
				break
			}
			if err != nil {
				http.Error(w, "invalid YAML: "+err.Error(), http.StatusBadRequest)
				return
			}
			if !reflect.ValueOf(conf).IsZero() {
				count++
			}
		}
		if count == 0 {
			http.Error(w, "configuration must contain at least one non-empty configuration block", http.StatusBadRequest)
			return
		}
		if err := os.MkdirAll(filepath.Dir(files[idx]), 0o700); err != nil {
			http.Error(w, err.Error(), http.StatusInternalServerError)
			return
		}
		if err := os.WriteFile(files[idx], body, 0o600); err != nil {
			http.Error(w, err.Error(), http.StatusInternalServerError)
			return
		}
		w.WriteHeader(http.StatusOK)

	default:
		http.Error(w, "method not allowed", http.StatusMethodNotAllowed)
	}
}