package router

import (
	"bytes"
	"encoding/json"
	"fmt"
	"io"
	"net/http"
	"net/http/httptest"
	"testing"

	"github.com/stretchr/testify/require"
)

func TestRouterStatus(t *testing.T) {

	reqPath := "/hello"
	handler := NewRouter()
	helloHandler := func(rctx *Context) {
		rctx.SetStatus(http.StatusOK)
	}
	handler.Get(reqPath, helloHandler)

	request, err := http.NewRequest("GET", reqPath, nil)
	require.NoError(t, err)

	recorder := httptest.NewRecorder()
	handler.ServeHTTP(recorder, request)
	require.Equal(t, http.StatusOK, recorder.Code)

	fmt.Printf("Response code: %d\n", recorder.Code)
}

func TestRouterSendText(t *testing.T) {
	testText := "hello, world"
	reqPath := "/hello"
	rout := NewRouter()
	helloHandler := func(rctx *Context) {
		rctx.SendText(testText)
	}
	rout.Get(reqPath, helloHandler)

	request, err := http.NewRequest("GET", reqPath, nil)
	require.NoError(t, err)

	recorder := httptest.NewRecorder()
	rout.ServeHTTP(recorder, request)
	fmt.Printf("Response code: %d\n", recorder.Code)
	require.Equal(t, http.StatusOK, recorder.Code)

	bodyReader := recorder.Body
	bodyBytes, err := io.ReadAll(bodyReader)

	fmt.Printf("Response body: %s\n", string(bodyBytes))
	require.Equal(t, string(bodyBytes), testText)
}

func TestRouterSendJSON(t *testing.T) {

	type testStruct struct {
		Message string `json:"message"`
	}
	testVar := testStruct{
		Message: "hello, world",
	}
	testData, err := json.Marshal(testVar)
	require.NoError(t, err)

	reqPath := "/hello"
	handler := NewRouter()
	helloHandler := func(rctx *Context) {
		rctx.SendJSON(&testVar)
	}
	handler.Get(reqPath, helloHandler)

	request, err := http.NewRequest("GET", reqPath, nil)
	require.NoError(t, err)

	recorder := httptest.NewRecorder()
	handler.ServeHTTP(recorder, request)
	fmt.Printf("Response code: %d\n", recorder.Code)
	require.Equal(t, http.StatusOK, recorder.Code)

	bodyReader := recorder.Body
	bodyBytes, err := io.ReadAll(bodyReader)
	require.NoError(t, err)
	bodyBytes = bytes.Trim(bodyBytes, "\n\r")

	fmt.Printf("Response body: %s\n", string(bodyBytes))
	require.Equal(t, string(testData), string(bodyBytes))
}

func BenchmarkLoggerL(b *testing.B) {
	reqPath := "/hello"
	helloHandler := func(rctx *Context) {
		rctx.SetStatus(http.StatusOK)
	}
	handler := NewRouter()
	handler.Get(reqPath, helloHandler)

	request, err := http.NewRequest("GET", reqPath, nil)
	require.NoError(b, err)

	for i := 0; i < b.N; i++ {
		recorder := httptest.NewRecorder()
		handler.ServeHTTP(recorder, request)
		require.Equal(b, http.StatusOK, recorder.Code)
	}
}