#!/usr/bin/env python

"""
MudServer author: Mark Frimston - mfrimston@gmail.com

Micropython port and expansion author: Barry Ruffner - barryruffner@gmail.com
"""

from time import sleep
from math import floor
from sys import platform
from os import listdir, remove

from mudserver import MudServer
from commandhandler import CommandHandler
from utils import load_object_from_file, save_object_to_file, password_hash, calc_att, get_att

if 'esp' in platform:
    from gc import collect, mem_free
    from machine import Pin
# import the MUD server class


print('STARTING MUD\r\n\r\n\r\n')

# Setup button so when pressed the mud goes in to wifi hotspot mode on 192.168.4.1
if 'esp' in platform:
    flash_button = Pin(0, Pin.IN, Pin.PULL_UP)

# stores the players in the game
players = {}

# start the server
globals()['mud'] = MudServer()

races = {
    "Android": {
        "desc":"Androids are artificial creatures with both biological and mechanical elements, originally created by humanity as servants and now free to chart their own destiny among the stars.",
        "mods": {"dex": 2, "int": 2, "cha": -2, "hp": 4}
    },
    "Human": {
        "desc":"Ambitious, creative, and endlessly curious, humans have shown more drive to explore their system and the universe beyond than any of their neighbor races for better and for worse. They've helped usher in a new era of system-wide communication and organization and are admired for their passion and tenacity, but their tendency to shoot first and think about the consequences later can make them a liability for those races otherwise inclined to work with them.",
        "mods": {"hp": 4}
    },
    "Kasathas": {
        "desc":"Originally from a planet orbiting a dying star far beyond the Pact Worlds, the four-armed kasathas maintain a reputation as a noble and mysterious people. They are famous for their anachronistic warriors, ancient wisdom, and strange traditions.",
        "mods": {"str": 2, "wis": 2, "int": -2, "hp": 4}
    },
    "Korasha Lashuntas": {
        "desc":"Idealized by many other humanoid races and gifted with innate psychic abilities, lashuntas are at once consummate scholars and enlightened warriors, naturally divided into two specialized subraces with different abilities and societal roles.",
        "mods": {"str": 2, "cha": 2, "wis": -2, "hp": 4}
    },
    "Damaya Lashuntas": {
        "desc":"Idealized by many other humanoid races and gifted with innate psychic abilities, lashuntas are at once consummate scholars and enlightened warriors, naturally divided into two specialized subraces with different abilities and societal roles.",
        "mods": {"int": 2, "cha": 2, "con": -2, "hp": 4}
    },
    "Shirrens": {
        "desc":"Once part of a ravenous hive of locust-like predators, the insectile shirrens only recently broke with their hive mind to become a race of telepaths physically addicted to their own individualism, yet dedicated to the idea of community and harmony with other races",
        "mods": {"con": 2, "wis": 2, "cha": -2, "hp": 6}
    },
    "Vesk": {
        "desc":"Heavily muscled and covered with thick scales and short, sharp horns, the reptilian vesk are exactly as predatory and warlike as they appear. Originally hailing from a star system near the Pact Worlds, they sought to conquer and subdue their stellar neighbors, as they had all the other intelligent races in their own system, until an overwhelming threat forced them into a grudging alliance with the Pact Worlds for now.",
        "mods": {"str": 2, "con": 2, "int": -2, "hp": 6}
    },
    "Ysoki": {
        "desc":"Small and furtive, the ysoki are often overlooked by larger races. Yet through wit and technological prowess, they've spread throughout the solar system, giving truth to the old adage that every starship needs a few rats.",
        "mods": {"dex": 2, "int": 2, "str": -2, "hp": 2}
    }
}

