package main

import (
	http "net/http"

	"github.com/aws/aws-sdk-go/aws"
	"github.com/aws/aws-sdk-go/aws/session"
	"github.com/aws/aws-sdk-go/service/s3"
)

// s3Transport implements RoundTripper for the 's3' protocol.
type s3Transport struct {
	svc *s3.S3
}

func newS3Transport() http.RoundTripper {
	s3Conf := aws.NewConfig()

	if len(conf.S3Region) != 0 {
		s3Conf.Region = aws.String(conf.S3Region)
	}

	if len(conf.S3Endpoint) != 0 {
		s3Conf.Endpoint = aws.String(conf.S3Endpoint)
		s3Conf.S3ForcePathStyle = aws.Bool(true)
	}

	sess, err := session.NewSession()
	if err != nil {
		logFatal("Can't create S3 session: %s", err)
	}

	if sess.Config.Region == nil || len(*sess.Config.Region) == 0 {
		sess.Config.Region = aws.String("us-west-1")
	}

	return s3Transport{s3.New(sess, s3Conf)}
}

func (t s3Transport) RoundTrip(req *http.Request) (resp *http.Response, err error) {
	input := &s3.GetObjectInput{
		Bucket: aws.String(req.URL.Host),
		Key:    aws.String(req.URL.Path),
	}

	if len(req.URL.RawQuery) > 0 {
		input.VersionId = aws.String(req.URL.RawQuery)
	}

	s3req, _ := t.svc.GetObjectRequest(input)

	if err := s3req.Send(); err != nil {
		return nil, err
	}

	return s3req.HTTPResponse, nil
}