package awsauth

import (
	"io/ioutil"
	"net/http"
	"net/url"
	"os"
	"strings"
	"testing"
	"time"

	. "github.com/smartystreets/goconvey/convey"
)

func TestIntegration(t *testing.T) {
	if testing.Short() {
		t.Skip("Skipping long-running integration test.")
	}

	Convey("Given real credentials from environment variables", t, func() {
		Convey("A request (with out-of-order query string) with to IAM should succeed (assuming Administrator Access policy)", func() {
			request := newRequest("GET", "https://iam.amazonaws.com/?Version=2010-05-08&Action=ListRoles", nil)

			if !credentialsSet() {
				SkipSo(http.StatusOK, ShouldEqual, http.StatusOK)
			} else {
				response := sign4AndDo(request)
				if response.StatusCode != http.StatusOK {
					message, _ := ioutil.ReadAll(response.Body)
					t.Error(string(message))
				}
				So(response.StatusCode, ShouldEqual, http.StatusOK)
			}
		})

		Convey("A request to S3 should succeed", func() {
			request, _ := http.NewRequest("GET", "https://s3.amazonaws.com", nil)

			if !credentialsSet() {
				SkipSo(http.StatusOK, ShouldEqual, http.StatusOK)
			} else {
				response := sign4AndDo(request)
				if response.StatusCode != http.StatusOK {
					message, _ := ioutil.ReadAll(response.Body)
					t.Error(string(message))
				}
				So(response.StatusCode, ShouldEqual, http.StatusOK)
			}
		})

		Convey("A request to EC2 should succeed", func() {
			request := newRequest("GET", "https://ec2.amazonaws.com/?Version=2013-10-15&Action=DescribeInstances", nil)

			if !credentialsSet() {
				SkipSo(http.StatusOK, ShouldEqual, http.StatusOK)
			} else {
				response := sign2AndDo(request)
				if response.StatusCode != http.StatusOK {
					message, _ := ioutil.ReadAll(response.Body)
					t.Error(string(message))
				}
				So(response.StatusCode, ShouldEqual, http.StatusOK)
			}
		})

		Convey("A request to SQS should succeed", func() {
			request := newRequest("POST", "https://sqs.us-west-2.amazonaws.com", url.Values{
				"Action": []string{"ListQueues"},
			})

			if !credentialsSet() {
				SkipSo(http.StatusOK, ShouldEqual, http.StatusOK)
			} else {
				response := sign4AndDo(request)
				if response.StatusCode != http.StatusOK {
					message, _ := ioutil.ReadAll(response.Body)
					t.Error(string(message))
				}
				So(response.StatusCode, ShouldEqual, http.StatusOK)
			}
		})

		Convey("A request to SES should succeed", func() {
			request := newRequest("GET", "https://email.us-east-1.amazonaws.com/?Action=GetSendStatistics", nil)

			if !credentialsSet() {
				SkipSo(http.StatusOK, ShouldEqual, http.StatusOK)
			} else {
				response := sign3AndDo(request)
				if response.StatusCode != http.StatusOK {
					message, _ := ioutil.ReadAll(response.Body)
					t.Error(string(message))
				}
				So(response.StatusCode, ShouldEqual, http.StatusOK)
			}
		})

		Convey("A request to Route 53 should succeed", func() {
			request := newRequest("GET", "https://route53.amazonaws.com/2013-04-01/hostedzone?maxitems=1", nil)

			if !credentialsSet() {
				SkipSo(http.StatusOK, ShouldEqual, http.StatusOK)
			} else {
				response := sign3AndDo(request)
				if response.StatusCode != http.StatusOK {
					message, _ := ioutil.ReadAll(response.Body)
					t.Error(string(message))
				}
				So(response.StatusCode, ShouldEqual, http.StatusOK)
			}
		})

		Convey("A request to SimpleDB should succeed", func() {
			request := newRequest("GET", "https://sdb.amazonaws.com/?Action=ListDomains&Version=2009-04-15", nil)

			if !credentialsSet() {
				SkipSo(http.StatusOK, ShouldEqual, http.StatusOK)
			} else {
				response := sign2AndDo(request)
				if response.StatusCode != http.StatusOK {
					message, _ := ioutil.ReadAll(response.Body)
					t.Error(string(message))
				}
				So(response.StatusCode, ShouldEqual, http.StatusOK)
			}
		})

		Convey("If S3Resource env variable is set", func() {
			s3res := os.Getenv("S3Resource")

			Convey("A URL-signed request to that S3 resource should succeed", func() {
				request, _ := http.NewRequest("GET", s3res, nil)

				if !credentialsSet() || s3res == "" {
					SkipSo(http.StatusOK, ShouldEqual, http.StatusOK)
				} else {
					response := signS3UrlAndDo(request)
					if response.StatusCode != http.StatusOK {
						message, _ := ioutil.ReadAll(response.Body)
						t.Error(string(message))
					}
					So(response.StatusCode, ShouldEqual, http.StatusOK)
				}
			})
		})
	})
}

