#!/usr/bin/env python
# coding: utf-8

# Copyright (C) 2010 Arthur Furlan <afurlan@afurlan.org>
#
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation; either version 3 of the License, or
# (at your option) any later version.
#
# On Debian systems, you can find the full text of the license in
# /usr/share/common-licenses/GPL-3

JABBER = "jabber"
CHAT = "chat"
DELAY = 0.3

CHAT_IN_USE = CHAT + ".used"


import xmpp
from jabberbot import JabberBot, botcmd
from datetime import datetime
import re
import codecs
import time
import os, os.path

class MUCJabberBot(JabberBot):

    ''' Add features in JabberBot to allow it to handle specific
    caractheristics of multiple users chatroom (MUC). '''

    def __init__(self, *args, **kwargs):
        ''' Initialize variables. '''

        # answer only direct messages or not?
        self.only_direct = kwargs.get('only_direct', False)
        try:
            del kwargs['only_direct']
        except KeyError:
            pass

        # initialize jabberbot
        super(MUCJabberBot, self).__init__(*args, **kwargs)

        # create a regex to check if a message is a direct message
        user, domain = str(self.jid).split('@')
        self.direct_message_re = re.compile('^%s(@%s)?[^\w]? ' \
                % (user, domain))

        self.timer = time.time()

    def callback_message(self, conn, mess):
        ''' Changes the behaviour of the JabberBot in order to allow
        it to answer direct messages. This is used often when it is
        connected in MUCs (multiple users chatroom). '''

        message = mess.getBody()
        if not message:
            return

        if mess.getType() != "groupchat":
            return
        if xmpp.NS_DELAY in mess.getProperties():
            return
        if self.jid.bareMatch(mess.getFrom()):
            return
        if not message:
            return
        user = self.get_sender_username(mess)
        if user == nickname:
            return
        if len(message) > 256:
            self.send_simple_reply(mess, "Message is too long")
            return

        # it's time to send message to lua
        f = codecs.open(JABBER, "a", "utf-8")
        f.write("<xmpp://%s> %s\n" % (user, " ".join(message.splitlines())))
        f.close()

    def idle_proc(self):
        super(MUCJabberBot, self)
        cur = time.time()
        if cur - self.timer > DELAY:
            # let's grab messages from lua folder!
            self.timer = cur
            if not os.path.exists(CHAT):
                return
            os.rename(CHAT, CHAT_IN_USE)
            f = codecs.open(CHAT_IN_USE, "r", "utf-8")
            for x in f.readlines():
                self.send("game@minetest.ru", " ".join(x.splitlines()), message_type='groupchat')
            f.close()

if __name__ == '__main__':

    username = 'game@users.minetest.ru'
    password = '---'
    nickname = 'game'
    chatroom = 'game@minetest.ru'

    mucbot = MUCJabberBot(username, password, only_direct=False)
    mucbot.join_room(chatroom, nickname)
    mucbot.serve_forever()