package run_test

import (
	"context"
	"errors"
	"testing"

	"gitea.stevedudenhoeffer.com/steve/majordomo/llm"
	"gitea.stevedudenhoeffer.com/steve/majordomo/provider/fake"

	"gitea.stevedudenhoeffer.com/steve/executus/deliver"
	"gitea.stevedudenhoeffer.com/steve/executus/run"
	"gitea.stevedudenhoeffer.com/steve/executus/tool"
)

type recordingDelivery struct {
	target   deliver.Target
	output   string
	errored  error
	delivers int
}

func (d *recordingDelivery) Deliver(_ context.Context, t deliver.Target, output string, _ []deliver.Artifact) (string, error) {
	d.target, d.output, d.delivers = t, output, d.delivers+1
	return "msg-1", nil
}
func (d *recordingDelivery) DeliverError(_ context.Context, t deliver.Target, e error) error {
	d.target, d.errored = t, e
	return nil
}

func TestDeliveryWired(t *testing.T) {
	d := &recordingDelivery{}
	fp := fake.New("fake")
	fp.Enqueue("m", fake.Reply("the output"))
	m, _ := fp.Model("m")
	ex := run.New(run.Config{
		Registry: tool.NewRegistry(),
		Models:   func(ctx context.Context, _ string) (context.Context, llm.Model, error) { return ctx, m, nil },
		Ports:    run.Ports{Delivery: d},
	})
	// With a delivery target, the executor posts the output.
	ex.Run(context.Background(),
		run.RunnableAgent{Name: "x", ModelTier: "m"},
		tool.Invocation{RunID: "r", DeliveryKind: "channel", DeliveryID: "chan-9"}, "go")
	if d.delivers != 1 || d.output != "the output" || d.target.ID != "chan-9" || d.target.Kind != "channel" {
		t.Fatalf("delivery wrong: %+v out=%q", d.target, d.output)
	}
}

func TestNoDeliveryWithoutTarget(t *testing.T) {
	d := &recordingDelivery{}
	fp := fake.New("fake")
	fp.Enqueue("m", fake.Reply("x"))
	m, _ := fp.Model("m")
	ex := run.New(run.Config{
		Registry: tool.NewRegistry(),
		Models:   func(ctx context.Context, _ string) (context.Context, llm.Model, error) { return ctx, m, nil },
		Ports:    run.Ports{Delivery: d},
	})
	// No DeliveryID → executor delivers nothing (caller reads Result.Output).
	ex.Run(context.Background(),
		run.RunnableAgent{Name: "x", ModelTier: "m"},
		tool.Invocation{RunID: "r"}, "go")
	if d.delivers != 0 {
		t.Errorf("no target should mean no delivery, got %d", d.delivers)
	}
}

func TestDeliveryErrorPath(t *testing.T) {
	d := &recordingDelivery{}
	ex := run.New(run.Config{
		Registry: tool.NewRegistry(),
		Models: func(ctx context.Context, _ string) (context.Context, llm.Model, error) {
			return ctx, nil, errors.New("resolve boom") // forces a run error
		},
		Ports: run.Ports{Delivery: d},
	})
	ex.Run(context.Background(),
		run.RunnableAgent{Name: "x", ModelTier: "m"},
		tool.Invocation{RunID: "r", DeliveryKind: "channel", DeliveryID: "chan-9"}, "go")
	// A model-resolve error returns before the run context exists, so delivery
	// isn't reached — assert no spurious Deliver. (DeliverError on in-loop errors
	// is exercised by the wiring; this guards the early-return path.)
	if d.delivers != 0 {
		t.Errorf("early failure should not Deliver, got %d", d.delivers)
	}
}