o11y-foundry/internal/casting/coolifycasting/enricher.go

package coolifycasting

import (
	"context"
	"path/filepath"
	"strings"

	"github.com/hanzoai/o11y-foundry/api/v1alpha1"
	"github.com/hanzoai/o11y-foundry/api/v1alpha1/installation"
	rootcasting "github.com/hanzoai/o11y-foundry/internal/casting"
	"github.com/hanzoai/o11y-foundry/internal/domain"
	"github.com/hanzoai/o11y-foundry/internal/errors"
	"github.com/hanzoai/o11y-foundry/internal/molding"
)

var _ molding.MoldingEnricher = (*coolifyMoldingEnricher)(nil)

type coolifyMoldingEnricher struct {
	material domain.StructuredMaterial
}

func newCoolifyMoldingEnricher(config *installation.Casting) (*coolifyMoldingEnricher, error) {
	material, err := getCoolifyMaterial(config, filepath.Join(rootcasting.DeploymentDir, "coolify.yaml"))
	if err != nil {
		return nil, errors.Wrapf(err, errors.TypeInternal, "failed to get coolify yaml material")
	}
	return &coolifyMoldingEnricher{material: material}, nil
}

func (enricher *coolifyMoldingEnricher) EnrichStatus(ctx context.Context, kind v1alpha1.MoldingKind, config *installation.Casting) error {
	switch kind {
	case v1alpha1.MoldingKindTelemetryStore:
		containerNames, err := enricher.material.GetStringSlice("services|@keys")
		if err != nil {
			return errors.Wrapf(err, errors.TypeInternal, "failed to get telemetrystore container names")
		}

		var telemetrystoreContainerNames []string
		for _, containerName := range containerNames {
			if strings.Contains(containerName, "telemetrystore-clickhouse") && !strings.Contains(containerName, "user-scripts") {
				telemetrystoreContainerNames = append(telemetrystoreContainerNames, domain.MustNewAddress("tcp", containerName, 9000).String())
			}
		}
		config.Spec.TelemetryStore.Status.Addresses.TCP = telemetrystoreContainerNames

	case v1alpha1.MoldingKindO11y:
		containerNames, err := enricher.material.GetStringSlice("services|@keys")
		if err != nil {
			return errors.Wrapf(err, errors.TypeInternal, "failed to get signoz container names")
		}

		var apiServerAddr []string
		var opampAddr []string
		for _, containerName := range containerNames {
			if strings.Contains(containerName, "-signoz") {
				apiServerAddr = append(apiServerAddr, domain.MustNewAddress("tcp", containerName, 8080).String())
				opampAddr = append(opampAddr, domain.MustNewAddress("ws", containerName, 4320).String())
			}
		}
		config.Spec.O11y.Status.Addresses.APIServer = apiServerAddr
		config.Spec.O11y.Status.Addresses.Opamp = opampAddr

	case v1alpha1.MoldingKindTelemetryKeeper:
		containerNames, err := enricher.material.GetStringSlice("services|@keys")
		if err != nil {
			return errors.Wrapf(err, errors.TypeInternal, "failed to get telemetrykeeper container names")
		}

		var telemetrykeeperContainerNames []string
		for _, containerName := range containerNames {
			if strings.Contains(containerName, "telemetrykeeper") {
				telemetrykeeperContainerNames = append(telemetrykeeperContainerNames, domain.MustNewAddress("tcp", containerName, 9181).String())
			}
		}
		config.Spec.TelemetryKeeper.Status.Addresses.Client = telemetrykeeperContainerNames

		var telemetryRaftaddress []string
		for _, containerName := range containerNames {
			if strings.Contains(containerName, "telemetrykeeper") {
				telemetryRaftaddress = append(telemetryRaftaddress, domain.MustNewAddress("tcp", containerName, 9234).String())
			}
		}
		config.Spec.TelemetryKeeper.Status.Addresses.Raft = telemetryRaftaddress

	case v1alpha1.MoldingKindMetaStore:
		// Skip molding enrichment if sqlite
		if config.Spec.MetaStore.Kind == installation.MetaStoreKindSQLite {
			return nil
		}
		containerNames, err := enricher.material.GetStringSlice("services|@keys")
		if err != nil {
			return errors.Wrapf(err, errors.TypeInternal, "failed to get metastore container names")
		}

		var metastoreContainerNames []string
		for _, containerName := range containerNames {
			if strings.Contains(containerName, "metastore") {
				metastoreContainerNames = append(metastoreContainerNames, domain.MustNewAddress("tcp", containerName, 5432).String())
			}
		}
		config.Spec.MetaStore.Status.Addresses.DSN = metastoreContainerNames

	case v1alpha1.MoldingKindIngester:
		// The ingester is scaled via `deploy.replicas` and reached through the
		// `<metadata.name>-ingester` network alias on the default network,
		// which compose load-balances across all replicas.
		config.Spec.Ingester.Status.Addresses.OTLP = []string{
			domain.MustNewAddress("tcp", config.Metadata.Name+"-ingester", 4318).String(),
			domain.MustNewAddress("tcp", config.Metadata.Name+"-ingester", 4317).String(),
		}
	}

	return nil
}