from pathlib import Path
from time import time

import yaml

from src import shared
from src.game import Game
from src.importer.sources.source import (
    LinuxSource,
    Source,
    SourceIterationResult,
    SourceIterator,
)
from src.utils.decorators import replaced_by_env_path, replaced_by_path


class BottlesSourceIterator(SourceIterator):
    source: "BottlesSource"

    def generator_builder(self) -> SourceIterationResult:
        """Generator method producing games"""

        data = (self.source.location / "library.yml").read_text("utf-8")
        library: dict = yaml.safe_load(data)

        for entry in library.values():
            # Build game
            values = {
                "version": shared.SPEC_VERSION,
                "source": self.source.id,
                "added": int(time()),
                "name": entry["name"],
                "game_id": self.source.game_id_format.format(game_id=entry["id"]),
                "executable": self.source.executable_format.format(
                    bottle_name=entry["bottle"]["name"], game_name=entry["name"]
                ),
            }
            game = Game(values, allow_side_effects=False)

            # Get official cover path
            bottle_path = entry["bottle"]["path"]
            image_name = entry["thumbnail"].split(":")[1]
            image_path = (
                self.source.location / "bottles" / bottle_path / "grids" / image_name
            )
            additional_data = {"local_image_path": image_path}

            # Produce game
            yield (game, additional_data)


class BottlesSource(Source):
    """Generic Bottles source"""

    name = "Bottles"
    location_key = "bottles-location"

    def __iter__(self) -> SourceIterator:
        return BottlesSourceIterator(self)


class BottlesLinuxSource(BottlesSource, LinuxSource):
    variant = "linux"
    executable_format = 'xdg-open bottles:run/"{bottle_name}"/"{game_name}"'

    @property
    @BottlesSource.replaced_by_schema_key()
    @replaced_by_path("~/.var/app/com.usebottles.bottles/data/bottles/")
    @replaced_by_env_path("XDG_DATA_HOME", "bottles/")
    @replaced_by_path("~/.local/share/bottles/")
    def location(self) -> Path:
        raise FileNotFoundError()