import praw
import config
import time
import os
import random
import re

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

def run_bot():
    global r
    global comments_dealt_with
    for subreddit in config.subreddits:
        didsomething = False
        print ("Obtaining 25 comments in " + subreddit)
        for comment in r.subreddit(subreddit).comments(limit=25):
            thecomment = comment.body
            flair = next(r.subreddit(config.flair_subreddit).flair(comment.author))
            if flair["flair_text"] is None:
                    r.subreddit(config.flair_subreddit).flair.set(comment.author, "1000")
                    flair["flair_text"] = "1000"
                    print ("New user, setting balance to 1000")
            try:
                currentMoney = int(flair["flair_text"])
            except ValueError:
                print ("Flair is not an integer, ignoring, flair = " + flair["flair_text"])
                comments_dealt_with.append(comment.id)
                with open ("comments_dealt_with.txt", "a") as a:
                    a.write(comment.id + "\n")
                continue


            if ("!status" in comment.body and comment.id not in comments_dealt_with and comment.author != r.user.me()):
                print ("'!STATUS' FOUND IN COMMENT " + comment.id)
                comment.reply(config.status_text + config.footer_text)
                didsomething = True

            if ("!resetbalance" in comment.body and comment.id not in comments_dealt_with and comment.author != r.user.me()):
                print ("resetbalance called, resetting balance")
                r.subreddit(config.flair_subreddit).flair.set(comment.author, "1000")
                comment.reply(config.reset_text + config.footer_text)
                didsomething = True

            if ("!checkbalance" in comment.body and comment.id not in comments_dealt_with and comment.author != r.user.me()):
                print ("checkbalance called, checking balance")
                comment.reply(config.checkbal_text(currentMoney) + config.footer_text)
                didsomething = True

            try:
                if "!coinflip" in thecomment:
                    betSize, betChoice = re.match(r".*!coinflip \[(\d+)] \(([HT])\)", thecomment).groups()
                    betSize = int(betSize)
                    if currentMoney < betSize:
                        comment.reply(config.low_money_text(currentMoney) + config.footer_text)
                        print ("Not enough coins, replied")
                        didsomething = True
                    else:
                        flip = random.choice(["Heads", "Tails"])
                        if flip(0) == betChoice:
                            r.subreddit(config.flair_subreddit).flair.set(comment.author, str(currentMoney + betSize))
                            comment.reply(config.win_text(flip, betSize, currentMoney + betSize) + config.footer_text)
                        else:
                            r.subreddit(config.flair_subreddit).flair.set(comment.author, str(currentMoney - betSize))
                            comment.reply(config.lose_text(flip, betSize, currentMoney - betSize) + config.footer_text)
                        didsomething = True
            except (AttributeError, ValueError):
                print ("Invalid formatting, ignoring")
                comment.reply("Incorrect formatting, try !coinflip [amount] (H or T)" + config.footer_text)
                didsomething = True

            if didsomething:
                print ("Replied to comment " + comment.id)
                comments_dealt_with.append(comment.id)
                with open ("comments_dealt_with.txt", "a") as a:
                    a.write(comment.id + "\n")

    print ("Sleeping for 10 seconds...")
    time.sleep(10)

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


bot_login()
get_saved_comments()
print (comments_dealt_with)

while True:
    run_bot()