package worker

import (
	"fmt"

	"code.nonshy.com/nonshy/website/pkg/log"
	"code.nonshy.com/nonshy/website/pkg/models"
	"code.nonshy.com/nonshy/website/pkg/photo"
)

// Vacuum runs database cleanup tasks for data consistency. Run it like `nonshy vacuum` from the CLI.
func Vacuum(dryrun bool) error {

	var steps = []struct {
		Label string
		Fn    func(bool) (int64, error)
	}{
		{"Comment Photos", VacuumOrphanedCommentPhotos},
		{"Photos", VacuumOrphanedPhotos},
		{"Polls", VacuumOrphanedPolls},
	}

	for _, step := range steps {
		log.Warn("Vacuum: %s", step.Label)
		if total, err := step.Fn(dryrun); err != nil {
			log.Error("%s: %s", step.Label, err)
		} else {
			log.Info("Removed %d rows", total)
		}
	}

	return nil
}

// VacuumOrphanedPolls removes any polls with forum threads no longer pointing to them.
func VacuumOrphanedPolls(dryrun bool) (int64, error) {
	polls, count, err := models.GetOrphanedPolls()
	if err != nil {
		return count, err
	}

	if dryrun {
		return count, nil
	}

	for _, row := range polls {
		log.Info("    #%d: %s", row.ID, row.Choices)
		if err := row.Delete(); err != nil {
			return count, fmt.Errorf("deleting orphaned poll (%d): %s", row.ID, err)
		}
	}

	return count, nil
}

// VacuumOrphanedPhotos removes any lingering photo from failed account deletion.
func VacuumOrphanedPhotos(dryrun bool) (int64, error) {
	photos, count, err := models.GetOrphanedPhotos()
	if err != nil {
		return count, err
	}

	if dryrun {
		return count, nil
	}

	for _, row := range photos {
		log.Info("    #%d: %s", row.ID, row.Filename)
		if err := photo.Delete(row.Filename); err != nil {
			return count, fmt.Errorf("photo ID %d: removing file %s: %s", row.ID, row.Filename, err)
		}

		if row.CroppedFilename != "" {
			if err := photo.Delete(row.CroppedFilename); err != nil {
				return count, fmt.Errorf("photo ID %d: removing file %s: %s", row.ID, row.Filename, err)
			}
		}

		if err := row.Delete(); err != nil {
			return count, fmt.Errorf("deleting orphaned photo (%d): %s", row.ID, err)
		}
	}

	return count, nil
}

// VacuumOrphanedCommentPhotos cleans up comment photos that weren't associated to a post, returning the count removed.
func VacuumOrphanedCommentPhotos(dryrun bool) (int64, error) {
	// Do the needful.
	photos, total, err := models.GetOrphanedCommentPhotos()
	if err != nil {
		return total, err
	}

	if dryrun {
		return total, nil
	}

	for _, row := range photos {
		if err := photo.Delete(row.Filename); err != nil {
			return total, fmt.Errorf("photo ID %d: removing file %s: %s", row.ID, row.Filename, err)
		}

		if err := row.Delete(); err != nil {
			return total, fmt.Errorf("deleting orphaned comment photo (%d): %s", row.ID, err)
		}
	}

	return total, nil
}