from pathlib import Path

from app.actions import BaseAction
from app.schemas import ActionParams, ActionContext
from app.chain.media import MediaChain
from app.chain.storage import StorageChain
from app.core.metainfo import MetaInfoPath
from app.log import logger


class ScrapeFileParams(ActionParams):
    """
    刮削文件参数
    """
    pass


class ScrapeFileAction(BaseAction):
    """
    刮削文件
    """

    _scraped_files = []
    _has_error = False

    def __init__(self):
        super().__init__()
        self.storagechain = StorageChain()
        self.mediachain = MediaChain()

    @classmethod
    @property
    def name(cls) -> str:
        return "刮削文件"

    @classmethod
    @property
    def description(cls) -> str:
        return "刮削媒体信息和图片"

    @classmethod
    @property
    def data(cls) -> dict:
        return ScrapeFileParams().dict()

    @property
    def success(self) -> bool:
        return not self._has_error

    def execute(self, params: dict, context: ActionContext) -> ActionContext:
        """
        刮削fileitems中的所有文件
        """
        for fileitem in context.fileitems:
            if fileitem in self._scraped_files:
                continue
            if not self.storagechain.exists(fileitem):
                continue
            meta = MetaInfoPath(Path(fileitem.path))
            mediainfo = self.mediachain.recognize_media(meta)
            if not mediainfo:
                self._has_error = True
                logger.info(f"{fileitem.path} 未识别到媒体信息，无法刮削")
                continue
            self.mediachain.scrape_metadata(fileitem=fileitem, meta=meta, mediainfo=mediainfo)
            self._scraped_files.append(fileitem)

        self.job_done()
        return context