package cloudflare

import (
	"context"
	"fmt"
	"net/http"
	"testing"
	"time"

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

const (
	testPagesDeploymentResponse = `
	{
		"id": "0012e50b-fa5d-44db-8cb5-1f372785dcbe",
		"short_id": "0012e50b",
		"project_id": "80776025-b1bd-4181-993f-8238c27d226f",
		"project_name": "test",
		"environment": "production",
		"url": "https://0012e50b.test.pages.dev",
		"created_on": "2021-01-01T00:00:00Z",
		"modified_on": "2021-01-01T00:00:00Z",
		"latest_stage": {
			"name": "deploy",
			"started_on": "2021-01-01T00:00:00Z",
			"ended_on": "2021-01-01T00:00:00Z",
			"status": "success"
		},
		"deployment_trigger": {
			"type": "ad_hoc",
			"metadata": {
				"branch": "main",
				"commit_hash": "20fb65fa9d7fd2a11f7fa3ebdc44137b263ee835",
				"commit_message": "Test commit"
			}
		},
		"stages": [
			{
				"name": "queued",
				"started_on": "2021-01-01T00:00:00Z",
				"ended_on": "2021-01-01T00:00:00Z",
				"status": "success"
			},
			{
				"name": "initialize",
				"started_on": "2021-01-01T00:00:00Z",
				"ended_on": "2021-01-01T00:00:00Z",
				"status": "success"
			},
			{
				"name": "clone_repo",
				"started_on": "2021-01-01T00:00:00Z",
				"ended_on": "2021-01-01T00:00:00Z",
				"status": "success"
			},
			{
				"name": "build",
				"started_on": "2021-01-01T00:00:00Z",
				"ended_on": "2021-01-01T00:00:00Z",
				"status": "success"
			},
			{
				"name": "deploy",
				"started_on": "2021-01-01T00:00:00Z",
				"ended_on": "2021-01-01T00:00:00Z",
				"status": "success"
			}
		],
		"build_config": {
			"build_command": "bash test.sh",
			"destination_dir": "",
			"root_dir": "",
			"web_analytics_tag": null,
			"web_analytics_token": null
		},
		"source": {
			"type": "github",
			"config": {
				"owner": "coudflare",
				"repo_name": "Test",
				"production_branch": "main",
				"pr_comments_enabled": false
			}
		},
		"env_vars": {
			"NODE_VERSION": {
				"value": "16"
			}
		},
		"aliases": null
	}`

	testPagesDeploymentLogsResponse = `
	{
		"total": 6,
		"includes_container_logs": true,
		"data": [
			{
				"ts": "2021-01-01T00:00:00Z",
				"line": "Installing dependencies"
			},
			{
				"ts": "2021-01-01T00:00:00Z",
				"line": "Verify run directory"
			},
			{
				"ts": "2021-01-01T00:00:00Z",
				"line": "Executing user command: bash test.sh"
			},
			{
				"ts": "2021-01-01T00:00:00Z",
				"line": "Finished"
			},
			{
				"ts": "2021-01-01T00:00:00Z",
				"line": "Building functions..."
			},
			{
				"ts": "2021-01-01T00:00:00Z",
				"line": "Validating asset output directory"
			},
			{
				"ts": "2021-01-01T00:00:00Z",
				"line": "Parsed 2 valid header rules."
			}
		]
	}`
)

var (
	pagesDeploymentDummyTime, _ = time.Parse(time.RFC3339, "2021-01-01T00:00:00Z")

	expectedPagesDeployment = &PagesProjectDeployment{
		ID:          "0012e50b-fa5d-44db-8cb5-1f372785dcbe",
		ShortID:     "0012e50b",
		ProjectID:   "80776025-b1bd-4181-993f-8238c27d226f",
		ProjectName: "test",
		Environment: "production",
		URL:         "https://0012e50b.test.pages.dev",
		CreatedOn:   &pagesDeploymentDummyTime,
		ModifiedOn:  &pagesDeploymentDummyTime,
		Aliases:     nil,
		LatestStage: *expectedPagesDeploymentLatestStage,
		EnvVars: EnvironmentVariableMap{
			"NODE_VERSION": &EnvironmentVariable{
				Value: "16",
			},
		},
		DeploymentTrigger: PagesProjectDeploymentTrigger{
			Type: "ad_hoc",
			Metadata: &PagesProjectDeploymentTriggerMetadata{
				Branch:        "main",
				CommitHash:    "20fb65fa9d7fd2a11f7fa3ebdc44137b263ee835",
				CommitMessage: "Test commit",
			},
		},
		Stages: expectedPagesDeploymentStages,
		BuildConfig: PagesProjectBuildConfig{
			BuildCommand:   "bash test.sh",
			DestinationDir: "",
			RootDir:        "",
		},
		Source: PagesProjectSource{
			Type: "github",
			Config: &PagesProjectSourceConfig{
				Owner:             "coudflare",
				RepoName:          "Test",
				ProductionBranch:  "main",
				PRCommentsEnabled: false,
			},
		},
	}

	expectedPagesDeploymentStages = []PagesProjectDeploymentStage{
		{
			Name:      "queued",
			StartedOn: &pagesDeploymentDummyTime,
			EndedOn:   &pagesDeploymentDummyTime,
			Status:    "success",
		},
		{
			Name:      "initialize",
			StartedOn: &pagesDeploymentDummyTime,
			EndedOn:   &pagesDeploymentDummyTime,
			Status:    "success",
		},
		{
			Name:      "clone_repo",
			StartedOn: &pagesDeploymentDummyTime,
			EndedOn:   &pagesDeploymentDummyTime,
			Status:    "success",
		},
		{
			Name:      "build",
			StartedOn: &pagesDeploymentDummyTime,
			EndedOn:   &pagesDeploymentDummyTime,
			Status:    "success",
		},
		*expectedPagesDeploymentLatestStage,
	}

	expectedPagesDeploymentLatestStage = &PagesProjectDeploymentStage{
		Name:      "deploy",
		StartedOn: &pagesDeploymentDummyTime,
		EndedOn:   &pagesDeploymentDummyTime,
		Status:    "success",
	}

	expectedPagesDeploymentLogs = &PagesDeploymentLogs{
		Total:                 6,
		IncludesContainerLogs: true,
		Data:                  expectedPagesDeploymentLogEntries,
	}

	expectedPagesDeploymentLogEntries = []PagesDeploymentLogEntry{
		{
			Timestamp: &pagesDeploymentDummyTime,
			Line:      "Installing dependencies",
		},
		{
			Timestamp: &pagesDeploymentDummyTime,
			Line:      "Verify run directory",
		},
		{
			Timestamp: &pagesDeploymentDummyTime,
			Line:      "Executing user command: bash test.sh",
		},
		{
			Timestamp: &pagesDeploymentDummyTime,
			Line:      "Finished",
		},
		{
			Timestamp: &pagesDeploymentDummyTime,
			Line:      "Building functions...",
		},
		{
			Timestamp: &pagesDeploymentDummyTime,
			Line:      "Validating asset output directory",
		},
		{
			Timestamp: &pagesDeploymentDummyTime,
			Line:      "Parsed 2 valid header rules.",
		},
	}
)

func TestListPagesDeployments(t *testing.T) {
	setup()
	defer teardown()

	handler := func(w http.ResponseWriter, r *http.Request) {
		assert.Equal(t, http.MethodGet, r.Method, "Expected method 'GET', got %s", r.Method)
		assert.Equal(t, "25", r.URL.Query().Get("per_page"))
		assert.Equal(t, "1", r.URL.Query().Get("page"))

		w.Header().Set("content-type", "application/json")
		fmt.Fprintf(w, `{
			"success": true,
			"errors": [],
			"messages": [],
			"result": [
				%s
			],
			"result_info": {
				"page": 1,
				"per_page": 100,
				"count": 1,
				"total_count": 1
			  }
		}`, testPagesDeploymentResponse)
	}

	mux.HandleFunc("/accounts/"+testAccountID+"/pages/projects/test/deployments", handler)

	expectedPagesDeployments := []PagesProjectDeployment{
		*expectedPagesDeployment,
	}
	expectedResultInfo := ResultInfo{
		Page:    1,
		PerPage: 100,
		Count:   1,
		Total:   1,
	}
	actual, resultInfo, err := client.ListPagesDeployments(context.Background(), AccountIdentifier(testAccountID), ListPagesDeploymentsParams{
		ProjectName: "test",
		ResultInfo:  ResultInfo{},
	})
	if assert.NoError(t, err) {
		assert.Equal(t, expectedPagesDeployments, actual)
		assert.Equal(t, &expectedResultInfo, resultInfo)
	}
}

func TestListPagesDeploymentsPagination(t *testing.T) {
	setup()
	defer teardown()
	var page1Called, page2Called bool
	handler := func(w http.ResponseWriter, r *http.Request) {
		page := r.URL.Query().Get("page")
		w.Header().Set("content-type", "application/json")
		switch page {
		case "1":
			page1Called = true
			fmt.Fprintf(w, `{
				"success": true,
				"errors": [],
				"messages": [],
				"result": [
					%s
				],
				"result_info": {
					"page": 1,
					"per_page": 25,
					"total_count": 26,
					"total_pages": 2
				  }
			}`, testPagesDeploymentResponse)
		case "2":
			page2Called = true
			fmt.Fprintf(w, `{
				"success": true,
				"errors": [],
				"messages": [],
				"result": [
					%s
				],
				"result_info": {
					"page": 2,
					"per_page": 25,
					"total_count": 26,
					"total_pages": 2
				  }
			}`, testPagesDeploymentResponse)
		default:
			assert.Failf(t, "Unexpected page number", "Expected page 1 or 2, got %s", page)
			return
		}
	}
	mux.HandleFunc("/accounts/"+testAccountID+"/pages/projects/test/deployments", handler)
	actual, resultInfo, err := client.ListPagesDeployments(context.Background(), AccountIdentifier(testAccountID), ListPagesDeploymentsParams{
		ProjectName: "test",
		ResultInfo:  ResultInfo{},
	})
	if assert.NoError(t, err) {
		assert.True(t, page1Called)
		assert.True(t, page2Called)
		assert.Equal(t, 2, len(actual))
		assert.Equal(t, 26, resultInfo.Total)
	}
}

func TestGetPagesDeploymentInfo(t *testing.T) {
	setup()
	defer teardown()

	handler := func(w http.ResponseWriter, r *http.Request) {
		assert.Equal(t, http.MethodGet, r.Method, "Expected method 'GET', got %s", r.Method)

		w.Header().Set("content-type", "application/json")
		fmt.Fprintf(w, `{
			"success": true,
			"errors": [],
			"messages": [],
			"result": %s
		}`, testPagesDeploymentResponse)
	}

	mux.HandleFunc("/accounts/"+testAccountID+"/pages/projects/test/deployments/0012e50b-fa5d-44db-8cb5-1f372785dcbe", handler)

	actual, err := client.GetPagesDeploymentInfo(context.Background(), AccountIdentifier(testAccountID), "test", "0012e50b-fa5d-44db-8cb5-1f372785dcbe")
	if assert.NoError(t, err) {
		assert.Equal(t, *expectedPagesDeployment, actual)
	}
}

func TestGetPagesDeploymentLogs(t *testing.T) {
	setup()
	defer teardown()

	handler := func(w http.ResponseWriter, r *http.Request) {
		assert.Equal(t, http.MethodGet, r.Method, "Expected method 'GET', got %s", r.Method)

		w.Header().Set("content-type", "application/json")
		fmt.Fprintf(w, `{
			"success": true,
			"errors": [],
			"messages": [],
			"result": %s
		}`, testPagesDeploymentLogsResponse)
	}

	mux.HandleFunc("/accounts/"+testAccountID+"/pages/projects/test/deployments/0012e50b-fa5d-44db-8cb5-1f372785dcbe/history/logs", handler)

	actual, err := client.GetPagesDeploymentLogs(context.Background(), AccountIdentifier(testAccountID), GetPagesDeploymentLogsParams{
		ProjectName:  "test",
		DeploymentID: "0012e50b-fa5d-44db-8cb5-1f372785dcbe",
		SizeOptions:  SizeOptions{},
	})
	if assert.NoError(t, err) {
		assert.Equal(t, *expectedPagesDeploymentLogs, actual)
	}
}

func TestDeletePagesDeployment(t *testing.T) {
	setup()
	defer teardown()

	handler := func(w http.ResponseWriter, r *http.Request) {
		assert.Equal(t, http.MethodDelete, r.Method, "Expected method 'DELETE', got %s", r.Method)
		assert.Equal(t, "true", r.URL.Query().Get("force"))
		w.Header().Set("content-type", "application/json")
		fmt.Fprintf(w, `{
			"success": true,
			"errors": [],
			"messages": [],
			"result": null
		}`)
	}

	mux.HandleFunc("/accounts/"+testAccountID+"/pages/projects/test/deployments/0012e50b-fa5d-44db-8cb5-1f372785dcbe", handler)

	err := client.DeletePagesDeployment(context.Background(), AccountIdentifier(testAccountID), DeletePagesDeploymentParams{ProjectName: "test", DeploymentID: "0012e50b-fa5d-44db-8cb5-1f372785dcbe", Force: true})
	assert.NoError(t, err)
}

func TestCreatePagesDeployment(t *testing.T) {
	setup()
	defer teardown()

	handler := func(w http.ResponseWriter, r *http.Request) {
		assert.Equal(t, http.MethodPost, r.Method, "Expected method 'POST', got %s", r.Method)

		w.Header().Set("content-type", "application/json")
		fmt.Fprintf(w, `{
			"success": true,
			"errors": [],
			"messages": [],
			"result": %s
		}`, testPagesDeploymentResponse)
	}

	mux.HandleFunc("/accounts/"+testAccountID+"/pages/projects/test/deployments", handler)

	actual, err := client.CreatePagesDeployment(context.Background(), AccountIdentifier(testAccountID), CreatePagesDeploymentParams{
		ProjectName: "test",
	})

	if assert.NoError(t, err) {
		assert.Equal(t, *expectedPagesDeployment, actual)
	}
}

func TestRetryPagesDeployment(t *testing.T) {
	setup()
	defer teardown()

	handler := func(w http.ResponseWriter, r *http.Request) {
		assert.Equal(t, http.MethodPost, r.Method, "Expected method 'POST', got %s", r.Method)

		w.Header().Set("content-type", "application/json")
		fmt.Fprintf(w, `{
			"success": true,
			"errors": [],
			"messages": [],
			"result": %s
		}`, testPagesDeploymentResponse)
	}

	mux.HandleFunc("/accounts/"+testAccountID+"/pages/projects/test/deployments/0012e50b-fa5d-44db-8cb5-1f372785dcbe/retry", handler)

	actual, err := client.RetryPagesDeployment(context.Background(), AccountIdentifier(testAccountID), "test", "0012e50b-fa5d-44db-8cb5-1f372785dcbe")
	if assert.NoError(t, err) {
		assert.Equal(t, *expectedPagesDeployment, actual)
	}
}

func TestRollbackPagesDeployment(t *testing.T) {
	setup()
	defer teardown()

	handler := func(w http.ResponseWriter, r *http.Request) {
		assert.Equal(t, http.MethodPost, r.Method, "Expected method 'POST', got %s", r.Method)

		w.Header().Set("content-type", "application/json")
		fmt.Fprintf(w, `{
			"success": true,
			"errors": [],
			"messages": [],
			"result": %s
		}`, testPagesDeploymentResponse)
	}

	mux.HandleFunc("/accounts/"+testAccountID+"/pages/projects/test/deployments/0012e50b-fa5d-44db-8cb5-1f372785dcbe/rollback", handler)

	actual, err := client.RollbackPagesDeployment(context.Background(), AccountIdentifier(testAccountID), "test", "0012e50b-fa5d-44db-8cb5-1f372785dcbe")
	if assert.NoError(t, err) {
		assert.Equal(t, *expectedPagesDeployment, actual)
	}
}