table.py

from parse import get_table_info

from db.option import Option
from models.options.OptionSet import OptionSet
from models.options.MysteryOptionSet import MysteryOptionSet

class Table:
    instance = None
    default_db = {}
    db = {}
    info = {}

    def __init__(self):
        if Table.instance is None:
            Table.instance = self
        else:
            self = Table.instance

    def __getitem__(self, key):
        return self.db[key]

    def generate_pairs(self, **kwargs):
        table_data = []

        # Options
        optionset = kwargs.get("options")

        option_dbtuples = optionset.get_dbtuples()

        for option_data in option_dbtuples:
            if isinstance(option_data, dict) and "key" in option_data:
                option_data_value = option_data.get("value")
                if isinstance(option_data_value, int) and option_data_value < 0:
                    option_data_value = 0x100000000 + option_data_value
                table_data.append({
                    "key": option_data.get("key"),
                    "value": option_data_value,
                })

        # temp fix for multiworld
        table_data.append({
            "key": 0xAF050000,
            "value": 0x00000000
        })

        # Quizzes
        quizzes = kwargs.get("quiz_data")

        for key, value in quizzes:
            table_data.append({
                "key": key,
                "value": value
            })

        # Items
        placed_items = kwargs.get("items")
        for node in placed_items:
            if node.key_name_item is not None and node.current_item is not None:
                assert not node.current_item.unplaceable # sanity check

                table_data.append({
                    "key": node.get_item_key(),
                    "value": node.current_item.value,
                })

            # Item Prices
            if (    node.key_name_price is not None
                and (   node.key_name_price.startswith("ShopPrice")
                     or node.key_name_price.startswith("RewardAmount")
                )
            ):
                table_data.append({
                    "key": node.get_price_key(),
                    "value": node.current_item.base_price
                })

        # Blocks
        placed_blocks = kwargs.get("blocks")
        for key, value in placed_blocks:
            table_data.append({
                "key": key,
                "value": value
            })

        # Entrances
        entrances = kwargs.get("entrances")
        for key, value in entrances:
            table_data.append({
                "key": key,
                "value": value
            })

        # Battle IDs
        battles = kwargs.get("battle_list")
        for key, value in battles:
            table_data.append({
                "key": key,
                "value": value
            })

        # Actor Attributes
        actor_attributes = kwargs.get("actor_data")
        for key, value in actor_attributes:
            table_data.append({
                "key": key,
                "value": value
            })

        palettes = kwargs.get("palettes")
        for key, value in palettes:
            table_data.append({
                "key": key,
                "value": value
            })

        # Move Costs
        move_costs = kwargs.get("move_costs")
        for key, value in move_costs:
            table_data.append({
                "key": key,
                "value": value
            })

        # Audio
        music_list = kwargs.get("music_list")
        for key, value in music_list:
            table_data.append({
                "key": key,
                "value": value
            })

        # Map mirroring
        mapmirror_list = kwargs.get("mapmirror_list")
        for key, value in mapmirror_list:
            table_data.append({
                "key": key,
                "value": value
            })

        # Puzzles & Minigames
        puzzle_list = kwargs.get("puzzle_list")
        for key, value in puzzle_list:
            table_data.append({
                "key": key,
                "value": value
            })

        table_data.sort(key=lambda pair: pair["key"])
        return table_data

    def generate_palettes_pairs(self, **kwargs):
        table_data = []

        palettes = kwargs.get("palettes")
        for key, value in palettes:
            table_data.append({
                "key": key,
                "value": value
            })

        table_data.sort(key=lambda pair: pair["key"])
        return table_data

    def generate_cosmetics_pairs(self, **kwargs):
        table_data = []

        cosmetic_options = kwargs.get("cosmetics")
        for option in cosmetic_options:
            table_data.append({
                "key": Option.get(Option.name == option).get_key(),
                "value": cosmetic_options[option]
            })

        table_data.sort(key=lambda pair: pair["key"])
        return table_data
    
    def generate_audio_option_pairs(self, **kwargs):
        table_data = []

        cosmetic_options = kwargs.get("audio_options")
        for option in cosmetic_options:
            table_data.append({
                "key": Option.get(Option.name == option).get_key(),
                "value": cosmetic_options[option]
            })

        table_data.sort(key=lambda pair: pair["key"])
        return table_data
    
    def generate_music_pairs(self, **kwargs):
        table_data = []
        
        music_list = kwargs.get("music_list")
        for key, value in music_list:
            table_data.append({
                "key": key,
                "value": value
            })

        table_data.sort(key=lambda pair: pair["key"])
        return table_data


    def create(self):
        self.info = get_table_info()