refactor progress

fluxer_devops/livekitctl/internal/state/state.go

@@ -0,0 +1,170 @@
+/*
+ * Copyright (C) 2026 Fluxer Contributors
+ *
+ * This file is part of Fluxer.
+ *
+ * Fluxer is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU Affero General Public License as published by
+ * the Free Software Foundation, either version 3 of the License, or
+ * (at your option) any later version.
+ *
+ * Fluxer is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+ * GNU Affero General Public License for more details.
+ *
+ * You should have received a copy of the GNU Affero General Public License
+ * along with Fluxer. If not, see <https://www.gnu.org/licenses/>.
+ */
+
+package state
+
+import (
+	"encoding/json"
+	"os"
+	"sort"
+
+	"github.com/fluxerapp/fluxer/fluxer_devops/livekitctl/internal/constants"
+	"github.com/fluxerapp/fluxer/fluxer_devops/livekitctl/internal/util"
+)
+
+type Versions struct {
+	LiveKit string `json:"livekit"`
+	Caddy   string `json:"caddy"`
+	CaddyL4 string `json:"caddy_l4"`
+	Xcaddy  string `json:"xcaddy"`
+}
+
+type Domains struct {
+	LiveKit string `json:"livekit"`
+	TURN    string `json:"turn"`
+}
+
+type Paths struct {
+	ConfigDir         string `json:"config_dir"`
+	StatePath         string `json:"state_path"`
+	SecretsPath       string `json:"secrets_path"`
+	LiveKitInstallDir string `json:"livekit_install_dir"`
+	LiveKitBinDir     string `json:"livekit_bin_dir"`
+	CaddyBin          string `json:"caddy_bin"`
+	CaddyStorageDir   string `json:"caddy_storage_dir"`
+	CaddyLogDir       string `json:"caddy_log_dir"`
+	LiveKitLogDir     string `json:"livekit_log_dir"`
+	KVDataDir         string `json:"kv_data_dir"`
+	KVLogDir          string `json:"kv_log_dir"`
+	UnitDir           string `json:"unit_dir"`
+}
+
+func DefaultPaths() Paths {
+	return Paths{
+		ConfigDir:         "/etc/livekit",
+		StatePath:         "/etc/livekit/livekitctl-state.json",
+		SecretsPath:       "/etc/livekit/livekitctl-secrets.json",
+		LiveKitInstallDir: "/opt/livekit",
+		LiveKitBinDir:     "/opt/livekit/bin",
+		CaddyBin:          "/usr/local/bin/caddy",
+		CaddyStorageDir:   "/var/lib/caddy",
+		CaddyLogDir:       "/var/log/caddy",
+		LiveKitLogDir:     "/var/log/livekit",
+		KVDataDir:         "/var/lib/livekit/kv",
+		KVLogDir:          "/var/log/livekit",
+		UnitDir:           "/etc/systemd/system",
+	}
+}
+
+type KVConfig struct {
+	BindHost string `json:"bind_host"`
+	Port     int    `json:"port"`
+}
+
+type FirewallConfig struct {
+	Enabled bool   `json:"enabled"`
+	Tool    string `json:"tool"`
+}
+
+type BootstrapState struct {
+	SchemaVersion int             `json:"schema_version"`
+	CreatedAt     string          `json:"created_at"`
+	UpdatedAt     string          `json:"updated_at"`
+	ACMEEmail     string          `json:"acme_email"`
+	Domains       Domains         `json:"domains"`
+	Ports         constants.Ports `json:"ports"`
+	Versions      Versions        `json:"versions"`
+	KV            KVConfig        `json:"kv"`
+	Webhooks      []string        `json:"webhooks"`
+	Paths         Paths           `json:"paths"`
+	Firewall      FirewallConfig  `json:"firewall"`
+}
+
+type NewStateParams struct {
+	ACMEEmail string
+	Domains   Domains
+	Ports     constants.Ports
+	Versions  Versions
+	KV        KVConfig
+	Webhooks  []string
+	Firewall  FirewallConfig
+	Paths     *Paths
+}
+
+func NewState(params NewStateParams) *BootstrapState {
+	ts := util.NowRFC3339()
+
+	webhooks := params.Webhooks
+	if webhooks == nil {
+		webhooks = []string{}
+	}
+	unique := make(map[string]bool)
+	for _, w := range webhooks {
+		unique[w] = true
+	}
+	sorted := make([]string, 0, len(unique))
+	for w := range unique {
+		sorted = append(sorted, w)
+	}
+	sort.Strings(sorted)
+
+	paths := DefaultPaths()
+	if params.Paths != nil {
+		paths = *params.Paths
+	}
+
+	return &BootstrapState{
+		SchemaVersion: 1,
+		CreatedAt:     ts,
+		UpdatedAt:     ts,
+		ACMEEmail:     params.ACMEEmail,
+		Domains:       params.Domains,
+		Ports:         params.Ports,
+		Versions:      params.Versions,
+		KV:            params.KV,
+		Webhooks:      sorted,
+		Paths:         paths,
+		Firewall:      params.Firewall,
+	}
+}
+
+func (st *BootstrapState) Touch() {
+	st.UpdatedAt = util.NowRFC3339()
+}
+
+func LoadState(path string) (*BootstrapState, error) {
+	data, err := os.ReadFile(path)
+	if err != nil {
+		if os.IsNotExist(err) {
+			return nil, nil
+		}
+		return nil, err
+	}
+
+	var st BootstrapState
+	if err := json.Unmarshal(data, &st); err != nil {
+		return nil, err
+	}
+
+	return &st, nil
+}
+
+func SaveState(st *BootstrapState) error {
+	return util.WriteJSON(st.Paths.StatePath, st, 0600, -1, -1)
+}