package auth0

import (
	"context"
	"net/http/httptest"
	"strings"
	"sync"
	"testing"

	"golang.org/x/oauth2"

	"git.citc.tech/go/web/auth/auth0/authenticator"
)

type mockLogger struct{}

func (m *mockLogger) Debug(msg string, args ...any) {}
func (m *mockLogger) Info(msg string, args ...any)  {}
func (m *mockLogger) Error(msg string, args ...any) {}

type mockSessionManager struct {
	store map[string]any
	mu    sync.RWMutex
}

func (m *mockSessionManager) Get(ctx context.Context, key string) any {
	m.mu.RLock()
	defer m.mu.RUnlock()
	return m.store[key]
}

func (m *mockSessionManager) Put(ctx context.Context, key string, value any) {
	m.mu.Lock()
	defer m.mu.Unlock()
	if value == nil {
		delete(m.store, key)
	} else {
		m.store[key] = value
	}
}

func TestHandleLogic(t *testing.T) {
	tests := []struct {
		name             string
		existingState    any
		wantRedirect     string
		wantSessionKey   string
		wantSessionValue string
	}{
		{
			name:             "generates and stores state",
			wantRedirect:     "https://",
			wantSessionKey:   StateKey,
			wantSessionValue: "",
		},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			mockSessions := &mockSessionManager{
				store: make(map[string]any),
			}
			if tt.existingState != nil {
				mockSessions.store[StateKey] = tt.existingState
			}

			d := &deps{
				log:      &mockLogger{},
				sessions: mockSessions,
				auth: &authenticator.Authenticator{
					Config: oauth2.Config{
						Endpoint: oauth2.Endpoint{
							AuthURL: "https://test.auth0.com/authorize",
						},
					},
				},
			}

			rr := httptest.NewRecorder()
			req := httptest.NewRequest("GET", "/login", nil)

			HandleLogin(d)(rr, req)

			if !strings.HasPrefix(rr.Header().Get("Location"), "https://test.auth0.com/authorize") {
				t.Errorf("wrong redirect: %s", rr.Header().Get("Location"))
			}

			state := mockSessions.store[StateKey].(string)
			if len(state) == 0 {
				t.Fatal("state not stored or empty")
			}
		})
	}
}