package daemon

import (
	"context"
	"crypto/rand"
	"testing"

	"github.com/stretchr/testify/require"

	"github.com/bnema/vev/internal/domain"
	"github.com/bnema/vev/internal/persist"
	"github.com/bnema/vev/internal/ports"
	recoveryusecase "github.com/bnema/vev/internal/usecase/recovery"
)

type testLifecycleRepository struct{ ports.SnapshotRepository }

func (testLifecycleRepository) DeleteIncarnation(context.Context, domain.IncarnationID) error {
	return nil
}
func (testLifecycleRepository) SnapshotIncarnations(context.Context) ([]domain.IncarnationID, error) {
	return nil, nil
}
func (testLifecycleRepository) CollectIncarnationGarbage(context.Context, domain.IncarnationID, *domain.CheckpointRef) error {
	return nil
}

// WithStore is package-test composition glue. Production injects the same
// recovery coordinator explicitly from internal/app.
func WithStore(t testing.TB, store ports.Store) Option {
	t.Helper()
	return func(d *Daemon) {
		if store == nil {
			return
		}
		catalogue := persist.New(store)
		records, err := catalogue.Records()
		require.NoError(t, err)
		d.catalogue = catalogue
		d.persistEnabled = true
		d.catalogueRecords = records
		d.catalogueRecordsProvided = true
		repository := d.snapshotRepository
		if repository == nil {
			repository = testLifecycleRepository{}
		}
		d.recovery = recoveryusecase.NewCoordinator(catalogue, repository, rand.Reader)
	}
}

func testPersister(t testing.TB, d *Daemon) *persist.Persister {
	t.Helper()
	persister, ok := d.catalogue.(*persist.Persister)
	require.True(t, ok, "catalogue type %T is not *persist.Persister", d.catalogue)
	return persister
}