#!/usr/bin/env python
# -*- coding: utf-8 -*-
import requests
import discord
import random
import datetime
import time
import re
import pickle
import logging
import thread

import traceback
import sys
import wikipedia
from subprocess import call

from dateutil.parser import parse

from discord.object import Object
from discord.channel import PrivateChannel
from ago import human
import simplejson as json
from collections import defaultdict
from nltk.tag import pos_tag
import wolframalpha
import sqlite3
from blackjack import Blackjack
import data

VERSION = 2.2

conn = sqlite3.connect('db.sqlite3')

credentials = 'creds.json'

muted_until = datetime.datetime.now()

client = discord.Client()
wolf = {}
logging.basicConfig(filename='hellsbot.log', level=logging.WARNING)

registered_commands = {'!help': 'do_help', '!commands': 'do_help',
                       '!shutup': 'do_shutup',
                       '!roll': 'do_roll',
                       '!lastseen': 'do_lastseen',
                       '!youtube': 'do_youtube',
                       '!image': 'do_image',
                       '!gif': 'do_gif',
                       '!gameslist': 'do_gameslist', '!gamelist': 'do_gameslist',
                       '!aliases': 'do_alias', '!alias': 'do_alias',
                       '!addalias': 'do_addalias',
                       '!games': 'do_games',
                       '!reloadbot': 'do_reload', '!restartbot': 'do_reload', '!rebootbot': 'do_reload',
                       '!whoplayed': 'do_whoplayed',
                       '!gimmecredits': 'do_gimmecredits', '!gimmecredit': 'do_gimmecredits',
                       '!grantcredits': 'do_grantcredits',
                       '!ticketrank': 'do_ticketrank',
                       '!startraffle': 'do_startraffle',
                       '!raffle': 'do_raffle',
                       '!buyticket': 'do_buyticket',
                       '!balance': 'do_balance',
                       '!slotsrules': 'do_slotsrules',
                       '!slots': 'do_slots',
                       '!hit': 'do_bj_hit', '!draw': 'do_bj_hit',
                       '!stay': 'do_bj_stay', '!stand': 'do_bj_stay',
                       '!bet': 'do_bj_bet',
                       '!msg': 'do_msg',
                       '!addfortune': 'do_addfortune',
                       '!fortune': 'do_fortune',
                       '!question': 'do_question',
                       '!addjoke': 'do_addjoke',
                       '!joke': 'do_joke',
                       '!secret': 'do_secret',
                       '!bemyirlwaifu': 'do_waifu',
                       'HILLARY 2016': 'do_hillary',
                       '!squid': 'do_squid',
                       '!stars': 'do_stars',
                       }


#####################
# Utility Functions
#####################
def log(message):
    try:
        logging.warning("{} - {}".format(datetime.datetime.now().strftime('%Y/%m/%d %H:%M:%S'), message))
    except:
        pass


def format_exception(e):
    exception_list = traceback.format_stack()
    exception_list = exception_list[:-2]
    exception_list.extend(traceback.format_tb(sys.exc_info()[2]))
    exception_list.extend(traceback.format_exception_only(sys.exc_info()[0], sys.exc_info()[1]))

    exception_str = "Traceback (most recent call last):\n"
    exception_str += "".join(exception_list)
    # Removing the last \n
    exception_str = exception_str[:-1]

    return exception_str


def leaders(xs, top=20):
    counts = defaultdict(int)
    for x in xs:
        counts[x] += 1
    return sorted(counts.items(), reverse=True, key=lambda tup: tup[1])[:top]


def byteify(input):
    if isinstance(input, dict):
        return {byteify(key): byteify(value) for key, value in input.iteritems()}
    elif isinstance(input, list):
        return [byteify(element) for element in input]
    elif isinstance(input, unicode):
        return input.encode('utf-8')
    else:
        return input


def search_youtube(query):
    query_string = {"search_query": query}
    r = requests.get("http://www.youtube.com/results", params=query_string)
    search_results = re.findall(r'href=\"\/watch\?v=(.{11})', r.content)
    log("http://www.youtube.com/watch?v=" + search_results[0])
    return "http://www.youtube.com/watch?v=" + search_results[0]


def search_google_images(query, animated=False):
    headers = {'User-Agent': "Mozilla/5.0 (X11; FreeBSD amd64; rv:12.0) Gecko/20100101 Firefox/12.0"}
    query_string = {"safe": "off", "tbm": "isch", "q": query}
    if animated:
        query_string = {"safe": "off", "tbm": "isch", "q": query, 'tbs': 'itp:animated'}

    r = requests.get("http://www.google.com/search", params=query_string, headers=headers)

    start_idx = r.content.find('imgurl=') + 7
    if start_idx > 0:
        search_result = r.content[start_idx:r.content.find('&', start_idx)]
        if '/revision/' in search_result:
            search_result = search_result[:search_result.find('/revision/')]
        if '%' in search_result:
            search_result = search_result[:search_result.find('%')]
        log(search_result)
        return search_result
    return "boo you fail.."


