// Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
// SPDX-License-Identifier: MIT

package pipeline

import (
	"errors"

	"go.opentelemetry.io/collector/component"
	"go.opentelemetry.io/collector/confmap"
	"go.opentelemetry.io/collector/service"

	"github.com/aws/amazon-cloudwatch-agent/translator/translate/otel/common"
)

var (
	ErrNoPipelines = errors.New("no valid pipelines")
)

type Translator common.Translator[*common.ComponentTranslators]

type Translation struct {
	// Pipelines is a map of component IDs to service pipelines.
	Pipelines   map[component.ID]*service.PipelineConfig
	Translators common.ComponentTranslators
}

type translator struct {
	translators common.TranslatorMap[*common.ComponentTranslators]
}

var _ common.Translator[*Translation] = (*translator)(nil)

func NewTranslator(translators common.TranslatorMap[*common.ComponentTranslators]) common.Translator[*Translation] {
	return &translator{translators: translators}
}

func (t *translator) ID() component.ID {
	return component.NewID("")
}

// Translate creates the pipeline configuration.
func (t *translator) Translate(conf *confmap.Conf) (*Translation, error) {
	translation := Translation{
		Pipelines: make(map[component.ID]*service.PipelineConfig),
		Translators: common.ComponentTranslators{
			Receivers:  common.NewTranslatorMap[component.Config](),
			Processors: common.NewTranslatorMap[component.Config](),
			Exporters:  common.NewTranslatorMap[component.Config](),
			Extensions: common.NewTranslatorMap[component.Config](),
		},
	}
	t.translators.Range(func(pt common.Translator[*common.ComponentTranslators]) {
		if pipeline, _ := pt.Translate(conf); pipeline != nil {
			translation.Pipelines[pt.ID()] = &service.PipelineConfig{
				Receivers:  pipeline.Receivers.Keys(),
				Processors: pipeline.Processors.Keys(),
				Exporters:  pipeline.Exporters.Keys(),
			}
			translation.Translators.Receivers.Merge(pipeline.Receivers)
			translation.Translators.Processors.Merge(pipeline.Processors)
			translation.Translators.Exporters.Merge(pipeline.Exporters)
			translation.Translators.Extensions.Merge(pipeline.Extensions)
		}
	})
	if len(translation.Pipelines) == 0 {
		return nil, ErrNoPipelines
	}
	return &translation, nil
}