WIP

Signed-off-by: Alejandro Mery <amery@jpi.io>

cmd/jpictl/config.go

@@ -1,7 +1,5 @@
 package main
 
-import "git.jpi.io/amery/jpictl/pkg/zones"
-
 // Config describes the repository
 type Config struct {
 	Base   string
@@ -12,8 +10,3 @@ var cfg = &Config{
 	Base:   "./m",
 	Domain: "m.jpi.cloud",
 }
-
-// LoadZones loads all zones and machines in the config directory
-func (cfg *Config) LoadZones() (*zones.Zones, error) {
-	return zones.New(cfg.Base, cfg.Domain)
-}

cmd/jpictl/dump.go

@@ -9,6 +9,8 @@ import (
 	"github.com/burntSushi/toml"
 	"github.com/spf13/cobra"
 	"gopkg.in/yaml.v3"
+
+	"git.jpi.io/amery/jpictl/pkg/zones"
 )
 
 // Encoder represents an object that encodes another internally
@@ -58,7 +60,7 @@ var dumpCmd = &cobra.Command{
 		var buf bytes.Buffer
 		var enc Encoder
 
-		m, err := cfg.LoadZones()
+		m, err := zones.New(cfg.Base, cfg.Domain)
 		if err != nil {
 			return err
 		}

cmd/jpictl/env.go

@@ -4,6 +4,8 @@ import (
 	"os"
 
 	"github.com/spf13/cobra"
+
+	"git.jpi.io/amery/jpictl/pkg/zones"
 )
 
 // Command
@@ -11,7 +13,7 @@ var envCmd = &cobra.Command{
 	Use:   "env",
 	Short: "generates environment variables for shell scripts",
 	RunE: func(_ *cobra.Command, _ []string) error {
-		m, err := cfg.LoadZones()
+		m, err := zones.New(cfg.Base, cfg.Domain)
 		if err != nil {
 			return err
 		}

cmd/jpictl/write.go

@@ -1,23 +0,0 @@
-package main
-
-import (
-	"github.com/spf13/cobra"
-)
-
-// Command
-var writeCmd = &cobra.Command{
-	Use:   "write",
-	Short: "rewrites all config files",
-	RunE: func(_ *cobra.Command, _ []string) error {
-		m, err := cfg.LoadZones()
-		if err != nil {
-			return err
-		}
-
-		return m.SyncAll()
-	},
-}
-
-func init() {
-	rootCmd.AddCommand(writeCmd)
-}

pkg/wireguard/config.go

@@ -37,20 +37,20 @@ type Config struct {
 }
 
 // GetAddress is a shortcut to the interface's address
-func (f *Config) GetAddress() netip.Addr {
+func (wg *Config) GetAddress() netip.Addr {
-	return f.Interface.Address
+	return wg.Interface.Address
 }
 
 // Peers tells how many peers are described
-func (f *Config) Peers() int {
+func (wg *Config) Peers() int {
-	return len(f.Peer)
+	return len(wg.Peer)
 }
 
 // WriteTo writes a Wireguard [Config] onto the provided [io.Writer]
-func (f *Config) WriteTo(w io.Writer) (int64, error) {
+func (wg *Config) WriteTo(w io.Writer) (int64, error) {
 	var buf bytes.Buffer
 
-	if err := configTemplate.Execute(&buf, f); err != nil {
+	if err := configTemplate.Execute(&buf, wg); err != nil {
 		return 0, err
 	}
 

pkg/zones/machine_rings.go

@@ -92,12 +92,12 @@ func (m *Machine) WriteWireguardKeys(ring int) error {
 		pub = ri.Keys.PrivateKey.Public().String()
 	}
 
-	err = m.WriteStringFile(key+"\n", "wg%v.key", ring)
+	err = m.WriteStringFile(key, "wg%v.key", ring)
 	if err != nil {
 		return err
 	}
 
-	err = m.WriteStringFile(pub+"\n", "wg%v.pub", ring)
+	err = m.WriteStringFile(pub, "wg%v.pub", ring)
 	if err != nil {
 		return err
 	}

pkg/zones/sync.go

@@ -1,33 +0,0 @@
-package zones
-
-// SyncAll updates all config files
-func (m *Zones) SyncAll() error {
-	for _, fn := range []func() error{
-		m.SyncAllWireguard,
-	} {
-		if err := fn(); err != nil {
-			return err
-		}
-	}
-
-	return nil
-}
-
-// SyncAllWireguard updates all wireguard config files
-func (m *Zones) SyncAllWireguard() error {
-	var err error
-
-	for ring := 0; ring < RingsCount; ring++ {
-		err = m.WriteWireguardKeys(ring)
-		if err != nil {
-			return err
-		}
-
-		err = m.SyncWireguardConfig(ring)
-		if err != nil {
-			return err
-		}
-	}
-
-	return nil
-}

pkg/zones/zone_rings.go

@@ -1,10 +1,6 @@
 package zones
 
-import (
+import "git.jpi.io/amery/jpictl/pkg/wireguard"
-	"os"
-
-	"git.jpi.io/amery/jpictl/pkg/wireguard"
-)
 
 type Ring struct {
 	Ring int
@@ -56,55 +52,3 @@ func (z *Zone) PruneWireguardConfig(ring int) error {
 
 	return err
 }
-
-// WriteWireguardKeys rewrites all wgN.{key,pub} files on this zone
-func (z *Zone) WriteWireguardKeys(ring int) error {
-	var err error
-
-	z.ForEachMachine(func(p *Machine) bool {
-		err = p.WriteWireguardKeys(ring)
-		if os.IsNotExist(err) {
-			// ignore
-			err = nil
-		}
-
-		return err != nil
-	})
-
-	return err
-}
-
-// PruneWireguardConfig removes wgN.conf files of machines with
-// the corresponding ring disabled on all zones
-func (m *Zones) PruneWireguardConfig(ring int) error {
-	var err error
-
-	m.ForEachZone(func(z *Zone) bool {
-		err = z.PruneWireguardConfig(ring)
-		return err != nil
-	})
-
-	return err
-}
-
-// WriteWireguardKeys rewrites all wgN.{key,pub} files
-func (m *Zones) WriteWireguardKeys(ring int) error {
-	var err error
-
-	m.ForEachZone(func(z *Zone) bool {
-		err = z.WriteWireguardKeys(ring)
-		return err != nil
-	})
-
-	return err
-}
-
-// SyncWireguardConfig updates all wgN.conf files for the specified
-// ring
-func (m *Zones) SyncWireguardConfig(ring int) error {
-	err := m.PruneWireguardConfig(ring)
-	if err != nil {
-		return err
-	}
-
-}