func TestSign(t *testing.T) {
	Convey("Requests to services using Version 2 should be signed accordingly", t, func() {
		reqs := []*http.Request{
			newRequest("GET", "https://ec2.amazonaws.com", url.Values{}),
			newRequest("GET", "https://elasticache.amazonaws.com/", url.Values{}),
		}
		for _, request := range reqs {
			signedReq := Sign(request)
			So(signedReq.URL.Query().Get("SignatureVersion"), ShouldEqual, "2")
		}
	})

	Convey("Requests to services using Version 3 should be signed accordingly", t, func() {
		reqs := []*http.Request{
			newRequest("GET", "https://route53.amazonaws.com", url.Values{}),
			newRequest("GET", "https://email.us-east-1.amazonaws.com/", url.Values{}),
		}
		for _, request := range reqs {
			signedReq := Sign(request)
			So(signedReq.Header.Get("X-Amzn-Authorization"), ShouldNotBeBlank)
		}
	})

	Convey("Requests to services using Version 4 should be signed accordingly", t, func() {
		reqs := []*http.Request{
			newRequest("POST", "https://sqs.amazonaws.com/", url.Values{}),
			newRequest("GET", "https://iam.amazonaws.com", url.Values{}),
			newRequest("GET", "https://s3.amazonaws.com", url.Values{}),
		}
		for _, request := range reqs {
			signedReq := Sign(request)
			So(signedReq.Header.Get("Authorization"), ShouldContainSubstring, ", Signature=")
		}
	})

	var keys Credentials
	keys = newKeys()
	Convey("Requests to services using existing credentials Version 2 should be signed accordingly", t, func() {
		reqs := []*http.Request{
			newRequest("GET", "https://ec2.amazonaws.com", url.Values{}),
			newRequest("GET", "https://elasticache.amazonaws.com/", url.Values{}),
		}
		for _, request := range reqs {
			signedReq := Sign(request, keys)
			So(signedReq.URL.Query().Get("SignatureVersion"), ShouldEqual, "2")
		}
	})

	Convey("Requests to services using existing credentials Version 3 should be signed accordingly", t, func() {
		reqs := []*http.Request{
			newRequest("GET", "https://route53.amazonaws.com", url.Values{}),
			newRequest("GET", "https://email.us-east-1.amazonaws.com/", url.Values{}),
		}
		for _, request := range reqs {
			signedReq := Sign(request, keys)
			So(signedReq.Header.Get("X-Amzn-Authorization"), ShouldNotBeBlank)
		}
	})

	Convey("Requests to services using existing credentials Version 4 should be signed accordingly", t, func() {
		reqs := []*http.Request{
			newRequest("POST", "https://sqs.amazonaws.com/", url.Values{}),
			newRequest("GET", "https://iam.amazonaws.com", url.Values{}),
			newRequest("GET", "https://s3.amazonaws.com", url.Values{}),
		}
		for _, request := range reqs {
			signedReq := Sign(request, keys)
			So(signedReq.Header.Get("Authorization"), ShouldContainSubstring, ", Signature=")
		}
	})
}

func TestExpiration(t *testing.T) {
	var credentials = &Credentials{}

	Convey("Credentials without an expiration can't expire", t, func() {
		So(credentials.expired(), ShouldBeFalse)
	})

	Convey("Credentials that expire in 5 minutes aren't expired", t, func() {
		credentials.Expiration = time.Now().Add(5 * time.Minute)
		So(credentials.expired(), ShouldBeFalse)
	})

	Convey("Credentials that expire in 1 minute are expired", t, func() {
		credentials.Expiration = time.Now().Add(1 * time.Minute)
		So(credentials.expired(), ShouldBeTrue)
	})

	Convey("Credentials that expired 2 hours ago are expired", t, func() {
		credentials.Expiration = time.Now().Add(-2 * time.Hour)
		So(credentials.expired(), ShouldBeTrue)
	})
}

func credentialsSet() bool {
	var keys Credentials
	keys = newKeys()
	if keys.AccessKeyID == "" {
		return false
	} else {
		return true
	}
}

func newRequest(method string, url string, v url.Values) *http.Request {
	request, _ := http.NewRequest(method, url, strings.NewReader(v.Encode()))
	return request
}

func sign2AndDo(request *http.Request) *http.Response {
	Sign2(request)
	response, _ := client.Do(request)
	return response
}

func sign3AndDo(request *http.Request) *http.Response {
	Sign3(request)
	response, _ := client.Do(request)
	return response
}

func sign4AndDo(request *http.Request) *http.Response {
	Sign4(request)
	response, _ := client.Do(request)
	return response
}

func signS3AndDo(request *http.Request) *http.Response {
	SignS3(request)
	response, _ := client.Do(request)
	return response
}

func signS3UrlAndDo(request *http.Request) *http.Response {
	SignS3Url(request, time.Now().AddDate(0, 0, 1))
	response, _ := client.Do(request)
	return response
}

var client = &http.Client{}