#################
# Client Events
#################
@client.event
def on_socket_raw_send(payload, binary=False):
    check_msg_queue()


@client.event
def on_status(member):
    for member in client.get_all_members():
        try:
            db_member = data.db_get_member(member.id)

            if not db_member:
                log("Creating new member: {}".format(member))
                data.db_create_member(member)
            else:
                data.db_update_member(member, db_member)

            check_msg_queue()
        except Exception as e:
            log("Exception: {}".format(format_exception(e)))
            pass


def check_msg_queue():
    messages = data.db_get_messages()
    if messages:
        for message in messages:
            try:
                member = data.db_get_member(filter(unicode.isalnum, message['message_to']))
                if member:
                    if message['message_to'] == member['discord_mention']:
                        log("Found Message: {} - {} - Status: {} Channel: {}".format(message['message_to'], member['discord_mention'], member['status'], message['channel']))
                        if member['status'] == 'online':
                            client.send_message(Object(message['channel']), '{}, {} asked me to tell you "{}"'.format(message['message_to'], message['message_from'], message['message']))
                            data.db_delete_sent_message(message['message_id'])
            except Exception as e:
                log("{}\nFailed to send message: {}".format(format_exception(e), message['message_id'],))
    return


def do_roll(client, message_parts, message):
    request = message_parts[0]
    count = 1
    dice = 100
    if request.strip() != '':
        if 'd' in request:
            dice_parts = request.split('d')
            if len(dice_parts) == 2:
                if dice_parts[0].isdigit() and dice_parts[1].isdigit():
                    count = int(dice_parts[0])
                    dice = int(dice_parts[1])
    if count > 100000000000000000:
        client.send_message(message.channel, '{} stop fucking around with those stupid numbers...'.format(message.author.mention()))
        return

    if count > 100:
        client.send_message(message.channel, '{} 100 is the largest number of dice I will roll.'.format(message.author.mention()))
        return
    roll_results = []
    for i in range(count):
        roll_results.append(random.randint(1, dice))
    out_string = '{} your roll {}d{}: {} = {}'.format(message.author.mention(), count, dice, '+'.join(str(r) for r in roll_results), sum(roll_results))
    client.send_message(message.channel, out_string)
    return


def do_lastseen(client, message_parts, message):
    username = ' '.join(message_parts).replace('@', '').lower()
    member = data.db_get_member(username=username)
    if member:
        out_string = ''
        if member['is_afk'] == 1:
            out_string = 'Went AFK at: {}\n'.format(member['afk_at'])
        elif member['status'] == 'offline':
            out_string = 'Currently Offline\n'
        else:
            out_string = 'Currently Online\n'
        out_string += 'Last Status: {} at {} which was {}\nPrevious Status: {}\n'.format(member['status'], 
            member['status_change_at'],
            human(datetime.datetime.strptime(member['status_change_at'], '%Y/%m/%d %H:%M:%S')),
            member['prev_status'])

        client.send_message(message.channel, 'Last Information on {}:\n{}'.format(byteify(username), out_string))
    else:
        client.send_message(message.channel, 'I don\'t have any data on {} yet {}'.format(byteify(username), message.author.mention()))
    return


def do_help(client, message_parts, message):
        client.send_message(message.channel, """{} Available Commands:
You can ask compound or questions and I will choose. Example: HellsBot Rui is a Faggot or Rui is a faggot?

User Info:
    !aliases <username> - Returns a list of all aliases a user has set for themselves.
    !addalias <alias> - Adds an alias to your list of aliases.
    !lastseen <username> - Returns info on when the user was last seen and their status.
Messages:
    !msg <username> in 5 minutes Tea is ready
    !msg <username> in 45 seconds Your finished masterbating
    !msg <username> in 2 hours The movie is over
    !msg <username> on 12/22/2015 Happy Birthday!
Games:
    !games <username> - Returns a list of games played for a username.
    !gameslist <count> - Returns a list of the top 20 games and the number of people who have played that game. if you pass a limit it will show that many games instead.
    !whoplayed <gamename> - Returns a list of players who have played the game.
Minigames:
    !gimmecredits - Gives you some extra credits in case you run out.
    !bet <amount> - Start a game of BlackJack.
    !slots <amount> - Spin the slot machine (max 10 credit bet).
    !hit - Draw a card
    !stand - Show the cards
""".format(message.author.mention()))
        client.send_message(message.channel, """
    !balance - Lists your current credits and tickets
    !buyticket - Purchases a raffle ticket for 100 credits
    !raffle - Shows information about the current raffle
    !ticketrank - Shows the percentage of the tickets on the system
Spam:
    !youtube <search term> - Returns the first video from the search results for the search term.
    !gif <search term> - Returns the first gif from the search results.
    !image <search term> - Returns the first image from the search results.
Stuff:
    !addfortune <fortune> - Adds a new fortune.
    !fortune - Returns your fortune.
    !addjoke <joke> - Adds a new joke (it can be multiline but must all be in a single message).
    !joke - Returns a random joke.
    !roll <1d20> - Roll X number of dice of size X. 1d20 returns 1 roll 1-20. 3d6 returns 3 rolls of 1-6 etc...
    !secret
    !shutup - disables all image / gif / youtube spam for 5 minutes
    !bemyirlwaifu""".format(message.author.mention()))