themes = {
    "Ace Pilot": {
        "desc": "Skillful operator of starships and other vehicles who is obsessed with all related knowledge and lore.",
        "mods": {"dex": 1}
    },
    "Bounty Hunter": {
        "desc": "Unstoppable tracker who knows how to stay hot on the trail of those who flee.",
        "mods": {"con": 1}
    },
    "Icon": {
        "desc": "Popular and respected celebrity who can leverage the public's adoration for specific needs.",
        "mods": {"cha": 1}
    },
    "Mercenary": {
        "desc": "Well-trained soldier of fortune who can work equally well as a combat grunt or a squad leader.",
        "mods": {"str": 1}
    },
    "Outlaw": {
        "desc": "Wanted criminal with back-alley connections to black markets and associates who can fend off legal trouble.",
        "mods": {"dex": 1}
    },
    "Priest": {
        "desc": "Dedicated and knowledgeable adherent to a philosophy or religion who commands clout among other followers.",
        "mods": {"wis": 1}
    },
    "Scholar": {
        "desc": "Skilled researcher and cutting-edge thinker with a broad base of knowledge and a thirst to expand it.",
        "mods": {"int": 1}
    },
    "Spacefarer": {
        "desc": "Restless explorer who has strong intuition and has collected deep knowledge about alien biology and topology.",
        "mods": {"con": 1}
    },
    "Xenoseeker": {
        "desc": "Guru of alien life-forms who finds that meeting them is one of life's most rewarding accomplishments.",
        "mods": {"cha": 1}
    }
    # "Themeless":{
    #     "desc": "One who doesn't fit into any niche above but forges a personal path of determination and training."
    #     "mods": {"": 1}
    # },
}


classes = {
    "Envoy": {
        "desc": "Charismatic people person good at a wide range of skills who inspires allies to accomplish great heroic feats.",
        "skills": ["Acrobatics", "Intimidate", "Athletics", "Medicine", "Bluff", "Perception", "Computers", "Piloting", "Culture", "Profession", "Diplomacy", "Sense Motive", "Disguise", "Sleight of Hand", "Engineering", "Stealth"],
        "keyability": "cha",
        "proficiencies": ["light armor", "basic melee", "grenades", "small arms"],
        "skillperlevel": 8,
    }, 
    "Mechanic": {
        "desc": "Master of machines and technology whose tinkering produces a drone companion or a powerful brain implant.",
        "skills": ["Athletics", "Perception", "Computers", "Physical Science", "Engineering", "Piloting", "Medicine", "Profession"],
        "keyability": "int",
        "proficiencies": ["light armor", "basic melee", "grenades", "small arms"],
        "skillperlevel": 4,
    }, 
    "Mystic": {
        "desc": "Magic user whose mysterious connection to a powerful force grants abilities that break the laws of the universe.",
        "skills": ["Bluff", "Medicine", "Culture", "Mysticism", "Diplomacy", "Perception", "Disguise", "Profession", "Intimidate", "Sense Motive", "Life Science", "Survival"],
        "keyability": "wis",
        "proficiencies": ["light armor", "basic melee", "small arms"],
        "skillperlevel": 6,
    }, 
    "Operative": {
        "desc": "Stealthy combatant with wide-ranging know-how who is adept at taking advantage of unprepared foes.",
        "skills": ["Acrobatics", "Medicine", "Athletics", "Perception", "Bluff", "Piloting", "Computers", "Profession", "Culture", "Sense Motive", "Disguise", "Sleight of Hand", "Engineering", "Stealth", "Intimidate", "Survival"],
        "keyability": "dex",
        "proficiencies": ["light armor", "basic melee", "small arms", "sniper weapons"],
        "skillperlevel": 8,
    }, 
    "Solarian": {
        "desc": "Disciplined warrior whose mastery of the stars grants either a weapon or armor made of stellar power.",
        "skills": ["Acrobatics", "Perception", "Athletics", "Physical Science", "Diplomacy", "Profession", "Intimidate", "Sense Motive", "Mysticism", "Stealth"],
        "keyability": "cha",
        "proficiencies": ["light armor", "basic melee", "advanced melee", "small arms"],
        "skillperlevel": 4,
    }, 
    "Soldier": {
        "desc": "Expert with a huge range of armor, guns, and melee weapons who specializes in certain types of gear.",
        "skills": ["Acrobatics", "Medicine", "Athletics", "Piloting", "Engineering", "Profession", "Intimidate", "Survival"],
        "keyability": "str",
        "proficiencies": ["light armor", "heavy armor", "basic melee", "advanced melee", "small arms", "long arms", "heavy weapons", "sniper weapons", "grenades"],
        "skillperlevel": 4,
    }, 
    "Technomancer": {
        "desc": "Magic user who is preternaturally attuned to technology and can use it to unlock powerful effects.",
        "skills": ["Computers", "Physical Science", "Engineering", "Piloting", "Life Science", "Profession", "Mysticism", "Sleight of Hand"],
        "keyability": "int",
        "proficiencies": ["light armor", "basic melee", "small arms"],
        "skillperlevel": 4,
    } 
}


