"fmt"
"log"
"os"
+ "os/signal"
"sort"
"strings"
"sync"
optionDryRun := flag.Bool("n", false,
"dry-run, show diff but don't perform any changes")
+ optionQuiet := flag.Bool("q", false,
+ "hide successful, non-trigger commands with no output from host changes listing")
optionLog := flag.String("log", "info", "set log `level`; "+
"levels: error, info, verbose, debug, debug2, debug3")
+ optionSshConfig := flag.String("sshconfig", "",
+ "`path` to ssh configuration file; used for tests")
flag.CommandLine.Parse(args[2:])
return err
}
cfg.DryRun = *optionDryRun
+ cfg.Quiet = *optionQuiet
cfg.LogLevel = level
+ cfg.SshConfig = *optionSshConfig
toSync, err := hostsToSync(names, allHosts, allGroups)
if err != nil {
done <- failed
}()
+ hostsLeft := make(map[string]bool)
+ for _, x := range toSync {
+ hostsLeft[x.Name] = true
+ }
+ var hostsLeftMutex sync.Mutex // protects hostsLeft
+
+ // Show unfinished hosts on Ctrl-C
+ sigint := make(chan os.Signal, 1) // buffered for Notify()
+ signal.Notify(sigint, os.Interrupt) // = SIGINT = Ctrl-C
+ go func() {
+ // Running `ssh` processes get killed by SIGINT which is sent
+ // to all processes
+
+ <-sigint
+ log.Print("Received SIGINT, aborting ...")
+
+ // Print all queued events
+ events <- Event{} // poison pill
+ <-done
+ // "races" with <-done in the main function and will hang here
+ // if the other is faster. This is fine because then all hosts
+ // were synced successfully.
+
+ hostsLeftMutex.Lock()
+ var hosts []string
+ for x := range hostsLeft {
+ hosts = append(hosts, x)
+ }
+ sort.Strings(hosts)
+ log.Fatalf("Failed to sync %s", strings.Join(hosts, ", "))
+ }()
+
// Sync all hosts concurrently
var wg sync.WaitGroup
for _, x := range toSync {
}
}
wg.Done()
+
+ hostsLeftMutex.Lock()
+ defer hostsLeftMutex.Unlock()
+ delete(hostsLeft, x.Name)
}()
}
func hostsToSync(names []string, allHosts *config.Hosts,
allGroups map[string][]string) ([]*config.Host, error) {
+ detectedMap := make(map[string]bool)
+ for _, x := range config.TransitivelyDetectedGroups(allGroups) {
+ detectedMap[x] = true
+ }
+
+ const detectedErr = `
+
+Groups depending on "detected" groups cannot be used to select hosts as these
+are only available after the hosts were contacted.
+`
+
nameMap := make(map[string]bool)
for _, x := range names {
+ if detectedMap[x] {
+ return nil, fmt.Errorf(
+ "group %q depends on \"detected\" groups%s",
+ x, detectedErr)
+ }
nameMap[x] = true
}
nameMatched := make(map[string]bool)
- // To detect typos we must check all given names but only want to add
- // each match once
+ // To detect typos we must check all given names but one host can be
+ // matched by multiple names (e.g. two groups with overlapping hosts)
hostMatched := make(map[string]bool)
var res []*config.Host
nameMatched[host.Name] = true
}
- // TODO: don't permit groups which contain "detected" groups
- // because these are not available yet
groups, err := config.ResolveHostGroups(host.Name,
allGroups, nil)
if err != nil {
}()
// Connect to remote host
- err := conn.DialSSH(s.host.Name)
+ err := conn.DialSSH(s.host.SshUser, s.host.Name, s.config.SshConfig)
if err != nil {
return err
}