def do_shutup(client, message_parts, message):
    global muted_until
    muted_until = datetime.datetime.now() + datetime.timedelta(minutes=5)
    client.send_message(message.channel, 'All image / gif / youtube spam disabled for 5 minutes')
    return


def do_youtube(client, message_parts, message):
    if datetime.datetime.now() < muted_until:
        return
    client.send_message(message.channel, search_youtube(' '.join(message_parts)))
    return


def do_image(client, message_parts, message):
    if datetime.datetime.now() < muted_until:
        return
    client.send_message(message.channel, search_google_images(' '.join(message_parts)))
    return


def do_gif(client, message_parts, message):
    if datetime.datetime.now() < muted_until:
        return
    client.send_message(message.channel, search_google_images(' '.join(message_parts), True))
    return


def do_gameslist(client, message_parts, message):
    limit = 20
    if len(message_parts) > 0 and message_parts[0].isdigit():
        limit = int(message_parts[0])
    games_list = data.db_get_games_list(limit)

    out_string = ''
    for game in games_list:
        out_string += '    {} - {}\n'.format(game[1], byteify(game[0]))
    client.send_message(message.channel, 'The games I have seen people playing are: ')
    while len(out_string) > 0:
        client.send_message(message.channel, out_string[:1900])
        out_string = out_string[1900:]
    return


def do_alias(client, message_parts, message):
    if len(message_parts) > 0:
        username = ' '.join(message_parts).replace('@', '').lower()
        member = data.db_get_member(username=username)
    else:
        username = message.author.name
        member = data.db_get_member(message.author.id)
    if member:
        aliases = data.db_get_aliases(member['member_id'])
        if aliases:
            client.send_message(message.channel, '{} has the following aliases: {}'.format(byteify(username), byteify(', '.join(aliases))))
        else:
            client.send_message(message.channel, 'No known alises for {} yet {}'.format(byteify(username), message.author.mention()))
    else:
        client.send_message(message.channel, 'I don\'t know who you are speaking of {}!'.format(message.author.mention()))
    return


def do_addalias(client, message_parts, message):
    alias = ' '.join(message_parts)
    username = message.author.name.lower()
    member = data.db_get_member(username=username)
    if member:
        data.db_add_aliases(member['member_id'], alias)
        client.send_message(message.channel, '{} has been added to your aliases'.format(byteify(alias)))
    else:
        client.send_message(message.channel, 'Something horrible happened and it is all your fault, try logging out / on again or play a game. (or fuck off i dunno i\'m just an error message. Who am I to tell you how to run your life...)')
    return


def do_games(client, message_parts, message):
    if len(message_parts) > 0:
        username = ' '.join(message_parts).replace('@', '').lower()
    else:
        username = message.author.name
    games_list = data.db_get_games(username)
    if games_list:
        games = ', '.join(games_list)
        games = games.replace("FINAL FANTASY XIV", "**FINAL FANTASY XIV**")
        log(byteify(games))
        client.send_message(message.channel, 'I have seen {} playing: {}'.format(byteify(username), byteify(games)))
    else:
        client.send_message(message.channel, 'I don\'t have any data on {} yet {}'.format(byteify(username), message.author.mention()))


def do_reload(client, message_parts, message):
    if message.author.id != '78767557628133376':
        client.send_message(message.channel, "You shouldn't be calling this. You are clearly looking to piss Hellsbreath off.")
        return
    call(["service", "hellsbot", "restart"])


def do_whoplayed(client, message_parts, message):
    game_name = ' '.join(message_parts)
    member_list = data.db_get_whoplayed(game_name)
    if not member_list:
        client.send_message(message.channel, 'I don\'t have any data on {} yet {}'.format(byteify(game_name), message.author.mention()))
    else:
        out_string = ''
        for member in member_list:
            out_string += '    {} - {}\n'.format(byteify(member[1]), byteify(member[0]))
        client.send_message(message.channel, 'Below is a list of people who have played {} and the number of times they have launched the game:'.format(byteify(game_name),))
        while len(out_string) > 0:
            client.send_message(message.channel, out_string[:1900])
            out_string = out_string[1900:]
    return


