package ofrep

import (
	"context"
	"net/http"

	"github.com/grafana/grafana/pkg/infra/tracing"
	goffmodel "github.com/thomaspoignant/go-feature-flag/cmd/relayproxy/model"
	"go.opentelemetry.io/otel/attribute"
)

func (b *APIBuilder) evalAllFlagsStatic(ctx context.Context, isAuthedUser bool, w http.ResponseWriter) {
	_, span := tracing.Start(ctx, "ofrep.static.evalAllFlags")
	defer span.End()

	result, err := b.staticEvaluator.EvalAllFlags(ctx)
	if err != nil {
		err = tracing.Error(span, err)
		b.logger.Error("Failed to evaluate all static flags", "error", err)
		http.Error(w, "failed to evaluate flags", http.StatusInternalServerError)
		return
	}

	span.SetAttributes(attribute.Int("total_flags_count", len(result.Flags)))

	if !isAuthedUser {
		var publicOnly []goffmodel.OFREPFlagBulkEvaluateSuccessResponse

		for _, flag := range result.Flags {
			if isPublicFlag(flag.Key) {
				publicOnly = append(publicOnly, flag)
			}
		}

		result.Flags = publicOnly
		span.SetAttributes(attribute.Int("public_flags_count", len(publicOnly)))
	}

	writeResponse(http.StatusOK, result, b.logger, w)
}

func (b *APIBuilder) evalFlagStatic(ctx context.Context, flagKey string, w http.ResponseWriter) {
	_, span := tracing.Start(ctx, "ofrep.static.evalFlag")
	defer span.End()

	span.SetAttributes(attribute.String("flag_key", flagKey))

	result, err := b.staticEvaluator.EvalFlag(ctx, flagKey)
	if err != nil {
		err = tracing.Error(span, err)
		b.logger.Error("Failed to evaluate static flag", "key", flagKey, "error", err)
		http.Error(w, "failed to evaluate flag", http.StatusInternalServerError)
		return
	}

	writeResponse(http.StatusOK, result, b.logger, w)
}