from htmlentitydefs import name2codepoint as n2cp

from htmlentitydefs import codepoint2name as cp2n

import twitter

import re, time

####

# Module declarations

##

twitter_settings = { "password": None,

                     "username": None,

                     "channel": None,

                     "last_friend_update": 0,

                     "last_reply_update": 0,

                     "last_directed": 0}

twitter_requires = ['webhelpers']

do_twitter_command = { 'description': "Different twitter functionalities",

                       'long help': """\

A set of commands to handle twitter streams and tweeting from a IRC-channel via the bot.""",

                       'arguments': [("update <update string>", "Update twitter status (>70)"),

                                     ("subscribe <user> [<user2> [<user3..]]", "Subscribe to user on twitter (>80)"),

                                     ("unsubscribe <user> [<user2> [<user3..]]", "Unsubscribe to user on twitter (>80)"),

                                     ("get friends", "List all friends on twitter. (>70)"),

                                     ("get last [nr]", "List nr (default 1) of updates listed in the channel. Only works in private."),

                                     ("get queue size", "Show the number of posts in queue to be outputed to the channel. (>70)"),

                                     ("clear queue", "Remove all posts in queue to be outputed to the channel.(>70)")],

                       'public': True,

                       'level': 0}

twitter_alias = {'t': 'twitter update',

                 'tsub': 'twitter subscribe',

                 'tun': 'twitter unsubscribe',

                 'tl': 'twitter get friends'}

update_twitter_delay = 50

update_twitter_time_left = 0

##

# End of module declarations

####

# If an update from twitter.com fails twitter_timeout is increased and lowered 1 each

# update. If twitter_timeout >0 nothing is done on update.

twitter_timeout = 0

# Next update is set according to which update function that's supposed to be called on

# next update iteration.

next_update = 0

# twitter_api is the twitter api handle, it needs to be initalized before use.

twitter_api = None

# left_over_updates is a list of updates generated by the update functions. Each update

# three of the lines are sent to the twitter channel.

left_over_updates = []

### last_updates stores the max_last_updates last update strings. last_updates are ###

### sorted from oldest to newest.                                                  ###

max_last_updates = 10

last_updates = []

### Keep a cache of old twitter friends. ###

old_friend_list = []

def init_twitter(bot, server, sender = None):

    """Initiate twitter and set user and password.

       This function is called on loading the module."""

    ### Check if all settings are set correctly. ###

    fail = False

    if not bot.settings.twitter_username:

        print " - No username set."

        if sender:

            server.privmsg(sender, "No twitter username.")

        fail = True

    if not bot.settings.twitter_password:

        print " - No password set."

        if sender:

            server.privmsg(sender, "No twitter password.")

        fail = True

    if not bot.settings.twitter_channel:

        print " - No channel set."

        if sender:

            server.privmsg(sender, "No twitter channel.")

        fail = True

    if fail:

        return False

    ### Login to twitter and return the api handle. ###

    print "Login to twitter:"

    return init_twitter_api(bot)

def update_twitter(bot, server):

    """Update twitter on regular intervals. On each update one api call will be made at

       most."""

    global twitter_timeout

    global next_update

    global left_over_updates

    print " * %s Twitter tick!" % time.strftime('[%H:%M:%S]')

    if twitter_timeout > 0:

        print " (timeouts: %d)" % twitter_timeout

    ### update_funcs are the functions that should be update. To add more just extend ###

    ### the list with a function of the same type as the once already there.          ###

    update_funcs = [get_friend_timeline, get_directed_messages, get_replies]

    if twitter_timeout == 0:

        if update_funcs[next_update](bot, server):

            next_update += 1

            if next_update == len(update_funcs):

                next_update = 0

        else:

            twitter_timeout += 1

    elif twitter_timeout > 0:

        twitter_timeout -= 1

        print "  - Timeout, waiting %d seconds." % (twitter_timeout*update_twitter.delay)

    else:

        twitter_timeout = 0

        print "  - Set timeout to 0."

    ### Print the update lines generated by the update funcs ###

    if left_over_updates:

        for i in xrange(0,2):

            if left_over_updates:

                status_string = left_over_updates.pop(0)

                print status_string

                send_to_channel(bot, server, status_string)

                last_updates.insert(0, status_string)

                if len(last_updates) > max_last_updates:

                    last_updates.pop()

            else:

                break

        print " + %d status updates left." % len(left_over_updates)

