package cluster

import (
	"errors"

	"github.com/rancher/k3k/pkg/apis/k3k.io/v1alpha1"
	"github.com/urfave/cli/v2"
)

func NewCreateFlags(config *CreateConfig) []cli.Flag {
	return []cli.Flag{
		&cli.IntFlag{
			Name:        "servers",
			Usage:       "number of servers",
			Destination: &config.servers,
			Value:       1,
			Action: func(ctx *cli.Context, value int) error {
				if value <= 0 {
					return errors.New("invalid number of servers")
				}
				return nil
			},
		},
		&cli.IntFlag{
			Name:        "agents",
			Usage:       "number of agents",
			Destination: &config.agents,
		},
		&cli.StringFlag{
			Name:        "token",
			Usage:       "token of the cluster",
			Destination: &config.token,
		},
		&cli.StringFlag{
			Name:        "cluster-cidr",
			Usage:       "cluster CIDR",
			Destination: &config.clusterCIDR,
		},
		&cli.StringFlag{
			Name:        "service-cidr",
			Usage:       "service CIDR",
			Destination: &config.serviceCIDR,
		},
		&cli.StringFlag{
			Name:        "persistence-type",
			Usage:       "persistence mode for the nodes (ephemeral, static, dynamic)",
			Value:       string(v1alpha1.DynamicNodesType),
			Destination: &config.persistenceType,
			Action: func(ctx *cli.Context, value string) error {
				switch v1alpha1.PersistenceMode(value) {
				case v1alpha1.EphemeralNodeType, v1alpha1.DynamicNodesType:
					return nil
				default:
					return errors.New(`persistence-type should be one of "ephemeral", "static" or "dynamic"`)
				}
			},
		},
		&cli.StringFlag{
			Name:        "storage-class-name",
			Usage:       "storage class name for dynamic persistence type",
			Destination: &config.storageClassName,
		},
		&cli.StringSliceFlag{
			Name:        "server-args",
			Usage:       "servers extra arguments",
			Destination: &config.serverArgs,
		},
		&cli.StringSliceFlag{
			Name:        "agent-args",
			Usage:       "agents extra arguments",
			Destination: &config.agentArgs,
		},
		&cli.StringFlag{
			Name:        "version",
			Usage:       "k3s version",
			Destination: &config.version,
		},
		&cli.StringFlag{
			Name:        "mode",
			Usage:       "k3k mode type (shared, virtual)",
			Destination: &config.mode,
			Value:       "shared",
			Action: func(ctx *cli.Context, value string) error {
				switch value {
				case string(v1alpha1.VirtualClusterMode), string(v1alpha1.SharedClusterMode):
					return nil
				default:
					return errors.New(`mode should be one of "shared" or "virtual"`)
				}
			},
		},
		&cli.StringFlag{
			Name:        "kubeconfig-server",
			Usage:       "override the kubeconfig server host",
			Destination: &config.kubeconfigServerHost,
		},
	}
}