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()
}
}
}
func label(label_name string) {
session := sh.NewSession()
session.ShowCMD = true
session.Command("kubectl", "-s", kube_master, "label", "node", node, label_name, "--overwrite").Run()
}