package cli

import (
	"context"

	"github.com/pkg/errors"

	"github.com/kopia/kopia/repo"
	"github.com/kopia/kopia/repo/maintenance"
	"github.com/kopia/kopia/snapshot/snapshotmaintenance"
)

type commandMaintenanceRun struct {
	maintenanceRunFull  bool
	maintenanceRunForce bool
	safety              maintenance.SafetyParameters
}

func (c *commandMaintenanceRun) setup(svc appServices, parent commandParent) {
	cmd := parent.Command("run", "Run repository maintenance")
	cmd.Flag("full", "Full maintenance").BoolVar(&c.maintenanceRunFull)
	cmd.Flag("force", "Run maintenance even if not owned (unsafe)").Hidden().BoolVar(&c.maintenanceRunForce)
	safetyFlagVar(cmd, &c.safety)

	cmd.Action(svc.directRepositoryWriteAction(c.run))
}

func (c *commandMaintenanceRun) run(ctx context.Context, rep repo.DirectRepositoryWriter) error {
	mode := maintenance.ModeQuick

	_, supportsEpochManager, err := rep.ContentManager().EpochManager(ctx)
	if err != nil {
		return errors.Wrap(err, "EpochManager")
	}

	if c.maintenanceRunFull || supportsEpochManager {
		mode = maintenance.ModeFull
	}

	//nolint:wrapcheck
	return snapshotmaintenance.Run(ctx, rep, mode, c.maintenanceRunForce, c.safety)
}