pss_lookups.py

from typing import Any, Dict, List, Optional, Tuple

import emojis


ALLIANCE_MEMBERSHIP: Dict[str, str] = {
    'Candidate': 'Candidate',
    'Ensign': 'Ensign',
    'Lieutenant': 'Lieutenant',
    'Major': 'Major',
    'Commander': 'Commander',
    'ViceAdmiral': 'Vice Admiral',
    'FleetAdmiral': 'Fleet Admiral'
}


ALLIANCE_MEMBERSHIP_LOOKUP: List[str] = [
    'FleetAdmiral',
    'ViceAdmiral',
    'Commander',
    'Major',
    'Lieutenant',
    'Ensign',
    'Candidate',
    None # Must always be the last element, for Tourney Data
]


COLLECTION_PERK_LOOKUP: Dict[str, str] = {
    'BloodThirstSkill': 'Vampirism',
    'EmpSkill': 'EMP Discharge',
    'FreezeAttackSkill': 'Cryo Field',
    'InstantKillSkill': 'Headshot',
    'MedicalSkill': 'Combat Medic',
    'ResurrectSkill': 'Resurrection',
    'SharpShooterSkill': 'Sharpshooter'
}


CURRENCY_EMOJI_LOOKUP: Dict[str, str] = {
    'android': 'droids',
    'capacity': 'rounds',
    'equipment': 'items',
    'gas': emojis.pss_gas,
    'item101': emojis.pss_gold,
    'item102': emojis.pss_carbon,
    'item103': emojis.pss_iron,
    'item104': emojis.pss_titanium,
    'item105': emojis.pss_silicon,
    'mineral': emojis.pss_min,
    'starbux': emojis.pss_bux,
    'supply': emojis.pss_supply,
}

CURRENCY_EMOJI_LOOKUP_REVERSE: Dict[str, str] = {value: key for key, value in CURRENCY_EMOJI_LOOKUP.items()}


DELETE_ON_CHANGE_ORDER: List[Optional[bool]] = [True, None, False]


DIVISION_CHAR_TO_DESIGN_ID: Dict[str, str] = {
    '-': '0',
    'A': '1',
    'B': '2',
    'C': '3',
    'D': '4'
}

DIVISION_CUTOFF_LOOKUP: Dict[str, Tuple[int, int]] = {
    '-': (-1, -1),
    'A': (1, 8),
    'B': (9, 20),
    'C': (21, 50),
    'D': (51, 100)
}

DIVISION_DESIGN_ID_TO_CHAR: Dict[str, str] = dict([(value, key) for key, value in DIVISION_CHAR_TO_DESIGN_ID.items()])

DIVISION_MAX_COUNT_TARGETS_TOP: Dict[str, int] = {
    '1': 20,
    '2': 14,
    '3': 5,
    '4': 3,
}


DMG_TYPES: List[str] = [
    'SystemDamage',
    'CharacterDamage',
    'ShieldDamage',
    'HullDamage',
    'DirectSystemDamage'
]


EQUIPMENT_MASK_LOOKUP: Dict[int, str] = {
    1: 'head',
    2: 'body',
    4: 'leg',
    8: 'weapon',
    16: 'accessory',
    32: 'pet'
}


EQUIPMENT_SLOTS_ORDER_LOOKUP: List[str] = [
    'EquipmentHead',
    'EquipmentAccessory',
    'EquipmentBody',
    'EquipmentWeapon',
    'EquipmentLeg',
    'EquipmentPet',
    'Module'
]


EQUIPMENT_SLOTS_LOOKUP: Dict[str, str] = {
    'head': 'EquipmentHead',
    'hat': 'EquipmentHead',
    'helm': 'EquipmentHead',
    'helmet': 'EquipmentHead',
    'body': 'EquipmentBody',
    'chest': 'EquipmentBody',
    'shirt': 'EquipmentBody',
    'armor': 'EquipmentBody',
    'leg': 'EquipmentLeg',
    'pant': 'EquipmentLeg',
    'pants': 'EquipmentLeg',
    'weapon': 'EquipmentWeapon',
    'hand': 'EquipmentWeapon',
    'gun': 'EquipmentWeapon',
    'accessory': 'EquipmentAccessory',
    'shoulder': 'EquipmentAccessory',
    'pet': 'EquipmentPet',
    'module': 'Module',
    'mod': 'Module'
}