def do_gimmecredits(client, message_parts, message):
    member = data.db_get_member(message.author.id)
    if not member:
        client.send_message(message.author, "There was a problem looking up your information.")
    else:
        credits = data.db_get_credit(member['member_id'])
        if credits < 5:
            amount = random.randint(5, 50)
            data.db_update_credit(member['member_id'], amount)
            client.send_message(message.author, "You have been given {} credits.".format(amount,))
        else:
            client.send_message(message.author, "You already have credits. Stop begging.")
    return


def do_grantcredits(clients, message_parts, message):
    if message.author.id != '78767557628133376':
        client.send_message(message.channel, "You are not Hellsbreath. Use !gimmecredits to get a few extra if you run out.")
        return
    members = data.db_get_all_members()
    if len(members) < 0:
        client.send_message(message.channel, "There was a problem looking up your information.")
    else:
        for member in members:
            credits = data.db_get_credit(member['member_id'])
            if credits < 100:
                data.db_update_credit(member['member_id'], 100)
                client.send_message(message.channel, "{} has been given {} credits.".format(member['member_name'], 100))
    return


def do_ticketrank(clients, message_parts, message):
    members = data.db_get_all_members()
    if len(members) < 0:
        client.send_message(message.channel, "There was a problem looking up your information.")
    else:
        ticket_count = 0
        for member in members:
            if member['discord_id'] != '78767557628133376':
                ticket_count += member['tickets']
        if ticket_count == 0:
            client.send_message(message.channel, "No Tickets have been sold for this raffle.")
            return

        out_string = ""
        for member in members:
            if member['tickets'] > 0 and member['discord_id'] != '78767557628133376':
                percent = (float(member['tickets']) / float(ticket_count)) * 100.0
                out_string += "{} - {}%\n".format(byteify(member['member_name']), int(percent))

        client.send_message(message.channel, out_string)
    return


def do_startraffle(client, message_parts, message):
    if message.author.id != '78767557628133376':
        client.send_message(message.channel, "You are not Hellsbreath. Go die in an especially hot fire.")
        return
    members = data.db_get_all_members()
    if len(members) < 0:
        client.send_message(message.channel, "There was a problem looking up your information.")
    else:
        ticket_count = 0
        for member in members:
            if member['discord_id'] != '78767557628133376':
                ticket_count += member['tickets']
        if ticket_count == 0:
            client.send_message(message.channel, "No Tickets have been sold for this raffle.")
            return

        out_string = "The final standings are as follows: \n\n"
        ticket_reel = []
        for member in members:
            if member['tickets'] > 0 and member['discord_id'] != '78767557628133376':
                ticket_reel += [member['discord_mention']] * member['tickets']
                percent = (float(member['tickets']) / float(ticket_count)) * 100.0
                out_string += "{} - {}%\n".format(byteify(member['member_name']), int(percent))
        if len(ticket_reel) > 0:
            winner = random.choice(ticket_reel)
            while winner in ticket_reel:
                ticket_reel.remove(winner)
            client.send_message(message.channel, "\n\n\n**The winner is....\n\n{}!**".format(byteify(winner)))
        if len(ticket_reel) > 0:
            second = random.choice(ticket_reel)
            while second in ticket_reel:
                ticket_reel.remove(second)
            time.sleep(0.5)
            client.send_message(message.channel, "\n\n*2nd Place:* {}".format(byteify(second)))
        if len(ticket_reel) > 0:
            third = random.choice(ticket_reel)
            time.sleep(0.5)
            client.send_message(message.channel, "\n*3rd Place:* {}".format(byteify(third)))
    return


def do_raffle(client, message_parts, message):
    client.send_message(message.channel, """Current Raffle Item:

**1st Place**

Game: **Lightning Returns**
Description:
*Lightning Returns is the concluding chapter of the Final Fantasy XIII saga and series heroine Lightning's final battle. The grand finale of the trilogy brings a world reborn as well as free character customization and stunning action based battles.*

http://store.steampowered.com/app/345350/

**2nd Place**

2 Random Steam Keys

**3rd Place**

1 Random Steam Key

Raffle Date: **2/9/2016 00:00:00 (ish)**

You will be contacted if you win. To win you must purchase tickets with the !buyticket command for 100 credits.
You can get extra credits by playing !slots <amount> and !bet <amount> on BlackJack.

**Disclaimer:** *If anything should go wrong you get no refund and there is no guarantee or warrantee on anything. 1 prize per person no matter how many tickets you have.*
""")
    return


def do_buyticket(client, message_parts, message):
    member = data.db_get_member(message.author.id)
    if not member:
        client.send_message(message.author, "There was a problem looking up your information.")
    else:
        result, response = data.db_buy_ticket(member['member_id'], 1)
        if not result:
            client.send_message(message.author, response)
            return

        credits = data.db_get_credit(member['member_id'])
        client.send_message(message.author, "Raffle ticket purchased. Tickets: {} Credits: {}".format(response, credits))
    return


