package cli

import (
	"context"
	"fmt"
	"os"

	"git.eeqj.de/sneak/vaultik/internal/globals"
	"github.com/spf13/cobra"
	"go.uber.org/fx"
)

// PruneOptions contains options for the prune command
type PruneOptions struct {
	Bucket string
	Prefix string
	DryRun bool
}

// NewPruneCommand creates the prune command
func NewPruneCommand() *cobra.Command {
	opts := &PruneOptions{}

	cmd := &cobra.Command{
		Use:   "prune <bucket> <prefix>",
		Short: "Remove unreferenced blobs",
		Long:  `Delete blobs that are no longer referenced by any snapshot`,
		Args:  cobra.ExactArgs(2),
		RunE: func(cmd *cobra.Command, args []string) error {
			opts.Bucket = args[0]
			opts.Prefix = args[1]
			return runPrune(cmd.Context(), opts)
		},
	}

	cmd.Flags().BoolVar(&opts.DryRun, "dry-run", false, "Show what would be deleted without actually deleting")

	return cmd
}

func runPrune(ctx context.Context, opts *PruneOptions) error {
	if os.Getenv("VAULTIK_PRIVATE_KEY") == "" {
		return fmt.Errorf("VAULTIK_PRIVATE_KEY environment variable must be set")
	}

	app := fx.New(
		fx.Supply(opts),
		fx.Provide(globals.New),
		// Additional modules will be added here
		fx.Invoke(func(g *globals.Globals) error {
			// TODO: Implement prune logic
			fmt.Printf("Pruning bucket %s with prefix %s\n", opts.Bucket, opts.Prefix)
			if opts.DryRun {
				fmt.Println("Running in dry-run mode")
			}
			return nil
		}),
		fx.NopLogger,
	)

	if err := app.Start(ctx); err != nil {
		return fmt.Errorf("failed to start prune: %w", err)
	}
	defer func() {
		if err := app.Stop(ctx); err != nil {
			fmt.Printf("error stopping app: %v\n", err)
		}
	}()

	return nil
}