package cwapi

import (
	"fmt"
	"io/ioutil"
	"net/http"
	"strings"
	"time"
)

const apiVersion = "v1"
const apiURL = "https://my_test_endpoint"

var apiBaseURL = fmt.Sprintf("%s/%s/", apiURL, apiVersion)

var pullResponse = `{
	"statusCode" : 200,
	"message":
	[
		{
			"range_ip" : "1.2.3.4",
			"country" : "FR",
			"as_org" : "Test", 
			"as_num" : "1234",
			"action" : "ban",
			"reason": "crowdsec/test"
		},
		{
			"range_ip" : "1.2.3.5",
			"country" : "FR",
			"as_org" : "Test", 
			"as_num" : "1235",
			"action" : "ban",
			"reason": "crowdsec/test"
		},
		{
			"range_ip" : "1.2.3.6",
			"country" : "FR",
			"as_org" : "Test", 
			"as_num" : "1236",
			"action" : "ban",
			"reason": "crowdsec/test"
		}
	]
}`

type mockTransport struct {
	nbTryPushTokenOK  int // to test token expiration
	nbTryPushTokenNOK int
}

func newMockTransport() http.RoundTripper {
	return &mockTransport{}
}

func newMockClient() *http.Client {
	return &http.Client{
		Transport: newMockTransport(),
		Timeout:   time.Second * 20,
	}
}

// Implement http.RoundTripper
func (t *mockTransport) RoundTrip(req *http.Request) (*http.Response, error) {
	var responseBody string
	var statusCode int
	// Create mocked http.Response
	response := &http.Response{
		Header:  make(http.Header),
		Request: req,
	}
	response.Header.Set("Content-Type", "application/json")
	switch req.URL.Path {
	case "/v1/signin":
		responseBody = `{"statusCode": 200, "message": "crowdsec_api_token"}`
		statusCode = 200
	case "/v1/register":
		responseBody = `{"statusCode": 200, "message": "OK"}`
		statusCode = 200
	case "/v1/signals":
		responseBody = `{"statusCode": 200, "message": "OK"}`
		statusCode = 200
	case "/v1/pull":
		responseBody = pullResponse
		statusCode = 200
	case "/v1/signals_token_expired":
		if t.nbTryPushTokenOK == 0 {
			responseBody = `{"statusCode": 200, "message": "crowdsec_api_token"}`
			statusCode = 401
			t.nbTryPushTokenOK++
		} else {
			responseBody = `{"statusCode": 200, "message": "OK"}`
			statusCode = 200
		}
	case "/v1/signals_token_renew_fail":
		if t.nbTryPushTokenNOK == 0 {
			responseBody = `{"statusCode": 200, "message": "crowdsec_api_token"}`
			statusCode = 401
			t.nbTryPushTokenNOK++
		} else {
			responseBody = `{"statusCode": 500, "message": "token expired"}`
			statusCode = 500
		}
	case "/v1/signals_bad_response_code":
		responseBody = `{"statusCode": 200, "message": "OK"}`
		statusCode = 500
	case "/v1/enroll":
		responseBody = `{"statusCode": 200, "message": "OK"}`
		statusCode = 200
	case "/v1/resetpassword":
		responseBody = `{"statusCode": 200, "message": "password updated successfully"}`
		statusCode = 200
	case "/v1/resetpassword_unknown_user":
		responseBody = `{"statusCode": 500, "message": "User not found"}`
		statusCode = 200
	case "/v1/unknown_path":
		statusCode = 404
		responseBody = `{"error": "unknown URI"}`
	case "/v1/malformed_response":
		statusCode = 200
		responseBody = `{"statusCode" : 200, "msg" : "api_token"`
	case "/v1/bad_response":
		statusCode = 200
		responseBody = `{"statusCode" : 200, "msg" : "api_token"}`
	}
	response.StatusCode = statusCode
	response.Body = ioutil.NopCloser(strings.NewReader(responseBody))
	return response, nil
}