Permalink
Switch branches/tags
Find file Copy path
Fetching contributors…
Cannot retrieve contributors at this time
99 lines (86 sloc) 3.8 KB
package main
import (
"strconv"
"strings"
log "github.com/sirupsen/logrus"
"github.com/skeema/mybase"
"github.com/skeema/tengo"
)
func init() {
summary := "Add a new named environment to an existing host directory"
desc := `Modifies the .skeema file in an existing host directory to add a new named
environment. For example, if ` + "`" + `skeema init` + "`" + ` was previously used to create a dir
for a host with the default "production" environment, ` + "`" + `skeema add-environment` + "`" + `
could be used to define a "staging" or "development" environment pointing at a different
host and port, or perhaps a "local" environment pointing at localhost and a
socket path.`
cmd := mybase.NewCommand("add-environment", summary, desc, AddEnvHandler)
cmd.AddOption(mybase.StringOption("host", 'h', "", "Database hostname or IP address"))
cmd.AddOption(mybase.StringOption("port", 'P', "3306", "Port to use for database host"))
cmd.AddOption(mybase.StringOption("socket", 'S', "/tmp/mysql.sock", "Absolute path to Unix socket file used if host is localhost"))
cmd.AddOption(mybase.StringOption("dir", 'd', ".", "Base dir for this host's schemas"))
cmd.AddArg("environment", "", true)
CommandSuite.AddSubCommand(cmd)
}
// AddEnvHandler is the handler method for `skeema add-environment`
func AddEnvHandler(cfg *mybase.Config) error {
AddGlobalConfigFiles(cfg)
dir, err := NewDir(cfg.Get("dir"), cfg)
if err != nil {
return err
}
if !dir.Exists() {
return NewExitValue(CodeBadConfig, "In add-environment, --dir must refer to a directory that already exists")
}
if !dir.HasOptionFile() {
return NewExitValue(CodeBadConfig, "Dir %s does not have an existing .skeema file! Can only use `skeema add-environment` on a dir previously created by `skeema init`", dir)
}
hostOptionFile, err := dir.OptionFile()
if err != nil {
return NewExitValue(CodeBadInput, "Unable to read .skeema file for %s: %s", dir, err)
}
environment := cfg.Get("environment")
if environment == "" || strings.ContainsAny(environment, "[]\n\r") {
return NewExitValue(CodeBadConfig, "Environment name \"%s\" is invalid", environment)
}
if hostOptionFile.HasSection(environment) {
return NewExitValue(CodeBadConfig, "Environment name \"%s\" already defined in %s", environment, hostOptionFile.Path())
}
if !hostOptionFile.SomeSectionHasOption("host") {
return NewExitValue(CodeBadConfig, "This command should be run against a --dir whose .skeema file already defines a host for another environment")
}
// Create a tengo.Instance representing the supplied host. We intentionally
// don't actually test connectivity here though, since this command only
// manipulates the option file. We can't use dir.FirstInstance() here since
// that checks connectivity.
var inst *tengo.Instance
if !cfg.OnCLI("host") {
return NewExitValue(CodeBadConfig, "`skeema add-environment` requires --host to be supplied on CLI")
}
if instances, err := dir.Instances(); err != nil {
return err
} else if len(instances) == 0 {
return NewExitValue(CodeBadConfig, "Command line did not specify which instance to connect to")
} else {
inst = instances[0]
}
hostOptionFile.SetOptionValue(environment, "host", inst.Host)
if inst.Host == "localhost" && inst.SocketPath != "" {
hostOptionFile.SetOptionValue(environment, "socket", inst.SocketPath)
} else {
hostOptionFile.SetOptionValue(environment, "port", strconv.Itoa(inst.Port))
}
if flavor := inst.Flavor(); flavor != tengo.FlavorUnknown {
hostOptionFile.SetOptionValue(environment, "flavor", flavor.String())
}
if cfg.OnCLI("user") {
hostOptionFile.SetOptionValue(environment, "user", cfg.Get("user"))
}
// Write the option file
if err := hostOptionFile.Write(true); err != nil {
return err
}
dir.Config.MarkDirty()
log.Infof("Added environment [%s] to %s", environment, hostOptionFile.Path())
return nil
}