mirror of
https://github.com/garethgeorge/backrest.git
synced 2025-12-14 09:35:41 +00:00
77 lines
1.5 KiB
Go
77 lines
1.5 KiB
Go
package config
|
|
|
|
import (
|
|
"bytes"
|
|
"errors"
|
|
"fmt"
|
|
"os"
|
|
"path/filepath"
|
|
"sync"
|
|
|
|
v1 "github.com/garethgeorge/resticui/gen/go/v1"
|
|
"github.com/natefinch/atomic"
|
|
"google.golang.org/protobuf/encoding/protojson"
|
|
)
|
|
|
|
type JsonFileStore struct {
|
|
Path string
|
|
mu sync.Mutex
|
|
}
|
|
|
|
var _ ConfigStore = &JsonFileStore{}
|
|
|
|
func (f *JsonFileStore) Get() (*v1.Config, error) {
|
|
f.mu.Lock()
|
|
defer f.mu.Unlock()
|
|
|
|
data, err := os.ReadFile(f.Path)
|
|
if err != nil {
|
|
if errors.Is(err, os.ErrNotExist) {
|
|
return nil, ErrConfigNotFound
|
|
}
|
|
return nil, fmt.Errorf("failed to read config file: %w", err)
|
|
}
|
|
|
|
var config v1.Config
|
|
|
|
if err = protojson.Unmarshal(data, &config); err != nil {
|
|
return nil, fmt.Errorf("failed to unmarshal config: %w", err)
|
|
}
|
|
|
|
if err := ValidateConfig(&config); err != nil {
|
|
return nil, fmt.Errorf("invalid config: %w", err)
|
|
}
|
|
|
|
return &config, nil
|
|
}
|
|
|
|
func (f *JsonFileStore) Update(config *v1.Config) error {
|
|
f.mu.Lock()
|
|
defer f.mu.Unlock()
|
|
|
|
if err := ValidateConfig(config); err != nil {
|
|
return fmt.Errorf("invalid config: %w", err)
|
|
}
|
|
|
|
data, err := protojson.MarshalOptions{
|
|
Indent: " ",
|
|
Multiline: true,
|
|
EmitUnpopulated: true,
|
|
}.Marshal(config)
|
|
if err != nil {
|
|
return fmt.Errorf("failed to marshal config: %w", err)
|
|
}
|
|
|
|
err = os.MkdirAll(filepath.Dir(f.Path), 0755)
|
|
if err != nil {
|
|
return fmt.Errorf("failed to create config directory: %w", err)
|
|
}
|
|
|
|
err = atomic.WriteFile(f.Path, bytes.NewReader(data))
|
|
if err != nil {
|
|
return fmt.Errorf("failed to write config file: %w", err)
|
|
}
|
|
|
|
return nil
|
|
}
|