import praw
import os.path
import json
import time

# Create default config/config.py if it doesn't exist and exit to prompt manual editing
import sys

default_config_path = os.path.join('config', 'config.py')
def env_or_default(var, default):
    return os.environ.get(var, default)

def write_config_from_env():
    os.makedirs('config', exist_ok=True)
    with open(default_config_path, 'w') as f:
        f.write(
            f'username = "{env_or_default("USERNAME", "")}"\n'
            f'password = "{env_or_default("PASSWORD", "")}"\n'
            f'client_id = "{env_or_default("CLIENT_ID", "")}"\n'
            f'client_secret = "{env_or_default("CLIENT_SECRET", "")}"\n'
            f'user_agent = "{env_or_default("USER_AGENT", "Flair Timer Comment Bot" )}"\n'
            '\n'
            f'subreddit = "{env_or_default("SUBREDDIT", "")}"\n'
            f'flair_text = "{env_or_default("FLAIR_TEXT", "Waiting for OP")}"\n'
            f'interval = {env_or_default("INTERVAL", "30")}\n'
            f'hours = {env_or_default("HOURS", "48")}\n'
            f'messagetitle = "{env_or_default("MESSAGETITLE", "Modmail Notification")}"\n'
            f'searchlimit = {env_or_default("SEARCHLIMIT", "600")}\n'
        )
    print(f"Configuration file auto-populated from environment variables at {default_config_path}.")

# Check if config file is missing or empty
populate_config = False
if not os.path.exists(default_config_path):
    populate_config = True
else:
    try:
        with open(default_config_path, 'r') as f:
            content = f.read().strip()
            if not content:
                populate_config = True
    except Exception:
        populate_config = True

if populate_config:
    write_config_from_env()
    sys.exit(0)

if not os.path.exists(default_config_path):
    sys.exit(0)
import config
 
def authentication():
    print ("Authenticating...")
    reddit = praw.Reddit(username = config.username,
            password = config.password,
            client_id = config.client_id,
            client_secret = config.client_secret,
            user_agent = config.user_agent)
    print ("Authenticated as {}.".format(reddit.user.me()))
    return reddit
 
def main(reddit, posts: dict):
    while True:
        for submission in reddit.subreddit(config.subreddit).new(limit=config.searchlimit):
            if not submission.saved:
                if submission.id not in posts.keys() and submission.link_flair_text == config.flair_text:
                    posts[submission.id] = time.time()
                    print(f"Post {submission} has been flaired {config.flair_text}")
                if submission.id in posts.keys() and submission.link_flair_text != config.flair_text:
                    posts.pop(submission.id)
                    print(f"Post {submission} has been unflaired {config.flair_text}")
 
        for submission in posts:
                if time.time() > posts[submission] + (config.hours * 60 * 60):
                    posts.pop(submission)
                    reddit.submission(submission).save()
                    data = {
                        "subject": config.messagetitle,
                        "text": f"It has been {config.hours/24} day/s since this was flaired [{config.flair_text}](https://old.reddit.com{reddit.submission(submission).permalink})",
                        "to": "/r/{}".format(config.subreddit),
                    }
                    reddit.post("api/compose/", data=data)
                    print(f"Post {submission} has been flaired {config.flair_text} for {config.hours * 60} minutes, sent modmail")
                    break
 
        save_posts(posts)
        time.sleep(config.interval)
 
def load_posts():
    if not os.path.exists("config/posts.json"):
        with open("config/posts.json", "w+") as file:
            json.dump({}, file)
    with open("config/posts.json", "r+") as file:
        data = json.load(file)
        return data
 
def save_posts(data):
    with open('config/posts.json', 'w+') as file:
        json.dump(data, file)
 
 
while True:
    try:
        posts = load_posts()
        main(reddit = authentication(), posts = posts)
    except Exception as e:
        print(e)