package server_test

import (
	"errors"
	"fmt"
	"io"
	"net"
	"net/http"
	"net/http/httptest"
	"strings"
	"syscall"
	"testing"

	"git.akyoto.dev/go/assert"
	"git.akyoto.dev/go/server"
)

func TestRouter(t *testing.T) {
	s := server.New()

	s.Get("/", func(ctx server.Context) error {
		return ctx.Bytes([]byte("Hello"))
	})

	s.Get("/string", func(ctx server.Context) error {
		return ctx.String("Hello")
	})

	s.Get("/write", func(ctx server.Context) error {
		_, err := ctx.Write([]byte("Hello"))
		return err
	})

	s.Get("/writestring", func(ctx server.Context) error {
		_, err := io.WriteString(ctx, "Hello")
		return err
	})

	s.Get("/error", func(ctx server.Context) error {
		return ctx.Status(http.StatusUnauthorized).Error("Not logged in")
	})

	s.Get("/error2", func(ctx server.Context) error {
		return ctx.Status(http.StatusUnauthorized).Error("Not logged in", errors.New("Missing auth token"))
	})

	s.Get("/reader", func(ctx server.Context) error {
		return ctx.Reader(strings.NewReader("Hello"))
	})

	s.Get("/request/data", func(ctx server.Context) error {
		method := ctx.Method()
		protocol := ctx.Protocol()
		host := ctx.Host()
		path := ctx.Path()
		return ctx.String(fmt.Sprintf("%s %s %s %s", method, protocol, host, path))
	})

	s.Get("/request/header", func(ctx server.Context) error {
		acceptEncoding := ctx.RequestHeader("Accept-Encoding")
		return ctx.String(acceptEncoding)
	})

	s.Get("/response/header", func(ctx server.Context) error {
		ctx.Header("Content-Type", "text/plain")
		contentType := ctx.ResponseHeader("Content-Type")
		return ctx.String(contentType)
	})

	s.Get("/blog/:article", func(ctx server.Context) error {
		article := ctx.Get("article")
		return ctx.String(article)
	})

	s.Get("/missing-parameter", func(ctx server.Context) error {
		missing := ctx.Get("missing")
		return ctx.String(missing)
	})

	s.Get("/scheme", func(ctx server.Context) error {
		return ctx.String(ctx.Scheme())
	})

	s.Post("/", func(ctx server.Context) error {
		return ctx.String("Post")
	})

	s.Delete("/", func(ctx server.Context) error {
		return ctx.String("Delete")
	})

	s.Put("/", func(ctx server.Context) error {
		return ctx.String("Put")
	})

	tests := []struct {
		Method string
		URL    string
		Status int
		Body   string
	}{
		{Method: "GET", URL: "/", Status: http.StatusOK, Body: "Hello"},
		{Method: "GET", URL: "/error", Status: http.StatusUnauthorized, Body: "Not logged in"},
		{Method: "GET", URL: "/error2", Status: http.StatusUnauthorized, Body: "Not logged in\nMissing auth token"},
		{Method: "GET", URL: "/not-found", Status: http.StatusNotFound, Body: http.StatusText(http.StatusNotFound)},
		{Method: "GET", URL: "/request/data", Status: http.StatusOK, Body: "GET HTTP/1.1 example.com /request/data"},
		{Method: "GET", URL: "/request/header", Status: http.StatusOK, Body: ""},
		{Method: "GET", URL: "/response/header", Status: http.StatusOK, Body: "text/plain"},
		{Method: "GET", URL: "/reader", Status: http.StatusOK, Body: "Hello"},
		{Method: "GET", URL: "/string", Status: http.StatusOK, Body: "Hello"},
		{Method: "GET", URL: "/scheme", Status: http.StatusOK, Body: "http"},
		{Method: "GET", URL: "/write", Status: http.StatusOK, Body: "Hello"},
		{Method: "GET", URL: "/writestring", Status: http.StatusOK, Body: "Hello"},
		{Method: "GET", URL: "/blog/testing-my-router", Status: http.StatusOK, Body: "testing-my-router"},
		{Method: "GET", URL: "/missing-parameter", Status: http.StatusOK, Body: ""},
		{Method: "POST", URL: "/", Status: http.StatusOK, Body: "Post"},
		{Method: "DELETE", URL: "/", Status: http.StatusOK, Body: "Delete"},
		{Method: "PUT", URL: "/", Status: http.StatusOK, Body: "Put"},
	}

	for _, test := range tests {
		t.Run("example.com"+test.URL, func(t *testing.T) {
			request := httptest.NewRequest(test.Method, "http://example.com"+test.URL, nil)
			response := httptest.NewRecorder()
			s.ServeHTTP(response, request)

			result := response.Result()
			assert.Equal(t, result.StatusCode, test.Status)

			body, err := io.ReadAll(result.Body)
			assert.Nil(t, err)
			assert.DeepEqual(t, string(body), test.Body)
		})
	}
}

func TestMiddleware(t *testing.T) {
	s := server.New()

	s.Use(func(ctx server.Context) error {
		ctx.Header("Middleware", "true")
		return ctx.Next()
	})

	request := httptest.NewRequest(http.MethodGet, "/", nil)
	response := httptest.NewRecorder()
	s.ServeHTTP(response, request)

	assert.Equal(t, response.Header().Get("Middleware"), "true")
}

func TestPanic(t *testing.T) {
	s := server.New()

	s.Router().Add(http.MethodGet, "/panic", func(ctx server.Context) error {
		panic("Something unbelievable happened")
	})

	t.Run("example.com/panic", func(t *testing.T) {
		defer func() {
			r := recover()

			if r == nil {
				t.Error("Didn't panic")
			}
		}()

		request := httptest.NewRequest(http.MethodGet, "/panic", nil)
		response := httptest.NewRecorder()
		s.ServeHTTP(response, request)
	})
}

func TestRun(t *testing.T) {
	s := server.New()

	go func() {
		_, err := http.Get("http://127.0.0.1:8080/")
		assert.Nil(t, err)
		err = syscall.Kill(syscall.Getpid(), syscall.SIGTERM)
		assert.Nil(t, err)
	}()

	s.Run(":8080")
}

func TestUnavailablePort(t *testing.T) {
	listener, err := net.Listen("tcp", ":8080")
	assert.Nil(t, err)
	defer listener.Close()

	s := server.New()
	s.Run(":8080")
}