package boot

import (
	"os"
	"testing"
	"time"

	"reasonix/internal/completioneval"
	"reasonix/internal/provider"
)

// mainConversationRequests drops the completion validator's isolated requests
// so prefix assertions stay aimed at the main conversation.
func mainConversationRequests(reqs []provider.Request) []provider.Request {
	out := make([]provider.Request, 0, len(reqs))
	for _, req := range reqs {
		if len(req.Messages) > 0 && req.Messages[0].Role == provider.RoleSystem &&
			req.Messages[0].Content == completioneval.PolicyPrompt {
			continue
		}
		out = append(out, req)
	}
	return out
}

// robustTempDir is a drop-in for t.TempDir whose cleanup retries RemoveAll for a
// short window. Tests here build a full Controller (Build / control.New); at
// teardown a background resource — a job goroutine draining after its context is
// cancelled, or an MCP stats/schema writer flushing — can still hold a file
// under the dir for a few milliseconds after Close returns. On Windows that
// surfaces as "being used by another process"; on Linux a write racing RemoveAll
// surfaces as "directory not empty". Plain t.TempDir turns that teardown race
// into a red test even though every assertion passed (this is the recurring
// main-v2 CI flake that #3371 only papered over). Retrying absorbs the race; a
// dir that never frees is logged, not fatal, so a genuine leak stays visible
// without reintroducing the flake.
func robustTempDir(t *testing.T) string {
	t.Helper()
	dir, err := os.MkdirTemp("", "reasonix-test-*")
	if err != nil {
		t.Fatalf("robustTempDir: %v", err)
	}
	t.Cleanup(func() {
		var rmErr error
		for range 100 {
			if rmErr = os.RemoveAll(dir); rmErr == nil {
				return
			}
			time.Sleep(20 * time.Millisecond)
		}
		t.Logf("robustTempDir: cleanup did not converge for %s: %v", dir, rmErr)
	})
	return dir
}