def show_prompt(pid):
    if "prompt" not in players[pid]:
        players[pid]["prompt"] = "> "
    if 'hp' not in players[pid]:
        players[pid]["hp"] = 100
    if 'mp' not in players[pid]:
        players[pid]["mp"] = 100
    if 'sta' not in players[pid]:
        players[pid]["sta"] = 10

    prompt = players[pid]["prompt"].replace('%st', str(floor(players[pid]['sta']))).replace('%hp', str(floor(players[pid]["hp"]))).replace('%mp', str(floor(players[pid]["mp"])))
    mud.send_message(pid, "\r\n" + prompt, '')

tick = 0.0
spawn = 0.0
cmd_handler = CommandHandler()

def spawn_mobs(players):
    rooms = listdir('rooms')
    for room in rooms:
        if '_monsters.json' not in room:
            continue
        room_monsters = load_object_from_file('rooms/{}'.format(room))
        for mon_name, monster in room_monsters.items():
            monster_template = load_object_from_file('mobs/{}.json'.format(mon_name))
            if not monster_template:
                continue
            while len(room_monsters[mon_name]['active']) < monster['max']:
                print('Spawning {} in {}'.format(mon_name, room))
                mp = get_att(monster_template['spawn']["mp"])
                hp = get_att(monster_template['spawn']["hp"])
                sta = get_att(monster_template['spawn']["sta"])
                new_active = {
                    "mp": mp, 
                    "maxhp": hp,
                    "hp": hp,
                    "sta": sta,
                    "maxmp": mp,
                    "target": "",
                    "action": "attack",
                    "maxsta": sta}
                room_monsters[mon_name]['active'].append(new_active)
                for pid, pl in players.items():
                    if players[pid]['room'].lower() == room.split('_')[0]:
                        mud.send_message(pid, "a {} arrived".format(mon_name))
        save_object_to_file(room_monsters, 'rooms/{}'.format(room))

def run_mobs(players, mud):
    for pid, player in players.items():
        if not player or not player.get("name") or not player.get('password'):
            continue
        if player['mp'] < player['maxmp']:
            players[pid]['mp'] += player['mpr']
        if player['sta'] < player['maxsta']:
            players[pid]['sta'] += player['star']
        room_monsters = load_object_from_file('rooms/{}_monsters.json'.format(player['room']))
        for mon_name, monster in room_monsters.items():
            monster_template = load_object_from_file('mobs/{}.json'.format(mon_name))
            for active_monster_idx, active_monster in enumerate(monster['active']):
                sta = active_monster['sta']
                if active_monster['mp'] < active_monster['maxmp']:
                    active_monster['mp'] += monster_template['mpr']
                if sta < active_monster['maxsta']:
                    sta += monster_template['star']
                    active_monster['sta'] = sta
                if active_monster['action'] == "attack" and active_monster['target'] == player['name']:
                    if player.get("weapon"):
                        weapon = load_object_from_file('inventory/{}.json'.format(player['weapon']))
                        att = get_att(weapon['damage'])
                        mud.send_message(pid, "Your %s strikes the %s for %d" % (weapon['title'], mon_name, att,), color='yellow')
                    else:
                        att = get_att(player['aa'])
                        mud.send_message(pid, "You hit the %s for %d" % (mon_name, att,), color='yellow')
                    active_monster['hp'] -= att
                    if active_monster['hp'] <= 0:
                        del room_monsters[mon_name]['active'][active_monster_idx]
                        mud.send_message(pid, "The %s dies." % (mon_name,), color=['bold', 'blue'])
                        break
                    if active_monster.get("weapon"):
                        weapon = load_object_from_file('inventory/{}.json'.format(active_monster['weapon']))
                        att = get_att(weapon['damage'])
                        mud.send_message(pid, "The %s strikes you with a %s for %d" % (mon_name, weapon['title'], att,), color='magenta')
                    else:
                        att = get_att(monster_template['aa'])
                        mud.send_message(pid, "You were hit by a %s for %d" % (mon_name, att,), color='magenta')
                    players[pid]['hp'] -= att
                    if (active_monster['hp']/active_monster['maxhp'] < 0.25) or (active_monster['mp'] > 0 and active_monster['mp']/active_monster['maxmp'] > 0.5) or (sta > 0 and sta/active_monster['maxsta'] > 0.5):
                        magic_cast = False
                        if active_monster['mp'] > 0:
                            att, active_monster['mp'] = calc_att(mud, pid, monster_template['sp'], active_monster['mp'])
                            players[pid]['hp'] -= att
                            if att > 0:
                                magic_cast = True                            
                        if not magic_cast:
                            if sta > 0:
                                att, sta = calc_att(mud, pid, monster_template['at'], sta)
                                active_monster['sta'] = sta
                                players[pid]['hp'] -= att

                    room_monsters[mon_name]['active'][active_monster_idx] = active_monster
            
        save_object_to_file(room_monsters, 'rooms/{}_monsters.json'.format(player['room']))

