rename machine config to state

This commit is contained in:
Pavel Sviderski
2024-09-06 09:26:45 +10:00
parent 945eff471b
commit c58c99e061
5 changed files with 60 additions and 55 deletions
+1 -4
View File
@@ -5,9 +5,6 @@ import (
) )
type MachineConnection struct { type MachineConnection struct {
User string `toml:"user,omitempty"`
Host string `toml:"host"` Host string `toml:"host"`
Port int `toml:"port"` PublicKey secret.Secret `toml:"public_key"`
SSHKey string `toml:"ssh_key,omitempty"`
PublicKey secret.Secret `toml:"public_key,omitempty"`
} }
+11 -5
View File
@@ -22,19 +22,25 @@ const (
StateFile = "cluster.pb" StateFile = "cluster.pb"
) )
type Config struct {
APIAddr string
APISockPath string
}
type Cluster struct { type Cluster struct {
// TODO: implement grpc ClusterServer // TODO: implement grpc ClusterServer
pb.UnimplementedClusterServer pb.UnimplementedClusterServer
config Config
state *State state *State
apiAddr string
server *grpc.Server server *grpc.Server
} }
func NewCluster(state *State, apiAddr string) *Cluster { func NewCluster(config *Config, state *State) *Cluster {
c := &Cluster{ c := &Cluster{
config: *config,
state: state, state: state,
apiAddr: apiAddr,
server: grpc.NewServer(), server: grpc.NewServer(),
} }
pb.RegisterClusterServer(c.server, c) pb.RegisterClusterServer(c.server, c)
@@ -42,11 +48,11 @@ func NewCluster(state *State, apiAddr string) *Cluster {
} }
func (c *Cluster) Run() error { func (c *Cluster) Run() error {
listener, err := net.Listen("tcp", c.apiAddr) listener, err := net.Listen("tcp", c.config.APIAddr)
if err != nil { if err != nil {
return fmt.Errorf("listen API port: %w", err) return fmt.Errorf("listen API port: %w", err)
} }
slog.Info("Starting API server.", "addr", c.apiAddr) slog.Info("Starting API server.", "addr", c.config.APIAddr)
if err = c.server.Serve(listener); err != nil { if err = c.server.Serve(listener); err != nil {
return fmt.Errorf("API server failed: %w", err) return fmt.Errorf("API server failed: %w", err)
} }
+23 -21
View File
@@ -32,7 +32,7 @@ func InitCluster(dataDir, machineName string, netPrefix netip.Prefix, users []*p
} }
state := cluster.NewState(cluster.StatePath(dataDir)) state := cluster.NewState(cluster.StatePath(dataDir))
c := cluster.NewCluster(state, "") c := cluster.NewCluster(&cluster.Config{}, state)
if err = c.SetNetwork(netPrefix); err != nil { if err = c.SetNetwork(netPrefix); err != nil {
return fmt.Errorf("set cluster network: %w", err) return fmt.Errorf("set cluster network: %w", err)
} }
@@ -69,7 +69,7 @@ func InitCluster(dataDir, machineName string, netPrefix netip.Prefix, users []*p
if err != nil { if err != nil {
return err return err
} }
mcfg := &machine.Config{ mcfg := &machine.State{
ID: m.Id, ID: m.Id,
Name: m.Name, Name: m.Name,
Network: &network.Config{ Network: &network.Config{
@@ -97,7 +97,7 @@ func InitCluster(dataDir, machineName string, netPrefix netip.Prefix, users []*p
} }
mcfg.Network.Peers = peers mcfg.Network.Peers = peers
mcfg.SetPath(machine.ConfigPath(dataDir)) mcfg.SetPath(machine.StatePath(dataDir))
if err = mcfg.Save(); err != nil { if err = mcfg.Save(); err != nil {
return fmt.Errorf("save machine config: %w", err) return fmt.Errorf("save machine config: %w", err)
} }
@@ -107,55 +107,57 @@ func InitCluster(dataDir, machineName string, netPrefix netip.Prefix, users []*p
} }
type Daemon struct { type Daemon struct {
config *machine.Config state *machine.State
cluster *cluster.Cluster cluster *cluster.Cluster
} }
func New(dataDir string) (*Daemon, error) { func New(dataDir string) (*Daemon, error) {
cfgPath := machine.ConfigPath(dataDir) mstatePath := machine.StatePath(dataDir)
cfg, err := machine.ParseConfig(cfgPath) mstate, err := machine.ParseState(mstatePath)
if err != nil { if err != nil {
if !errors.Is(err, os.ErrNotExist) { if !errors.Is(err, os.ErrNotExist) {
return nil, fmt.Errorf("load machine config: %w", err) return nil, fmt.Errorf("load machine config: %w", err)
} }
// Generate an empty machine config with a new key pair. // Generate an empty machine config with a new key pair.
slog.Info("Machine config not found, creating a new one.", "path", cfgPath) slog.Info("Machine config not found, creating a new one.", "path", mstatePath)
privKey, pubKey, kErr := network.NewMachineKeys() privKey, pubKey, kErr := network.NewMachineKeys()
if kErr != nil { if kErr != nil {
return nil, fmt.Errorf("generate machine keys: %w", kErr) return nil, fmt.Errorf("generate machine keys: %w", kErr)
} }
slog.Info("Generated machine key pair.", "pubkey", pubKey) slog.Info("Generated machine key pair.", "pubkey", pubKey)
cfg = &machine.Config{ mstate = &machine.State{
Network: &network.Config{ Network: &network.Config{
PrivateKey: privKey, PrivateKey: privKey,
PublicKey: pubKey, PublicKey: pubKey,
}, },
} }
cfg.SetPath(cfgPath) mstate.SetPath(mstatePath)
if err = cfg.Save(); err != nil { if err = mstate.Save(); err != nil {
return nil, fmt.Errorf("save machine config: %w", err) return nil, fmt.Errorf("save machine config: %w", err)
} }
} }
statePath := cluster.StatePath(dataDir) cstatePath := cluster.StatePath(dataDir)
state := cluster.NewState(statePath) cstate := cluster.NewState(cstatePath)
if err = state.Load(); err != nil { if err = cstate.Load(); err != nil {
if !errors.Is(err, os.ErrNotExist) { if !errors.Is(err, os.ErrNotExist) {
return nil, fmt.Errorf("load cluster state: %w", err) return nil, fmt.Errorf("load cluster state: %w", err)
} }
slog.Info("Cluster state not found, creating a new one.", "path", statePath) slog.Info("Cluster state not found, creating a new one.", "path", cstatePath)
if err = state.Save(); err != nil { if err = cstate.Save(); err != nil {
return nil, fmt.Errorf("save cluster state: %w", err) return nil, fmt.Errorf("save cluster state: %w", err)
} }
} }
d := &Daemon{ d := &Daemon{
config: cfg, state: mstate,
} }
if cfg.Network.IsConfigured() { if mstate.Network.IsConfigured() {
apiAddr := net.JoinHostPort(cfg.Network.ManagementIP.String(), strconv.Itoa(machine.APIPort)) config := &cluster.Config{
d.cluster = cluster.NewCluster(state, apiAddr) APIAddr: net.JoinHostPort(mstate.Network.ManagementIP.String(), strconv.Itoa(machine.APIPort)),
}
d.cluster = cluster.NewCluster(config, cstate)
} }
return d, nil return d, nil
@@ -166,12 +168,12 @@ func (d *Daemon) Run(ctx context.Context) error {
errGroup, ctx := errgroup.WithContext(ctx) errGroup, ctx := errgroup.WithContext(ctx)
// Start the network only if it is configured. // Start the network only if it is configured.
if d.config.Network.IsConfigured() { if d.state.Network.IsConfigured() {
wgnet, err := network.NewWireGuardNetwork() wgnet, err := network.NewWireGuardNetwork()
if err != nil { if err != nil {
return fmt.Errorf("create WireGuard network: %w", err) return fmt.Errorf("create WireGuard network: %w", err)
} }
if err = wgnet.Configure(*d.config.Network); err != nil { if err = wgnet.Configure(*d.state.Network); err != nil {
return fmt.Errorf("configure WireGuard network: %w", err) return fmt.Errorf("configure WireGuard network: %w", err)
} }
+3 -3
View File
@@ -12,14 +12,14 @@ import (
// MachineToken returns the local machine's token that can be used for adding the machine to a cluster. // MachineToken returns the local machine's token that can be used for adding the machine to a cluster.
// TODO: ideally, this should be an RPC call to the daemon API to ensure the config is created and up-to-date. // TODO: ideally, this should be an RPC call to the daemon API to ensure the config is created and up-to-date.
func MachineToken(dataDir string) (machine.Token, error) { func MachineToken(dataDir string) (machine.Token, error) {
cfg, err := machine.ParseConfig(machine.ConfigPath(dataDir)) state, err := machine.ParseState(machine.StatePath(dataDir))
if err != nil { if err != nil {
if errors.Is(err, os.ErrNotExist) { if errors.Is(err, os.ErrNotExist) {
return machine.Token{}, fmt.Errorf("load machine config (is uncloudd daemon running?): %w", err) return machine.Token{}, fmt.Errorf("load machine config (is uncloudd daemon running?): %w", err)
} }
return machine.Token{}, fmt.Errorf("load machine config: %w", err) return machine.Token{}, fmt.Errorf("load machine config: %w", err)
} }
if len(cfg.Network.PublicKey) == 0 { if len(state.Network.PublicKey) == 0 {
return machine.Token{}, errors.New("public key is not set in machine config") return machine.Token{}, errors.New("public key is not set in machine config")
} }
@@ -37,5 +37,5 @@ func MachineToken(dataDir string) (machine.Token, error) {
for i, ip := range ips { for i, ip := range ips {
endpoints[i] = netip.AddrPortFrom(ip, network.WireGuardPort) endpoints[i] = netip.AddrPortFrom(ip, network.WireGuardPort)
} }
return machine.NewToken(cfg.Network.PublicKey, endpoints), nil return machine.NewToken(state.Network.PublicKey, endpoints), nil
} }
@@ -14,13 +14,13 @@ import (
const ( const (
DefaultDataDir = "/var/lib/uncloud" DefaultDataDir = "/var/lib/uncloud"
ConfigFileName = "machine.json" StateFileName = "machine.json"
APIPort = 51000 APIPort = 51000
) )
// Config defines the machine-specific configuration within a cluster for the Uncloud daemon. It encapsulates // State defines the machine-specific configuration within a cluster. It encapsulates essential identifiers
// essential identifiers and settings required to establish an overlay network and join the cluster. // and settings required to establish an overlay network and operate as a member of a cluster.
type Config struct { type State struct {
// ID uniquely identifies this machine in the cluster. // ID uniquely identifies this machine in the cluster.
ID string ID string
// Name provides a human-readable identifier for the machine. // Name provides a human-readable identifier for the machine.
@@ -32,18 +32,18 @@ type Config struct {
path string path string
} }
// ConfigPath returns the path to the machine configuration file within the given data directory. // StatePath returns the path to the machine state file within the given data directory.
func ConfigPath(dataDir string) string { func StatePath(dataDir string) string {
return filepath.Join(dataDir, ConfigFileName) return filepath.Join(dataDir, StateFileName)
} }
// ParseConfig reads and decodes a config from the file at the given path. // ParseState reads and decodes a state from the file at the given path.
func ParseConfig(path string) (*Config, error) { func ParseState(path string) (*State, error) {
data, err := os.ReadFile(path) data, err := os.ReadFile(path)
if err != nil { if err != nil {
return nil, fmt.Errorf("read config file: %w", err) return nil, fmt.Errorf("read config file: %w", err)
} }
var config Config var config State
if err = json.Unmarshal(data, &config); err != nil { if err = json.Unmarshal(data, &config); err != nil {
return nil, fmt.Errorf("parse config file %q: %w", path, err) return nil, fmt.Errorf("parse config file %q: %w", path, err)
} }
@@ -54,13 +54,13 @@ func ParseConfig(path string) (*Config, error) {
return &config, nil return &config, nil
} }
// SetPath sets the file path the config can be saved to. // SetPath sets the file path the state can be saved to.
func (c *Config) SetPath(path string) { func (c *State) SetPath(path string) {
c.path = path c.path = path
} }
// Encode returns the JSON encoded config data. // Encode returns the JSON encoded state data.
func (c *Config) Encode() ([]byte, error) { func (c *State) Encode() ([]byte, error) {
data, err := json.MarshalIndent(c, "", " ") data, err := json.MarshalIndent(c, "", " ")
if err != nil { if err != nil {
return nil, fmt.Errorf("marshal config: %w", err) return nil, fmt.Errorf("marshal config: %w", err)
@@ -68,8 +68,8 @@ func (c *Config) Encode() ([]byte, error) {
return data, nil return data, nil
} }
// Save writes the config data to the file at the given path. // Save writes the state data to the file at the given path.
func (c *Config) Save() error { func (c *State) Save() error {
if c.path == "" { if c.path == "" {
return fmt.Errorf("config path not set") return fmt.Errorf("config path not set")
} }
@@ -105,7 +105,7 @@ func NewRandomName() (string, error) {
} }
// NewBootstrapConfig returns a new machine configuration that should be applied to the first machine in a cluster. // NewBootstrapConfig returns a new machine configuration that should be applied to the first machine in a cluster.
func NewBootstrapConfig(name string, subnet netip.Prefix, peers ...network.PeerConfig) (*Config, error) { func NewBootstrapConfig(name string, subnet netip.Prefix, peers ...network.PeerConfig) (*State, error) {
mid, err := NewID() mid, err := NewID()
if err != nil { if err != nil {
return nil, fmt.Errorf("generate machine ID: %w", err) return nil, fmt.Errorf("generate machine ID: %w", err)
@@ -125,7 +125,7 @@ func NewBootstrapConfig(name string, subnet netip.Prefix, peers ...network.PeerC
subnet = netip.PrefixFrom(network.DefaultNetwork.Addr(), network.DefaultSubnetBits) subnet = netip.PrefixFrom(network.DefaultNetwork.Addr(), network.DefaultSubnetBits)
} }
return &Config{ return &State{
ID: mid, ID: mid,
Name: name, Name: name,
Network: &network.Config{ Network: &network.Config{