o11y-foundry/internal/foundry/foundry.go

package foundry

import (
	"context"
	"log/slog"

	"github.com/hanzoai/o11y-foundry/api/v1alpha1"
	"github.com/hanzoai/o11y-foundry/internal/config"
	"github.com/hanzoai/o11y-foundry/internal/config/yamlconfig"
	"github.com/hanzoai/o11y-foundry/internal/molding"
	"github.com/hanzoai/o11y-foundry/internal/molding/ingestermolding"
	"github.com/hanzoai/o11y-foundry/internal/molding/metastoremolding"
	"github.com/hanzoai/o11y-foundry/internal/molding/o11ymolding"
	"github.com/hanzoai/o11y-foundry/internal/molding/telemetrykeepermolding"
	"github.com/hanzoai/o11y-foundry/internal/molding/telemetrystoremolding"
)

type plannerCtor func(ctx context.Context, m v1alpha1.Machinery, logger *slog.Logger) (planner.Planner, error)

type Foundry struct {
	// Config for loading the casting configuration.
	Config config.Config

	// Patchers for applying patches to generated materials, keyed by patch type.
	Patchers map[string]patch.Patch

	// Logger for logging.
	Logger *slog.Logger

	// Planners for the different casting kinds.
	Planners map[v1alpha1.Kind]plannerCtor

	// InfrastructureGenerator for generating infrastructure-as-code manifests.
	InfrastructureGenerator infrastructure.Generator
}

func New(logger *slog.Logger) (*Foundry, error) {
	return &Foundry{
		Config:   yamlConfig,
		Logger:   logger,
		Registry: registry,
		Moldings: map[v1alpha1.MoldingKind]molding.Molding{
			v1alpha1.MoldingKindTelemetryStore:  telemetrystoremolding.New(logger),
			v1alpha1.MoldingKindTelemetryKeeper: telemetrykeepermolding.New(logger),
			v1alpha1.MoldingKindMetaStore:       metastoremolding.New(logger),
			v1alpha1.MoldingKindO11y:          o11ymolding.New(logger),
			v1alpha1.MoldingKindIngester:        ingestermolding.New(logger),
		},
		Logger: logger,
		Planners: map[v1alpha1.Kind]plannerCtor{
			v1alpha1.KindInstallation: func(ctx context.Context, m v1alpha1.Machinery, logger *slog.Logger) (planner.Planner, error) {
				return installationcasting.NewPlanner(ctx, m.(*installation.Casting), logger)
			},
			v1alpha1.KindCollectionAgent: func(ctx context.Context, m v1alpha1.Machinery, logger *slog.Logger) (planner.Planner, error) {
				return collectionagentcasting.NewPlanner(ctx, m.(*collectionagent.Casting), logger)
			},
		},
		InfrastructureGenerator: terraformgenerator.New(logger),
	}, nil
}

func (foundry *Foundry) newPlanner(ctx context.Context, m v1alpha1.Machinery) (planner.Planner, error) {
	ctor, ok := foundry.Planners[m.Kind()]
	if !ok {
		return nil, foundryerrors.Newf(foundryerrors.TypeUnsupported, "unsupported casting kind %q", m.Kind())
	}
	return ctor(ctx, m, foundry.Logger)
}