package monitors_test

import (
	"bytes"
	"context"
	"encoding/json"
	"io"
	"net/http"
	"testing"

	"github.com/openstatusHQ/cli/internal/config"
	"github.com/openstatusHQ/cli/internal/monitors"
)

func Test_UpdateMonitor(t *testing.T) {
	t.Parallel()

	t.Run("Update HTTP monitor successfully", func(t *testing.T) {
		body := `{"monitor":{"id":"123","name":"Updated Monitor","url":"https://updated.example.com","periodicity":"PERIODICITY_5M","method":"HTTP_METHOD_POST","regions":["REGION_FLY_IAD"],"active":true}}`
		r := io.NopCloser(bytes.NewReader([]byte(body)))

		interceptor := &interceptorHTTPClient{
			f: func(req *http.Request) (*http.Response, error) {
				if req.Header.Get("x-openstatus-key") != "test-api-key" {
					t.Errorf("Expected x-openstatus-key header, got %s", req.Header.Get("x-openstatus-key"))
				}
				return &http.Response{
					StatusCode: http.StatusOK,
					Body:       r,
					Header:     http.Header{"Content-Type": []string{"application/json"}},
				}, nil
			},
		}

		monitor := config.Monitor{
			Name:      "Updated Monitor",
			Active:    true,
			Frequency: config.The5M,
			Kind:      config.HTTP,
			Regions:   []config.Region{config.Iad},
			Request: config.Request{
				URL:    "https://updated.example.com",
				Method: config.Post,
			},
		}

		result, err := monitors.UpdateMonitor(context.Background(), interceptor.GetHTTPClient(), "test-api-key", 123, monitor)
		if err != nil {
			t.Fatalf("Expected no error, got %v", err)
		}

		if result.ID != 123 {
			t.Errorf("Expected ID 123, got %d", result.ID)
		}
		if result.Name != "Updated Monitor" {
			t.Errorf("Expected name 'Updated Monitor', got %s", result.Name)
		}
	})

	t.Run("Update TCP monitor successfully", func(t *testing.T) {
		body := `{"monitor":{"id":"456","name":"Updated TCP Monitor","uri":"updated.example.com:8080","periodicity":"PERIODICITY_1M","regions":["REGION_FLY_AMS"],"active":true}}`
		r := io.NopCloser(bytes.NewReader([]byte(body)))

		interceptor := &interceptorHTTPClient{
			f: func(req *http.Request) (*http.Response, error) {
				return &http.Response{
					StatusCode: http.StatusOK,
					Body:       r,
					Header:     http.Header{"Content-Type": []string{"application/json"}},
				}, nil
			},
		}

		monitor := config.Monitor{
			Name:      "Updated TCP Monitor",
			Active:    true,
			Frequency: config.The1M,
			Kind:      config.TCP,
			Regions:   []config.Region{config.Ams},
			Request: config.Request{
				Host: "updated.example.com",
				Port: 8080,
			},
		}

		result, err := monitors.UpdateMonitor(context.Background(), interceptor.GetHTTPClient(), "test-api-key", 456, monitor)
		if err != nil {
			t.Fatalf("Expected no error, got %v", err)
		}

		if result.ID != 456 {
			t.Errorf("Expected ID 456, got %d", result.ID)
		}
		if result.JobType != "tcp" {
			t.Errorf("Expected jobType 'tcp', got %s", result.JobType)
		}
	})

	t.Run("Update ICMP monitor successfully", func(t *testing.T) {
		body := `{"monitor":{"id":"457","name":"Updated ICMP Monitor","uri":"updated.example.com","periodicity":"PERIODICITY_1M","regions":["REGION_FLY_AMS"],"active":true}}`
		r := io.NopCloser(bytes.NewReader([]byte(body)))

		interceptor := &interceptorHTTPClient{
			f: func(req *http.Request) (*http.Response, error) {
				return &http.Response{
					StatusCode: http.StatusOK,
					Body:       r,
					Header:     http.Header{"Content-Type": []string{"application/json"}},
				}, nil
			},
		}

		monitor := config.Monitor{
			Name:      "Updated ICMP Monitor",
			Active:    true,
			Frequency: config.The1M,
			Kind:      config.ICMP,
			Regions:   []config.Region{config.Ams},
			Request: config.Request{
				Host: "updated.example.com",
			},
		}

		result, err := monitors.UpdateMonitor(context.Background(), interceptor.GetHTTPClient(), "test-api-key", 457, monitor)
		if err != nil {
			t.Fatalf("Expected no error, got %v", err)
		}

		if result.ID != 457 {
			t.Errorf("Expected ID 457, got %d", result.ID)
		}
		if result.JobType != "icmp" {
			t.Errorf("Expected jobType 'icmp', got %s", result.JobType)
		}
		if result.URL != "updated.example.com" {
			t.Errorf("Expected URL 'updated.example.com', got %s", result.URL)
		}
	})

	t.Run("Update monitor fails with error response", func(t *testing.T) {
		body := `{"code":"not_found","message":"monitor not found"}`
		r := io.NopCloser(bytes.NewReader([]byte(body)))

		interceptor := &interceptorHTTPClient{
			f: func(req *http.Request) (*http.Response, error) {
				return &http.Response{
					StatusCode: http.StatusNotFound,
					Body:       r,
					Header:     http.Header{"Content-Type": []string{"application/json"}},
				}, nil
			},
		}

		monitor := config.Monitor{
			Name: "Test Monitor",
			Kind: config.HTTP,
		}

		_, err := monitors.UpdateMonitor(context.Background(), interceptor.GetHTTPClient(), "test-api-key", 999, monitor)
		if err == nil {
			t.Error("Expected error for not found status, got nil")
		}
	})

	t.Run("Unsupported monitor kind returns error", func(t *testing.T) {
		interceptor := &interceptorHTTPClient{
			f: func(req *http.Request) (*http.Response, error) {
				return &http.Response{
					StatusCode: http.StatusOK,
					Body:       io.NopCloser(bytes.NewReader([]byte(`{}`))),
					Header:     http.Header{"Content-Type": []string{"application/json"}},
				}, nil
			},
		}

		monitor := config.Monitor{
			Name: "Test Monitor",
			Kind: "unsupported",
		}

		_, err := monitors.UpdateMonitor(context.Background(), interceptor.GetHTTPClient(), "test-api-key", 123, monitor)
		if err == nil {
			t.Error("Expected error for unsupported monitor kind, got nil")
		}
	})

	t.Run("Update HTTP monitor returns correct Monitor struct", func(t *testing.T) {
		body := `{"monitor":{"id":"789","name":"Full Monitor","description":"Test description","url":"https://test.example.com","periodicity":"PERIODICITY_5M","method":"HTTP_METHOD_POST","regions":["REGION_FLY_IAD","REGION_FLY_AMS","REGION_FLY_SYD"],"active":true,"public":true,"timeout":30000,"retry":2}}`
		r := io.NopCloser(bytes.NewReader([]byte(body)))

		interceptor := &interceptorHTTPClient{
			f: func(req *http.Request) (*http.Response, error) {
				return &http.Response{
					StatusCode: http.StatusOK,
					Body:       r,
					Header:     http.Header{"Content-Type": []string{"application/json"}},
				}, nil
			},
		}

		monitor := config.Monitor{
			Name: "Full Monitor",
			Kind: config.HTTP,
		}

		result, err := monitors.UpdateMonitor(context.Background(), interceptor.GetHTTPClient(), "test-api-key", 789, monitor)
		if err != nil {
			t.Fatalf("Expected no error, got %v", err)
		}

		if result.ID != 789 {
			t.Errorf("Expected ID 789, got %d", result.ID)
		}
		if result.Name != "Full Monitor" {
			t.Errorf("Expected name 'Full Monitor', got %s", result.Name)
		}
		if result.Description != "Test description" {
			t.Errorf("Expected description 'Test description', got %s", result.Description)
		}
		if result.URL != "https://test.example.com" {
			t.Errorf("Expected URL 'https://test.example.com', got %s", result.URL)
		}
		if result.Periodicity != "5m" {
			t.Errorf("Expected periodicity '5m', got %s", result.Periodicity)
		}
		if result.Method != http.MethodPost {
			t.Errorf("Expected method 'POST', got %s", result.Method)
		}
		if result.Active != true {
			t.Errorf("Expected active true, got %v", result.Active)
		}
		if result.Public != true {
			t.Errorf("Expected public true, got %v", result.Public)
		}
		if result.Timeout != 30000 {
			t.Errorf("Expected timeout 30000, got %d", result.Timeout)
		}
		if result.Retry != 2 {
			t.Errorf("Expected retry 2, got %d", result.Retry)
		}
		if result.JobType != "http" {
			t.Errorf("Expected jobType 'http', got %s", result.JobType)
		}
	})
}