def do_twitter(bot, server, sender, target, args):

    """This is one of the commands, twitter in the name is not the module name but the

       commandname. The description of the command is specified in do_twitter_command"""

    if not twitter_api:

        init_twitter_api(bot)

    if args:

        sender_level = bot.get_level(sender)

        if sender_level > 70:

            if args[0] == 'update':

                return do_twitter_update(bot, server, sender, target, args[1:])

        if sender_level > 80:

            if args[0] == 'subscribe':

                return do_twitter_subscribe(bot, server, sender, target, args[1:])

            elif args[0] == 'unsubscribe':

                return do_twitter_unsubscribe(bot, server, sender, target, args[1:])

        if args[0] == 'get':

            return do_twitter_get(bot, server, sender, target, args[1:])

        elif args[0] == 'clear':

            return do_twitter_clear(bot, server, sender, target, args[1:])

    else:

        bot.respond(server, sender, target, "Not enough arguments.")

        return False

def handle_posts(bot, server, post_list):

    """Handle posts from twitter. Posts are returned by friend timeline, replies and

       public timeline."""

    global left_over_updates

    ### Iterate through the list of status updates and format it to a string that will ###

    ### end up on IRC.                                                                 ###

    for post in post_list:

        ### Get all the values from the post to shorten the rest of the codes. ###

        text         = post.GetText()

        realname     = post.GetUser().GetName()

        screename    = post.GetUser().GetScreenName()

        in_reply_to  = post.GetInReplyToScreenName()

        since_posted = post.GetRelativeCreatedAt()

        posted_at    = post.GetCreatedAtInSeconds()

        source       = post.GetSource()

        ### Try to expand shortened URL. ###

        for match in re.finditer(bot.regex_url, text):

            url = match.group(0)

            lurl = bot.expand_url(url)

            if len(lurl) > 70:

                print " + Expanding url '%s' to '%s'." % (url, lurl)

                text = text.replace(url, lurl)

        ### Look for html tags in the source string and strip them away to avoid ###

        ### junk in the IRC output.                                              ###

        m = re.match(r'<.*>(.*)<.*>', source)

        if m:

            source = m.group(1)

            print "Stripped source of html to '%s'." % source

        update_string = "\x02%s\x02 - Posted by %s [%s] from %s" % (text, realname, screename, source)

        if in_reply_to:

            update_string += " in reply to %s" % in_reply_to

        ### Only add time string if time exceed 5 minutes. ###

        if (time.time() - posted_at) > 5*60:

            update_string += " %s" % since_posted

        left_over_updates.append(update_string)

def get_friend_timeline(bot, server):

    """This update function fetches the timeline of all friends.

       Returns True if successfull, return False if got error from twitter.com"""

    global left_over_updates

    print " - Get twitter friend timeline."

    try:

        ### Make a call to the api and reverse the list so that the oldest message is ###

        ### first.                                                                    ###

        friend_timeline = twitter_api.GetFriendsTimeline(since_id = bot.settings.twitter_last_friend_update)

        friend_timeline.reverse()

    except Exception, e:

        global twitter_timeout

        twitter_timeout = (twitter_timeout + 1) * 2

        print "Failed to get friend timeline:", e

        return False

    ### Handle the post list and add them to the queue of posts to get posted on IRC ###

    handle_posts(bot, server, friend_timeline)

    ### If got any updates get the id for the last one and use that to look for updates ###

    ### later on. But only change the value in settings if post id is higher.           ###

    if friend_timeline:

        bot.settings.twitter_last_friend_update = max(friend_timeline[-1].id, bot.settings.twitter_last_friend_update)

        print " * Update twitter from '%d'." % bot.settings.twitter_last_friend_update

    return True

