mirror of https://github.com/k3d-io/k3d
parent
8278e8d20b
commit
08f7a47119
@ -0,0 +1,54 @@ |
||||
/* |
||||
Copyright © 2020 The k3d Author(s) |
||||
|
||||
Permission is hereby granted, free of charge, to any person obtaining a copy |
||||
of this software and associated documentation files (the "Software"), to deal |
||||
in the Software without restriction, including without limitation the rights |
||||
to use, copy, modify, merge, publish, distribute, sublicense, and/or sell |
||||
copies of the Software, and to permit persons to whom the Software is |
||||
furnished to do so, subject to the following conditions: |
||||
|
||||
The above copyright notice and this permission notice shall be included in |
||||
all copies or substantial portions of the Software. |
||||
|
||||
THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR |
||||
IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, |
||||
FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE |
||||
AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER |
||||
LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, |
||||
OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN |
||||
THE SOFTWARE. |
||||
*/ |
||||
package cluster |
||||
|
||||
import ( |
||||
log "github.com/sirupsen/logrus" |
||||
"github.com/spf13/cobra" |
||||
) |
||||
|
||||
// NewCmdCluster returns a new cobra command
|
||||
func NewCmdCluster() *cobra.Command { |
||||
|
||||
// create new cobra command
|
||||
cmd := &cobra.Command{ |
||||
Use: "cluster", |
||||
Short: "Manage cluster(s)", |
||||
Long: `Manage cluster(s)`, |
||||
Run: func(cmd *cobra.Command, args []string) { |
||||
if err := cmd.Help(); err != nil { |
||||
log.Errorln("Couldn't get help text") |
||||
log.Fatalln(err) |
||||
} |
||||
}, |
||||
} |
||||
|
||||
// add subcommands
|
||||
cmd.AddCommand(NewCmdClusterCreate()) |
||||
cmd.AddCommand(NewCmdClusterDelete()) |
||||
cmd.AddCommand(NewCmdClusterList()) |
||||
|
||||
// add flags
|
||||
|
||||
// done
|
||||
return cmd |
||||
} |
@ -0,0 +1,408 @@ |
||||
/* |
||||
|
||||
Copyright © 2020 The k3d Author(s) |
||||
|
||||
Permission is hereby granted, free of charge, to any person obtaining a copy |
||||
of this software and associated documentation files (the "Software"), to deal |
||||
in the Software without restriction, including without limitation the rights |
||||
to use, copy, modify, merge, publish, distribute, sublicense, and/or sell |
||||
copies of the Software, and to permit persons to whom the Software is |
||||
furnished to do so, subject to the following conditions: |
||||
|
||||
The above copyright notice and this permission notice shall be included in |
||||
all copies or substantial portions of the Software. |
||||
|
||||
THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR |
||||
IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, |
||||
FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE |
||||
AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER |
||||
LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, |
||||
OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN |
||||
THE SOFTWARE. |
||||
*/ |
||||
package cluster |
||||
|
||||
import ( |
||||
"fmt" |
||||
"os" |
||||
"runtime" |
||||
"time" |
||||
|
||||
"github.com/spf13/cobra" |
||||
|
||||
cliutil "github.com/rancher/k3d/v3/cmd/util" |
||||
"github.com/rancher/k3d/v3/pkg/cluster" |
||||
k3dCluster "github.com/rancher/k3d/v3/pkg/cluster" |
||||
"github.com/rancher/k3d/v3/pkg/runtimes" |
||||
k3d "github.com/rancher/k3d/v3/pkg/types" |
||||
"github.com/rancher/k3d/v3/version" |
||||
|
||||
log "github.com/sirupsen/logrus" |
||||
) |
||||
|
||||
const clusterCreateDescription = ` |
||||
Create a new k3s cluster with containerized nodes (k3s in docker). |
||||
Every cluster will consist of at least 2 containers: |
||||
- 1 master node container (k3s) |
||||
- 1 loadbalancer container as the entrypoint to the cluster (nginx) |
||||
` |
||||
|
||||
// NewCmdClusterCreate returns a new cobra command
|
||||
func NewCmdClusterCreate() *cobra.Command { |
||||
|
||||
createClusterOpts := &k3d.CreateClusterOpts{} |
||||
var updateKubeconfig, updateCurrentContext bool |
||||
|
||||
// create new command
|
||||
cmd := &cobra.Command{ |
||||
Use: "create NAME", |
||||
Short: "Create a new cluster", |
||||
Long: clusterCreateDescription, |
||||
Args: cobra.RangeArgs(0, 1), // exactly one cluster name can be set (default: k3d.DefaultClusterName)
|
||||
Run: func(cmd *cobra.Command, args []string) { |
||||
// parse args and flags
|
||||
cluster := parseCreateClusterCmd(cmd, args, createClusterOpts) |
||||
|
||||
// check if a cluster with that name exists already
|
||||
if _, err := k3dCluster.GetCluster(cmd.Context(), runtimes.SelectedRuntime, cluster); err == nil { |
||||
log.Fatalf("Failed to create cluster '%s' because a cluster with that name already exists", cluster.Name) |
||||
} |
||||
|
||||
// create cluster
|
||||
if updateKubeconfig || updateCurrentContext { |
||||
log.Debugln("'--update-kubeconfig set: enabling wait-for-master") |
||||
cluster.CreateClusterOpts.WaitForMaster = true |
||||
} |
||||
if err := k3dCluster.CreateCluster(cmd.Context(), runtimes.SelectedRuntime, cluster); err != nil { |
||||
// rollback if creation failed
|
||||
log.Errorln(err) |
||||
log.Errorln("Failed to create cluster >>> Rolling Back") |
||||
if err := k3dCluster.DeleteCluster(cmd.Context(), runtimes.SelectedRuntime, cluster); err != nil { |
||||
log.Errorln(err) |
||||
log.Fatalln("Cluster creation FAILED, also FAILED to rollback changes!") |
||||
} |
||||
log.Fatalln("Cluster creation FAILED, all changes have been rolled back!") |
||||
} |
||||
log.Infof("Cluster '%s' created successfully!", cluster.Name) |
||||
|
||||
if updateKubeconfig || updateCurrentContext { |
||||
log.Debugf("Updating default kubeconfig with a new context for cluster %s", cluster.Name) |
||||
if _, err := k3dCluster.GetAndWriteKubeConfig(cmd.Context(), runtimes.SelectedRuntime, cluster, "", &k3dCluster.WriteKubeConfigOptions{UpdateExisting: true, OverwriteExisting: false, UpdateCurrentContext: updateCurrentContext}); err != nil { |
||||
log.Fatalln(err) |
||||
} |
||||
} |
||||
|
||||
// print information on how to use the cluster with kubectl
|
||||
log.Infoln("You can now use it like this:") |
||||
if updateKubeconfig && !updateCurrentContext { |
||||
fmt.Printf("kubectl config use-context %s\n", fmt.Sprintf("%s-%s", k3d.DefaultObjectNamePrefix, cluster.Name)) |
||||
} else if !updateCurrentContext { |
||||
if runtime.GOOS == "windows" { |
||||
fmt.Printf("$env:KUBECONFIG=(%s get kubeconfig %s)\n", os.Args[0], cluster.Name) |
||||
} else { |
||||
fmt.Printf("export KUBECONFIG=$(%s get kubeconfig %s)\n", os.Args[0], cluster.Name) |
||||
} |
||||
} |
||||
fmt.Println("kubectl cluster-info") |
||||
}, |
||||
} |
||||
|
||||
/********* |
||||
* Flags * |
||||
*********/ |
||||
cmd.Flags().StringP("api-port", "a", "random", "Specify the Kubernetes API server port exposed on the LoadBalancer (Format: `--api-port [HOST:]HOSTPORT`)\n - Example: `k3d create -m 3 -a 0.0.0.0:6550`") |
||||
cmd.Flags().IntP("masters", "m", 1, "Specify how many masters you want to create") |
||||
cmd.Flags().IntP("workers", "w", 0, "Specify how many workers you want to create") |
||||
cmd.Flags().StringP("image", "i", fmt.Sprintf("%s:%s", k3d.DefaultK3sImageRepo, version.GetK3sVersion(false)), "Specify k3s image that you want to use for the nodes") |
||||
cmd.Flags().String("network", "", "Join an existing network") |
||||
cmd.Flags().String("token", "", "Specify a cluster token. By default, we generate one.") |
||||
cmd.Flags().StringArrayP("volume", "v", nil, "Mount volumes into the nodes (Format: `--volume [SOURCE:]DEST[@NODEFILTER[;NODEFILTER...]]`\n - Example: `k3d create -w 2 -v /my/path@worker[0,1] -v /tmp/test:/tmp/other@master[0]`") |
||||
cmd.Flags().StringArrayP("port", "p", nil, "Map ports from the node containers to the host (Format: `[HOST:][HOSTPORT:]CONTAINERPORT[/PROTOCOL][@NODEFILTER]`)\n - Example: `k3d create -w 2 -p 8080:80@worker[0] -p 8081@worker[1]`") |
||||
cmd.Flags().BoolVar(&createClusterOpts.WaitForMaster, "wait", true, "Wait for the master(s) to be ready before returning. Use '--timeout DURATION' to not wait forever.") |
||||
cmd.Flags().DurationVar(&createClusterOpts.Timeout, "timeout", 0*time.Second, "Rollback changes if cluster couldn't be created in specified duration.") |
||||
cmd.Flags().BoolVar(&updateKubeconfig, "update-kubeconfig", false, "Directly update the default kubeconfig with the new cluster's context") |
||||
cmd.Flags().BoolVar(&updateCurrentContext, "switch", false, "Directly switch the default kubeconfig's current-context to the new cluster's context (implies --update-kubeconfig)") |
||||
cmd.Flags().BoolVar(&createClusterOpts.DisableLoadBalancer, "no-lb", false, "Disable the creation of a LoadBalancer in front of the master nodes") |
||||
|
||||
/* Image Importing */ |
||||
cmd.Flags().BoolVar(&createClusterOpts.DisableImageVolume, "no-image-volume", false, "Disable the creation of a volume for importing images") |
||||
|
||||
/* Multi Master Configuration */ |
||||
|
||||
// multi-master - datastore
|
||||
// TODO: implement multi-master setups with external data store
|
||||
// cmd.Flags().String("datastore-endpoint", "", "[WIP] Specify external datastore endpoint (e.g. for multi master clusters)")
|
||||
/* |
||||
cmd.Flags().String("datastore-network", "", "Specify container network where we can find the datastore-endpoint (add a connection)") |
||||
|
||||
// TODO: set default paths and hint, that one should simply mount the files using --volume flag
|
||||
cmd.Flags().String("datastore-cafile", "", "Specify external datastore's TLS Certificate Authority (CA) file") |
||||
cmd.Flags().String("datastore-certfile", "", "Specify external datastore's TLS certificate file'") |
||||
cmd.Flags().String("datastore-keyfile", "", "Specify external datastore's TLS key file'") |
||||
*/ |
||||
|
||||
/* k3s */ |
||||
cmd.Flags().StringArrayVar(&createClusterOpts.K3sServerArgs, "k3s-server-arg", nil, "Additional args passed to the `k3s server` command on master nodes (new flag per arg)") |
||||
cmd.Flags().StringArrayVar(&createClusterOpts.K3sAgentArgs, "k3s-agent-arg", nil, "Additional args passed to the `k3s agent` command on worker nodes (new flag per arg)") |
||||
|
||||
/* Subcommands */ |
||||
|
||||
// done
|
||||
return cmd |
||||
} |
||||
|
||||
// parseCreateClusterCmd parses the command input into variables required to create a cluster
|
||||
func parseCreateClusterCmd(cmd *cobra.Command, args []string, createClusterOpts *k3d.CreateClusterOpts) *k3d.Cluster { |
||||
|
||||
/******************************** |
||||
* Parse and validate arguments * |
||||
********************************/ |
||||
|
||||
clustername := k3d.DefaultClusterName |
||||
if len(args) != 0 { |
||||
clustername = args[0] |
||||
} |
||||
if err := cluster.CheckName(clustername); err != nil { |
||||
log.Fatal(err) |
||||
} |
||||
|
||||
/**************************** |
||||
* Parse and validate flags * |
||||
****************************/ |
||||
|
||||
// --image
|
||||
image, err := cmd.Flags().GetString("image") |
||||
if err != nil { |
||||
log.Errorln("No image specified") |
||||
log.Fatalln(err) |
||||
} |
||||
if image == "latest" { |
||||
image = version.GetK3sVersion(true) |
||||
} |
||||
|
||||
// --masters
|
||||
masterCount, err := cmd.Flags().GetInt("masters") |
||||
if err != nil { |
||||
log.Fatalln(err) |
||||
} |
||||
|
||||
// --workers
|
||||
workerCount, err := cmd.Flags().GetInt("workers") |
||||
if err != nil { |
||||
log.Fatalln(err) |
||||
} |
||||
|
||||
// --network
|
||||
networkName, err := cmd.Flags().GetString("network") |
||||
if err != nil { |
||||
log.Fatalln(err) |
||||
} |
||||
network := k3d.ClusterNetwork{} |
||||
if networkName != "" { |
||||
network.Name = networkName |
||||
network.External = true |
||||
} |
||||
if networkName == "host" && (masterCount+workerCount) > 1 { |
||||
log.Fatalln("Can only run a single node in hostnetwork mode") |
||||
} |
||||
|
||||
// --token
|
||||
token, err := cmd.Flags().GetString("token") |
||||
if err != nil { |
||||
log.Fatalln(err) |
||||
} |
||||
|
||||
// --timeout
|
||||
if cmd.Flags().Changed("timeout") && createClusterOpts.Timeout <= 0*time.Second { |
||||
log.Fatalln("--timeout DURATION must be >= 1s") |
||||
} |
||||
|
||||
// --api-port
|
||||
apiPort, err := cmd.Flags().GetString("api-port") |
||||
if err != nil { |
||||
log.Fatalln(err) |
||||
} |
||||
|
||||
// parse the port mapping
|
||||
exposeAPI, err := cliutil.ParseAPIPort(apiPort) |
||||
if err != nil { |
||||
log.Fatalln(err) |
||||
} |
||||
if exposeAPI.Host == "" { |
||||
exposeAPI.Host = k3d.DefaultAPIHost |
||||
} |
||||
if exposeAPI.HostIP == "" { |
||||
exposeAPI.HostIP = k3d.DefaultAPIHost |
||||
} |
||||
if networkName == "host" { |
||||
// in hostNetwork mode, we're not going to map a hostport. Here it should always use 6443.
|
||||
// Note that hostNetwork mode is super inflexible and since we don't change the backend port (on the container), it will only be one hostmode cluster allowed.
|
||||
exposeAPI.Port = k3d.DefaultAPIPort |
||||
} |
||||
|
||||
// --volume
|
||||
volumeFlags, err := cmd.Flags().GetStringArray("volume") |
||||
if err != nil { |
||||
log.Fatalln(err) |
||||
} |
||||
|
||||
// volumeFilterMap will map volume mounts to applied node filters
|
||||
volumeFilterMap := make(map[string][]string, 1) |
||||
for _, volumeFlag := range volumeFlags { |
||||
|
||||
// split node filter from the specified volume
|
||||
volume, filters, err := cliutil.SplitFiltersFromFlag(volumeFlag) |
||||
if err != nil { |
||||
log.Fatalln(err) |
||||
} |
||||
|
||||
// validate the specified volume mount and return it in SRC:DEST format
|
||||
volume, err = cliutil.ValidateVolumeMount(runtimes.SelectedRuntime, volume) |
||||
if err != nil { |
||||
log.Fatalln(err) |
||||
} |
||||
|
||||
// create new entry or append filter to existing entry
|
||||
if _, exists := volumeFilterMap[volume]; exists { |
||||
volumeFilterMap[volume] = append(volumeFilterMap[volume], filters...) |
||||
} else { |
||||
volumeFilterMap[volume] = filters |
||||
} |
||||
} |
||||
|
||||
// --port
|
||||
portFlags, err := cmd.Flags().GetStringArray("port") |
||||
if err != nil { |
||||
log.Fatalln(err) |
||||
} |
||||
portFilterMap := make(map[string][]string, 1) |
||||
for _, portFlag := range portFlags { |
||||
// split node filter from the specified volume
|
||||
portmap, filters, err := cliutil.SplitFiltersFromFlag(portFlag) |
||||
if err != nil { |
||||
log.Fatalln(err) |
||||
} |
||||
|
||||
if len(filters) > 1 { |
||||
log.Fatalln("Can only apply a Portmap to one node") |
||||
} |
||||
|
||||
// the same portmapping can't be applied to multiple nodes
|
||||
|
||||
// validate the specified volume mount and return it in SRC:DEST format
|
||||
portmap, err = cliutil.ValidatePortMap(portmap) |
||||
if err != nil { |
||||
log.Fatalln(err) |
||||
} |
||||
|
||||
// create new entry or append filter to existing entry
|
||||
if _, exists := portFilterMap[portmap]; exists { |
||||
log.Fatalln("Same Portmapping can not be used for multiple nodes") |
||||
} else { |
||||
portFilterMap[portmap] = filters |
||||
} |
||||
} |
||||
|
||||
log.Debugf("PortFilterMap: %+v", portFilterMap) |
||||
|
||||
/******************** |
||||
* * |
||||
* generate cluster * |
||||
* * |
||||
********************/ |
||||
|
||||
cluster := &k3d.Cluster{ |
||||
Name: clustername, |
||||
Network: network, |
||||
Token: token, |
||||
CreateClusterOpts: createClusterOpts, |
||||
ExposeAPI: exposeAPI, |
||||
} |
||||
|
||||
// generate list of nodes
|
||||
cluster.Nodes = []*k3d.Node{} |
||||
|
||||
// MasterLoadBalancer
|
||||
if !createClusterOpts.DisableLoadBalancer { |
||||
cluster.MasterLoadBalancer = &k3d.Node{ |
||||
Role: k3d.LoadBalancerRole, |
||||
} |
||||
} |
||||
|
||||
/**************** |
||||
* Master Nodes * |
||||
****************/ |
||||
|
||||
for i := 0; i < masterCount; i++ { |
||||
node := k3d.Node{ |
||||
Role: k3d.MasterRole, |
||||
Image: image, |
||||
Args: createClusterOpts.K3sServerArgs, |
||||
MasterOpts: k3d.MasterOpts{}, |
||||
} |
||||
|
||||
// TODO: by default, we don't expose an API port: should we change that?
|
||||
// -> if we want to change that, simply add the exposeAPI struct here
|
||||
|
||||
// first master node will be init node if we have more than one master specified but no external datastore
|
||||
if i == 0 && masterCount > 1 { |
||||
node.MasterOpts.IsInit = true |
||||
cluster.InitNode = &node |
||||
} |
||||
|
||||
// append node to list
|
||||
cluster.Nodes = append(cluster.Nodes, &node) |
||||
} |
||||
|
||||
/**************** |
||||
* Worker Nodes * |
||||
****************/ |
||||
|
||||
for i := 0; i < workerCount; i++ { |
||||
node := k3d.Node{ |
||||
Role: k3d.WorkerRole, |
||||
Image: image, |
||||
Args: createClusterOpts.K3sAgentArgs, |
||||
} |
||||
|
||||
cluster.Nodes = append(cluster.Nodes, &node) |
||||
} |
||||
|
||||
// append volumes
|
||||
for volume, filters := range volumeFilterMap { |
||||
nodes, err := cliutil.FilterNodes(cluster.Nodes, filters) |
||||
if err != nil { |
||||
log.Fatalln(err) |
||||
} |
||||
for _, node := range nodes { |
||||
node.Volumes = append(node.Volumes, volume) |
||||
} |
||||
} |
||||
|
||||
// append ports
|
||||
nodeCount := masterCount + workerCount |
||||
nodeList := cluster.Nodes |
||||
if !createClusterOpts.DisableLoadBalancer { |
||||
nodeCount++ |
||||
nodeList = append(nodeList, cluster.MasterLoadBalancer) |
||||
} |
||||
for portmap, filters := range portFilterMap { |
||||
if len(filters) == 0 && (nodeCount) > 1 { |
||||
log.Fatalf("Malformed portmapping '%s' lacks a node filter, but there is more than one node (including the loadbalancer, if there is any).", portmap) |
||||
} |
||||
nodes, err := cliutil.FilterNodes(nodeList, filters) |
||||
if err != nil { |
||||
log.Fatalln(err) |
||||
} |
||||
for _, node := range nodes { |
||||
node.Ports = append(node.Ports, portmap) |
||||
} |
||||
} |
||||
|
||||
/********************** |
||||
* Utility Containers * |
||||
**********************/ |
||||
// ...
|
||||
|
||||
return cluster |
||||
} |
@ -0,0 +1,106 @@ |
||||
/* |
||||
Copyright © 2020 The k3d Author(s) |
||||
|
||||
Permission is hereby granted, free of charge, to any person obtaining a copy |
||||
of this software and associated documentation files (the "Software"), to deal |
||||
in the Software without restriction, including without limitation the rights |
||||
to use, copy, modify, merge, publish, distribute, sublicense, and/or sell |
||||
copies of the Software, and to permit persons to whom the Software is |
||||
furnished to do so, subject to the following conditions: |
||||
|
||||
The above copyright notice and this permission notice shall be included in |
||||
all copies or substantial portions of the Software. |
||||
|
||||
THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR |
||||
IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, |
||||
FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE |
||||
AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER |
||||
LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, |
||||
OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN |
||||
THE SOFTWARE. |
||||
*/ |
||||
package cluster |
||||
|
||||
import ( |
||||
"github.com/rancher/k3d/v3/cmd/util" |
||||
"github.com/rancher/k3d/v3/pkg/cluster" |
||||
"github.com/rancher/k3d/v3/pkg/runtimes" |
||||
k3d "github.com/rancher/k3d/v3/pkg/types" |
||||
log "github.com/sirupsen/logrus" |
||||
|
||||
"github.com/spf13/cobra" |
||||
) |
||||
|
||||
// NewCmdClusterDelete returns a new cobra command
|
||||
func NewCmdClusterDelete() *cobra.Command { |
||||
|
||||
// create new cobra command
|
||||
cmd := &cobra.Command{ |
||||
Use: "delete (NAME | --all)", |
||||
Aliases: []string{"del", "rm"}, |
||||
Short: "Delete cluster(s).", |
||||
Long: `Delete cluster(s).`, |
||||
Args: cobra.MinimumNArgs(0), // 0 or n arguments; 0 only if --all is set
|
||||
ValidArgsFunction: util.ValidArgsAvailableClusters, |
||||
Run: func(cmd *cobra.Command, args []string) { |
||||
clusters := parseDeleteClusterCmd(cmd, args) |
||||
|
||||
if len(clusters) == 0 { |
||||
log.Infoln("No clusters found") |
||||
} else { |
||||
for _, c := range clusters { |
||||
if err := cluster.DeleteCluster(cmd.Context(), runtimes.SelectedRuntime, c); err != nil { |
||||
log.Fatalln(err) |
||||
} |
||||
log.Infoln("Removing cluster details from default kubeconfig") |
||||
if err := cluster.RemoveClusterFromDefaultKubeConfig(cmd.Context(), c); err != nil { |
||||
log.Warnln("Failed to remove cluster details from default kubeconfig") |
||||
log.Warnln(err) |
||||
} |
||||
|
||||
log.Infof("Successfully deleted cluster %s!", c.Name) |
||||
} |
||||
} |
||||
|
||||
}, |
||||
} |
||||
|
||||
// add subcommands
|
||||
|
||||
// add flags
|
||||
cmd.Flags().BoolP("all", "a", false, "Delete all existing clusters") |
||||
|
||||
// done
|
||||
return cmd |
||||
} |
||||
|
||||
// parseDeleteClusterCmd parses the command input into variables required to delete clusters
|
||||
func parseDeleteClusterCmd(cmd *cobra.Command, args []string) []*k3d.Cluster { |
||||
|
||||
// --all
|
||||
var clusters []*k3d.Cluster |
||||
|
||||
if all, err := cmd.Flags().GetBool("all"); err != nil { |
||||
log.Fatalln(err) |
||||
} else if all { |
||||
clusters, err = cluster.GetClusters(cmd.Context(), runtimes.SelectedRuntime) |
||||
if err != nil { |
||||
log.Fatalln(err) |
||||
} |
||||
return clusters |
||||
} |
||||
|
||||
if len(args) < 1 { |
||||
log.Fatalln("Expecting at least one cluster name if `--all` is not set") |
||||
} |
||||
|
||||
for _, name := range args { |
||||
cluster, err := cluster.GetCluster(cmd.Context(), runtimes.SelectedRuntime, &k3d.Cluster{Name: name}) |
||||
if err != nil { |
||||
log.Fatalln(err) |
||||
} |
||||
clusters = append(clusters, cluster) |
||||
} |
||||
|
||||
return clusters |
||||
} |
@ -0,0 +1,128 @@ |
||||
/* |
||||
Copyright © 2020 The k3d Author(s) |
||||
|
||||
Permission is hereby granted, free of charge, to any person obtaining a copy |
||||
of this software and associated documentation files (the "Software"), to deal |
||||
in the Software without restriction, including without limitation the rights |
||||
to use, copy, modify, merge, publish, distribute, sublicense, and/or sell |
||||
copies of the Software, and to permit persons to whom the Software is |
||||
furnished to do so, subject to the following conditions: |
||||
|
||||
The above copyright notice and this permission notice shall be included in |
||||
all copies or substantial portions of the Software. |
||||
|
||||
THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR |
||||
IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, |
||||
FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE |
||||
AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER |
||||
LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, |
||||
OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN |
||||
THE SOFTWARE. |
||||
*/ |
||||
package cluster |
||||
|
||||
import ( |
||||
"context" |
||||
"fmt" |
||||
"os" |
||||
"strings" |
||||
|
||||
"github.com/rancher/k3d/v3/cmd/util" |
||||
k3cluster "github.com/rancher/k3d/v3/pkg/cluster" |
||||
"github.com/rancher/k3d/v3/pkg/runtimes" |
||||
k3d "github.com/rancher/k3d/v3/pkg/types" |
||||
"github.com/spf13/cobra" |
||||
|
||||
log "github.com/sirupsen/logrus" |
||||
|
||||
"github.com/liggitt/tabwriter" |
||||
) |
||||
|
||||
// TODO : deal with --all flag to manage differentiate started cluster and stopped cluster like `docker ps` and `docker ps -a`
|
||||
type clusterFlags struct { |
||||
noHeader bool |
||||
token bool |
||||
} |
||||
|
||||
// NewCmdClusterList returns a new cobra command
|
||||
func NewCmdClusterList() *cobra.Command { |
||||
|
||||
clusterFlags := clusterFlags{} |
||||
|
||||
// create new command
|
||||
cmd := &cobra.Command{ |
||||
Use: "list [NAME [NAME...]]", |
||||
Aliases: []string{"ls", "get"}, |
||||
Short: "List cluster(s)", |
||||
Long: `List cluster(s).`, |
||||
Run: func(cmd *cobra.Command, args []string) { |
||||
clusters := buildClusterList(cmd.Context(), args) |
||||
PrintClusters(clusters, clusterFlags) |
||||
}, |
||||
ValidArgsFunction: util.ValidArgsAvailableClusters, |
||||
} |
||||
|
||||
// add flags
|
||||
cmd.Flags().BoolVar(&clusterFlags.noHeader, "no-headers", false, "Disable headers") |
||||
cmd.Flags().BoolVar(&clusterFlags.token, "token", false, "Print k3s cluster token") |
||||
|
||||
// add subcommands
|
||||
|
||||
// done
|
||||
return cmd |
||||
} |
||||
|
||||
func buildClusterList(ctx context.Context, args []string) []*k3d.Cluster { |
||||
var clusters []*k3d.Cluster |
||||
var err error |
||||
|
||||
if len(args) == 0 { |
||||
// cluster name not specified : get all clusters
|
||||
clusters, err = k3cluster.GetClusters(ctx, runtimes.SelectedRuntime) |
||||
if err != nil { |
||||
log.Fatalln(err) |
||||
} |
||||
} else { |
||||
for _, clusterName := range args { |
||||
// cluster name specified : get specific cluster
|
||||
retrievedCluster, err := k3cluster.GetCluster(ctx, runtimes.SelectedRuntime, &k3d.Cluster{Name: clusterName}) |
||||
if err != nil { |
||||
log.Fatalln(err) |
||||
} |
||||
clusters = append(clusters, retrievedCluster) |
||||
} |
||||
} |
||||
|
||||
return clusters |
||||
} |
||||
|
||||
// PrintPrintClusters : display list of cluster
|
||||
func PrintClusters(clusters []*k3d.Cluster, flags clusterFlags) { |
||||
|
||||
tabwriter := tabwriter.NewWriter(os.Stdout, 6, 4, 3, ' ', tabwriter.RememberWidths) |
||||
defer tabwriter.Flush() |
||||
|
||||
if !flags.noHeader { |
||||
headers := []string{"NAME", "MASTERS", "WORKERS"} // TODO: getCluster: add status column
|
||||
if flags.token { |
||||
headers = append(headers, "TOKEN") |
||||
} |
||||
_, err := fmt.Fprintf(tabwriter, "%s\n", strings.Join(headers, "\t")) |
||||
if err != nil { |
||||
log.Fatalln("Failed to print headers") |
||||
} |
||||
} |
||||
|
||||
k3cluster.SortClusters(clusters) |
||||
|
||||
for _, cluster := range clusters { |
||||
masterCount := cluster.MasterCount() |
||||
workerCount := cluster.WorkerCount() |
||||
|
||||
if flags.token { |
||||
fmt.Fprintf(tabwriter, "%s\t%d\t%d\t%s\n", cluster.Name, masterCount, workerCount, cluster.Token) |
||||
} else { |
||||
fmt.Fprintf(tabwriter, "%s\t%d\t%d\n", cluster.Name, masterCount, workerCount) |
||||
} |
||||
} |
||||
} |
@ -0,0 +1,53 @@ |
||||
/* |
||||
Copyright © 2020 The k3d Author(s) |
||||
|
||||
Permission is hereby granted, free of charge, to any person obtaining a copy |
||||
of this software and associated documentation files (the "Software"), to deal |
||||
in the Software without restriction, including without limitation the rights |
||||
to use, copy, modify, merge, publish, distribute, sublicense, and/or sell |
||||
copies of the Software, and to permit persons to whom the Software is |
||||
furnished to do so, subject to the following conditions: |
||||
|
||||
The above copyright notice and this permission notice shall be included in |
||||
all copies or substantial portions of the Software. |
||||
|
||||
THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR |
||||
IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, |
||||
FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE |
||||
AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER |
||||
LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, |
||||
OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN |
||||
THE SOFTWARE. |
||||
*/ |
||||
package kubeconfig |
||||
|
||||
import ( |
||||
log "github.com/sirupsen/logrus" |
||||
"github.com/spf13/cobra" |
||||
) |
||||
|
||||
// NewCmdKubeconfig returns a new cobra command
|
||||
func NewCmdKubeconfig() *cobra.Command { |
||||
|
||||
// create new cobra command
|
||||
cmd := &cobra.Command{ |
||||
Use: "kubeconfig", |
||||
Short: "Manage kubeconfig(s)", |
||||
Long: `Manage kubeconfig(s)`, |
||||
Run: func(cmd *cobra.Command, args []string) { |
||||
if err := cmd.Help(); err != nil { |
||||
log.Errorln("Couldn't get help text") |
||||
log.Fatalln(err) |
||||
} |
||||
}, |
||||
} |
||||
|
||||
// add subcommands
|
||||
cmd.AddCommand(NewCmdKubeconfigGet()) |
||||
cmd.AddCommand(NewCmdKubeconfigMerge()) |
||||
|
||||
// add flags
|
||||
|
||||
// done
|
||||
return cmd |
||||
} |
@ -0,0 +1,107 @@ |
||||
/* |
||||
Copyright © 2020 The k3d Author(s) |
||||
|
||||
Permission is hereby granted, free of charge, to any person obtaining a copy |
||||
of this software and associated documentation files (the "Software"), to deal |
||||
in the Software without restriction, including without limitation the rights |
||||
to use, copy, modify, merge, publish, distribute, sublicense, and/or sell |
||||
copies of the Software, and to permit persons to whom the Software is |
||||
furnished to do so, subject to the following conditions: |
||||
|
||||
The above copyright notice and this permission notice shall be included in |
||||
all copies or substantial portions of the Software. |
||||
|
||||
THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR |
||||
IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, |
||||
FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE |
||||
AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER |
||||
LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, |
||||
OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN |
||||
THE SOFTWARE. |
||||
*/ |
||||
package kubeconfig |
||||
|
||||
import ( |
||||
"fmt" |
||||
"os" |
||||
|
||||
"github.com/rancher/k3d/v3/cmd/util" |
||||
"github.com/rancher/k3d/v3/pkg/cluster" |
||||
"github.com/rancher/k3d/v3/pkg/runtimes" |
||||
k3d "github.com/rancher/k3d/v3/pkg/types" |
||||
"github.com/spf13/cobra" |
||||
|
||||
log "github.com/sirupsen/logrus" |
||||
) |
||||
|
||||
type getKubeconfigFlags struct { |
||||
all bool |
||||
} |
||||
|
||||
// NewCmdKubeconfigGet returns a new cobra command
|
||||
func NewCmdKubeconfigGet() *cobra.Command { |
||||
|
||||
writeKubeConfigOptions := cluster.WriteKubeConfigOptions{ |
||||
UpdateExisting: true, |
||||
UpdateCurrentContext: true, |
||||
OverwriteExisting: true, |
||||
} |
||||
|
||||
getKubeconfigFlags := getKubeconfigFlags{} |
||||
|
||||
// create new command
|
||||
cmd := &cobra.Command{ |
||||
Use: "get [CLUSTER [CLUSTER [...]] | --all]", |
||||
Short: "Get kubeconfig from cluster(s).", |
||||
Long: `Get kubeconfig from cluster(s).`, |
||||
ValidArgsFunction: util.ValidArgsAvailableClusters, |
||||
Args: func(cmd *cobra.Command, args []string) error { |
||||
if (len(args) < 1 && !getKubeconfigFlags.all) || (len(args) > 0 && getKubeconfigFlags.all) { |
||||
return fmt.Errorf("Need to specify one or more cluster names *or* set `--all` flag") |
||||
} |
||||
return nil |
||||
}, |
||||
Run: func(cmd *cobra.Command, args []string) { |
||||
var clusters []*k3d.Cluster |
||||
var err error |
||||
|
||||
// generate list of clusters
|
||||
if getKubeconfigFlags.all { |
||||
clusters, err = cluster.GetClusters(cmd.Context(), runtimes.SelectedRuntime) |
||||
if err != nil { |
||||
log.Fatalln(err) |
||||
} |
||||
} else { |
||||
for _, clusterName := range args { |
||||
retrievedCluster, err := cluster.GetCluster(cmd.Context(), runtimes.SelectedRuntime, &k3d.Cluster{Name: clusterName}) |
||||
if err != nil { |
||||
log.Fatalln(err) |
||||
} |
||||
clusters = append(clusters, retrievedCluster) |
||||
} |
||||
} |
||||
|
||||
// get kubeconfigs from all clusters
|
||||
errorGettingKubeconfig := false |
||||
for _, c := range clusters { |
||||
log.Debugf("Getting kubeconfig for cluster '%s'", c.Name) |
||||
fmt.Println("---") // YAML document separator
|
||||
if _, err := cluster.GetAndWriteKubeConfig(cmd.Context(), runtimes.SelectedRuntime, c, "-", &writeKubeConfigOptions); err != nil { |
||||
log.Errorln(err) |
||||
errorGettingKubeconfig = true |
||||
} |
||||
} |
||||
|
||||
// return with non-zero exit code, if there was an error for one of the clusters
|
||||
if errorGettingKubeconfig { |
||||
os.Exit(1) |
||||
} |
||||
}, |
||||
} |
||||
|
||||
// add flags
|
||||
cmd.Flags().BoolVarP(&getKubeconfigFlags.all, "all", "a", false, "Get kubeconfigs from all existing clusters") |
||||
|
||||
// done
|
||||
return cmd |
||||
} |
@ -0,0 +1,116 @@ |
||||
/* |
||||
Copyright © 2020 The k3d Author(s) |
||||
|
||||
Permission is hereby granted, free of charge, to any person obtaining a copy |
||||
of this software and associated documentation files (the "Software"), to deal |
||||
in the Software without restriction, including without limitation the rights |
||||
to use, copy, modify, merge, publish, distribute, sublicense, and/or sell |
||||
copies of the Software, and to permit persons to whom the Software is |
||||
furnished to do so, subject to the following conditions: |
||||
|
||||
The above copyright notice and this permission notice shall be included in |
||||
all copies or substantial portions of the Software. |
||||
|
||||
THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR |
||||
IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, |
||||
FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE |
||||
AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER |
||||
LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, |
||||
OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN |
||||
THE SOFTWARE. |
||||
*/ |
||||
package kubeconfig |
||||
|
||||
import ( |
||||
"fmt" |
||||
"os" |
||||
|
||||
"github.com/rancher/k3d/v3/cmd/util" |
||||
"github.com/rancher/k3d/v3/pkg/cluster" |
||||
"github.com/rancher/k3d/v3/pkg/runtimes" |
||||
k3d "github.com/rancher/k3d/v3/pkg/types" |
||||
"github.com/spf13/cobra" |
||||
"k8s.io/client-go/tools/clientcmd" |
||||
|
||||
log "github.com/sirupsen/logrus" |
||||
) |
||||
|
||||
type mergeKubeconfigFlags struct { |
||||
all bool |
||||
output string |
||||
} |
||||
|
||||
// NewCmdKubeconfigMerge returns a new cobra command
|
||||
func NewCmdKubeconfigMerge() *cobra.Command { |
||||
|
||||
writeKubeConfigOptions := cluster.WriteKubeConfigOptions{} |
||||
|
||||
mergeKubeconfigFlags := mergeKubeconfigFlags{} |
||||
|
||||
// create new command
|
||||
cmd := &cobra.Command{ |
||||
Use: "merge [CLUSTER [CLUSTER [...]] | --all]", |
||||
Long: `Merge kubeconfig from cluster(s) into existing kubeconfig/file.`, |
||||
Short: "Merge kubeconfig from cluster(s) into existing kubeconfig/file.", |
||||
ValidArgsFunction: util.ValidArgsAvailableClusters, |
||||
Args: func(cmd *cobra.Command, args []string) error { |
||||
if (len(args) < 1 && !mergeKubeconfigFlags.all) || (len(args) > 0 && mergeKubeconfigFlags.all) { |
||||
return fmt.Errorf("Need to specify one or more cluster names *or* set `--all` flag") |
||||
} |
||||
return nil |
||||
}, |
||||
Run: func(cmd *cobra.Command, args []string) { |
||||
var clusters []*k3d.Cluster |
||||
var err error |
||||
|
||||
// generate list of clusters
|
||||
if mergeKubeconfigFlags.all { |
||||
clusters, err = cluster.GetClusters(cmd.Context(), runtimes.SelectedRuntime) |
||||
if err != nil { |
||||
log.Fatalln(err) |
||||
} |
||||
} else { |
||||
for _, clusterName := range args { |
||||
retrievedCluster, err := cluster.GetCluster(cmd.Context(), runtimes.SelectedRuntime, &k3d.Cluster{Name: clusterName}) |
||||
if err != nil { |
||||
log.Fatalln(err) |
||||
} |
||||
clusters = append(clusters, retrievedCluster) |
||||
} |
||||
} |
||||
|
||||
// get kubeconfigs from all clusters
|
||||
errorGettingKubeconfig := false |
||||
for _, c := range clusters { |
||||
log.Debugf("Getting kubeconfig for cluster '%s'", c.Name) |
||||
if mergeKubeconfigFlags.output, err = cluster.GetAndWriteKubeConfig(cmd.Context(), runtimes.SelectedRuntime, c, mergeKubeconfigFlags.output, &writeKubeConfigOptions); err != nil { |
||||
log.Errorln(err) |
||||
errorGettingKubeconfig = true |
||||
} |
||||
} |
||||
|
||||
// only print kubeconfig file path if output is not stdout ("-")
|
||||
if mergeKubeconfigFlags.output != "-" { |
||||
fmt.Println(mergeKubeconfigFlags.output) |
||||
} |
||||
|
||||
// return with non-zero exit code, if there was an error for one of the clusters
|
||||
if errorGettingKubeconfig { |
||||
os.Exit(1) |
||||
} |
||||
}, |
||||
} |
||||
|
||||
// add flags
|
||||
cmd.Flags().StringVarP(&mergeKubeconfigFlags.output, "output", "o", "", fmt.Sprintf("Define output [ - | FILE ] (default from $KUBECONFIG or %s", clientcmd.RecommendedHomeFile)) |
||||
if err := cmd.MarkFlagFilename("output"); err != nil { |
||||
log.Fatalln("Failed to mark flag --output as filename") |
||||
} |
||||
cmd.Flags().BoolVarP(&writeKubeConfigOptions.UpdateExisting, "update", "u", true, "Update conflicting fields in existing KubeConfig") |
||||
cmd.Flags().BoolVarP(&writeKubeConfigOptions.UpdateCurrentContext, "switch", "s", false, "Switch to new context") |
||||
cmd.Flags().BoolVar(&writeKubeConfigOptions.OverwriteExisting, "overwrite", false, "[Careful!] Overwrite existing file, ignoring its contents") |
||||
cmd.Flags().BoolVarP(&mergeKubeconfigFlags.all, "all", "a", false, "Get kubeconfigs from all existing clusters") |
||||
|
||||
// done
|
||||
return cmd |
||||
} |
Loading…
Reference in new issue