#!/usr/bin/env python3

import sqlite3
import sys
import datetime

import discord

client = discord.Client()

@client.async_event
async def on_ready():
    if __name__ == '__main__':
        print("Logged in")
        if len(sys.argv) == 2:
            await make_logs(sys.argv[1],sys.argv[1])
        elif len(sys.argv) > 2:
            await make_logs(sys.argv[1],sys.argv[2])
        await client.logout()

async def make_logs(guildid, filename):
    database = None
    def scrub(s):
        while "-" in s:
            s = s.replace("-", "")
        return s
    def check_table_exists(tablename):
        dbcur = database.cursor()
        dbcur.execute("""
            SELECT name FROM sqlite_master WHERE type='table' AND name='{0}';
            """.format(str(tablename).replace('\'', '\'\'')))
        if dbcur.fetchone():
            dbcur.close()
            return True

        dbcur.close()
        return False

    guild = client.get_guild(int(sys.argv[1]))
    database = sqlite3.connect("{}.sqlite".format(filename))

    cursor = database.cursor()
    if check_table_exists('channels'):
        cursor.execute("""DROP TABLE channels""")
    cursor.execute("""CREATE TABLE channels(cid INTEGER, position INTEGER, name TEXT, topic TEXT, type TEXT)""")
    cursor.close()
    database.commit()
    for channel in guild.channels:
        if isinstance(channel, discord.TextChannel):
            cursor = database.cursor()
            cursor.execute(
                """INSERT INTO channels(cid, position, name, topic, type) VALUES (?, ?, ?, ?, ?)""",
                (channel.id, channel.position, channel.name, str(channel.topic), 'discord.TextChannel')
            )
            cursor.close()
            database.commit()
            if channel.permissions_for(guild.get_member(client.user.id)).read_message_history:
                sys.stdout.write("Logging {0}: Counting".format(channel.name))
                sys.stdout.flush()
                cursor = database.cursor()
                after = datetime.datetime(2015,3,1)
                if check_table_exists(channel.name):
                    cursor.execute("""SELECT timestamp FROM `{0}` ORDER BY timestamp DESC LIMIT 1""".format(channel.name))
                    try:
                        after = datetime.datetime.strptime(cursor.fetchone()[0], "%Y-%m-%d %H:%M:%S.%f")
                    except TypeError:
                        pass
                else:
                    cursor.execute("""CREATE TABLE `{0}`(uid INTEGER, uname TEXT, mid INTEGER, message TEXT, files TEXT, timestamp TEXT)""".format(channel.name))
                database.commit()
                count = 0
                msg_c = 0
                async for message in channel.history(limit=None,after=after):
                    msg_c += 1
                print("\rLogging {0}: 0/{1}         ".format(channel.name, msg_c), end="")
                async for message in channel.history(limit=None,after=after):
                    at = ",".join([i.url for i in message.attachments])
                    cursor.execute("""
                        INSERT INTO `{0}`(uid, uname, mid, message, files, timestamp)
                        VALUES (?, ?, ?, ?, ?, ?)""".format(channel.name), (message.author.id, message.author.name, message.id, message.content, at, message.created_at))
                    count += 1
                    if count % 200 == 0:
                        database.commit()
                        print("\rLogging {0}: {1}/{2}".format(channel.name, count, msg_c), end="")
                database.commit()
                print("\rLogging {0}: [DONE]            ".format(channel.name))
        elif isinstance(channel, discord.VoiceChannel):
            cursor = database.cursor()
            cursor.execute(
                """INSERT INTO channels(cid, position, name, topic, type) VALUES (?, ?, ?, ?, ?)""",
                (channel.id, channel.position, channel.name, '', 'discord.VoiceChannel')
            )
            cursor.close()
            database.commit()
    print("LOGS FINISHED!")

if __name__ == '__main__':
    #bot=False may have to be changed if you are using a bot
    client.run(open('token.txt','r').read().split('\n')[0], bot=False)