package layer

import (
	"crypto/md5"
	"encoding/xml"
	"testing"

	"git.frostfs.info/TrueCloudLab/frostfs-s3-gw/api/data"
	apierr "git.frostfs.info/TrueCloudLab/frostfs-s3-gw/api/errors"
	frosterr "git.frostfs.info/TrueCloudLab/frostfs-s3-gw/internal/frostfs/errors"
	"git.frostfs.info/TrueCloudLab/frostfs-sdk-go/object"
	oid "git.frostfs.info/TrueCloudLab/frostfs-sdk-go/object/id"
	"github.com/stretchr/testify/require"
)

func TestBucketLifecycle(t *testing.T) {
	tc := prepareContext(t)

	lifecycle := &data.LifecycleConfiguration{
		XMLName: xml.Name{
			Space: `http://s3.amazonaws.com/doc/2006-03-01/`,
			Local: "LifecycleConfiguration",
		},
		Rules: []data.LifecycleRule{
			{
				Status: data.LifecycleStatusEnabled,
				Expiration: &data.LifecycleExpiration{
					Days: ptr(21),
				},
			},
		},
	}
	lifecycleBytes, err := xml.Marshal(lifecycle)
	require.NoError(t, err)
	hash := md5.New()
	hash.Write(lifecycleBytes)

	_, err = tc.layer.GetBucketLifecycleConfiguration(tc.ctx, tc.bktInfo)
	require.Equal(t, apierr.GetAPIError(apierr.ErrNoSuchLifecycleConfiguration), frosterr.UnwrapErr(err))

	err = tc.layer.DeleteBucketLifecycleConfiguration(tc.ctx, tc.bktInfo)
	require.NoError(t, err)

	err = tc.layer.PutBucketLifecycleConfiguration(tc.ctx, &PutBucketLifecycleParams{
		BktInfo:      tc.bktInfo,
		LifecycleCfg: lifecycle,
	})
	require.NoError(t, err)

	cfg, err := tc.layer.GetBucketLifecycleConfiguration(tc.ctx, tc.bktInfo)
	require.NoError(t, err)
	require.Equal(t, *lifecycle, *cfg)

	err = tc.layer.DeleteBucketLifecycleConfiguration(tc.ctx, tc.bktInfo)
	require.NoError(t, err)

	_, err = tc.layer.GetBucketLifecycleConfiguration(tc.ctx, tc.bktInfo)
	require.Equal(t, apierr.GetAPIError(apierr.ErrNoSuchLifecycleConfiguration), frosterr.UnwrapErr(err))
}

func TestLifecycleCopiesNumber(t *testing.T) {
	tc := prepareCORSContext(t)

	lifecycle := &data.LifecycleConfiguration{
		XMLName: xml.Name{
			Space: `http://s3.amazonaws.com/doc/2006-03-01/`,
			Local: "LifecycleConfiguration",
		},
		Rules: []data.LifecycleRule{
			{
				Status: data.LifecycleStatusEnabled,
				Expiration: &data.LifecycleExpiration{
					Days: ptr(21),
				},
			},
		},
	}
	copies := []uint32{2, 0}

	err := tc.layer.PutBucketLifecycleConfiguration(tc.ctx, &PutBucketLifecycleParams{
		BktInfo:       tc.bktInfo,
		LifecycleCfg:  lifecycle,
		CopiesNumbers: copies,
	})
	require.NoError(t, err)

	cfg, err := tc.layer.GetBucketLifecycleConfiguration(tc.ctx, tc.bktInfo)
	require.NoError(t, err)
	require.Equal(t, *lifecycle, *cfg)

	objs := tc.testFrostFS.Objects()
	require.Len(t, objs, 1)

	require.EqualValues(t, copies, tc.testFrostFS.CopiesNumbers(addrFromObject(objs[0]).EncodeToString()))
}

func ptr[T any](t T) *T {
	return &t
}
func addrFromObject(obj *object.Object) oid.Address {
	var addr oid.Address
	cnrID, _ := obj.ContainerID()
	objID, _ := obj.ID()

	addr.SetContainer(cnrID)
	addr.SetObject(objID)

	return addr
}