// Copyright 2023 Versity Software
// This file is licensed under the Apache License, Version 2.0
// (the "License"); you may not use this file except in compliance
// with the License.  You may obtain a copy of the License at
//
//   http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing,
// software distributed under the License is distributed on an
// "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
// KIND, either express or implied.  See the License for the
// specific language governing permissions and limitations
// under the License.

package integration

import (
	"context"
	"net/http"

	"github.com/aws/aws-sdk-go-v2/service/s3"
	"github.com/aws/aws-sdk-go-v2/service/s3/types"
	"github.com/versity/versitygw/s3err"
)

func GetBucketCors_non_existing_bucket(s *S3Conf) error {
	testName := "GetBucketCors_non_existing_bucket"
	return actionHandler(s, testName, func(s3client *s3.Client, bucket string) error {
		ctx, cancel := context.WithTimeout(context.Background(), shortTimeout)
		_, err := s3client.GetBucketCors(ctx, &s3.GetBucketCorsInput{
			Bucket: getPtr("non-existing-bucket"),
		})
		cancel()

		return checkApiErr(err, s3err.GetAPIError(s3err.ErrNoSuchBucket))
	})
}

func GetBucketCors_no_such_bucket_cors(s *S3Conf) error {
	testName := "GetBucketCors_no_such_bucket_cors"
	return actionHandler(s, testName, func(s3client *s3.Client, bucket string) error {
		ctx, cancel := context.WithTimeout(context.Background(), shortTimeout)
		_, err := s3client.GetBucketCors(ctx, &s3.GetBucketCorsInput{
			Bucket: &bucket,
		})
		cancel()
		return checkApiErr(err, s3err.GetAPIError(s3err.ErrNoSuchCORSConfiguration))
	})
}

func GetBucketCors_success(s *S3Conf) error {
	testName := "GetBucketCors_success"
	return actionHandler(s, testName, func(s3client *s3.Client, bucket string) error {
		cfg := &types.CORSConfiguration{
			CORSRules: []types.CORSRule{
				{
					AllowedOrigins: []string{"http://origin.com", "helloworld.net"},
					AllowedMethods: []string{http.MethodPost, http.MethodPut, http.MethodHead},
					AllowedHeaders: []string{"X-Amz-Date", "X-Amz-Meta-Something"},
					ExposeHeaders:  []string{"Authorization", "Content-Disposition"},
					MaxAgeSeconds:  getPtr(int32(125)),
				},
				{
					AllowedOrigins: []string{"*"},
					AllowedMethods: []string{http.MethodDelete, http.MethodGet, http.MethodHead},
					AllowedHeaders: []string{"Content-*"},
					ExposeHeaders:  []string{"Authorization", "X-Amz-Date", "X-Amz-Conten-Sha256"},
					ID:             getPtr("my_extra_unique_id"),
					MaxAgeSeconds:  getPtr(int32(-200)),
				},
			},
		}

		err := putBucketCors(s3client, &s3.PutBucketCorsInput{
			Bucket:            &bucket,
			CORSConfiguration: cfg,
		})
		if err != nil {
			return err
		}

		ctx, cancel := context.WithTimeout(context.Background(), shortTimeout)
		res, err := s3client.GetBucketCors(ctx, &s3.GetBucketCorsInput{
			Bucket: &bucket,
		})
		cancel()
		if err != nil {
			return err
		}

		return compareCorsConfig(cfg.CORSRules, res.CORSRules)
	})
}