package s3

import (
	"bytes"
	"context"
	"fmt"
	"os"
	"path"
	"path/filepath"
	"testing"
	"time"

	pkg_aws "github.com/aws/aws-k8s-tester/pkg/aws"
	"github.com/aws/aws-k8s-tester/pkg/randutil"
	"github.com/aws/aws-sdk-go/aws"
	"github.com/aws/aws-sdk-go/service/s3"
	"go.uber.org/zap"
)

func TestS3(t *testing.T) {
	lg := zap.NewExample()
	ss, _, _, err := pkg_aws.New(&pkg_aws.Config{
		Logger:    lg,
		Partition: "aws",
		Region:    "us-west-2",
	})
	if err != nil {
		t.Skip(err)
	}
	s3API := s3.New(ss)

	bucket := randutil.String(15)
	dir := filepath.Join("hello", "world")
	if err = CreateBucket(lg, s3API, bucket, "us-west-2", "", 0); err != nil {
		t.Fatal(err)
	}
	defer func() {
		t.Logf("EmptyBucket: %v", EmptyBucket(lg, s3API, bucket))
		t.Logf("DeleteBucket: %v", DeleteBucket(lg, s3API, bucket))
	}()

	testKey := randutil.String(30)
	ch := PollUntilExist(
		context.Background(),
		make(chan struct{}),
		lg,
		s3API,
		bucket,
		testKey,
		10*time.Second,
		5*time.Second,
	)

	s3Key := ""
	for i := 0; i < 10; i++ {
		s3Key = filepath.Join(dir, randutil.String(10))
		if err = UploadBody(
			lg,
			s3API,
			bucket,
			filepath.Join(filepath.Clean(dir)+"-raw", randutil.String(10)),
			bytes.NewReader(randutil.Bytes(10)),
		); err != nil {
			t.Fatal(err)
		}
		time.Sleep(100 * time.Millisecond)
		if err = UploadBody(
			lg,
			s3API,
			bucket,
			s3Key,
			bytes.NewReader(randutil.Bytes(10)),
		); err != nil {
			t.Fatal(err)
		}
		time.Sleep(100 * time.Millisecond)
	}
	if ok, err := Exist(lg, s3API, bucket, s3Key); !ok || err != nil {
		t.Fatalf("unexpected ok %v, err %v", ok, err)
	}
	localPath, err := DownloadToTempFile(lg, s3API, bucket, s3Key)
	if err != nil {
		t.Fatal(err)
	}
	defer os.RemoveAll(localPath)
	fmt.Println("localPath:", localPath)

	targetDir, err := DownloadDir(lg, s3API, bucket, dir)
	fmt.Println("targetDir:", targetDir)
	defer os.RemoveAll(targetDir)
	if err != nil {
		t.Fatal(err)
	}

	s3Objects, err := ListInDescendingLastModified(lg, s3API, bucket, path.Clean(dir)+"/")
	if err != nil {
		t.Fatal(err)
	}
	for _, obj := range s3Objects {
		fmt.Println(aws.StringValue(obj.Key), aws.TimeValue(obj.LastModified), aws.Int64Value(obj.Size))
	}

	if err = UploadBody(
		lg,
		s3API,
		bucket,
		testKey,
		bytes.NewReader(randutil.Bytes(10)),
	); err != nil {
		t.Fatal(err)
	}
	for cur := range ch {
		if cur.HeadObject == nil {
			fmt.Println(nil, cur.Error)
		} else {
			fmt.Println(aws.Int64Value(cur.HeadObject.ContentLength), cur.Error)
		}
	}
}