Compare commits
17 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
| a8a5d60e31 | |||
| e46eec3c2a | |||
| cdb8f1ea53 | |||
| 032342279d | |||
| 7f1286c101 | |||
| 20ee1328e3 | |||
| 046c9a508b | |||
|
f6766547f9
|
|||
| c9314b3b7c | |||
| 52774336c2 | |||
| 36f1619d69 | |||
| f1a3606016 | |||
| f18202b332 | |||
| 6aec17d079 | |||
| ab8ffdd507 | |||
| 422e119f88 | |||
|
d1198328f6
|
Vendored
+2
-1
@@ -1,10 +1,11 @@
|
|||||||
{
|
{
|
||||||
"cSpell.words": [
|
"cSpell.words": [
|
||||||
|
"asciigoat",
|
||||||
"ceph",
|
"ceph",
|
||||||
|
"cyclomatic",
|
||||||
"darvaza",
|
"darvaza",
|
||||||
"gofrs",
|
"gofrs",
|
||||||
"jpictl",
|
"jpictl",
|
||||||
"Lookuper",
|
|
||||||
"Wrapf",
|
"Wrapf",
|
||||||
"zerolog"
|
"zerolog"
|
||||||
]
|
]
|
||||||
|
|||||||
+10
-9
@@ -19,24 +19,19 @@ type Config struct {
|
|||||||
Base string
|
Base string
|
||||||
Domain string
|
Domain string
|
||||||
|
|
||||||
ConfigFile *string
|
ConfigFile string
|
||||||
}
|
}
|
||||||
|
|
||||||
var cfg = &Config{
|
var cfg = &Config{
|
||||||
Base: "m",
|
Base: "m",
|
||||||
Domain: "jpi.cloud",
|
Domain: "jpi.cloud",
|
||||||
|
|
||||||
ConfigFile: rootCmd.PersistentFlags().
|
|
||||||
StringP("config-file", "f", DefaultConfigFile, "config file (JSON or YAML)"),
|
|
||||||
}
|
}
|
||||||
|
|
||||||
// LoadZones loads all zones and machines in the config directory
|
// LoadZones loads all zones and machines in the config directory
|
||||||
// or file
|
// or file
|
||||||
func (cfg *Config) LoadZones(resolve bool) (*cluster.Cluster, error) {
|
func (cfg *Config) LoadZones(resolve bool) (*cluster.Cluster, error) {
|
||||||
configFile := *cfg.ConfigFile
|
|
||||||
|
|
||||||
// try config file first
|
// try config file first
|
||||||
zones, err := cluster.NewFromConfig(configFile,
|
zones, err := cluster.NewFromConfig(cfg.ConfigFile,
|
||||||
cluster.ResolvePublicAddresses(resolve),
|
cluster.ResolvePublicAddresses(resolve),
|
||||||
cluster.WithLogger(log),
|
cluster.WithLogger(log),
|
||||||
)
|
)
|
||||||
@@ -45,9 +40,9 @@ func (cfg *Config) LoadZones(resolve bool) (*cluster.Cluster, error) {
|
|||||||
case err == nil:
|
case err == nil:
|
||||||
// file was good
|
// file was good
|
||||||
return zones, nil
|
return zones, nil
|
||||||
case !os.IsNotExist(err) || configFile != DefaultConfigFile:
|
case !os.IsNotExist(err) || cfg.ConfigFile != DefaultConfigFile:
|
||||||
// file was bad
|
// file was bad
|
||||||
return nil, core.Wrapf(err, "NewFromConfig(%q)", configFile)
|
return nil, core.Wrapf(err, "NewFromConfig(%q)", cfg.ConfigFile)
|
||||||
}
|
}
|
||||||
|
|
||||||
// default file doesn't exist. scan instead.
|
// default file doesn't exist. scan instead.
|
||||||
@@ -56,3 +51,9 @@ func (cfg *Config) LoadZones(resolve bool) (*cluster.Cluster, error) {
|
|||||||
cluster.WithLogger(log),
|
cluster.WithLogger(log),
|
||||||
)
|
)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func init() {
|
||||||
|
rootCmd.PersistentFlags().
|
||||||
|
StringVarP(&cfg.ConfigFile, "config-file", "f",
|
||||||
|
DefaultConfigFile, "config file (JSON or YAML)")
|
||||||
|
}
|
||||||
|
|||||||
@@ -1,48 +0,0 @@
|
|||||||
package main
|
|
||||||
|
|
||||||
import (
|
|
||||||
"github.com/spf13/cobra"
|
|
||||||
)
|
|
||||||
|
|
||||||
// Command
|
|
||||||
var dnsCmd = &cobra.Command{
|
|
||||||
Use: "dns",
|
|
||||||
}
|
|
||||||
|
|
||||||
var dnsSyncCmd = &cobra.Command{
|
|
||||||
Use: "sync",
|
|
||||||
RunE: func(_ *cobra.Command, _ []string) error {
|
|
||||||
_, err := cfg.LoadZones(true)
|
|
||||||
return err
|
|
||||||
},
|
|
||||||
}
|
|
||||||
|
|
||||||
var dnsWriteCmd = &cobra.Command{
|
|
||||||
Use: "write",
|
|
||||||
RunE: func(_ *cobra.Command, _ []string) error {
|
|
||||||
return nil
|
|
||||||
},
|
|
||||||
}
|
|
||||||
|
|
||||||
var dnsAddCmd = &cobra.Command{
|
|
||||||
Use: "add",
|
|
||||||
RunE: func(_ *cobra.Command, _ []string) error {
|
|
||||||
return nil
|
|
||||||
},
|
|
||||||
}
|
|
||||||
|
|
||||||
var dnsRemoveCmd = &cobra.Command{
|
|
||||||
Use: "remove",
|
|
||||||
RunE: func(_ *cobra.Command, _ []string) error {
|
|
||||||
return nil
|
|
||||||
},
|
|
||||||
}
|
|
||||||
|
|
||||||
func init() {
|
|
||||||
rootCmd.AddCommand(dnsCmd)
|
|
||||||
|
|
||||||
dnsCmd.AddCommand(dnsSyncCmd)
|
|
||||||
dnsCmd.AddCommand(dnsWriteCmd)
|
|
||||||
dnsCmd.AddCommand(dnsAddCmd)
|
|
||||||
dnsCmd.AddCommand(dnsRemoveCmd)
|
|
||||||
}
|
|
||||||
+3
-2
@@ -45,8 +45,9 @@ const encoding = YAMLEncoding
|
|||||||
|
|
||||||
// Command
|
// Command
|
||||||
var dumpCmd = &cobra.Command{
|
var dumpCmd = &cobra.Command{
|
||||||
Use: "dump",
|
Use: "dump",
|
||||||
Short: "generates a text representation of the config",
|
Short: "generates a text representation of the config",
|
||||||
|
PreRun: setVerbosity,
|
||||||
RunE: func(_ *cobra.Command, _ []string) error {
|
RunE: func(_ *cobra.Command, _ []string) error {
|
||||||
var buf bytes.Buffer
|
var buf bytes.Buffer
|
||||||
var enc Encoder
|
var enc Encoder
|
||||||
|
|||||||
+3
-2
@@ -8,8 +8,9 @@ import (
|
|||||||
|
|
||||||
// Command
|
// Command
|
||||||
var envCmd = &cobra.Command{
|
var envCmd = &cobra.Command{
|
||||||
Use: "env",
|
Use: "env",
|
||||||
Short: "generates environment variables for shell scripts",
|
Short: "generates environment variables for shell scripts",
|
||||||
|
PreRun: setVerbosity,
|
||||||
RunE: func(_ *cobra.Command, _ []string) error {
|
RunE: func(_ *cobra.Command, _ []string) error {
|
||||||
m, err := cfg.LoadZones(false)
|
m, err := cfg.LoadZones(false)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
|
|||||||
@@ -20,8 +20,9 @@ var gatewayCmd = &cobra.Command{
|
|||||||
|
|
||||||
// gateway set
|
// gateway set
|
||||||
var gatewaySetCmd = &cobra.Command{
|
var gatewaySetCmd = &cobra.Command{
|
||||||
Use: "set",
|
Use: "set",
|
||||||
Short: "gateway set sets machines as gateways",
|
Short: "gateway set sets machines as gateways",
|
||||||
|
PreRun: setVerbosity,
|
||||||
RunE: func(_ *cobra.Command, args []string) error {
|
RunE: func(_ *cobra.Command, args []string) error {
|
||||||
m, err := cfg.LoadZones(false)
|
m, err := cfg.LoadZones(false)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
|
|||||||
@@ -3,15 +3,9 @@ package main
|
|||||||
import (
|
import (
|
||||||
"fmt"
|
"fmt"
|
||||||
|
|
||||||
"darvaza.org/sidecar/pkg/logger/zerolog"
|
|
||||||
"darvaza.org/slog"
|
"darvaza.org/slog"
|
||||||
)
|
)
|
||||||
|
|
||||||
// TODO: make log level configurable via flags
|
|
||||||
var (
|
|
||||||
log = zerolog.New(nil, slog.Debug)
|
|
||||||
)
|
|
||||||
|
|
||||||
// fatal is a convenience wrapper for slog.Logger.Fatal().Print()
|
// fatal is a convenience wrapper for slog.Logger.Fatal().Print()
|
||||||
func fatal(err error, msg string, args ...any) {
|
func fatal(err error, msg string, args ...any) {
|
||||||
l := log.Fatal()
|
l := log.Fatal()
|
||||||
|
|||||||
+23
-4
@@ -2,6 +2,8 @@
|
|||||||
package main
|
package main
|
||||||
|
|
||||||
import (
|
import (
|
||||||
|
"darvaza.org/sidecar/pkg/logger/zerolog"
|
||||||
|
"darvaza.org/slog"
|
||||||
"github.com/spf13/cobra"
|
"github.com/spf13/cobra"
|
||||||
)
|
)
|
||||||
|
|
||||||
@@ -10,13 +12,30 @@ const (
|
|||||||
CmdName = "jpictl"
|
CmdName = "jpictl"
|
||||||
)
|
)
|
||||||
|
|
||||||
var rootCmd = &cobra.Command{
|
var (
|
||||||
Use: CmdName,
|
log = zerolog.New(nil, slog.Error)
|
||||||
Short: "control tool for jpi.cloud",
|
verbosity int
|
||||||
}
|
rootCmd = &cobra.Command{
|
||||||
|
Use: CmdName,
|
||||||
|
Short: "control tool for jpi.cloud",
|
||||||
|
}
|
||||||
|
)
|
||||||
|
|
||||||
func main() {
|
func main() {
|
||||||
if err := rootCmd.Execute(); err != nil {
|
if err := rootCmd.Execute(); err != nil {
|
||||||
fatal(err, "")
|
fatal(err, "")
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func init() {
|
||||||
|
rootCmd.PersistentFlags().CountVarP(&verbosity, "verbosity", "v",
|
||||||
|
"increase the verbosity level to Warn, Info or Debug")
|
||||||
|
}
|
||||||
|
|
||||||
|
func setVerbosity(_ *cobra.Command, _ []string) {
|
||||||
|
desired := int8(slog.Error) + int8(verbosity)
|
||||||
|
if desired > 6 {
|
||||||
|
desired = 6
|
||||||
|
}
|
||||||
|
log = log.WithLevel(slog.LogLevel(desired))
|
||||||
|
}
|
||||||
|
|||||||
+3
-2
@@ -6,8 +6,9 @@ import (
|
|||||||
|
|
||||||
// Command
|
// Command
|
||||||
var writeCmd = &cobra.Command{
|
var writeCmd = &cobra.Command{
|
||||||
Use: "write",
|
Use: "write",
|
||||||
Short: "rewrites all config files",
|
Short: "rewrites all config files",
|
||||||
|
PreRun: setVerbosity,
|
||||||
RunE: func(_ *cobra.Command, _ []string) error {
|
RunE: func(_ *cobra.Command, _ []string) error {
|
||||||
m, err := cfg.LoadZones(false)
|
m, err := cfg.LoadZones(false)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
|
|||||||
@@ -2,6 +2,8 @@ module git.jpi.io/amery/jpictl
|
|||||||
|
|
||||||
go 1.19
|
go 1.19
|
||||||
|
|
||||||
|
replace asciigoat.org/ini => ../../../asciigoat.org/ini
|
||||||
|
|
||||||
require (
|
require (
|
||||||
asciigoat.org/ini v0.2.5
|
asciigoat.org/ini v0.2.5
|
||||||
darvaza.org/core v0.9.8
|
darvaza.org/core v0.9.8
|
||||||
@@ -11,8 +13,6 @@ require (
|
|||||||
darvaza.org/slog/handlers/discard v0.4.5
|
darvaza.org/slog/handlers/discard v0.4.5
|
||||||
github.com/gofrs/uuid/v5 v5.0.0
|
github.com/gofrs/uuid/v5 v5.0.0
|
||||||
github.com/hack-pad/hackpadfs v0.2.1
|
github.com/hack-pad/hackpadfs v0.2.1
|
||||||
github.com/libdns/cloudflare v0.1.0
|
|
||||||
github.com/libdns/libdns v0.2.1
|
|
||||||
github.com/mgechev/revive v1.3.3
|
github.com/mgechev/revive v1.3.3
|
||||||
github.com/spf13/cobra v1.7.0
|
github.com/spf13/cobra v1.7.0
|
||||||
golang.org/x/crypto v0.12.0
|
golang.org/x/crypto v0.12.0
|
||||||
|
|||||||
@@ -1,7 +1,5 @@
|
|||||||
asciigoat.org/core v0.3.9 h1:hgDDz4ecm3ZvehX++m8A/IzAt+B5oDPiRtxatzfUHPQ=
|
asciigoat.org/core v0.3.9 h1:hgDDz4ecm3ZvehX++m8A/IzAt+B5oDPiRtxatzfUHPQ=
|
||||||
asciigoat.org/core v0.3.9/go.mod h1:CAaHwyw8MpAq4a1MYtN2dxJrsK+hmIdW50OndaQZYPI=
|
asciigoat.org/core v0.3.9/go.mod h1:CAaHwyw8MpAq4a1MYtN2dxJrsK+hmIdW50OndaQZYPI=
|
||||||
asciigoat.org/ini v0.2.5 h1:4gRIp9rU+XQt8+HMqZO5R7GavMv9Yl2+N+je6djDIAE=
|
|
||||||
asciigoat.org/ini v0.2.5/go.mod h1:gmXzJ9XFqf1NLk5nQkj04USQ4tMtdRJHNQX6vp3DzjU=
|
|
||||||
darvaza.org/core v0.9.8 h1:luLxgfUc2pzuusYPo/Z/dC/qr9XZPKpSQw8/kS7zNUM=
|
darvaza.org/core v0.9.8 h1:luLxgfUc2pzuusYPo/Z/dC/qr9XZPKpSQw8/kS7zNUM=
|
||||||
darvaza.org/core v0.9.8/go.mod h1:Dbme64naxeshQfxcVJX9ZT7AiGyIY8kldfuELVtf8mw=
|
darvaza.org/core v0.9.8/go.mod h1:Dbme64naxeshQfxcVJX9ZT7AiGyIY8kldfuELVtf8mw=
|
||||||
darvaza.org/resolver v0.5.4 h1:dlSBNV14yYsp7Kg7ipwYOMNsLbrpeXa8Z0HBTa0Ryxs=
|
darvaza.org/resolver v0.5.4 h1:dlSBNV14yYsp7Kg7ipwYOMNsLbrpeXa8Z0HBTa0Ryxs=
|
||||||
@@ -36,11 +34,6 @@ github.com/hack-pad/hackpadfs v0.2.1 h1:FelFhIhv26gyjujoA/yeFO+6YGlqzmc9la/6iKMI
|
|||||||
github.com/hack-pad/hackpadfs v0.2.1/go.mod h1:khQBuCEwGXWakkmq8ZiFUvUZz84ZkJ2KNwKvChs4OrU=
|
github.com/hack-pad/hackpadfs v0.2.1/go.mod h1:khQBuCEwGXWakkmq8ZiFUvUZz84ZkJ2KNwKvChs4OrU=
|
||||||
github.com/inconshreveable/mousetrap v1.1.0 h1:wN+x4NVGpMsO7ErUn/mUI3vEoE6Jt13X2s0bqwp9tc8=
|
github.com/inconshreveable/mousetrap v1.1.0 h1:wN+x4NVGpMsO7ErUn/mUI3vEoE6Jt13X2s0bqwp9tc8=
|
||||||
github.com/inconshreveable/mousetrap v1.1.0/go.mod h1:vpF70FUmC8bwa3OWnCshd2FqLfsEA9PFc4w1p2J65bw=
|
github.com/inconshreveable/mousetrap v1.1.0/go.mod h1:vpF70FUmC8bwa3OWnCshd2FqLfsEA9PFc4w1p2J65bw=
|
||||||
github.com/libdns/cloudflare v0.1.0 h1:93WkJaGaiXCe353LHEP36kAWCUw0YjFqwhkBkU2/iic=
|
|
||||||
github.com/libdns/cloudflare v0.1.0/go.mod h1:a44IP6J1YH6nvcNl1PverfJviADgXUnsozR3a7vBKN8=
|
|
||||||
github.com/libdns/libdns v0.2.0/go.mod h1:yQCXzk1lEZmmCPa857bnk4TsOiqYasqpyOEeSObbb40=
|
|
||||||
github.com/libdns/libdns v0.2.1 h1:Wu59T7wSHRgtA0cfxC+n1c/e+O3upJGWytknkmFEDis=
|
|
||||||
github.com/libdns/libdns v0.2.1/go.mod h1:yQCXzk1lEZmmCPa857bnk4TsOiqYasqpyOEeSObbb40=
|
|
||||||
github.com/mattn/go-colorable v0.1.12/go.mod h1:u5H1YNBxpqRaxsYJYSkiCWKzEfiAb1Gb520KVy5xxl4=
|
github.com/mattn/go-colorable v0.1.12/go.mod h1:u5H1YNBxpqRaxsYJYSkiCWKzEfiAb1Gb520KVy5xxl4=
|
||||||
github.com/mattn/go-colorable v0.1.13 h1:fFA4WZxdEF4tXPZVKMLwD8oUnCTTo08duU7wxecdEvA=
|
github.com/mattn/go-colorable v0.1.13 h1:fFA4WZxdEF4tXPZVKMLwD8oUnCTTo08duU7wxecdEvA=
|
||||||
github.com/mattn/go-colorable v0.1.13/go.mod h1:7S9/ev0klgBDR4GtXTXX8a3vIGJpMovkB8vQcUbaXHg=
|
github.com/mattn/go-colorable v0.1.13/go.mod h1:7S9/ev0klgBDR4GtXTXX8a3vIGJpMovkB8vQcUbaXHg=
|
||||||
|
|||||||
@@ -2,6 +2,7 @@ package cluster
|
|||||||
|
|
||||||
import (
|
import (
|
||||||
"encoding/json"
|
"encoding/json"
|
||||||
|
"fmt"
|
||||||
"os"
|
"os"
|
||||||
|
|
||||||
"gopkg.in/yaml.v3"
|
"gopkg.in/yaml.v3"
|
||||||
@@ -22,25 +23,58 @@ func (m *Cluster) init(opts *ScanOptions) error {
|
|||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (m *Cluster) initZones(_ *ScanOptions) error {
|
func (m *Cluster) initZones(opts *ScanOptions) error {
|
||||||
var err error
|
var err error
|
||||||
|
|
||||||
m.ForEachZone(func(z *Zone) bool {
|
m.ForEachZone(func(z *Zone) bool {
|
||||||
z.zones = m
|
err = m.initZone(z, opts)
|
||||||
z.logger = m
|
|
||||||
|
|
||||||
z.ForEachMachine(func(p *Machine) bool {
|
|
||||||
p.zone = z
|
|
||||||
p.logger = z
|
|
||||||
return err != nil
|
|
||||||
})
|
|
||||||
|
|
||||||
return err != nil
|
return err != nil
|
||||||
})
|
})
|
||||||
|
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func (m *Cluster) initZone(z *Zone, _ *ScanOptions) error {
|
||||||
|
var hasMissing bool
|
||||||
|
var lastMachineID int
|
||||||
|
|
||||||
|
z.zones = m
|
||||||
|
z.logger = m
|
||||||
|
|
||||||
|
z.ForEachMachine(func(p *Machine) bool {
|
||||||
|
p.zone = z
|
||||||
|
p.logger = z
|
||||||
|
|
||||||
|
switch {
|
||||||
|
case p.ID == 0:
|
||||||
|
hasMissing = true
|
||||||
|
case p.ID > lastMachineID:
|
||||||
|
lastMachineID = z.ID
|
||||||
|
}
|
||||||
|
|
||||||
|
return false
|
||||||
|
})
|
||||||
|
|
||||||
|
if hasMissing {
|
||||||
|
next := lastMachineID + 1
|
||||||
|
|
||||||
|
z.ForEachMachine(func(p *Machine) bool {
|
||||||
|
if p.ID == 0 {
|
||||||
|
p.ID, next = next, next+1
|
||||||
|
}
|
||||||
|
|
||||||
|
return false
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
z.ForEachMachine(func(p *Machine) bool {
|
||||||
|
p.Name = fmt.Sprintf("%s-%v", z.Name, p.ID)
|
||||||
|
return false
|
||||||
|
})
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
func decodeConfigData(data []byte) (out *Cluster, err error) {
|
func decodeConfigData(data []byte) (out *Cluster, err error) {
|
||||||
// try JSON first
|
// try JSON first
|
||||||
out = new(Cluster)
|
out = new(Cluster)
|
||||||
|
|||||||
@@ -15,8 +15,3 @@ func DefaultLogger() slog.Logger {
|
|||||||
func DefaultLookuper() resolver.Lookuper {
|
func DefaultLookuper() resolver.Lookuper {
|
||||||
return resolver.NewCloudflareLookuper()
|
return resolver.NewCloudflareLookuper()
|
||||||
}
|
}
|
||||||
|
|
||||||
// DefaultResolver returns a [resolver.Resolver] using Cloudflare's 1.1.1.1
|
|
||||||
func DefaultResolver() resolver.Resolver {
|
|
||||||
return resolver.NewResolver(DefaultLookuper())
|
|
||||||
}
|
|
||||||
|
|||||||
@@ -1,12 +0,0 @@
|
|||||||
// Package dns manages DNS entries for the cluster
|
|
||||||
package dns
|
|
||||||
|
|
||||||
// A Region describes where a Zone is
|
|
||||||
type Region struct {
|
|
||||||
// Name is the identifier of this Region
|
|
||||||
Name string
|
|
||||||
// Regions are a list of (sub)regions that belong to this Region
|
|
||||||
Regions []string
|
|
||||||
// Zones are a list of Zones that directly belong to this Region
|
|
||||||
Zones []string
|
|
||||||
}
|
|
||||||
@@ -1,121 +0,0 @@
|
|||||||
package dns
|
|
||||||
|
|
||||||
import (
|
|
||||||
"darvaza.org/core"
|
|
||||||
"darvaza.org/resolver"
|
|
||||||
"darvaza.org/slog"
|
|
||||||
|
|
||||||
"git.jpi.io/amery/jpictl/pkg/cluster"
|
|
||||||
)
|
|
||||||
|
|
||||||
// Manager is a DNS Manager instance
|
|
||||||
type Manager struct {
|
|
||||||
p Provider
|
|
||||||
r resolver.Resolver
|
|
||||||
l slog.Logger
|
|
||||||
}
|
|
||||||
|
|
||||||
// ManagerOption configures a Manager
|
|
||||||
type ManagerOption func(*Manager) error
|
|
||||||
|
|
||||||
func newErrorManagerOption(err error, hint string) ManagerOption {
|
|
||||||
return func(*Manager) error {
|
|
||||||
if hint != "" {
|
|
||||||
err = core.Wrap(err, hint)
|
|
||||||
}
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// WithProvider attaches a libdns Provider to the Manager
|
|
||||||
func WithProvider(p Provider) ManagerOption {
|
|
||||||
var err error
|
|
||||||
|
|
||||||
if p == nil {
|
|
||||||
p, err = DefaultDNSProvider()
|
|
||||||
}
|
|
||||||
|
|
||||||
if err != nil {
|
|
||||||
return newErrorManagerOption(err, "WithProvider")
|
|
||||||
}
|
|
||||||
|
|
||||||
return func(mgr *Manager) error {
|
|
||||||
mgr.p = p
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// WithLookuper attaches a resolver.Lookuper to the Manager
|
|
||||||
func WithLookuper(h resolver.Lookuper) ManagerOption {
|
|
||||||
if h == nil {
|
|
||||||
h = cluster.DefaultLookuper()
|
|
||||||
}
|
|
||||||
|
|
||||||
return func(mgr *Manager) error {
|
|
||||||
mgr.r = resolver.NewResolver(h)
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// WithResolver attaches a resolver.Resolver to the Manager
|
|
||||||
func WithResolver(r resolver.Resolver) ManagerOption {
|
|
||||||
if r == nil {
|
|
||||||
r = cluster.DefaultResolver()
|
|
||||||
}
|
|
||||||
|
|
||||||
return func(mgr *Manager) error {
|
|
||||||
mgr.r = r
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// WithLogger attaches a logger to the Manager
|
|
||||||
func WithLogger(log slog.Logger) ManagerOption {
|
|
||||||
if log == nil {
|
|
||||||
log = cluster.DefaultLogger()
|
|
||||||
}
|
|
||||||
|
|
||||||
return func(mgr *Manager) error {
|
|
||||||
mgr.l = log
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func (mgr *Manager) setDefaults() error {
|
|
||||||
var opts []ManagerOption
|
|
||||||
|
|
||||||
if mgr.p == nil {
|
|
||||||
opts = append(opts, WithProvider(nil))
|
|
||||||
}
|
|
||||||
|
|
||||||
if mgr.r == nil {
|
|
||||||
opts = append(opts, WithResolver(nil))
|
|
||||||
}
|
|
||||||
|
|
||||||
if mgr.l == nil {
|
|
||||||
opts = append(opts, WithLogger(nil))
|
|
||||||
}
|
|
||||||
|
|
||||||
for _, opt := range opts {
|
|
||||||
if err := opt(mgr); err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
// NewManager creates a DNS manager with the
|
|
||||||
func NewManager(opts ...ManagerOption) (*Manager, error) {
|
|
||||||
mgr := new(Manager)
|
|
||||||
|
|
||||||
for _, opt := range opts {
|
|
||||||
if err := opt(mgr); err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
if err := mgr.setDefaults(); err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
return mgr, nil
|
|
||||||
}
|
|
||||||
@@ -1,36 +0,0 @@
|
|||||||
package dns
|
|
||||||
|
|
||||||
import (
|
|
||||||
"fmt"
|
|
||||||
"os"
|
|
||||||
|
|
||||||
"github.com/libdns/cloudflare"
|
|
||||||
"github.com/libdns/libdns"
|
|
||||||
)
|
|
||||||
|
|
||||||
const (
|
|
||||||
// CloudflareAPIToken is the environment variable
|
|
||||||
// containing the API Token
|
|
||||||
CloudflareAPIToken = "CLOUDFLARE_DNS_API_TOKEN"
|
|
||||||
)
|
|
||||||
|
|
||||||
// Provider manages DNS entries
|
|
||||||
type Provider interface {
|
|
||||||
libdns.RecordGetter
|
|
||||||
libdns.RecordDeleter
|
|
||||||
}
|
|
||||||
|
|
||||||
// DefaultDNSProvider returns a cloudflare DNS provider
|
|
||||||
// using an API Token from env [CloudflareAPIToken]
|
|
||||||
func DefaultDNSProvider() (*cloudflare.Provider, error) {
|
|
||||||
s := os.Getenv(CloudflareAPIToken)
|
|
||||||
if s == "" {
|
|
||||||
return nil, fmt.Errorf("%q: %s", CloudflareAPIToken, "not found")
|
|
||||||
}
|
|
||||||
|
|
||||||
p := &cloudflare.Provider{
|
|
||||||
APIToken: s,
|
|
||||||
}
|
|
||||||
|
|
||||||
return p, nil
|
|
||||||
}
|
|
||||||
Reference in New Issue
Block a user