83 lines
1.9 KiB
Go
83 lines
1.9 KiB
Go
package main
|
|
|
|
import (
|
|
"os"
|
|
|
|
"darvaza.org/core"
|
|
|
|
"git.jpi.io/amery/jpictl/pkg/cluster"
|
|
)
|
|
|
|
const (
|
|
// DefaultConfigFile is read if -f/--config-file isn't specified.
|
|
// If it doesn't exist, m/ will be scanned
|
|
DefaultConfigFile = "cloud.yaml"
|
|
|
|
// DefaultClusterDir is the directory we will scan and write
|
|
// unless something else is indicated
|
|
DefaultClusterDir = "m"
|
|
|
|
// DefaultDomain indicates the domain to use unless
|
|
// something else is specified
|
|
DefaultDomain = "jpi.cloud"
|
|
)
|
|
|
|
// Config describes the repository
|
|
type Config struct {
|
|
Base string
|
|
Domain string
|
|
|
|
ConfigFile string
|
|
}
|
|
|
|
var forceScan bool
|
|
|
|
var cfg = &Config{
|
|
Base: DefaultClusterDir,
|
|
Domain: DefaultDomain,
|
|
}
|
|
|
|
// LoadZones loads all zones and machines in the config directory
|
|
// or file
|
|
func (cfg *Config) LoadZones(resolve bool) (*cluster.Cluster, error) {
|
|
var zones *cluster.Cluster
|
|
var err error
|
|
|
|
if !forceScan {
|
|
// try config file first
|
|
zones, err = cluster.NewFromConfig(cfg.ConfigFile,
|
|
cluster.ResolvePublicAddresses(resolve),
|
|
cluster.WithLogger(log),
|
|
)
|
|
|
|
switch {
|
|
case err == nil:
|
|
// file was good
|
|
return zones, nil
|
|
case !os.IsNotExist(err) || cfg.ConfigFile != DefaultConfigFile:
|
|
// file was bad
|
|
return nil, core.Wrap(err, "NewFromConfig(%q)", cfg.ConfigFile)
|
|
}
|
|
}
|
|
|
|
// default file doesn't exist. scan instead.
|
|
return cluster.NewFromDirectory(cfg.Base, cfg.Domain,
|
|
cluster.ResolvePublicAddresses(resolve),
|
|
cluster.WithLogger(log),
|
|
)
|
|
}
|
|
|
|
func init() {
|
|
flags := rootCmd.PersistentFlags()
|
|
|
|
flags.StringVarP(&cfg.Base, "scan-dir", "d",
|
|
DefaultClusterDir, "directory to scan for cluster data")
|
|
flags.StringVarP(&cfg.Domain, "domain", "D",
|
|
DefaultDomain, "domain to use for scanned data")
|
|
flags.StringVarP(&cfg.ConfigFile, "config-file", "f",
|
|
DefaultConfigFile, "config file (JSON or YAML)")
|
|
|
|
flags.BoolVarP(&forceScan, "force-scan", "S",
|
|
false, "ignore config file and scan the directory instead")
|
|
}
|