main.py

import os
os.system('')

from items import iron_sword, wooden_staff
from entities import Character, hero1, hero2,hero3, slime, dragon, wizard

from spells import fireball, icebolt, light_heal

from items.create_db import create_db, add_item, show_items

from save.save_system import save_game, load_game

from maps.map_manager import MapManager

# Clears Screen
os.system('cls') 

def setup_party(party):
    hero3.learnt_spells.append(fireball)
    hero3.learnt_spells.append(icebolt)
    hero3.learnt_spells.append(light_heal)

    hero2.learnt_spells.append(light_heal)
    
    hero1.equip(iron_sword)
    hero2.equip(iron_sword)

    party.add_member(hero1)
    party.add_member(hero2)
    party.add_member(hero3)

    party.add_item_to_inventory('Light Healing Potion', 5)
    party.add_item_to_inventory('Mid Revive', 2)



from entities.party import Party   
party = Party()



from maps.overworld import Map
from utils.getch import getch

print('Press any key to continue...')
getch()


def clear_screen_ansi():
    print("\033[H\033[J", end='')

controls = { 'Q': 'Quit',
            'M': 'Spell Menu', 
            'O': 'Overworld Spells',
            'L': 'Add Light Healing Potion & Mid Revive',
            'I': 'Inventory Menu 🎒'}

debug_controls = { 'G': 'Add Gold'}

def print_controls():
    print('Controls:')
    print('Q - Quit')
    print('M - Spell Menu')
    print('O - Overworld Spells')
    print('I - Inventory Menu 🎒')
    print('F - Use Item 👝')
    print('P - Show Party Details')
    print('9 - Save Game, 0 - Load Game')

    # Debug controls
    print('B - Test Battle', end=' ') #Debug
    print('G - Add Gold', 'K - Kill Hero1 and Hero2') #Debug
    print('E - Gain Exp', end=' ') #Debug
    print('R - Reset Enemy', end= ' ') #Debug
    print('L - Add Light Healing Potion & Mid Revive') #Debug
    print('Y - Save Map') #Debug

def show_gold(party):
    gold_str = str(party.gold)
    border = '-------' + '-' * len(gold_str)
    print(border)
    print(f'|{gold_str} Gold|')
    print(border)

from battle.system import BattleSystem
def run() -> None:
    while True:

        game_map = map_manager.current_map

        
        clear_screen_ansi()
        party.show_party()

        show_gold(party)


        map_manager.current_map.display_map()

        print('Message Board: ')

        print_controls()

        move = getch().upper()

        if move == 'Q':
            break

        if move == '9':
            save_game(map_manager, party)
            print('Game saved!')
            input('Press Enter to continue...')

        if move == '0':
            print('Loading Game is currently not implemented correctly. Do you want to continue? (Y/N)')
            if getch().upper() == 'Y':

                load_game(map_manager, party)

        if move == 'C':
            map_manager.switch_map('Town')

        if move == 'P':
            party.show_all_details()
            input('Press Enter to continue...')
        
        if move == 'K':
            hero1.take_damage(100000)
            hero2.kill()

        if move == 'Y':
            game_map.save_map_to_file('map.txt')
            print('Map saved to map.txt')
            input('Press Enter to continue...')
            
        if move == 'R':
            slime.reset()
            input()

        if move == 'M':
            from ui.spell_menu import test_pick_spells
            test_pick_spells(hero1, slime)
            input('Press Enter to continue...')

        if move == 'O':
            from ui.spell_menu import overworld_spells
            overworld_spells(party)

        if move == 'L':
            party.add_item_to_inventory('Light Healing Potion', 5)
            party.add_item_to_inventory('Mid Revive', 5)
            input()

        if move == 'I': 
            from ui.inventory_menu import show_inventory_menu
            show_inventory_menu(hero1, party)
            input('Press Enter to continue...')

        if move == 'G':
            party.add_gold(50)
            getch() 

        if move == 'F':
            BattleSystem(party, slime).victory()

        if move == 'E':
            hero1.gain_exp(1000)
            hero2.gain_exp(2000)
            input()

        if move == '2':
            BattleSystem(party, wizard).start_battle()
        if move == '3':
            BattleSystem(party, dragon).start_battle()
        

        tile = map_manager.current_map.map_data[
            map_manager.current_map.player_y
        ][
            map_manager.current_map.player_x
        ]
        if tile.name == 'town':
            print('You entered a town!')
            map_manager.switch_map('Town', (7,6))

            input('Press Enter to continue...')
        elif tile.name == "shop" and map_manager.current_map_name == "Town":
            print("🛒 You entered the shop!")
            input()
        elif map_manager.current_map_name == "Town" and map_manager.current_map.player_x == 7 and map_manager.current_map.player_y == 7:
            map_manager.switch_map('overworld', (10,3))


        if move == 'B':
            print('BATTLE')
            enemy = slime

            BattleSystem(party, enemy).start_battle()
        
        
        result = game_map.move_player(move)

        if result['event'] == 'enemy_encounter':
            enemy = result['enemy']
            print(f'You encountered a {enemy.name}!')

            BattleSystem(party, enemy).start_battle()

if __name__ == '__main__' :
    setup_party(party)

    map_manager = MapManager()  


    map_manager.load_map('overworld', 'data/maps/map.txt', spawnable_enemies={slime: 0.6, dragon: 0.03, wizard: 0.1})
    map_manager.load_map('Town', 'data/maps/townmap.txt')


    map_manager.switch_map('overworld')
    
    run()