def do_balance(client, message_parts, message):
    member = data.db_get_member(message.author.id)
    if not member:
        client.send_message(message.author, "There was a problem looking up your information.")
    else:
        client.send_message(message.author, "Credits: {}\nTickets: {}".format(member['credits'], member['tickets']))
    return


def do_slotsrules(client, message_parts, message):
    client.send_message(message.channel, """Paying Combinations:

:moneybag:\t:moneybag:\t:moneybag:\t\t\t  pays\t250
:bell:\t:bell:\t:bell:/:moneybag:\tpays\t20
:diamonds:\t:diamonds:\t:diamonds:/:moneybag:\tpays\t14
:spades:\t:spades:\t:spades:/:moneybag:\tpays\t10
:cherries:\t:cherries:\t:cherries:\t\t\t  pays\t7
:cherries:\t:cherries:\t  -\t\t\t\t pays\t5
:cherries:\t  -\t\t  -\t\t\t\t pays\t2

:black_square_button:\tblank space

All payouts are in credits. Each pull costs 1 credit.

Max Bet: 10 credits

To Play: !slots <bet>""")
    return


def do_slots(client, message_parts, message):
    member = data.db_get_member(message.author.id)
    if not member:
        client.send_message(message.author, "There was a problem looking up your information.")
        return
    elif type(message.channel) is not discord.channel.PrivateChannel:
        client.send_message(message.author, "You must make all bets / gaming via private message.")
        return
    bet_amount = message.content[7:]
    log("Member: {} Slots Bet: {}".format(member['member_name'], bet_amount))

    if not bet_amount.isdigit() or int(bet_amount) > 10 or int(bet_amount) < 0:
        client.send_message(message.author, "Please provide a bet amount up to 10 credits.")
        return
    result, error_message = data.db_update_credit(member['member_id'], -int(bet_amount))
    if not result:
        client.send_message(message.author, error_message)
        return

    reel1 = [':black_square_button:', ':black_square_button:', ':black_square_button:', ':black_square_button:', ':black_square_button:', ':black_square_button:', ':black_square_button:', ':black_square_button:', ':black_square_button:', ':black_square_button:', ':cherries:', ':cherries:', ':cherries:', ':spades:', ':spades:', ':spades:', ':hearts:', ':hearts:', ':diamonds:', ':bell:', ':moneybag:']
    reel2 = [':black_square_button:', ':black_square_button:', ':black_square_button:', ':black_square_button:', ':black_square_button:', ':black_square_button:', ':black_square_button:', ':black_square_button:', ':black_square_button:', ':black_square_button:', ':cherries:', ':cherries:', ':cherries:', ':spades:', ':spades:', ':spades:', ':hearts:', ':hearts:', ':diamonds:', ':bell:', ':moneybag:']
    reel3 = [':black_square_button:', ':black_square_button:', ':black_square_button:', ':black_square_button:', ':black_square_button:', ':black_square_button:', ':black_square_button:', ':black_square_button:', ':black_square_button:', ':black_square_button:', ':cherries:', ':cherries:', ':cherries:', ':spades:', ':spades:', ':spades:', ':hearts:', ':hearts:', ':diamonds:', ':bell:', ':moneybag:']
    #:moneybag: :bell: :diamonds: :spades:  :hearts: 
    val1 = random.choice(reel1)
    val2 = random.choice(reel2)
    val3 = random.choice(reel3)
    winnings = 0
    reels = [val1, val2, val3]
    cherries = reels.count(":cherries:")
    spades = reels.count(":spades:")
    diamonds = reels.count(":diamonds:")
    bells = reels.count(":bell:")
    moneybags = reels.count(":moneybag:")
    out_string = ""
    if moneybags == 3:
        out_string += "JACKPOT!!"
        winnings = 250
    elif bells == 3 or (bells == 2 and moneybags == 1):
        winnings = 20
    elif diamonds == 3 or (diamonds == 2 and moneybags == 1):
        winnings = 14
    elif spades == 3 or (spades == 2 and moneybags == 1):
        winnings = 10
    elif cherries == 3:
        winnings = 7
    elif cherries == 2:
        winnings = 5
    elif cherries == 1:
        winnings = 2
    out_string = """| {} | {} | {} |\n\n""".format(val1, val2, val3)
    winnings = int(winnings * int(bet_amount))
    if winnings > 0:
        out_string += "You Won! Total Winnings: {}".format(winnings,)
        log("Member: {} Wins: {}".format(member['member_name'], winnings))

    else:
        out_string += "You lose. Total Winnings: {}".format(winnings,)
    if winnings > 0:
        result, error_message = data.db_update_credit(member['member_id'], winnings)
        if not result:
            client.send_message(message.author, error_message)
            return

    credits = data.db_get_credit(member['member_id'])
    out_string += "\nCredits: {}".format(credits)

    client.send_message(message.author, out_string)
    return


