Files
elmprodvpn/selective-vpn-api/app/transport_tokens_state_ownership.go

47 lines
1.1 KiB
Go

package app
import (
"encoding/json"
"os"
"path/filepath"
"time"
)
func loadTransportOwnershipState() TransportOwnershipState {
st := TransportOwnershipState{Version: transportStateVersion}
data, err := os.ReadFile(transportOwnershipStatePath)
if err != nil {
return st
}
if err := json.Unmarshal(data, &st); err != nil {
return TransportOwnershipState{Version: transportStateVersion}
}
if st.Version == 0 {
st.Version = transportStateVersion
}
if st.Items == nil {
st.Items = nil
}
st.Count = len(st.Items)
return st
}
func saveTransportOwnershipState(st TransportOwnershipState) error {
st.Version = transportStateVersion
st.UpdatedAt = time.Now().UTC().Format(time.RFC3339)
st.Count = len(st.Items)
data, err := json.MarshalIndent(st, "", " ")
if err != nil {
return err
}
if err := os.MkdirAll(filepath.Dir(transportOwnershipStatePath), 0o755); err != nil {
return err
}
tmp := transportOwnershipStatePath + ".tmp"
if err := os.WriteFile(tmp, data, 0o644); err != nil {
return err
}
return os.Rename(tmp, transportOwnershipStatePath)
}