steve/pkg/server/cli/clicontext.go

package cli

import (
	authcli "github.com/rancher/steve/pkg/auth/cli"
	"github.com/rancher/steve/pkg/server"
	"github.com/rancher/wrangler/pkg/kubeconfig"
	"github.com/urfave/cli"
)

type Config struct {
	KubeConfig      string
	HTTPSListenPort int
	HTTPListenPort  int
	Namespace       string

	WebhookConfig authcli.WebhookConfig
}

func (c *Config) MustServerConfig() *server.Server {
	cc, err := c.ToServerConfig()
	if err != nil {
		panic(err)
	}
	return cc
}

func (c *Config) ToServerConfig() (*server.Server, error) {
	restConfig, err := kubeconfig.GetNonInteractiveClientConfig(c.KubeConfig).ClientConfig()
	if err != nil {
		return nil, err
	}

	auth, err := c.WebhookConfig.WebhookMiddleware()
	if err != nil {
		return nil, err
	}

	return &server.Server{
		Namespace:      c.Namespace,
		RestConfig:     restConfig,
		AuthMiddleware: auth,
		HTTPPort:       c.HTTPListenPort,
		HTTPSPort:      c.HTTPSListenPort,
	}, nil
}

func Flags(config *Config) []cli.Flag {
	flags := []cli.Flag{
		cli.StringFlag{
			Name:        "kubeconfig",
			EnvVar:      "KUBECONFIG",
			Destination: &config.KubeConfig,
		},
		cli.IntFlag{
			Name:        "https-listen-port",
			Value:       8443,
			Destination: &config.HTTPSListenPort,
		},
		cli.IntFlag{
			Name:        "http-listen-port",
			Value:       8080,
			Destination: &config.HTTPListenPort,
		},
		cli.StringFlag{
			Name:        "namespace",
			EnvVar:      "NAMESPACE",
			Value:       "steve",
			Destination: &config.Namespace,
		},
	}

	return append(flags, authcli.Flags(&config.WebhookConfig)...)
}