mirror of https://github.com/rancher/cli.git
110 lines
2.3 KiB
Go
110 lines
2.3 KiB
Go
package cmd
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"github.com/rancher/cli/cliclient"
|
|
capiClient "github.com/rancher/rancher/pkg/client/generated/cluster/v1alpha4"
|
|
"github.com/urfave/cli"
|
|
)
|
|
|
|
type MachineData struct {
|
|
ID string
|
|
Machine capiClient.Machine
|
|
Name string
|
|
}
|
|
|
|
func MachineCommand() cli.Command {
|
|
return cli.Command{
|
|
Name: "machines",
|
|
Aliases: []string{"machine"},
|
|
Usage: "Operations on machines",
|
|
Action: defaultAction(machineLs),
|
|
Subcommands: []cli.Command{
|
|
{
|
|
Name: "ls",
|
|
Usage: "List machines",
|
|
Description: "\nLists all machines in the current cluster.",
|
|
ArgsUsage: "None",
|
|
Action: machineLs,
|
|
Flags: []cli.Flag{
|
|
cli.StringFlag{
|
|
Name: "format",
|
|
Usage: "'json', 'yaml' or Custom format: '{{.Machine.ID}} {{.Machine.Name}}'",
|
|
},
|
|
quietFlag,
|
|
},
|
|
},
|
|
},
|
|
}
|
|
}
|
|
|
|
func machineLs(ctx *cli.Context) error {
|
|
c, err := GetClient(ctx)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
collection, err := getMachinesList(ctx, c)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
writer := NewTableWriter([][]string{
|
|
{"ID", "ID"},
|
|
{"NAME", "Name"},
|
|
{"PHASE", "Machine.Status.Phase"},
|
|
}, ctx)
|
|
|
|
defer writer.Close()
|
|
|
|
for _, item := range collection.Data {
|
|
writer.Write(&MachineData{
|
|
ID: item.ID,
|
|
Machine: item,
|
|
Name: getMachineName(item),
|
|
})
|
|
}
|
|
|
|
return writer.Err()
|
|
}
|
|
|
|
func getMachinesList(
|
|
ctx *cli.Context,
|
|
c *cliclient.MasterClient,
|
|
) (*capiClient.MachineCollection, error) {
|
|
filter := defaultListOpts(ctx)
|
|
return c.CAPIClient.Machine.List(filter)
|
|
}
|
|
|
|
func getMachineByNodeName(
|
|
ctx *cli.Context,
|
|
c *cliclient.MasterClient,
|
|
nodeName string,
|
|
) (capiClient.Machine, error) {
|
|
machineCollection, err := getMachinesList(ctx, c)
|
|
if err != nil {
|
|
return capiClient.Machine{}, err
|
|
}
|
|
|
|
for _, machine := range machineCollection.Data {
|
|
if machine.Status.NodeRef != nil && machine.Status.NodeRef.Name == nodeName {
|
|
return machine, nil
|
|
}
|
|
}
|
|
|
|
return capiClient.Machine{}, fmt.Errorf("no machine found with associated to node [%s], run "+
|
|
"`rancher machines` to see available nodes", nodeName)
|
|
}
|
|
|
|
func getMachineName(machine capiClient.Machine) string {
|
|
if machine.Name != "" {
|
|
return machine.Name
|
|
} else if machine.Status.NodeRef != nil {
|
|
return machine.Status.NodeRef.Name
|
|
} else if machine.InfrastructureRef != nil {
|
|
return machine.InfrastructureRef.Name
|
|
}
|
|
return machine.ID
|
|
}
|