def do_bj_hit(client, message_parts, message):
    member = data.db_get_member(message.author.id)
    if not member:
        client.send_message(message.author, "There was a problem looking up your information.")
    elif type(message.channel) is not discord.channel.PrivateChannel:
        client.send_message(message.author, "You must make all bets / gaming via private message.")
    else:
        state = data.db_get_minigame_state(member['member_id'], 'blackjack')
        if state:
            out_string = ""
            bj = pickle.loads(str(state))
            bj.draw()
            out_string += bj.print_hand()
            actions = bj.get_actions()
            if len(actions) > 0:
                out_string += '\n\nPlease choose an option [{}]\n'.format(', '.join(actions))
                data.db_add_minigame(member['member_id'], 'blackjack', bj.serialize())
            else:
                win, response = bj.is_win()
                out_string += "\n\n" + bj.print_hand(show_dealer=True)
                out_string += "\n\n" + response.format(win,)
                data.db_delete_minigame_state(member['member_id'], 'blackjack')
                log("{} - {}".format(member['member_name'], response.format(win,)))
                result, error_message = data.db_update_credit(member['member_id'], int(win))
                if not result:
                    client.send_message(message.author, error_message)
                    return

                credits = data.db_get_credit(member['member_id'])
                out_string += "\nCredits: {}".format(credits)

            client.send_message(message.author, out_string)

        else:
            client.send_message(message.author, "You must start a game with !bet before you can ask for a new card.")
    return


def do_bj_stay(client, message_parts, message):
    member = data.db_get_member(message.author.id)
    if not member:
        client.send_message(message.author, "There was a problem looking up your information.")
    elif type(message.channel) is not discord.channel.PrivateChannel:
        client.send_message(message.author, "You must make all bets / gaming via private message.")
    else:
        state = data.db_get_minigame_state(member['member_id'], 'blackjack')
        if state:
            out_string = ""
            bj = pickle.loads(str(state))

            win, response = bj.is_win()
            out_string += "\n\n" + bj.print_hand(show_dealer=True)
            out_string += "\n\n" + response.format(win,)
            data.db_delete_minigame_state(member['member_id'], 'blackjack')
            log("{} - {}".format(member['member_name'], response.format(win,)))
            result, error_message = data.db_update_credit(member['member_id'], int(win))
            if not result:
                client.send_message(message.author, error_message)
                return
            credits = data.db_get_credit(member['member_id'])
            out_string += "\nCredits: {}".format(credits)
            client.send_message(message.author, out_string)
    return


def do_bj_bet(client, message_parts, message):
    member = data.db_get_member(message.author.id)
    if not member:
        client.send_message(message.author, "There was a problem looking up your information.")
    elif type(message.channel) is not discord.channel.PrivateChannel:
        client.send_message(message.author, "You must make all bets / gaming via private message.")
    else:
        state = data.db_get_minigame_state(member['member_id'], 'blackjack')
        if state:
            client.send_message(message.author, "You are already playing a game!")

            out_string = ""
            bj = pickle.loads(str(state))
            out_string += bj.print_hand()
            actions = bj.get_actions()
            if len(actions) > 0:
                out_string += '\n\nPlease choose an option [{}]\n'.format(', '.join(actions))
                data.db_add_minigame(member['member_id'], 'blackjack', bj.serialize())
            else:
                win, response = bj.is_win()
                out_string += "\n\n" + bj.print_hand(show_dealer=True)
                out_string += "\n\n" + response.format(win,)
                data.db_delete_minigame_state(member['member_id'], 'blackjack')
                log("{} - {}".format(member['member_name'], response.format(win,)))
                result, error_message = data.db_update_credit(member['member_id'], int(win))
                if not result:
                    client.send_message(message.author, error_message)
                    return
                credits = data.db_get_credit(member['member_id'])
                out_string += "\nCredits: {}".format(credits)

            client.send_message(message.author, out_string)

            return

        out_string = ""
        bet_amount = message.content[5:]
        log("Member: {} Bet: {}".format(member['member_name'], bet_amount))

        if not bet_amount.isdigit():
            client.send_message(message.author, "Please provide a bet amount. !bet 10")
            return

        result, error_message = data.db_update_credit(member['member_id'], -int(bet_amount))
        if not result:
            client.send_message(message.author, error_message)
            return
        out_string += "Welcome to BlackJack! :flower_playing_cards: You have placed a bet of: {}\n".format(bet_amount)
        bj = Blackjack(bet_amount)
        out_string += bj.print_hand()
        actions = bj.get_actions()
        if len(actions) > 0:
            out_string += '\n\nPlease choose an option [{}]\n'.format(', '.join(actions))
            data.db_add_minigame(member['member_id'], 'blackjack', pickle.dumps(bj))
        else:
            win, response = bj.is_win()
            out_string += "\n\n" + bj.print_hand(show_dealer=True)
            out_string += "\n\n" + response.format(win,)
            data.db_delete_minigame_state(member['member_id'], 'blackjack')
            log("{} - {}".format(member['member_name'], response.format(win,)))
            result, error_message = data.db_update_credit(member['member_id'], int(win))
            if not result:
                client.send_message(message.author, error_message)
                return
            credits = data.db_get_credit(member['member_id'])
            out_string += "\nCredits: {}".format(credits)

        client.send_message(message.author, out_string)
    return


