package s3browser

import (
	"context"
	"fmt"
	"path/filepath"

	"github.com/graph-gophers/dataloader"
	"github.com/minio/minio-go/v7"
)

func deleteMutation(ctx context.Context, id string) error {
	s3Client, ok := ctx.Value("s3Client").(*minio.Client)

	if !ok {
		return fmt.Errorf("Failed to get s3Client from context")
	}

	// TODO: it is posible to remove multiple objects with a single call.
	// Is it better to batch this?
	err := s3Client.RemoveObject(ctx, bucketName, id, minio.RemoveObjectOptions{})

	if err != nil {
		return err
	}

	// Invalidate cache
	loader, ok := ctx.Value("loader").(map[string]*dataloader.Loader)
	if !ok {
		return fmt.Errorf("Failed to get loader from context")
	}

	loader["getFile"].Clear(ctx, dataloader.StringKey(id))
	loader["listObjects"].Clear(ctx, dataloader.StringKey(id))
	loader["getFiles"].Clear(ctx, dataloader.StringKey(filepath.Dir(id)))

	return nil
}

func copyMutation(ctx context.Context, src, dest string) (*File, error) {
	s3Client, ok := ctx.Value("s3Client").(*minio.Client)

	if !ok {
		return nil, fmt.Errorf("Failed to get s3Client from context")
	}

	info, err := s3Client.CopyObject(ctx, minio.CopyDestOptions{
		Bucket: bucketName,
		Object: dest,
	}, minio.CopySrcOptions{
		Bucket: bucketName,
		Object: src,
	})

	if err != nil {
		return nil, err
	}

	// Invalidate cache

	loader, ok := ctx.Value("loader").(map[string]*dataloader.Loader)

	// TODO: Do we want to error when the operation
	// has succeeded but the cache invalidation has failed ?
	if ok {
		loader["getFile"].Clear(ctx, dataloader.StringKey(info.Key))
		loader["listObjects"].Clear(ctx, dataloader.StringKey(info.Key))
		loader["getFiles"].Clear(ctx, dataloader.StringKey(filepath.Dir(info.Key)))
	}

	return &File{
		ID: info.Key,
	}, nil

}

func moveMutation(ctx context.Context, src, dest string) (*File, error) {
	s3Client, ok := ctx.Value("s3Client").(*minio.Client)

	if !ok {
		return nil, fmt.Errorf("Failed to get s3Client from context")
	}

	// There is no (spoon) move. Only copy and delete
	info, err := s3Client.CopyObject(ctx, minio.CopyDestOptions{
		Bucket: bucketName,
		Object: dest,
	}, minio.CopySrcOptions{
		Bucket: bucketName,
		Object: src,
	})

	if err != nil {
		return nil, err
	}

	err = deleteMutation(ctx, src)

	if err != nil {
		return nil, err
	}

	return &File{
		ID: info.Key,
	}, nil

}