from typing import Any, Dict

from constants import LOCKFILE_NAME

from ..base import BaseStep
from ..mixins import S3Mixin


__all__ = ["Cleanup"]


class Cleanup(S3Mixin, BaseStep):
    depends_on = {
        "aggregates.artist.videos.youtube.switch",
        "aggregates.track.videos.youtube.switch",
        "aggregates.track.videos.tiktok.switch",
        "aggregates.artist.external_links.switch",
        "aggregates.artist.tracks.spotify.switch",
        "aggregates.artist.albums.spotify.switch",
        "aggregates.artist.tracks.soundcloud.switch",
        "aggregates.track.external_links.switch",
        "aggregates.artist.videos.tiktok.switch",
        "search.artist.switch",
        "search.track.switch",
    }

    def process(self) -> Dict[str, Any]:
        deleted_keys = self.wipe_folder(excluded_keys=(LOCKFILE_NAME,), prefix=False)

        return {"deleted_keys": deleted_keys}