import praw
import config
import random
import os
import time

def bot_login():
    print "Logging in..."
    r = praw.Reddit(username = config.username,
            password = config.password,
            client_id = config.client_id,
            client_secret = config.client_secret,
            user_agent = "kemonomimi bot v0.4")
    print "Logged in!"

    return r

def run_bot(r, comments_replied_to, catgirls):
    #print catgirls
    print "Searching last 25 comments..."

    for comment in r.subreddit('KemonomimiCheerUpBot+anime_irl+animemes').comments(limit=25):
        if (("im sad" in comment.body.lower())  and \
            comment.id not in comments_replied_to and \
            comment.author != r.user.me()):

            # print "Im sad found in comment" + comment.id

            comment.reply(">Im sad"
                          "\n\n"
                          "---"
                          "\n\n"
                          "[Here](" + random.choice(catgirls) + ") is a "
                          "picture of a catgirl! Hopefully this will cheer you up!"
                          "\n\n"
                          "---"
                          "\n\n"
                          "I am a bot. For more info on me and how to use me, see r/KemonomimiCheerUpBot "
                          "Have I gone rogue? Reply \"!SHUTDOWN\" to stop me.")

            print "replied to comment " + comment.id

            comments_replied_to.append(comment.id)

            with open("comments_replied_to.txt", "a") as f:
                f.write(comment.id + "\n")

        if (("i'm sad" in comment.body.lower())  and \
            comment.id not in comments_replied_to and \
            comment.author != r.user.me()):

            # print "I'm sad found in comment" + comment.id

            comment.reply(">I'm sad"
                          "\n\n"
                          "---"
                          "\n\n"
                          "[Here](" + random.choice(catgirls) + ") is a "
                          "picture of a catgirl! Hopefully this will cheer you up!"
                          "\n\n"
                          "---"
                          "\n\n"
                          "I am a bot. For more info on me and how to use me, see r/KemonomimiCheerUpBot "
                          "Have I gone rogue? Reply \"!SHUTDOWN\" to stop me.")

            print "replied to comment " + comment.id

            comments_replied_to.append(comment.id)

            with open("comments_replied_to.txt", "a") as f:
                f.write(comment.id + "\n")

        if (("cheer me up" in comment.body.lower())  and \
            comment.id not in comments_replied_to and \
            comment.author != r.user.me()):

            # print "Cheer me up found in comment" + comment.id

            comment.reply(">Cheer me up"
                          "\n\n"
                          "---"
                          "\n\n"
                          "[Here](" + random.choice(catgirls) + ") is a "
                          "picture of a catgirl! Hopefully this will cheer you up!"
                          "\n\n"
                          "---"
                          "\n\n"
                          "I am a bot. For more info on me and how to use me, see r/KemonomimiCheerUpBot "
                          "Have I gone rogue? Reply \"!SHUTDOWN\" to stop me.")

            print "replied to comment " + comment.id

            comments_replied_to.append(comment.id)

            with open("comments_replied_to.txt", "a") as f:
                f.write(comment.id + "\n")


    print "searching inbox..."

    for reply in r.inbox.comment_replies(limit=3):
        if ("good bot" in reply.body.lower() and reply.id not in comments_replied_to):

            #print "Good bot found in reply " + reply.id

            reply.reply("[Thank You! :)](https://i.imgur.com/P3GRavv.gifv)")

            print "Thanked reply " + reply.id

            comments_replied_to.append(reply.id)

            with open("comments_replied_to.txt", "a") as f:
                f.write(reply.id+ "\n")

        # print "Searching inbox part 2..."

        if ("bad bot" in reply.body.lower() and reply.id not in comments_replied_to):

            print "TRASH WAIFU FOUND!!"

            reply.reply("[Your waifu](merriam-webster.com/dictionary/trash)")

            comments_replied_to.append(reply.id)

            with open("comments_replied_to.txt", "a") as f:
                f.write(reply.id+ "\n")

        # print "Searching inbox part 2 episode 1..."

        if ("who is your waifu" in reply.body.lower() and reply.id not in comments_replied_to):

            reply.reply("[Fixed Artillery-San](https://i.imgur.com/qrsKQCH.jpg)")

            print "bragged about waifu"

            comments_replied_to.append(reply.id)

            with open("comments_replied_to.txt", "a") as f:
                f.write(reply.id+ "\n")

        #This is to allow a shutdown when necessary.

        print "Checking for SHUTDOWN command"

        if ("!SHUTDOWN" in reply.body):
            reply.repl("")



    print "Time for a cat-nap!..."

    time.sleep(60)

# Load catgirls from a file (defaults to catgirls.txt)
def get_catgirls(fname='catgirls.txt'):

    print "collecting catgirls..."

    if not os.path.isfile(fname):
        raise FileNotFoundError("Can't get catgirls from " + fname)

    catgirls = []

    with open(fname) as catgirls_file:
        catgirls_contents = catgirls_file.read()

    catgirls = catgirls_contents.split("\n")
    catgirls = filter(None, catgirls)

    catgirls = [catgirl.strip() for catgirl in catgirls]

    return catgirls

def get_saved_comments():

    print "reading comments_replied_to..."

    if not os.path.isfile("comments_replied_to.txt"):
        comments_replied_to = []
    else:
        with open("comments_replied_to.txt", "r") as f:
            comments_replied_to = f.read()
            comments_replied_to = comments_replied_to.split("\n")
            comments_replied_to = filter(None, comments_replied_to)

    return comments_replied_to


if __name__ == '__main__':

    r = bot_login()
    catgirls = get_catgirls()
    comments_replied_to = get_saved_comments()

    while True:
        run_bot(r, comments_replied_to, catgirls)