def do_msg(client, message_parts, message):
    channel = message.channel
    author = message.author
    username = ''
    try:
        # TODO: Switch message_bits with message_parts
        message_bits = message.content.split(" ")
        msg_datetime = datetime.datetime.now()
        msg_idx = 2
        if message_bits[2] == 'in' and message_bits[3].isdigit():
            time = int(message_bits[3])
            msg_idx = 4
            if message_bits[4].startswith('sec'):
                msg_datetime = msg_datetime + datetime.timedelta(seconds=time)
                msg_idx = 5
            elif message_bits[4].startswith('hour'):
                msg_datetime = msg_datetime + datetime.timedelta(hours=time)
                msg_idx = 5
            else: # minutes by default
                msg_datetime = msg_datetime + datetime.timedelta(minutes=time)
                msg_idx = 5
        elif message_bits[2] == 'on':
            try:
                tmp_date = parse(message_bits[3])
                msg_datetime = tmp_date
                msg_idx = 4
            except ValueError:
                client.send_message(channel, 'Your shitty message has been rejected {}. Next time learn how to date...MM\\DD\\YYYY'.format(message.author.mention()))
                return

        username = message_bits[1]
        user_mention = ''
        # TODO: have it look in the database. Do this AFTER on startup we add all users.
        for member in client.get_all_members():
            if username.lower() == member.name.lower():
                user_mention = member.mention()
                user_id = member.id
        if user_mention == '':
            client.send_message(channel, 'Your shitty message has been rejected {}. That user does not exist.'.format(message.author.name))
            return

        msg_text = byteify(' '.join(message_bits[msg_idx:]))
        message = {'user_id': user_id, 'channel': channel.id, 'delivery_time': msg_datetime.strftime('%Y/%m/%d %H:%M:%S'), 'message': msg_text}
        log("Message: %s" % byteify(message))
        data.db_add_message(msg_text, msg_datetime.strftime('%Y-%m-%d %H:%M:%S'), channel.id, author.mention(), user_mention, user_id)
        # print("Data: %s" % data)
        #test_ch = Object(channel.id)
        #client.send_message(test_ch, 'Test Message {}.'.format(author))
    except Exception as e:
        client.send_message(channel, 'Your shitty message has been rejected {}. {}'.format(author.name, format_exception(e)))
        return
    if msg_datetime < datetime.datetime.now():
        client.send_message(channel, '{} your message will be delivered to {} as soon as they are available.'.format(author.name, user_mention))
    else:
        client.send_message(channel, '{} your message will be delivered to {} {}.'.format(author.name, user_mention, human(msg_datetime)))            
    check_msg_queue()
    return


def do_addfortune(client, message_parts, message):
    try:
        fortune = message.content[9:]
        if 'aa737a5846' in fortune:
            client.send_message(message.channel, '{} you stop it, you are a pedofile, stop looking at little girls.'.format(message.author.mention()))
            return
        date_added = datetime.datetime.now().strftime('%Y/%m/%d %H:%M:%S')
        c = conn.cursor()
        c.execute("INSERT INTO fortunes (fortune, date_added) VALUES (?, ?)", (fortune, date_added))
        conn.commit()
        log("Added fortune")
    except Exception as e:
        log(e.message)
        client.send_message(message.channel, 'Your shitty fortune has been rejected {}.'.format(message.author.mention()))
        return
    client.send_message(message.channel, 'Your shitty fortune has been added {}.'.format(message.author.mention()))
    return


def do_fortune(client, message_parts, message):
    fortune = None
    try:
        c = conn.cursor()
        # TODO: Move this shit to data
        fortune = c.execute("SELECT fortune FROM fortunes ORDER BY RANDOM() LIMIT 1;").fetchone()[0]
        log(fortune)
    except Exception as e:
        log(e)
        pass
    if not fortune:
        client.send_message(message.channel, 'Try adding a fortune with "!addfortune <fortune>" {}!'.format(message.author.mention()))
    else:
        client.send_message(message.channel, '{} Your fortune is... {}'.format(message.author.mention(), byteify(fortune)))
    return


