from direct.showbase.PythonUtil import invertDict
from toontown.toonbase import ToontownGlobals
from toontown.coghq import BossbotCountryClubFairwayRoom_Battle00_Cogs
from toontown.coghq import BossbotCountryClubMazeRoom_Battle00_Cogs
from toontown.coghq import BossbotCountryClubMazeRoom_Battle01_Cogs
from toontown.coghq import BossbotCountryClubMazeRoom_Battle02_Cogs
from toontown.coghq import BossbotCountryClubMazeRoom_Battle03_Cogs
from toontown.coghq import NullCogs
from toontown.coghq import BossbotCountryClubKartRoom_Battle00_Cogs
from toontown.coghq import BossbotCountryClubPresidentRoom_Battle00_Cogs

# Explicit imports...
from toontown.coghq import BossbotCountryClubEntrance_Action00
from toontown.coghq import BossbotCountryClubTeeOffRoom_Action00
from toontown.coghq import BossbotCountryClubFairwayRoom_Battle00
from toontown.coghq import BossbotCountryClubMazeRoom_Battle00
from toontown.coghq import BossbotCountryClubMazeRoom_Battle01
from toontown.coghq import BossbotCountryClubMazeRoom_Battle02
from toontown.coghq import BossbotCountryClubGreenRoom_Action00
from toontown.coghq import BossbotCountryClubKartRoom_Battle00
from toontown.coghq import BossbotCountryClubPresidentRoom_Battle00
from toontown.coghq import BossbotCountryClubTeeOffRoom_Action01
from toontown.coghq import BossbotCountryClubTeeOffRoom_Action02
from toontown.coghq import BossbotCountryClubGreenRoom_Action01
from toontown.coghq import BossbotCountryClubGreenRoom_Action02

def getCountryClubRoomSpecModule(roomId):
    return CashbotMintSpecModules[roomId]


def getCogSpecModule(roomId):
    roomName = BossbotCountryClubRoomId2RoomName[roomId]
    return CogSpecModules.get(roomName, NullCogs)


def getNumBattles(roomId):
    return roomId2numBattles[roomId]


BossbotCountryClubRoomId2RoomName = {0: 'BossbotCountryClubEntrance_Action00',
 2: 'BossbotCountryClubTeeOffRoom_Action00',
 4: 'BossbotCountryClubFairwayRoom_Battle00',
 5: 'BossbotCountryClubMazeRoom_Battle00',
 6: 'BossbotCountryClubMazeRoom_Battle01',
 7: 'BossbotCountryClubMazeRoom_Battle02',
 9: 'BossbotCountryClubGreenRoom_Action00',
 17: 'BossbotCountryClubKartRoom_Battle00',
 18: 'BossbotCountryClubPresidentRoom_Battle00',
 22: 'BossbotCountryClubTeeOffRoom_Action01',
 32: 'BossbotCountryClubTeeOffRoom_Action02',
 29: 'BossbotCountryClubGreenRoom_Action01',
 39: 'BossbotCountryClubGreenRoom_Action02'}
BossbotCountryClubRoomName2RoomId = invertDict(BossbotCountryClubRoomId2RoomName)
BossbotCountryClubEntranceIDs = (0,)
BossbotCountryClubMiddleRoomIDs = (2, 5, 6)
BossbotCountryClubFinalRoomIDs = (18,)
BossbotCountryClubConnectorRooms = ('phase_12/models/bossbotHQ/Connector_Tunnel_A', 'phase_12/models/bossbotHQ/Connector_Tunnel_B')
CashbotMintSpecModules = {}
for roomName, roomId in BossbotCountryClubRoomName2RoomId.items():
    CashbotMintSpecModules[roomId] = locals()[roomName]

CogSpecModules = {'BossbotCountryClubFairwayRoom_Battle00': BossbotCountryClubFairwayRoom_Battle00_Cogs,
 'BossbotCountryClubMazeRoom_Battle00': BossbotCountryClubMazeRoom_Battle00_Cogs,
 'BossbotCountryClubMazeRoom_Battle01': BossbotCountryClubMazeRoom_Battle01_Cogs,
 'BossbotCountryClubMazeRoom_Battle02': BossbotCountryClubMazeRoom_Battle02_Cogs,
 'BossbotCountryClubKartRoom_Battle00': BossbotCountryClubKartRoom_Battle00_Cogs,
 'BossbotCountryClubPresidentRoom_Battle00': BossbotCountryClubPresidentRoom_Battle00_Cogs}
roomId2numBattles = {}
for roomName, roomId in BossbotCountryClubRoomName2RoomId.items():
    if roomName not in CogSpecModules:
        roomId2numBattles[roomId] = 0
    else:
        cogSpecModule = CogSpecModules[roomName]
        roomId2numBattles[roomId] = len(cogSpecModule.BattleCells)

name2id = BossbotCountryClubRoomName2RoomId
roomId2numBattles[name2id['BossbotCountryClubTeeOffRoom_Action00']] = 1
del name2id
middleRoomId2numBattles = {}
for roomId in BossbotCountryClubMiddleRoomIDs:
    middleRoomId2numBattles[roomId] = roomId2numBattles[roomId]