[Feature] add --image/-i flag to specify custom k3s image

cli/commands.go

@@ -75,7 +75,7 @@ func CreateCluster(c *cli.Context) error {
 	log.Printf("Creating cluster [%s]", c.String("name"))
 	dockerID, err := createServer(
 		c.GlobalBool("verbose"),
-		fmt.Sprintf("docker.io/rancher/k3s:%s", c.String("version")),
+		fmt.Sprintf("%s:%s", c.String("image"), c.String("version")),
 		c.String("port"),
 		k3sServerArgs,
 		env,
@@ -137,7 +137,7 @@ func CreateCluster(c *cli.Context) error {
 		for i := 0; i < c.Int("workers"); i++ {
 			workerID, err := createWorker(
 				c.GlobalBool("verbose"),
-				fmt.Sprintf("docker.io/rancher/k3s:%s", c.String("version")),
+				fmt.Sprintf("%s:%s", c.String("image"), c.String("version")),
 				k3sWorkerArgs,
 				env,
 				c.String("name"),

main.go

@@ -56,7 +56,7 @@ func main() {
 					Usage: "Mount one or more volumes into every node of the cluster (Docker notation: `source:destination[,source:destination]`)",
 				},
 				cli.StringFlag{
-					Name:  "version",
+					Name:  "version, tag",
 					Value: version.GetK3sVersion(),
 					Usage: "Choose the k3s image version",
 				},
@@ -74,6 +74,11 @@ func main() {
 					Name:  "wait, w",
 					Usage: "Wait for the cluster to come up before returning",
 				},
+				cli.StringFlag{
+					Name:  "image, i",
+					Usage: "Specify a k3s image (repo only)",
+					Value: "docker.io/rancher/k3s",
+				},
 				cli.StringSliceFlag{
 					Name:  "server-arg, x",
 					Usage: "Pass an additional argument to k3s server (new flag per argument)",