import requests
from bs4 import BeautifulSoup
from datetime import datetime
import pandas as pd
import argparse
import os
from urlencode import urlencode
from dotenv import load_dotenv
import logging


##########################################
#Logging
logger = logging.getLogger('Uploadbot')
##########################################

XML_BEGIN_RECENT = """<?xml version="1.0" encoding="utf-8" ?><rss version="2.0" xmlns:atom="http://www.w3.org/2005/Atom"><channel><title>Torfiles Last 25</title>
	            <link>https://torfiles.fieryeagle.org</link>
	            <description>Torfiles RSS Feed</description>
	            <language>en</language>
				<ttl>5</ttl>
				<atom:link href="https://rss.torfiles.fieryeagle.org/recent_25.xml" rel="self" type="application/rss+xml" />"""

XML_BEGIN_LAST = """<?xml version="1.0" encoding="utf-8" ?><rss version="2.0" xmlns:atom="http://www.w3.org/2005/Atom"><channel><title>Torfiles Last 48h</title>
	            <link>https://torfiles.fieryeagle.org</link>
	            <description>Torfiles RSS Feed</description>
	            <language>en</language>
				<ttl>5</ttl>
				<atom:link href="https://rss.torfiles.fieryeagle.org/latest_48h.xml" rel="self" type="application/rss+xml" />"""

XML_END = """</channel></rss>"""

load_dotenv()
copyparty_token = os.getenv('COPYPARTY_TOKEN')

def make_torrent_entry(name, timestamp, category):
    if category:
        category = "DODI_Repack"
    else:
        category = "Non_Repack"
    return f"""<item>
	                <title><![CDATA[{name}]]></title>
					<pubDate>{timestamp} +0000</pubDate>
					<category>{category}</category>
                    <guid>{name}</guid>
	                <link><![CDATA[https://torfiles.fieryeagle.org/{urlencode(name)}]]></link>
	                <description><![CDATA[Category: {category} → {timestamp} +0000]]></description>
                </item>"""

def write_file(path, data):
    if not os.path.exists("RSS"):
        logger.warning('RSS folder not found, creating...')
        os.mkdir("RSS")
    with open(os.path.realpath(path), "wt") as f:
        f.write(data)

def create_25_recent_feed(df):
    most_recent_25 = df.sort_values("timestamp", ascending=False).head(25)
    latest_25_xml = "\n".join([make_torrent_entry(i[0], i[2], i[3]) for i in most_recent_25.values])
    write_file("RSS/recent_25.xml", XML_BEGIN_RECENT+latest_25_xml+XML_END)

def create_48h_latest(df):
    last_48_hours = df[df['timestamp'] > datetime.now() - pd.Timedelta(days=2)]
    last_48_hours = last_48_hours.sort_values("timestamp", ascending=False)
    last_48_xml = "\n".join([make_torrent_entry(i[0], i[2], i[3]) for i in last_48_hours.values])
    write_file("RSS/latest_48h.xml", XML_BEGIN_LAST+last_48_xml+XML_END)

def create_df():
    torfiles_scrape = requests.get("https://upload.fieryeagle.org/torfiles", auth=("torbot",copyparty_token))
    soup = BeautifulSoup(torfiles_scrape.content, "html.parser")
    rows = [i.find_all("td") for i in soup.tbody.find_all("tr")]
    filtered_rows = [[i[1].text, i[6].text, datetime.strptime(i[9].text, "%Y-%m-%d %H:%M:%S")] for i in rows]
    df = pd.DataFrame(
        data=filtered_rows, columns=["name", "upload_ip", "timestamp"]
    )

    df['is_repack'] = df['name'].str.contains('DODI', case=False)
    return df[df["upload_ip"] == "49.12.185.202"]