package cli

import (
	"context"
	"fmt"
	"os"

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

// RestoreOptions contains options for the restore command
type RestoreOptions struct {
	Bucket     string
	Prefix     string
	SnapshotID string
	TargetDir  string
}

// NewRestoreCommand creates the restore command
func NewRestoreCommand() *cobra.Command {
	cmd := &cobra.Command{
		Use:   "restore <bucket> <prefix> <snapshot_id> <target_dir>",
		Short: "Restore files from backup",
		Long:  `Download and decrypt files from a backup snapshot`,
		Args:  cobra.ExactArgs(4),
		RunE: func(cmd *cobra.Command, args []string) error {
			opts := &RestoreOptions{
				Bucket:     args[0],
				Prefix:     args[1],
				SnapshotID: args[2],
				TargetDir:  args[3],
			}
			return runRestore(cmd.Context(), opts)
		},
	}

	return cmd
}

func runRestore(ctx context.Context, opts *RestoreOptions) 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 restore logic
			fmt.Printf("Restoring snapshot %s to %s\n", opts.SnapshotID, opts.TargetDir)
			return nil
		}),
		fx.NopLogger,
	)

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

	return nil
}