// NewHealthCheck creates a new health check core plugin.
func NewHealthCheck(context context.T) *HealthCheck {
healthContext := context.With("[" + name + "]")
healthCheckStopPolicy := sdkutil.NewStopPolicy(name, 10)
svc := ssm.NewService()
return &HealthCheck{
context: healthContext,
healthCheckStopPolicy: healthCheckStopPolicy,
service: svc,
}
}
func main() {
defer log.Flush()
commandPtr := flag.String("c", "", "a command")
scriptFilePtr := flag.String("f", "", "a script file")
dirPtr := flag.String("d", "", "working directory")
bucketNamePtr := flag.String("b", "", "bucket name")
keyPrefixPtr := flag.String("k", "", "bucket key prefix")
cancelPtr := flag.Bool("cancel", false, "cancel command on key press")
typePtr := flag.String("type", "", "instance type (windows, ubuntu or aml)")
instanceIDPtr := flag.String("i", "", "instance id")
regionPtr := flag.String("r", "us-east-1", "instance region")
flag.Parse()
var timeout int64 = 10000
timeoutPtr := &timeout
var err error
err = platform.SetRegion(*regionPtr)
if err != nil {
log.Error("please specify the region to use.")
return
}
if *commandPtr == "" && *scriptFilePtr == "" {
fmt.Println("No commands specified (use either -c or -f).")
flag.Usage()
return
}
if *keyPrefixPtr == "" {
keyPrefixPtr = nil
}
if *bucketNamePtr == "" {
bucketNamePtr = nil
if keyPrefixPtr != nil {
defaultBucket := "ec2configservice-ssm-logs"
bucketNamePtr = &defaultBucket
}
}
var cc consoleConfig
err = jsonutil.UnmarshalFile("integration-cli.json", &cc)
if err != nil {
log.Error("error parsing consoleConfig ", err)
return
}
// specific instance is provided use only that
if *instanceIDPtr != "" {
cc.Instances = make(map[string]string)
if *typePtr != "" {
cc.Instances[*instanceIDPtr] = *typePtr
} else {
cc.Instances[*instanceIDPtr] = "aml"
}
} else {
// other wise select or filter from the consoleConfig file list
if *typePtr != "" {
for instanceID, instanceType := range cc.Instances {
if instanceType != *typePtr {
delete(cc.Instances, instanceID)
}
}
if len(cc.Instances) == 0 {
log.Error("no instances of type ", *typePtr)
return
}
}
}
ssmSvc := ssm.NewService()
if ssmSvc == nil {
log.Error("couldn't create ssm service.")
return
}
var instanceIDs []string
// first get windows instances (bug in SSM if first instance is not win)
for instanceID, instanceType := range cc.Instances {
if instanceType == "windows" {
instanceIDs = append(instanceIDs, instanceID)
}
}
// then get rest of the instances
for instanceID, instanceType := range cc.Instances {
if instanceType != "windows" {
instanceIDs = append(instanceIDs, instanceID)
}
}
docName := "AWS-BETA-RunShellScript"
if runtime.GOOS == "windows" {
docName = "AWS-RunPowerShellScript"
}
var commands []string
if *commandPtr != "" {
commands = []string{*commandPtr}
} else {
f, err := os.Open(*scriptFilePtr)
if err != nil {
log.Error(err)
return
}
scanner := bufio.NewScanner(f)
for scanner.Scan() {
commands = append(commands, scanner.Text())
}
}
for i := 0; i < len(commands); i++ {
// escape backslashes
commands[i] = strings.Replace(commands[i], `\`, `\\`, -1)
// escape double quotes
commands[i] = strings.Replace(commands[i], `"`, `\"`, -1)
// escape single quotes
// commands[i] = strings.Replace(commands[i], `'`, `\'`, -1)
}
parameters := map[string][]*string{
"commands": aws.StringSlice(commands),
"workingDirectory": aws.StringSlice([]string{*dirPtr}),
}
log.Infof("Sending command %v", parameters)
cmd, err := ssmSvc.SendCommand(log, docName, instanceIDs, parameters, timeoutPtr, bucketNamePtr, keyPrefixPtr)
if cmd == nil || cmd.Command == nil || cmd.Command.CommandId == nil {
log.Error("command was not created. Aborting!")
return
}
if *cancelPtr {
log.Info("Press any key to cancel command")
var b = make([]byte, 1)
os.Stdin.Read(b)
log.Info("Canceling command ")
ssmSvc.CancelCommand(log, *cmd.Command.CommandId, instanceIDs)
}
log.Info("================== Looping for results ================")
log.Flush()
time.Sleep(1000 * time.Millisecond)
for {
done := true
inst:
for instanceID, instanceType := range cc.Instances {
descr := fmt.Sprintf("%v [%v]", instanceID, instanceType)
out, err := ssmSvc.ListCommandInvocations(log, instanceID, *cmd.Command.CommandId)
if err != nil {
continue
}
for _, inv := range out.CommandInvocations {
if *inv.Status == "Pending" {
log.Infof("Instance %v is in status %v; waiting some more", descr, *inv.Status)
done = false
continue inst
}
data, err := json.Marshal(inv)
if err != nil {
log.Error(err)
continue
}
log.Debug(jsonutil.Indent(string(data)))
for _, cp := range inv.CommandPlugins {
if cp.Output == nil {
log.Errorf("Output Nil for %v", descr)
continue
}
var o interface{}
err := json.Unmarshal([]byte(*cp.Output), &o)
if err != nil {
log.Errorf("error parsing %v\n err=%v", *cp.Output, err)
}
log.Info(descr, " : ", prettyPrint(o, 0))
}
}
}
if done {
break
}
time.Sleep(3000 * time.Millisecond)
}
// c.Wait()*/
}