Ejemplo n.º 1
0
func Ssh(c *cli.Context) {
	if c.Bool("d") {
		fmt.Println("tag: ", c.String("t"))
		fmt.Println("value: ", c.String("v"))
		fmt.Println("args: ", c.Args()[0])
		fmt.Println("user: "******"u"))
	}
	svc := ec2.New(session.New(), &aws.Config{Region: aws.String("us-east-1")})
	cyan := color.New(color.FgCyan).SprintFunc()
	magenta := color.New(color.FgMagenta).SprintFunc()
	yellow := color.New(color.FgYellow).SprintFunc()
	green := color.New(color.FgGreen).SprintFunc()
	if len(c.String("v")) > 0 && len(c.String("t")) > 0 {
		params := &ec2.DescribeInstancesInput{
			Filters: []*ec2.Filter{
				{ // Required
					Name: aws.String(fmt.Sprintf("tag:%s", c.String("t"))),
					Values: []*string{
						aws.String(c.String("v")), // Required
						// More values...
					},
				},
				// More values...
			},
		}
		resp, err := svc.DescribeInstances(params)

		if err != nil {
			if awsErr, ok := err.(awserr.Error); ok {
				// Generic AWS error with Code, Message, and original error (if any)
				fmt.Println(awsErr.Code(), awsErr.Message(), awsErr.OrigErr())
				if reqErr, ok := err.(awserr.RequestFailure); ok {
					// A service error occurred
					fmt.Println(reqErr.Code(), reqErr.Message(), reqErr.StatusCode(), reqErr.RequestID())
				}
			} else {
				// This case should never be hit, the SDK should always return an
				// error which satisfies the awserr.Error interface.
				fmt.Println(err.Error())
			}
		}

		// Declare a wait group and set the count to two.

		//Start a Shell Session
		session := sh.NewSession()
		session.ShowCMD = c.Bool("d")

		fmt.Println(fmt.Sprintf("ReservationSets: %v", len(resp.Reservations)))
		for idx, _ := range resp.Reservations {
			var wg sync.WaitGroup
			for _, inst := range resp.Reservations[idx].Instances {
				for _, tag := range inst.Tags {
					wg.Add(1)
					go func() {
						if *tag.Key == c.String("t") {
							tag := *tag.Value
							id := *inst.InstanceId
							dns := *inst.PublicDnsName
							ip := ""
							if c.Bool("p") {
								ip += *inst.PublicIpAddress
							} else {
								ip += *inst.PrivateIpAddress
							}
							output, _ := session.Command("ssh", "-q", "-t", "-t", "-o", "StrictHostKeyChecking=no", fmt.Sprintf("%s@%s", c.String("u"), *inst.PrivateIpAddress), fmt.Sprintf("%s", c.Args()[0])).Output()
							fmt.Println(fmt.Sprintf("%s %s %s %s: \n%s", cyan(id), magenta(tag), yellow(ip), green(dns), output))
							if c.Bool("d") {
								//fmt.Println(awsutil.Prettify(*inst))
							}
						}
						wg.Done()
					}()
				}
			}
			// Wait for the goroutines to finish.
			wg.Wait()
		}
	}
}
Ejemplo n.º 2
0
func label(label_name string) {
	session := sh.NewSession()
	session.ShowCMD = true
	session.Command("kubectl", "-s", kube_master, "label", "node", node, label_name, "--overwrite").Run()
}