def get_replies(bot, server):

    """This update function fetches all replies from twitter. A reply is a tweet that

       contains @myscreenname.

       Returns True if successfull, return False if got error from twitter.com"""

    global left_over_updates

    print " - Get twitter replies."

    try:

        ### Make a call to the api and reverse the list so that the oldest message is ###

        ### first.                                                                    ###

        replies = twitter_api.GetReplies(since_id=bot.settings.twitter_last_reply_update)

        replies.reverse()

    except Exception, e:

        global twitter_timeout

        twitter_timeout = (twitter_timeout + 1) * 2

        print "Failed to get replies:", e

        return False

    ### Handle the reply list and add them to the queue of posts to get posted on IRC ###

    handle_posts(bot, server, replies)

    ### If got any updates get the id for the last one and use that to look for updates ###

    ### later on. But only change the value in settings if post id is higher.           ###

    if replies:

        bot.settings.twitter_last_reply_update = max(replies[-1].id, bot.settings.twitter_last_reply_update)

        print " * Update twitter from '%d'." % bot.settings.twitter_last_reply_update

    return True

def get_directed_messages(bot, server):

    """This update function fetches all direct messages from twitter.

       Direct messages are not posts and can not be handled by handle_posts.

       Returns True if successfull, return False if got error from twitter.com"""

    global left_over_updates

    print " - Get direct messages from twitter."

    try:

        directed_messages = twitter_api.GetDirectMessages(since_id=bot.settings.twitter_last_directed)

        directed_messages.reverse()

    except Exception, e:

        global twitter_timeout

        twitter_timeout = (twitter_timeout + 5) * 2

        print "Failed to get directed messages:", e

        return False

    ### Iterate through the list of messages and 

    for message in directed_messages:

        ### Assign variables to each value from message to simplify the code. ###

        text              = message.GetText()

        sender_screenname = message.GetSenderScreenName()

        recip_screenname  = message.GetRecipientScreenName()

        ### Try to expand shortened URL. ###

        for match in re.finditer(bot.regex_url, text):

            url = match.group(0)

            lurl = bot.expand_url(url)

            if len(surl) > 70:

                print " + Expanding url '%s' to '%s'." % (url, lurl)

                text = text.replace(url, lurl)

        update_string = "Direct message: \x02%s\x02 from %s to %s" % (text, sender_screenname, recip_screenname)

        left_over_updates.append(update_string)

    if directed_messages:

        bot.settings.twitter_last_directed = max(directed_messages[-1].id, bot.settings.twitter_last_directed)

        print " * Update twitter directed messages from '%d'." % bot.settings.twitter_last_directed

    return True

def do_twitter_get(bot, server, sender, target, args):

    """Subcommand to twitter, get values from twitter. Each argument to get should only

       make one API call."""

    global twitter_api

    global old_friend_list

    if not twitter_api:

        init_twitter_api(bot)

    if args:

        if args[0] == 'friends' and bot.get_level(sender) > 70:

            friend_string = "My twitter friends are:"

            users = twitter_api.GetFriends()

            user_names = [u.GetScreenName() for u in users]

            first_friend = True

            for screenname in user_names:

                if not screenname in old_friend_list:

                    friend_string += "%s \x02%s\x02" % (("", ",")[first_friend], screenname)

                else:

                    friend_string += "%s %s" % (("", ",")[first_friend], screenname)

            bot.respond(server, sender, target, bot.convert_to_irc_string(friend_string))

            old_friend_list = user_names

            return True

        elif args[0] == 'queue' and bot.get_level(sender) > 70:

            queue_size = len(left_over_updates)

            bot.respond(server, sender, target, "There are currently %d twitter message%s in queue." % (queue_size,

                                                                                                        ("", "s")[queue_size > 1]))

            return True

        ### If command is called in private also handle these subcommands to 'get'. ###

        elif target == bot.settings.bot_nick_name:

            if args[0] == 'last':

                max_updates = 1

                try:

                    if len(args) > 1:

                        max_updates = int(''.join(args[1:]))

                except Exception, e:

                    pass

                for update in last_updates[:max_updates]:

                    bot.respond(server, sender, target, bot.convert_to_irc_string(update))

                return True

            else:

                bot.respond(server, sender, target, "'%s' is not a valid argument." % args[0])

                return False

        else:

            bot.respond(server, sender, target, "'%s' is not a valid argument." % args[0])

            return False

    else:

        bot.respond(server, sender, target, "Not enough arguments.")

        return False

