package clients

import (
	"context"

	"github.com/rancher/os/pkg/generated/controllers/fleet.cattle.io"
	fleetcontrollers "github.com/rancher/os/pkg/generated/controllers/fleet.cattle.io/v1alpha1"
	"github.com/rancher/os/pkg/generated/controllers/management.cattle.io"
	ranchercontrollers "github.com/rancher/os/pkg/generated/controllers/management.cattle.io/v3"
	"github.com/rancher/os/pkg/generated/controllers/provisioning.cattle.io"
	provcontrollers "github.com/rancher/os/pkg/generated/controllers/provisioning.cattle.io/v1"
	"github.com/rancher/os/pkg/generated/controllers/rancheros.cattle.io"
	oscontrollers "github.com/rancher/os/pkg/generated/controllers/rancheros.cattle.io/v1"
	"github.com/rancher/wrangler/pkg/clients"
	"github.com/rancher/wrangler/pkg/generic"
	"k8s.io/client-go/rest"
)

const (
	SystemNamespace = "cattle-system"
)

type Clients struct {
	*clients.Clients
	Fleet        fleetcontrollers.Interface
	OS           oscontrollers.Interface
	Rancher      ranchercontrollers.Interface
	Provisioning provcontrollers.Interface
}

func NewFromConfig(restConfig *rest.Config) (*Clients, error) {
	c, err := clients.NewFromConfig(restConfig, nil)
	if err != nil {
		return nil, err
	}

	opts := &generic.FactoryOptions{
		SharedControllerFactory: c.SharedControllerFactory,
	}
	return &Clients{
		Clients:      c,
		Fleet:        fleet.NewFactoryFromConfigWithOptionsOrDie(restConfig, opts).Fleet().V1alpha1(),
		OS:           rancheros.NewFactoryFromConfigWithOptionsOrDie(restConfig, opts).Rancheros().V1(),
		Rancher:      management.NewFactoryFromConfigWithOptionsOrDie(restConfig, opts).Management().V3(),
		Provisioning: provisioning.NewFactoryFromConfigWithOptionsOrDie(restConfig, opts).Provisioning().V1(),
	}, nil
}

func (c *Clients) Start(ctx context.Context) error {
	return c.SharedControllerFactory.Start(ctx, 5)
}