package coordinator

import (
	"context"
	"testing"
	"time"

	"github.com/stretchr/testify/assert"

	"github.com/bnema/dumber/internal/application/port"
	"github.com/bnema/dumber/internal/domain/entity"
	"github.com/bnema/dumber/internal/infrastructure/webkit"
)

func TestComposeOnClose_Order(t *testing.T) {
	var calls []string

	composed := composeOnClose(
		func() { calls = append(calls, "existing") },
		func() { calls = append(calls, "next") },
	)
	composed()

	assert.Equal(t, []string{"existing", "next"}, calls)
}

func TestComposeOnURIChanged_Order(t *testing.T) {
	var calls []string

	composed := composeOnURIChanged(
		func(_ string) { calls = append(calls, "existing") },
		func(_ string) { calls = append(calls, "next") },
	)
	composed("https://example.com")

	assert.Equal(t, []string{"existing", "next"}, calls)
}

func TestComposeOnLoadChanged_Order(t *testing.T) {
	var calls []string

	composed := composeOnLoadChanged(
		func(_ webkit.LoadEvent) { calls = append(calls, "existing") },
		func(_ webkit.LoadEvent) { calls = append(calls, "next") },
	)
	composed(webkit.LoadCommitted)

	assert.Equal(t, []string{"existing", "next"}, calls)
}

func TestHandlePopupOAuthClose_SuccessSchedulesParentRefresh(t *testing.T) {
	parentPaneID := entity.PaneID("parent-pane")
	popupID := port.WebViewID(101)

	c := &ContentCoordinator{
		webViews:     make(map[entity.PaneID]*webkit.WebView),
		popupOAuth:   make(map[port.WebViewID]*popupOAuthState),
		popupRefresh: make(map[entity.PaneID]*time.Timer),
	}

	c.trackOAuthPopup(popupID, parentPaneID)
	c.capturePopupOAuthState(popupID, "https://app/callback?code=123")
	c.handlePopupOAuthClose(context.Background(), popupID)

	c.popupMu.RLock()
	_, exists := c.popupOAuth[popupID]
	refreshTimer := c.popupRefresh[parentPaneID]
	c.popupMu.RUnlock()

	assert.False(t, exists, "oauth state should be removed after close handling")
	assert.NotNil(t, refreshTimer, "successful oauth close should schedule parent refresh")

	waitFor(t, time.Second, func() bool {
		c.popupMu.RLock()
		defer c.popupMu.RUnlock()
		return c.popupRefresh[parentPaneID] == nil
	})
}

func TestHandlePopupOAuthClose_ErrorDoesNotScheduleParentRefresh(t *testing.T) {
	parentPaneID := entity.PaneID("parent-pane")
	popupID := port.WebViewID(102)

	c := &ContentCoordinator{
		webViews:     make(map[entity.PaneID]*webkit.WebView),
		popupOAuth:   make(map[port.WebViewID]*popupOAuthState),
		popupRefresh: make(map[entity.PaneID]*time.Timer),
	}

	c.trackOAuthPopup(popupID, parentPaneID)
	c.capturePopupOAuthState(popupID, "https://app/callback?error=access_denied")
	c.handlePopupOAuthClose(context.Background(), popupID)

	c.popupMu.RLock()
	defer c.popupMu.RUnlock()
	assert.Nil(t, c.popupRefresh[parentPaneID], "oauth errors should not refresh parent pane")
}

func waitFor(t *testing.T, timeout time.Duration, condition func() bool) {
	t.Helper()

	deadline := time.Now().Add(timeout)
	for time.Now().Before(deadline) {
		if condition() {
			return
		}
		time.Sleep(20 * time.Millisecond)
	}
	t.Fatalf("condition not met before timeout (%s)", timeout)
}