def do_twitter_clear(bot, server, sender, target, args):

    """Clear a variable from the module."""

    global left_over_updates

    if left_over_updates and bot.get_level(sender) > 70:

        queue_size = len(left_over_updates)

        left_over_updates = []

        bot.respond(server, sender, target, "Removed %d posts from queue." % queue_size)

        return True

    else:

        bot.respond(server, sender, target, "The queue is already empty.")

        return False

def do_twitter_update(bot, server, sender, target, args):

    """Subcommand to twitter, update the twitter status."""

    global twitter_api

    if not twitter_api:

        init_twitter_api(bot)

    if args:

        ### Join all the args to twitter update into one string. ###

        new_status = " ".join(args)

        ### If the status string is to long, shorten all the URL's in it and try again. ###

        if len(new_status) > 100:

            for match in re.finditer(bot.regex_url, new_status):

                url = match.group(0)

                surl = bot.shorten_url(url)

                if len(url) > len(surl):

                    print "Shortening url '%s' to '%s'." % (url, surl)

                    new_status = new_status.replace(url, surl)

            ### If the status is still to long don't even try to post to save api calls. ###

            if len(new_status) > 140:

                bot.respond(server, sender, target, "Message is %d characters too long." % (len(new_status)-140))

                print "Failed to update, too long message."

                return True

        try:

            ### Call the api and try to update status. ###

            status = twitter_api.PostUpdate(bot.convert_to_twitter_string(new_status))

        except Exception, e:

            print "Updating status failed:", e

            bot.respond(server, sender, target, "Twitter status update failed: %s" % str(e))

            return True

        print "Updated twitter status to %s by %s" % (bot.convert_to_irc_string(status.text), sender)

        server.notice(bot.settings.twitter_channel,

                      "Twitter status updated successfully to \x02'%s'\x02." % bot.convert_to_irc_string(status.text))

        return True

    else:

        bot.respond(server, sender, target, "Not enough arguments.")

        return False

def do_twitter_subscribe(bot, server, sender, target, args):

    global twitter_api

    if not twitter_api:

        init_twitter_api(bot)

    if args:

        print "Create friendship with:", ", ".join(args)

        for user in args:

            try:

                new_friend = twitter_api.CreateFriendship(user)

                if new_friend:

                    bot.respond(server, sender, target, "You are now friends with '%s'." % new_friend.GetName())

            except Exception, e:

                bot.respond(server, sender, target, "Could not make friends with '%s'." % user)

                print " - Could not make friends with '%s'." % user

        return True

    else:

        bot.respond(server, sender, target, "Not enough arguments.")

        return False

def do_twitter_unsubscribe(bot, server, sender, target, args):

    global twitter_api

    if not twitter_api:

        init_twitter_api(bot)

    if args:

        print "Destroy friendship with:", ", ".join(args)

        for user in args:

            try:

                old_friend = twitter_api.DestroyFriendship(user)

                if old_friend:

                    bot.respond(server, sender, target, "You are no longer friends with '%s'." % old_friend.GetName())

            except Exception, e:

                bot.respond(server, sender, target, "Could not remove friendship with '%s'." % user)

                print " - Could not remove friendship with '%s'." % user

        return True

    else:

        bot.respond(server, sender, target, "Not enough arguments.")

        return False

####

# The following declarations are only helper functions

##

def init_twitter_api(bot):

    """ [Internal] Initiate api"""

    global twitter_api

    global twitter_update_from

    twitter_api = twitter.Api(username = bot.settings.twitter_username,

                              password = bot.settings.twitter_password)

    if twitter_api:

        print " + Logged into twitter account '%s : %s'." % (bot.settings.twitter_username, bot.settings.twitter_password)

        twitter_api.SetUserAgent("KoFooBot")

        return True

    else:

        print " + Failed to login."

        return False

def send_to_channel(bot, server, msg):

    """ [Internal] Send string to channel with proper format."""

    line = ' '.join(msg.splitlines())

    line = re.sub(bot.regex_ahref, bot.rewrite_url, line)

    server.privmsg(bot.settings.twitter_channel, bot.convert_to_irc_string(line))