def do_question(client, message_parts, message):
    question = message.content[10:]
    if "is it gay" in question:
        client.send_message(message.channel, 'Yes {}, it is gay.'.format(message.author.mention()))
        return
    res = wolf.query(question)
    try:
        if len(res.pods):
            pod_text = []
            for pod in res.pods:
                if pod.text:
                    pod_text.append(pod.text)
            client.send_message(message.channel, '{} {}.'.format(message.author.mention(), byteify("\n".join(pod_text)[:1990])))
        else:
            tagged_sent = pos_tag(question.replace('?', '').split())
            proper_nouns = [word for word, pos in tagged_sent if pos == 'NNP']
            wiki_search = " ".join(proper_nouns)
            if wiki_search.strip() != "":
                log("Looking up {}".format(wiki_search))
                wiki_out = wikipedia.summary(wiki_search, sentences=3)
                client.send_message(message.channel, '{} {}.'.format(message.author.mention(), byteify(wiki_out)))
            else:
                client.send_message(message.channel, 'I don\'t know {}.'.format(message.author.mention()))
        return
    except Exception as e:
        log(format_exception(e))
        client.send_message(message.channel, 'I don\'t know {}.'.format(message.author.mention()))
        return


def do_addjoke(client, message_parts, message):
    try:
        joke = message.content[9:]
        if 'aa737a5846' in joke:
            client.send_message(message.channel, '{} you stop it, you are a pedofile, stop looking at little girls.'.format(message.author.mention()))
            return
        c = conn.cursor()
        c.execute("INSERT INTO jokes (joke) VALUES (?)", (joke,))
        conn.commit()
        log("Added joke")
    except Exception as e:
        log(e.message)
        client.send_message(message.channel, 'Your shitty joke has been rejected {}.'.format(message.author.mention()))
        return
    client.send_message(message.channel, 'Your shitty joke has been added {}.'.format(message.author.mention()))
    return


def do_joke(client, message_parts, message):
    joke = None
    try:
        c = conn.cursor()
        joke = c.execute("SELECT joke FROM jokes ORDER BY RANDOM() LIMIT 1;").fetchone()[0]
        log(joke)
    except Exception as e:
        log(e)
        pass
    if not joke:
        client.send_message(message.channel, 'Try adding a joke with "!addjoke <joke>" {}!'.format(message.author.mention()))
    else:
        client.send_message(message.channel, '{} {}'.format(message.author.mention(), byteify(joke)))
    return


def do_secret(client, message_parts, message):
    client.send_message(message.channel, 'git gud {}! My source is here: http://git.savsoul.com/barry/discordbot\nVersion: {}'.format(message.author.mention(), VERSION))
    return


def do_waifu(client, message_parts, message):
    client.send_message(message.channel, 'http://orig13.deviantart.net/b25e/f/2014/175/3/d/no_waifu_no_laifu_by_imtheonenexttome-d7nsx3b.gif {}!'.format(message.author.mention()))
    return


def do_hillary(client, message_parts, message):
    client.send_message(message.channel, ':bomb: Ohhhhhh, now you done it...:bomb:'.format(message.author.mention()))
    return


def do_squid(client, message_parts, message):
    client.send_message(message.channel, 'くコ:彡         くコ:彡            くコ:彡          くコ:彡')
    return


def do_stars(client, message_parts, message):
    client.send_message(message.channel, '✮═━┈　　✰═━┈　　✮═━┈　　✰═━┈　　✮═━┈　　✰═━┈　　✮═━┈　　✰═━┈　　✮═━┈　　✰═━┈　✰═━┈┈　✰═━┈┈')
    return


@client.event
def on_message(message):
    #print message.content
    #print message.author
    #print client.user
    # we do not want the bot to reply to itself
    if message.author == client.user:
        return

    message_parts = message.content.split(" ")
    for command, method in registered_commands.iteritems():
        if message_parts[0] == command:
            try:
                thread.start_new_thread(globals()[method], (client, message_parts[1:], message))
            except Exception as e:
                log("{} - {}".format(format_exception(e), e.message))
            return

    if message.content.lower().startswith(client.user.name.lower()):
        log('Someone is talking to %s' % (client.user.name.lower(),))
        if ' or ' in message.content:
            questions = message.content[len(client.user.name)+1:].replace('?', '').split(' or ')
            client.send_message(message.channel, '{} I choose: {}'.format(message.author.mention(), random.choice(questions).encode('utf-8', errors='ignore')))


@client.event
def on_ready():
    log('Logged in as')
    log(client.user.name)
    log(client.user.id)
    log('------')
    for member in client.get_all_members():
        if member.id == '78767557628133376':
            client.send_message(member, "Bot Started")
    check_msg_queue()

retries = 0
while retries < 1000:
    try:
        json_data = open(credentials).read()
        creds = json.loads(json_data)
        wolf = wolframalpha.Client(creds['wolframkey'])
        client.login(creds['username'], creds['password'])
        client.run()
    except KeyboardInterrupt:
        conn.close
        quit()
    except Exception as e:
        retries += 1
        log("Shit I crashed: {}\nRetry {}".format(e, retries))
        #time.sleep(1)