Refactor

pkg/server/config.go

@@ -0,0 +1,89 @@
+package server
+
+import (
+	"context"
+	"net/http"
+
+	"github.com/rancher/steve/pkg/auth"
+	"github.com/rancher/steve/pkg/schema"
+	"github.com/rancher/steve/pkg/schemaserver/types"
+	"github.com/rancher/steve/pkg/server/router"
+	"github.com/rancher/wrangler-api/pkg/generated/controllers/apiextensions.k8s.io"
+	apiextensionsv1beta1 "github.com/rancher/wrangler-api/pkg/generated/controllers/apiextensions.k8s.io/v1beta1"
+	"github.com/rancher/wrangler-api/pkg/generated/controllers/apiregistration.k8s.io"
+	apiregistrationv1 "github.com/rancher/wrangler-api/pkg/generated/controllers/apiregistration.k8s.io/v1"
+	"github.com/rancher/wrangler-api/pkg/generated/controllers/core"
+	corev1 "github.com/rancher/wrangler-api/pkg/generated/controllers/core/v1"
+	"github.com/rancher/wrangler-api/pkg/generated/controllers/rbac"
+	rbacv1 "github.com/rancher/wrangler-api/pkg/generated/controllers/rbac/v1"
+	"github.com/rancher/wrangler/pkg/start"
+	"k8s.io/client-go/kubernetes"
+	"k8s.io/client-go/rest"
+)
+
+type Server struct {
+	*Controllers
+
+	RestConfig *rest.Config
+
+	Namespace       string
+	HTTPSPort       int
+	HTTPPort        int
+	BaseSchemas     *types.APISchemas
+	SchemaTemplates []schema.Template
+	AuthMiddleware  auth.Middleware
+	Next            http.Handler
+	Router          router.RouterFunc
+	PostStartHooks  []func() error
+	StartHooks      []StartHook
+}
+
+type Controllers struct {
+	K8s      kubernetes.Interface
+	Core     corev1.Interface
+	RBAC     rbacv1.Interface
+	API      apiregistrationv1.Interface
+	CRD      apiextensionsv1beta1.Interface
+	starters []start.Starter
+}
+
+func NewController(cfg *rest.Config) (*Controllers, error) {
+	c := &Controllers{}
+
+	core, err := core.NewFactoryFromConfig(cfg)
+	if err != nil {
+		return nil, err
+	}
+	c.starters = append(c.starters, core)
+
+	rbac, err := rbac.NewFactoryFromConfig(cfg)
+	if err != nil {
+		return nil, err
+	}
+	c.starters = append(c.starters, rbac)
+
+	api, err := apiregistration.NewFactoryFromConfig(cfg)
+	if err != nil {
+		return nil, err
+	}
+	c.starters = append(c.starters, api)
+
+	crd, err := apiextensions.NewFactoryFromConfig(cfg)
+	if err != nil {
+		return nil, err
+	}
+	c.starters = append(c.starters, crd)
+
+	c.K8s, err = kubernetes.NewForConfig(cfg)
+	if err != nil {
+		return nil, err
+	}
+	c.Core = core.Core().V1()
+	c.RBAC = rbac.Rbac().V1()
+	c.API = api.Apiregistration().V1()
+	c.CRD = crd.Apiextensions().V1beta1()
+
+	return c, nil
+}
+
+type StartHook func(context.Context, *Server) error