GAS_COSTS_LEGENDARY_LOOKUP: List[int] = [
    0, 130000, 162500, 195000, 227500,
    260000, 292500, 325000, 357500, 390000,
    422500, 455000, 487500, 520000, 552500,
    585000, 617500, 650000, 682500, 715000,
    747500, 780000, 812500, 845000, 877500,
    910000, 942000, 975000, 1007500, 1040000,
    1072500, 1105000, 1137500, 1170000, 1202500,
    1235000, 1267500, 1300000, 1332500, 1365000]


GAS_COSTS_LOOKUP: List[int] = [
    0, 0, 17, 33, 65,
    130, 325, 650, 1300, 3200,
    6500, 9700, 13000, 19500, 26000,
    35700, 43800, 52000, 61700, 71500,
    84500, 104000, 117000, 130000, 156000,
    175000, 201000, 227000, 253000, 279000,
    312000, 351000, 383000, 422000, 468000,
    507000, 552000, 604000, 650000, 715000]


FLEET_TYPE_LOOKUP: Dict[bool, str] = {
    None: None,
    True: 'Private',
    False: 'Public'
}


GRID_TYPE_MASK_LOOKUP: Dict[int, str] = {
    1: 'A',
    2: 'B'
}


IAP_OPTIONS_MASK_LOOKUP: Dict[int, Tuple[str, int]] = {
    1: ('Clip', 500),
    2: ('Roll', 1200),
    4: ('Stash', 2500),
    8: ('Case', 6500),
    16: ('Vault', 14000)
}


ITEM_SUB_TYPES_TO_GET_PARENTS_FOR: List[str] = [
    'Module'
]


ITEM_SUB_TYPES_LOOKUP: Dict[str, str] = {
    'BlueprintShip': 'Ship Blueprint',
    'BlueprintRoom': 'Room Blueprint',
}


MODULE_TYPE_TO_STAT_LOOKUP: Dict[str, str] = {
    'Turret': 'Attack'
}


PROGRESSION_TYPES: Dict[str, float] = {
    'Linear': 1.0,
    'EaseIn': 2.0,
    'EaseOut': 0.5
}


RARITY_ORDER_LOOKUP: Dict[str, int] = {
    'Common': 70,
    'Elite': 60,
    'Unique': 50,
    'Epic': 40,
    'Hero': 30,
    'Special': 20,
    'Legendary': 10
}

RARITY_EMOJIS_LOOKUP: Dict[str, str] = {
    'Common': emojis.pss_rarity,
    'Elite': emojis.pss_rarity * 2,
    'Unique': emojis.pss_rarity * 3,
    'Epic': emojis.pss_rarity * 4,
    'Hero': emojis.pss_rarity * 5,
    'Special': emojis.pss_rarity_special,
    'Legendary': emojis.pss_rarity_legendary
}

RARITY_INDICTAOR_EMOJIS_LOOKUP: Dict[str, str] = {
    'Common': emojis.rarity_indicator_common,
    'Elite': emojis.rarity_indicator_elite,
    'Unique': emojis.rarity_indicator_unique,
    'Epic': emojis.rarity_indicator_epic,
    'Hero': emojis.rarity_indicator_hero,
    'Special': emojis.rarity_indicator_special,
    'Legendary': emojis.rarity_indicator_legendary
}


PROMO_REQUIREMENT_TYPE_LOOKUP: Dict[str, Dict[str, str]] = {
    'en': {
        'shipLevel': 'ship lvl',
        'purchaseRewardPoints': f'{emojis.pss_dove} purchased'
    }
}


SPECIAL_ABILITIES_LOOKUP: Dict[str, str] = {
    'AddReload': 'Rush Command',
    'DamageToCurrentEnemy': 'Critical Strike',
    'DamageToRoom': 'Ultra Dismantle',
    'DamageToSameRoomCharacters': 'Poison Gas',
    'DeductReload': 'System Hack',
    'FireWalk': 'Fire Walk',
    'Freeze': 'Freeze',
    'HealRoomHp': 'Urgent Repair',
    'HealSameRoomCharacters': 'Healing Rain',
    'HealSelfHp': 'First Aid',
    'Invulnerability': 'Phase Shift',
    'ProtectRoom': 'Stasis Shield',
    'SetFire': 'Arson'
}


EQUIPMENT_SLOTS_EMOJI_LOOKUP: Dict[str, str] = {
    'EquipmentHead': emojis.pss_slot_head,
    'EquipmentAccessory': emojis.pss_slot_accessory,
    'EquipmentBody': emojis.pss_slot_body,
    'EquipmentWeapon': emojis.pss_slot_weapon,
    'EquipmentLeg': emojis.pss_slot_leg,
    'EquipmentPet': emojis.pss_slot_pet,
}