def isalnum(c):
    for letter in c:
        if not letter.isalpha() and not letter.isdigit():
            return False
    return True

def handle_character_create(id, command, params):
    if players[id]["createstep"] == 1:
        players[id]["color_enabled"] = command[0] in ['y', 'Y']
        mud._clients[id].color_enabled = players[id]["color_enabled"]
        if players[id]["color_enabled"]:
            mud.send_message(id, '%bold%blueC%yellowo%redl%greeno%cyanr%reset %boldenabled.%reset\r\n')
        else:
            mud.send_message(id, 'Color disabled.\r\n')
        mud.send_message(id, "Are you at least 13 years of age?")
        players[id]["createstep"] = 2

    elif players[id]["createstep"] == 2:
        players[id]["over_13"] = command[0] in ['y', 'Y']
        if not players[id]["over_13"]:
            mud.send_message(id, "You must be at least 13 years old to play this game.")
            try:
                mud._clients[id].socket.close()
                remove("players/{}.json".format(players[id]["name"]))
                del(players[id])
                del(mud._clients[id])
            except:
                pass
            return
        else:
            mud.send_message(id, "  %green+------------=Races=------------+", nowrap=True)
            for idx, race in enumerate(races):
                mud.send_message(id, "  %%green|%%reset%%bold%%white%s%%reset%%green|" % ('{}. {:<28}'.format(idx, race),), nowrap=True)
            mud.send_message(id, "  %green+-------------------------------+", nowrap=True)

            mud.send_message(id, 'Please choose a race from the list by number: ')
            players[id]["createstep"] = 3

    elif players[id]["createstep"] == 3:
        if command.lower().startswith("help"):
            races_lower = {k.lower():v for k,v in races.items()}
            race = races_lower.get(params.lower().strip())
            if race:
                mods = '\r\n'.join("  %green{!s: <4} %bold{!r}%reset".format(key,val) for (key,val) in race.get("mods").items())

                mud.send_message(id, "\r\n%green{}\r\n%bold{}\r\n\n%resetStat Modifiers:".format(params.strip(), race.get("desc")))
                mud.send_message(id, mods + "\r\n", nowrap=True)
            else:
                mud.send_message(id, "Please type the name of the race you want to learn about.", color='cyan')
        if not command.isdigit():
            mud.send_message(id, "  %green+------------=Races=-----------+", nowrap=True)
            for idx, race in enumerate(races):
                mud.send_message(id, "  %%green|%%reset%%bold%%white%s%%reset%%green|" % ('{}. {:<27}'.format(idx, race),), nowrap=True)
            mud.send_message(id, "  %green+------------------------------+", nowrap=True)

            mud.send_message(id, '   (Type %boldhelp [race]%reset for details)\r\n')
            mud.send_message(id, 'Please choose a race from the list by number: ')
        else:
            race = list(races)[int(command)]
            mud.send_message(id, 'Race selected: {}'.format(race))

            players[id]["race"] = race
            players[id]["createstep"] = 4
            mud.send_message(id, "  %green+------------=Themes=-----------+", nowrap=True)
            for idx, theme in enumerate(themes):
                mud.send_message(id, "  %%green|%%reset%%bold%%white%s%%reset%%green|" % ('{}. {:<27}'.format(idx, theme),), nowrap=True)
            mud.send_message(id, "  %green+------------------------------+", nowrap=True)

            mud.send_message(id, '   (Type %boldhelp [theme]%reset for details)\r\n')
            mud.send_message(id, 'Please choose a theme from the list by number: ')
    elif players[id]["createstep"] == 4:
        if command.lower().startswith("help"):
            themes_lower = {k.lower():v for k,v in themes.items()}
            theme = themes_lower.get(params.lower().strip())
            if theme:
                mods = '\r\n'.join("  %green{!s: <4} %bold{!r}%reset".format(key,val) for (key,val) in theme.get("mods").items())

                mud.send_message(id, "\r\n%green{}\r\n%bold{}\r\n\n%resetStat Modifiers:".format(params.strip(), theme.get("desc")))
                mud.send_message(id, mods + "\r\n", nowrap=True)
            else:
                mud.send_message(id, "Please type the name of the theme you want to learn about.", color='cyan')
        if not command.isdigit():
            mud.send_message(id, "  %green+------------=Themes=-----------+", nowrap=True)
            for idx, theme in enumerate(themes):
                mud.send_message(id, "  %%green|%%reset%%bold%%white%s%%reset%%green|" % ('{}. {:<27}'.format(idx, theme),), nowrap=True)
            mud.send_message(id, "  %green+------------------------------+", nowrap=True)

            mud.send_message(id, '   (Type %boldhelp [theme]%reset for details)\r\n')
            mud.send_message(id, 'Please choose a theme from the list by number: ')
        else:
            theme = list(themes)[int(command)]
            mud.send_message(id, 'Theme selected: {}'.format(theme))

            players[id]["theme"] = theme
            players[id]["createstep"] = 5
    elif players[id]["createstep"] == 5:
        if command.lower().startswith("help"):
            themes_lower = {k.lower():v for k,v in themes.items()}
            theme = themes_lower.get(params.lower().strip())
            if theme:
                mods = '\r\n'.join("  %green{!s: <4} %bold{!r}%reset".format(key,val) for (key,val) in theme.get("mods").items())

                mud.send_message(id, "\r\n%green{}\r\n%bold{}\r\n\n%resetStat Modifiers:".format(params.strip(), theme.get("desc")))
                mud.send_message(id, mods + "\r\n", nowrap=True)
            else:
                mud.send_message(id, "Please type the name of the theme you want to learn about.", color='cyan')
        if not command.isdigit():
            mud.send_message(id, "  %green+------------=Themes=-----------+", nowrap=True)
            for idx, theme in enumerate(themes):
                mud.send_message(id, "  %%green|%%reset%%bold%%white%s%%reset%%green|" % ('{}. {:<27}'.format(idx, theme),), nowrap=True)
            mud.send_message(id, "  %green+------------------------------+", nowrap=True)

            mud.send_message(id, '   (Type %boldhelp [theme]%reset for details)\r\n')
            mud.send_message(id, 'Please choose a theme from the list by number: ')
        else:
            theme = list(themes)[int(command)]
            mud.send_message(id, 'Theme selected: {}'.format(theme))

            players[id]["theme"] = theme
            players[id]["createstep"] = 5

    save_object_to_file(players[id], "players/{}.json".format(players[id]["name"]))

