fftcgtool/fftcg/ttsdeck.py

from __future__ import annotations

import json
import os

import requests

from .carddb import CardDB
from .cards import Cards
from .code import Code
from .language import Language
from .utils import CARD_BACK_URL, DECKS_DIR_NAME


class TTSDeck(Cards):
    def __init__(self, codes: list[Code], name: str, description: str, face_down: bool):
        super().__init__(name)
        self.__description = description
        self.__face_down = face_down

        # get cards from carddb
        carddb = CardDB()
        self.extend([
            carddb[code]
            for code in codes
        ])

    __FFDECKS_API_URL = "https://ffdecks.com/api/deck"

    @classmethod
    def from_ffdecks_deck(cls, deck_id: str) -> TTSDeck:
        # api request
        req = requests.get(TTSDeck.__FFDECKS_API_URL, params={"deck_id": deck_id})

        # pre-extract the used data
        deck_cards = [{
            "code": card["card"]["serial_number"],
            "type": card["card"]["type"],
            "cost": int(card["card"]["cost"]),
            "count": int(card["quantity"]),
        } for card in req.json()["cards"]]

        # sort cards by type, then by cost
        def by_type(data: dict[str, str | int]) -> int:
            key_prios = {
                "Forward": 1,
                "Summon": 2,
                "Monster": 3,
                "Backup": 5,
            }

            if data["type"] in key_prios:
                return key_prios[data["type"]]
            else:
                return 4

        def by_cost(data: dict[str, str | int]) -> int:
            return data["cost"]

        deck_cards.sort(key=by_cost)
        deck_cards.sort(key=by_type)

        # ffdecks quirk: some full-art promos in database
        replace_full_arts = {
            # line format:
            # full-art-id: normal id,
            "PR-051": "11-083",
            "PR-055": "11-062",
        }

        # replace with normal-art cards
        for card in deck_cards:
            if card["code"] in replace_full_arts:
                card["code"] = replace_full_arts[card["code"]]

        codes = [
            # create list of code objects
            Code(card["code"])
            # for each card
            for card in deck_cards
            # repeat to meet count
            for _ in range(card["count"])
        ]

        # general metadata
        name = f"{req.json()['name']}"
        description = deck_id

        # create deck object
        return cls(codes, name, description, True)

    def tts_object(self, language: Language) -> dict[str, any]:
        carddb = CardDB()

        # unique face urls used
        unique_faces = set([
            card[language].face
            for card in self
        ])

        # lookup for indices of urls
        face_indices = {
            url: i + 1
            for i, url in enumerate(unique_faces)
        }

        # build the "CustomDeck" dictionary
        custom_deck = {
            str(i): {
                "NumWidth": "10",
                "NumHeight": "7",
                "FaceURL": carddb.get_face_url(face),
                "BackURL": CARD_BACK_URL,
            } for face, i in face_indices.items()
        }

        # values both in main deck and each contained card
        common_dict = {
            "Transform": {
                "scaleX": 2.17822933,
                "scaleY": 1.0,
                "scaleZ": 2.17822933,
                "rotY": 180.0,
            },
            "Locked": False,
            "Grid": True,
            "Snap": True,
            "Autoraise": True,
            "Sticky": True,
            "Tooltip": True,
            "GridProjection": False,
        }

        # cards contained in deck
        contained_objects = [
            {
                "Nickname": card[language].name,
                "Description": card[language].text,
                "CardID": 100 * face_indices[card[language].face] + card.index,

                "Name": "Card",
                "Hands": True,
                "SidewaysCard": False,
            } | common_dict for card in self
        ]

        # extract the card ids
        deck_ids = [
            contained_object["CardID"]
            for contained_object in contained_objects
        ]

        # create the deck dictionary
        deck_dict = {"ObjectStates": [
            {
                "Nickname": self.name,
                "Description": self.__description,
                "DeckIDs": deck_ids,
                "CustomDeck": custom_deck,
                "ContainedObjects": contained_objects,

                "Name": "Deck",
                "Hands": False,
                "SidewaysCard": False,
            } | common_dict
        ]}

        if self.__face_down:
            # flip the deck
            deck_dict["ObjectStates"][0]["Transform"]["rotZ"] = 180.0

        return deck_dict

    def save(self, language: Language) -> None:
        # only save if the deck contains cards
        if self:
            if not os.path.exists(DECKS_DIR_NAME):
                os.mkdir(DECKS_DIR_NAME)

            with open(os.path.join(DECKS_DIR_NAME, f"{self.file_name}.json"), "w") as file:
                json.dump(self.tts_object(language), file, indent=2)