STAT_EMOJI_LOOKUP: Dict[str, str] = {
    'Ability': emojis.pss_stat_ability,
    'Attack': emojis.pss_stat_attack,
    'Engine': emojis.pss_stat_engine,
    'FireResistance': emojis.pss_stat_fireresistance,
    'Hp': emojis.pss_stat_hp,
    'Pilot': emojis.pss_stat_pilot,
    'Repair': emojis.pss_stat_repair,
    'Science': emojis.pss_stat_science,
    'Stamina': emojis.pss_stat_stamina,
    'Weapon': emojis.pss_stat_weapon,
    'Xp': emojis.pss_stat_xp,
}


STAT_TYPES_LOOKUP: Dict[str, str] = {
    'hp': 'HP',
    'health': 'HP',
    'attack': 'Attack',
    'atk': 'Attack',
    'att': 'Attack',
    'damage': 'Attack',
    'dmg': 'Attack',
    'repair': 'Repair',
    'rep': 'Repair',
    'ability': 'Ability',
    'abl': 'Ability',
    'pilot': 'Pilot',
    'plt': 'Pilot',
    'science': 'Science',
    'sci': 'Science',
    'stamina': 'Stamina',
    'stam': 'Stamina',
    'stm': 'Stamina',
    'sta': 'Stamina',
    'engine': 'Engine',
    'eng': 'Engine',
    'weapon': 'Weapon',
    'wpn': 'Weapon',
    'fire': 'FireResistance',
    'fireresistance': 'FireResistance',
    'fireres': 'FireResistance',
    'fire res': 'FireResistance',
    'fire resistance': 'FireResistance',
    'fr': 'FireResistance',
}


STAT_UNITS_ENHANCEMENT_MODIFIER_LOOKUP: Dict[str, str] = {
    'Ability': '%',
}


STAT_UNITS_TRAINING_MODIFIER_LOOKUP: Dict[str, str] = {
    'Ability': '%',
    'Attack': '%',
    'Engine': '%',
    'FireResistance': '',
    'Hp': '%',
    'Pilot': '%',
    'Repair': '%',
    'Science': '%',
    'Stamina': '',
    'Weapon': '%',
    'Xp': ''
}


STATS_LEFT: List[str] = [
    'Hp',
    'Attack',
    'Repair',
    'Ability',
    'Stamina'
]


STATS_RIGHT: List[str] = [
    'Pilot',
    'Science',
    'Engine',
    'Weapon',
    'FireResistance'
]


TRADER_CURRENCY_EMOJI_LOOKUP: Dict[int, str] = { # Use ItemDesignId for lookup
    741: emojis.pss_stardust_common,
    742: emojis.pss_stardust_elite,
    743: emojis.pss_stardust_unique,
    744: emojis.pss_stardust_epic,
    778: emojis.pss_void_particle,
}


TRAINING_RANK_ROOM_LOOKUP: Dict[int, Tuple[str, str]] = {
    1: ('Gym', 'GYM'),
    2: ('Academy', 'ACA')
    # 100: Consumable
}


USER_STATUS: Dict[str, str] = {
    'Attacking': 'Attacking',
    'Defending': 'Defending / Immunity',
    'Offline': 'Offline'
}


USER_TYPE: Dict[str, str] = {
    'Administrator': 'Administrator',
    'Banned': 'Banned',
    'Mission': 'NPC',
    'UserTypeAlliance': 'Starbase',
    'UserTypeCommunityManager': 'Community Manager',
}


XP_COSTS_LOOKUP: List[int] = [
    0, 90, 270, 450, 630,
    810, 1020, 1230, 1440, 1650,
    1860, 2130, 2400, 2670, 2940,
    3210, 3540, 3870, 4200, 4530,
    4860, 5220, 5580, 5940, 6300,
    6660, 7050, 7440, 7830, 8220,
    8610, 9030, 9450, 9870, 10290,
    10710, 11160, 11610, 12060, 12510
]


XP_COSTS_LEGENDARY_LOOKUP: List[int] = [cost * 3 for cost in XP_COSTS_LOOKUP]





# ---------- Helper functions ----------

def get_lookup_value_or_default(lookup: Any, key: Any, default: Any = None) -> Any:
    if key in lookup.keys():
        result = lookup[key]
    else:
        result = default
    return result


def select_next_element(lookup: List[Any], current_element: Any) -> Any:
    if lookup is None:
        return None
    elif lookup:
        index_of_current_element = lookup.index(current_element)
        index_of_next_element = (index_of_current_element + 1) % len(lookup)
        return lookup[index_of_next_element]
    else:
        return current_element