package router

import (
	"context"
	"testing"

	"github.com/noahlann/nnet/internal/codec"
	"github.com/noahlann/nnet/internal/request"
	"github.com/noahlann/nnet/internal/response"
	ctxpkg "github.com/noahlann/nnet/pkg/context"
	routerpkg "github.com/noahlann/nnet/pkg/router"
	"github.com/stretchr/testify/assert"
	"github.com/stretchr/testify/require"
)

func TestStringMatcher(t *testing.T) {
	matcher := routerpkg.NewStringMatcher("hello")

	// 创建正确的 Context
	parentCtx := context.Background()
	conn := &mockConnection{id: "test-conn"}
	req := request.New([]byte("test"), nil)
	codecRegistry := codec.NewRegistry()
	resp := response.New(conn, nil, codecRegistry, "json")
	ctx := ctxpkg.New(parentCtx, conn, req, resp)

	// 测试精确匹配
	assert.True(t, matcher.Match(routerpkg.MatchInput{Raw: []byte("hello")}, ctx), "Expected match for 'hello'")

	// 测试不匹配
	assert.False(t, matcher.Match(routerpkg.MatchInput{Raw: []byte("world")}, ctx), "Expected no match for 'world'")
}

func TestRouterRegister(t *testing.T) {
	r := NewRouter()

	handler := func(ctx ctxpkg.Context) error {
		return ctx.Response().WriteString("OK")
	}

	route := r.RegisterString("/hello", handler)
	require.NotNil(t, route, "Expected route to be returned")
}

func TestRouterMatch(t *testing.T) {
	r := NewRouter()

	handler := func(ctx ctxpkg.Context) error {
		return ctx.Response().WriteString("OK")
	}

	r.RegisterString("/hello", handler)

	// 创建正确的 Context
	parentCtx := context.Background()
	conn := &mockConnection{id: "test-conn"}
	req := request.New([]byte("/hello"), nil)
	codecRegistry := codec.NewRegistry()
	resp := response.New(conn, nil, codecRegistry, "json")
	ctx := ctxpkg.New(parentCtx, conn, req, resp)

	route, matchedHandler, err := r.Match(routerpkg.MatchInput{Raw: []byte("/hello")}, ctx)
	require.NoError(t, err, "Expected no error")
	require.NotNil(t, route, "Expected route to be returned")
	require.NotNil(t, matchedHandler, "Expected handler to be returned")
}

func TestRouterMatchNotFound(t *testing.T) {
	r := NewRouter()

	handler := func(ctx ctxpkg.Context) error {
		return ctx.Response().WriteString("OK")
	}

	r.RegisterString("/hello", handler)

	// 创建正确的 Context
	parentCtx := context.Background()
	conn := &mockConnection{id: "test-conn"}
	req := request.New([]byte("/notfound"), nil)
	codecRegistry := codec.NewRegistry()
	resp := response.New(conn, nil, codecRegistry, "json")
	ctx := ctxpkg.New(parentCtx, conn, req, resp)

	_, _, err := r.Match(routerpkg.MatchInput{Raw: []byte("/notfound")}, ctx)
	assert.Error(t, err, "Expected error for unmatched route")
}

// TestRouterGroup moved to group_test.go

func TestRouterRegisterWithOptions(t *testing.T) {
	r := NewRouter()

	type RequestBody struct {
		Name string `json:"name"`
	}

	handler := func(ctx ctxpkg.Context) error {
		return ctx.Response().WriteString("OK")
	}

	route := r.RegisterString("/hello", handler, routerpkg.WithRequestType(&RequestBody{}))
	require.NotNil(t, route, "Expected route to be returned")
	assert.NotNil(t, route.RequestType(), "Expected RequestType to be set")
}

func TestRouterRegisterCustom(t *testing.T) {
	r := NewRouter()

	handler := func(ctx ctxpkg.Context) error {
		return ctx.Response().WriteString("OK")
	}

	customMatcher := func(input routerpkg.MatchInput, ctx ctxpkg.Context) bool {
		return len(input.Raw) > 0 && input.Raw[0] == 'X'
	}

	route := r.RegisterCustom(customMatcher, handler)
	require.NotNil(t, route, "Expected route to be returned")
}

// mockConnection 模拟连接
type mockConnection struct {
	id string
}

func (m *mockConnection) ID() string {
	return m.id
}

func (m *mockConnection) RemoteAddr() string {
	return "127.0.0.1:6995"
}

func (m *mockConnection) LocalAddr() string {
	return "127.0.0.1:6995"
}

func (m *mockConnection) Write(data []byte) error {
	return nil
}

func (m *mockConnection) Close() error {
	return nil
}