func Test_UpdateMonitor_FollowRedirects(t *testing.T) {
	t.Parallel()

	t.Run("followRedirects false is sent in update request body", func(t *testing.T) {
		var capturedBody map[string]json.RawMessage

		interceptor := &interceptorHTTPClient{
			f: func(req *http.Request) (*http.Response, error) {
				bodyBytes, _ := io.ReadAll(req.Body)
				json.Unmarshal(bodyBytes, &capturedBody)

				respBody := `{"monitor":{"id":"123","name":"Redirect Monitor","url":"https://example.com","periodicity":"PERIODICITY_10M","method":"HTTP_METHOD_GET","regions":["REGION_FLY_IAD"],"active":true,"followRedirects":false}}`
				return &http.Response{
					StatusCode: http.StatusOK,
					Body:       io.NopCloser(bytes.NewReader([]byte(respBody))),
					Header:     http.Header{"Content-Type": []string{"application/json"}},
				}, nil
			},
		}

		monitor := config.Monitor{
			Name:      "Redirect Monitor",
			Active:    true,
			Frequency: config.The10M,
			Kind:      config.HTTP,
			Regions:   []config.Region{config.Iad},
			Request: config.Request{
				URL:             "https://example.com",
				Method:          config.Get,
				FollowRedirects: boolPtr(false),
			},
		}

		_, err := monitors.UpdateMonitor(context.Background(), interceptor.GetHTTPClient(), "test-api-key", 123, monitor)
		if err != nil {
			t.Fatalf("Expected no error, got %v", err)
		}

		var monitorPayload map[string]json.RawMessage
		json.Unmarshal(capturedBody["monitor"], &monitorPayload)

		followRedirectsRaw, exists := monitorPayload["followRedirects"]
		if !exists {
			t.Fatal("Expected followRedirects to be present in update request body")
		}

		var followRedirects bool
		json.Unmarshal(followRedirectsRaw, &followRedirects)
		if followRedirects != false {
			t.Errorf("Expected followRedirects to be false, got %v", followRedirects)
		}
	})

	t.Run("followRedirects nil is omitted from update request body", func(t *testing.T) {
		var capturedBody map[string]json.RawMessage

		interceptor := &interceptorHTTPClient{
			f: func(req *http.Request) (*http.Response, error) {
				bodyBytes, _ := io.ReadAll(req.Body)
				json.Unmarshal(bodyBytes, &capturedBody)

				respBody := `{"monitor":{"id":"124","name":"Default Monitor","url":"https://example.com","periodicity":"PERIODICITY_10M","method":"HTTP_METHOD_GET","regions":["REGION_FLY_IAD"],"active":true}}`
				return &http.Response{
					StatusCode: http.StatusOK,
					Body:       io.NopCloser(bytes.NewReader([]byte(respBody))),
					Header:     http.Header{"Content-Type": []string{"application/json"}},
				}, nil
			},
		}

		monitor := config.Monitor{
			Name:      "Default Monitor",
			Active:    true,
			Frequency: config.The10M,
			Kind:      config.HTTP,
			Regions:   []config.Region{config.Iad},
			Request: config.Request{
				URL:    "https://example.com",
				Method: config.Get,
			},
		}

		_, err := monitors.UpdateMonitor(context.Background(), interceptor.GetHTTPClient(), "test-api-key", 124, monitor)
		if err != nil {
			t.Fatalf("Expected no error, got %v", err)
		}

		var monitorPayload map[string]json.RawMessage
		json.Unmarshal(capturedBody["monitor"], &monitorPayload)

		if _, exists := monitorPayload["followRedirects"]; exists {
			t.Error("Expected followRedirects to be absent from update request body when nil")
		}
	})
}