mirror of
https://github.com/rancher/k3k.git
synced 2026-02-14 18:10:01 +00:00
27730305c2
* added labels and annotations flags to cluster create * added labels and annotations flag to create policy command
66 lines
3.1 KiB
Go
66 lines
3.1 KiB
Go
package cmds
|
|
|
|
import (
|
|
"errors"
|
|
"time"
|
|
|
|
"github.com/spf13/cobra"
|
|
"k8s.io/apimachinery/pkg/api/resource"
|
|
|
|
"github.com/rancher/k3k/pkg/apis/k3k.io/v1beta1"
|
|
)
|
|
|
|
func createFlags(cmd *cobra.Command, cfg *CreateConfig) {
|
|
cmd.Flags().IntVar(&cfg.servers, "servers", 1, "number of servers")
|
|
cmd.Flags().IntVar(&cfg.agents, "agents", 0, "number of agents")
|
|
cmd.Flags().StringVar(&cfg.token, "token", "", "token of the cluster")
|
|
cmd.Flags().StringVar(&cfg.clusterCIDR, "cluster-cidr", "", "cluster CIDR")
|
|
cmd.Flags().StringVar(&cfg.serviceCIDR, "service-cidr", "", "service CIDR")
|
|
cmd.Flags().BoolVar(&cfg.mirrorHostNodes, "mirror-host-nodes", false, "Mirror Host Cluster Nodes")
|
|
cmd.Flags().StringVar(&cfg.persistenceType, "persistence-type", string(v1beta1.DynamicPersistenceMode), "persistence mode for the nodes (dynamic, ephemeral)")
|
|
cmd.Flags().StringVar(&cfg.storageClassName, "storage-class-name", "", "storage class name for dynamic persistence type")
|
|
cmd.Flags().StringVar(&cfg.storageRequestSize, "storage-request-size", "", "storage size for dynamic persistence type")
|
|
cmd.Flags().StringSliceVar(&cfg.serverArgs, "server-args", []string{}, "servers extra arguments")
|
|
cmd.Flags().StringSliceVar(&cfg.agentArgs, "agent-args", []string{}, "agents extra arguments")
|
|
cmd.Flags().StringSliceVar(&cfg.serverEnvs, "server-envs", []string{}, "servers extra Envs")
|
|
cmd.Flags().StringSliceVar(&cfg.agentEnvs, "agent-envs", []string{}, "agents extra Envs")
|
|
cmd.Flags().StringArrayVar(&cfg.labels, "labels", []string{}, "Labels to add to the cluster object (e.g. key=value)")
|
|
cmd.Flags().StringArrayVar(&cfg.annotations, "annotations", []string{}, "Annotations to add to the cluster object (e.g. key=value)")
|
|
cmd.Flags().StringVar(&cfg.version, "version", "", "k3s version")
|
|
cmd.Flags().StringVar(&cfg.mode, "mode", "shared", "k3k mode type (shared, virtual)")
|
|
cmd.Flags().StringVar(&cfg.kubeconfigServerHost, "kubeconfig-server", "", "override the kubeconfig server host")
|
|
cmd.Flags().StringVar(&cfg.policy, "policy", "", "The policy to create the cluster in")
|
|
cmd.Flags().StringVar(&cfg.customCertsPath, "custom-certs", "", "The path for custom certificate directory")
|
|
cmd.Flags().DurationVar(&cfg.timeout, "timeout", 3*time.Minute, "The timeout for waiting for the cluster to become ready (e.g., 10s, 5m, 1h).")
|
|
}
|
|
|
|
func validateCreateConfig(cfg *CreateConfig) error {
|
|
if cfg.servers <= 0 {
|
|
return errors.New("invalid number of servers")
|
|
}
|
|
|
|
if cfg.persistenceType != "" {
|
|
switch v1beta1.PersistenceMode(cfg.persistenceType) {
|
|
case v1beta1.EphemeralPersistenceMode, v1beta1.DynamicPersistenceMode:
|
|
return nil
|
|
default:
|
|
return errors.New(`persistence-type should be one of "dynamic" or "ephemeral"`)
|
|
}
|
|
}
|
|
|
|
if _, err := resource.ParseQuantity(cfg.storageRequestSize); err != nil {
|
|
return errors.New(`invalid storage size, should be a valid resource quantity e.g "10Gi"`)
|
|
}
|
|
|
|
if cfg.mode != "" {
|
|
switch cfg.mode {
|
|
case string(v1beta1.VirtualClusterMode), string(v1beta1.SharedClusterMode):
|
|
return nil
|
|
default:
|
|
return errors.New(`mode should be one of "shared" or "virtual"`)
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|