// Package dis provides the main distillery
package dis

import (
	"context"

	"github.com/FAU-CDI/wisski-distillery/internal/component"
	"github.com/FAU-CDI/wisski-distillery/internal/component/control"
	"github.com/FAU-CDI/wisski-distillery/internal/component/instances"
	"github.com/FAU-CDI/wisski-distillery/internal/component/resolver"
	"github.com/FAU-CDI/wisski-distillery/internal/component/snapshots"
	"github.com/FAU-CDI/wisski-distillery/internal/component/sql"
	"github.com/FAU-CDI/wisski-distillery/internal/component/ssh"
	"github.com/FAU-CDI/wisski-distillery/internal/component/triplestore"
)

// Distillery represents a WissKI Distillery
//
// It is the main structure used to interact with different components.
type Distillery struct {
	// core holds the core of the distillery
	component.Core

	// Upstream holds information to connect to the various running
	// distillery components.
	//
	// NOTE(twiesing): This is intended to eventually allow full remote management of the distillery.
	// But for now this will just hold upstream configuration.
	Upstream Upstream

	// Pool holds all the components in this pool
	pool component.Pool
}

// Upstream contains the configuration for accessing remote configuration.
type Upstream struct {
	SQL         string
	Triplestore string
}

// Context returns a new Context belonging to this distillery
func (dis *Distillery) Context() context.Context {
	return context.Background()
}

//
// PUBLIC COMPONENT GETTERS
//

// e is a convenience function to export a single component
func e[C component.Component](dis *Distillery) C {
	return component.ExportComponent[C](
		&dis.pool,
		dis.register,
	)
}

func ea[C component.Component](dis *Distillery) []C {
	return component.ExportComponents[C](
		&dis.pool,
		dis.register,
	)
}

func (dis *Distillery) Control() *control.Control {
	return e[*control.Control](dis)
}
func (dis *Distillery) Resolver() *resolver.Resolver {
	return e[*resolver.Resolver](dis)
}
func (dis *Distillery) SSH() *ssh.SSH {
	return e[*ssh.SSH](dis)
}
func (dis *Distillery) SQL() *sql.SQL {
	return e[*sql.SQL](dis)
}
func (dis *Distillery) Triplestore() *triplestore.Triplestore {
	return e[*triplestore.Triplestore](dis)
}
func (dis *Distillery) Instances() *instances.Instances {
	return e[*instances.Instances](dis)
}
func (dis *Distillery) SnapshotManager() *snapshots.Manager {
	return e[*snapshots.Manager](dis)
}

func (dis *Distillery) Installable() []component.Installable {
	return ea[component.Installable](dis)
}
func (dis *Distillery) Updatable() []component.Updatable {
	return ea[component.Updatable](dis)
}
func (dis *Distillery) Provisionable() []component.Provisionable {
	return ea[component.Provisionable](dis)
}