# main game loop. We loop forever (i.e. until the program is terminated)
while True:
    if 'esp' in platform:
        collect()
        if flash_button.value() == 0:
            break
    # pause for 1/5 of a second on each loop, so that we don't constantly
    sleep(0.002)
    if 'esp' in platform:
        tick += 0.001
    else:
        tick += 0.0005
    if spawn >= 10:
        spawn = 0
        try:
            spawn_mobs(players)
        except Exception as e:
            print('spawner error:')
            print(e)
        if 'esp' in platform:
            collect()
    if tick >= 1:
        if 'esp' in platform:
            print(mem_free())
        spawn += tick
        tick = 0
        try:
            run_mobs(players, mud)
        except Exception as e:
            print('mob error:')
            print(e)
    # 'update' must be called in the loop to keep the game running and give
    # us up-to-date information
    mud.update()

    # go through any newly connected players
    for id in mud.get_new_players():

        # add the new player to the dictionary, noting that they've not been
        # named yet.
        # The dictionary key is the player's id number. We set their room to
        # None initially until they have entered a name
        # Try adding more player stats - level, gold, inventory, etc
        players[id] = load_object_from_file("defaultplayer.json")
        with open('welcome.txt', 'r', encoding='utf-8') as f:
            for line in f:
                mud.send_message(id, line, "\r")
        # send the new player a prompt for their name
        #bytes_to_send = bytearray([mud._TN_INTERPRET_AS_COMMAND, mud._TN_WONT, mud._ECHO])
        #mud.raw_send(id, bytes_to_send)
        mud.send_message(id, "What is your name?")

    # go through any recently disconnected players
    for id in mud.get_disconnected_players():

        # if for any reason the player isn't in the player map, skip them and
        # move on to the next one
        if id not in players:
            continue

        # go through all the players in the game
        for pid, pl in players.items():
            # send each player a message to tell them about the diconnected
            # player
            if players[pid].get("name") is not None:
                mud.send_message(pid, "{} quit the game".format(players[pid]["name"]))

        # remove the player's entry in the player dictionary
        if players[id]["name"] is not None:
            save_object_to_file(players[id], "players/{}.json".format(players[id]["name"]))
        del(players[id])

    # go through any new commands sent from players
    for id, command, params in mud.get_commands():

        # if for any reason the player isn't in the player map, skip them and
        # move on to the next one
        if id not in players:
            continue

        # if the player hasn't given their name yet, use this first command as
        # their name and move them to the starting room.
        if players[id].get("name") is None:
            if command.strip() == '' or len(command) > 12 or not isalnum(command) or command is None:
                mud.send_message(id, "Invalid Name")
                print("Bad guy!")
                print(mud.get_remote_ip(id))
                continue
            already_logged_in = False
            for pid, pl in players.items():
                if players[pid].get("name") == command:
                    mud.send_message(id, "{} is already logged in.".format(command))
                    mud.disconnect_player(id)
                    continue
            loaded_player = load_object_from_file("players/{}.json".format(command))
            mud.remote_echo(id, False)
            players[id]["name"] = command
            if not loaded_player:
                # Player does not exist.
                mud.send_message(id, "Character does not exist. Please enter a password to create a new character: ")
            else:
                mud.send_message(id, "Enter Password: ")

        elif players[id]["password"] is None:
            if players[id].get("retries", 0) > 1:
                mud.send_message(id, "Too many attempts")
                mud.disconnect_player(id)
                continue

            if command.strip() == '' or command is None:
                players[id]["retries"] = players[id].get("retries", 0) + 1
                mud.send_message(id, "Invalid Password")
                continue

            loaded_player = load_object_from_file("players/{}.json".format(players[pid]["name"]))
            if loaded_player is None:
                players[id]["password"] = password_hash(players[pid]["name"], command)
                players[id]["room"] = "town/tavern"
                players[id]["createstep"] = 1
                mud.send_message(pid, "Can you view %bold%bluecolors%reset? ")
                continue
            else:
                if loaded_player["password"] == password_hash(players[pid]["name"], command):
                    players[id] = loaded_player
                    mud._clients[id].color_enabled = players[id]["color_enabled"]
                    if players[id]["createstep"]:
                        mud.send_message(pid, "Can you view %bold%bluecolors%reset? ")
                        continue
                else:
                    players[id]["retries"] = players[id].get("retries", 0) + 1
                    mud.send_message(id, "Invalid Password")
                    continue

            # go through all the players in the game
            for pid, pl in players.items():
                # send each player a message to tell them about the new player
                mud.send_message(pid, "{} entered the game".format(players[id]["name"]))

            mud.remote_echo(id, True)
            # send the new player a welcome message
            mud.send_message(id, "\r\n\r\nWelcome to the game, {}. ".format(players[id]["name"]) +
                             "\r\nType 'help' for a list of commands. Have fun!\r\n\r\n")

            # send the new player the description of their current room

            cmd_handler.parse(id, 'look', '', mud, players)
            show_prompt(id)
        elif players[id].get("createstep"):
            handle_character_create(id, command, params)                
        else:
            if 'esp' in platform:
                collect()

            cmd_handler.parse(id, command, params, mud, players)
            show_prompt(id)

# Start WIFI Setup
if 'esp' in platform:
    if flash_button.value() == 0:
        print('Starting WIFIWeb')
        import wifiweb