from textwrap import dedent

from model import DB, User

db = DB('mysql+pymysql://no:no@localhost/TMS-DB')


class TMSPlugin(PythonPlugin):
    def onEnable(self):            
        print "TMS MC Plugin enabled!"

    def onDisable(self):
        print "TMS MC Plugin disabled!"

    def onCommand(self, sender, command, label, args):
        sender.sendMessage("Received command: " + command.getName())
        return True

    @hook.event('player.PlayerJoinEvent', 'normal')
    def onJoin(event):
        player = event.getPlayer()
        nick = player.getName()
        if ' ' in nick:
            player.kickPlayer('No spaces allowed in name!')

        message = dedent('''\
            Welcome to TMS!
            ----------------
            New commands:
            /register <password>
            /login <password>
            /changepass <new_password>
        ''')
        player.sendMessage(message)

        with db as session:
            user = None
            try:
                user = session.query(User).filter_by(name=nick).first()
            except Exception as e:
                print 'CRASH:', str(e)
            if user:
                player.sendMessage('This account is registered, use /login <password> to log in.')