// Copyright (c) 2025 Nikolai Papin
//
// This file is part of Easywish
//
// This program is free software: you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.
//
// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See
// the GNU General Public License for more details.
//
// You should have received a copy of the GNU General Public License
// along with this program.  If not, see <https://www.gnu.org/licenses/>.

package minioclient

import (
	"context"
	"slices"

	"github.com/minio/minio-go/v7"
	"github.com/minio/minio-go/v7/pkg/lifecycle"
)

var Buckets map[string]string

func setupBuckets(client *minio.Client) {

	Buckets = map[string]string{
		"avatars": "avatars",
		"images": "images",
		"uploads": "uploads",
	}

	ctx := context.Background()
	var newBuckets []string
	for key, value := range Buckets {
		bucketExists, err := client.BucketExists(ctx, value); if err != nil {
			panic("Failure to check if bucket '" + value + "' exists: " + err.Error())
		} 

		if !bucketExists {
			err := client.MakeBucket(ctx, value, minio.MakeBucketOptions{}); if err != nil {
				panic("Failure to create bucket '" + value + "': " + err.Error())
			}
			newBuckets = append(newBuckets, key)
		}
	}

	if slices.Contains(newBuckets, "uploads") { 
		uploadsCfg := lifecycle.NewConfiguration()
		uploadsCfg.Rules = []lifecycle.Rule{
			{
				ID:		"expire-uploads",
				Status: "enabled",
				Expiration: lifecycle.Expiration{Days: 1},
			},
		}
		
		err := client.SetBucketLifecycle(ctx, Buckets["uploads"], uploadsCfg); if err != nil {
			errRm := client.RemoveBucket(ctx, Buckets["uploads"])

			if errRm != nil {
				panic("Failed to set lifecycle configuration for the uploads bucket: '" + err.Error() + "' and then failed to delete it: " + errRm.Error())
			}

			panic("Failed to set lifecycle configuration for the uploads bucket: " + err.Error())
		}
	}
}