###
# Copyright (c) 2019 oddluck
# All rights reserved.
#
#
###

import supybot.utils as utils
from supybot.commands import *
import supybot.plugins as plugins
import supybot.ircutils as ircutils
import supybot.callbacks as callbacks
import supybot.ircmsgs as ircmsgs
import os
import pexpect
import re

try:
    from supybot.i18n import PluginInternationalization
    _ = PluginInternationalization('Infocom')
except ImportError:
    # Placeholder that allows to run the plugin on a bot
    # without the i18n module
    _ = lambda x: x

class Frotz(callbacks.Plugin):
    """
    Play Infocom (Interactive Fiction, Z_machine) Games.
    """
    threaded = True

    def __init__(self, irc):
        self.__parent = super(Frotz, self)
        self.__parent.__init__(irc)
        self.game = {}
        self.game_path = "{0}/games/".format(os.path.dirname(os.path.abspath(__file__)))
        self.binary = '{0}/frotz/dfrotz'.format(os.path.dirname(os.path.abspath(__file__)))

    def load(self, irc, msg, args, input):
        """<game_name>
        Load <game_name.z*>.
        """
        channel = msg.args[0]
        if not irc.isChannel(channel):
            channel = msg.nick
        game_name = input
        self.game.setdefault(channel, None)
        if self.game[channel]:
            irc.reply("There is a game already in progress on {0}. Please end that game first.".format(channel))
        else:
            irc.reply("Starting {0} on {1}. Please wait...".format(game_name, channel))
            game_file= "{0}{1}".format(self.game_path, game_name)
            self.game[channel] = pexpect.spawn("{0} -S 0 {1}".format(self.binary, game_file))
            response = self.output(self.game[channel])
            if len(response) > 2:
                irc.reply(re.sub(' +', ' ', response[0]), prefixNick=False)
                irc.reply(" ".join(response[1:]).replace(' .', '.'), prefixNick=False)
            else:
                irc.reply(" ".join(response).replace(' .', '.'), prefixNick=False)
    load = wrap(load, ['text'])

    def output(self, output):
        response = []
        prompts = ["\n>", "\n> >", "to begin]", "\n\*\*\*MORE\*\*\*", pexpect.TIMEOUT]
        output.expect(prompts, timeout=2)
        for line in output.before.splitlines():
            if line.strip():
                response.append(line.decode().strip())
        return response

    def stop(self, irc, msg, args):
        """
        Stop game.
        """
        channel = msg.args[0]
        if not irc.isChannel(channel):
            channel = msg.nick
        self.game.setdefault(channel, None)
        if self.game[channel]:
            if self.game[channel].isalive() is True:
                irc.reply("Stopping Game. Thanks for playing.")
                self.game[channel].close()
                self.game[channel].terminate()
                del self.game[channel]
            else:
                irc.reply("No game running in {0}".format(channel))
        else:
            irc.reply("No game started in {0}".format(channel))
    stop = wrap(stop)

    def z(self, irc, msg, args, input):
        """[<input>]
        Send user input or blank line (ENTER/RETURN) to the game.
        """
        channel = msg.args[0]
        if not irc.isChannel(channel):
            channel = msg.nick
        self.game.setdefault(channel, None)
        if input:
            command = input.strip()
        else:
            command = None
        if self.game[channel]:
            if command:
                self.game[channel].sendline(command)
            else:
                self.game[channel].sendline()
            response = self.output(self.game[channel])
            if len(response) > 2:
                irc.reply(re.sub(' +', ' ', response[1]), prefixNick=False)
                irc.reply(" ".join(response[2:]).replace(' .', '.'), prefixNick=False)
            else:
                irc.reply(" ".join(response).replace(' .', '.'), prefixNick=False)
        else:
            irc.reply("No game running in {0}?".format(channel))
    z = wrap(z, [additional('text')])

    def games(self, irc, msg, args):
        """
        List files in the game directory.
        """
        reply = ", ".join(sorted(os.listdir(self.game_path)))
        irc.reply(reply, prefixNick=False)
    games = wrap(games)

Class = Frotz