mirror of
https://github.com/rancher/cli.git
synced 2026-02-05 09:48:36 +01:00
1a2f5d2bb5
Before this change, it was not possible to use the SSH command to connect to machines provisioned with v2 provisioning. After this change (and including the changes to Rancher), the CLI will use the new CAPI client to get the SSH key and config from Rancher for v2 provisioned machines. A side effect of this change is the addition of the new `rancher machines ls` command that lists all machines for the current cluster context.
130 lines
2.8 KiB
Go
130 lines
2.8 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 getMachineByID(
|
|
ctx *cli.Context,
|
|
c *cliclient.MasterClient,
|
|
machineID string,
|
|
) (capiClient.Machine, error) {
|
|
machineCollection, err := getMachinesList(ctx, c)
|
|
if err != nil {
|
|
return capiClient.Machine{}, err
|
|
}
|
|
|
|
for _, machine := range machineCollection.Data {
|
|
if machine.ID == machineID {
|
|
return machine, nil
|
|
}
|
|
}
|
|
|
|
return capiClient.Machine{}, fmt.Errorf("no machine found with the ID [%s], run "+
|
|
"`rancher machines` to see available nodes", machineID)
|
|
}
|
|
|
|
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
|
|
}
|