import gradio as gr
import os
import json
from pymongo.mongo_client import MongoClient
from pymongo.server_api import ServerApi
import urllib.parse
from bson.objectid import ObjectId

username = urllib.parse.quote_plus(os.getenv('MONGO_USERNAME'))
password = urllib.parse.quote_plus(os.getenv('MONGO_PASSWORD'))
restUri = os.getenv('REST_URI')
uri = f'mongodb+srv://{username}:{password}{restUri}'
client = MongoClient(uri, server_api=ServerApi('1'))
db = client['file_storage']
references_collection = db['references']

try:
    client.admin.command('ping')
    print("Pinged your deployment. You successfully connected to MongoDB!")
except Exception as e:
    print(e)

theme = gr.themes.Default(
    primary_hue="blue",
    secondary_hue="violet",
    neutral_hue="slate",
)

def upload(file):
    root_directory = '/tmp/gradio'
    folder_contents_dict = get_folder_contents_dict(root_directory)
    print(folder_contents_dict)

    base_url = "https://abhicodes-file-sharing-system.hf.space/file=/tmp/gradio"

    urls = [f"{base_url}/{folder}/{file}" for folder, files in folder_contents_dict.items() for file in files]

    print(json.dumps(urls, indent=4))

    references_collection.update_one({"_id": ObjectId('66531a797cbaa19ba4e441c5')}, {"$set": {"urls": urls}})

    gr.Info("Uploaded Successfully")

def get_folder_contents_dict(root_dir):
    folder_contents = {}
    for item in os.listdir(root_dir):
        item_path = os.path.join(root_dir, item)
        if os.path.isdir(item_path):
            contents = os.listdir(item_path)
            folder_contents[item] = contents
    
    return folder_contents

def generate_markdown(json_data):
    urls = json_data.get("urls", [])
    markdown_lines = []
    for url in urls:
        # Encode URL components
        encoded_url = urllib.parse.quote(url, safe=':/')
        filename = url.split('/')[-1]
        encoded_filename = urllib.parse.quote(filename)
        markdown_lines.append(f'- <a style="text-decoration:none; font-size:18px" href={encoded_url} target="_blank">{encoded_filename}</a>')
    return "\n".join(markdown_lines)

def get_uploads(secret):
    if secret == os.environ.get('SECRET_KEY'):
        result = references_collection.find_one({"_id": ObjectId('66531a797cbaa19ba4e441c5')}, {"_id": 0})
        if result:
            markdown_output = generate_markdown(result)
            return markdown_output
        else:
            return {"error": "No result found"}
    else:
        return {"error": "Unauthorized"}

with gr.Blocks(theme=theme) as demo:
    gr.Markdown('''<h1 style="text-align:center;">File Storing and Sharing System</h1>''')
    gr.Markdown('''This project is a file storing and sharing system built using Gradio for the front-end interface and MongoDB for the back-end database. The system allows users to upload files, which are then stored and managed within a specified directory. URLs for accessing these files are generated and stored in a MongoDB collection, making it easy to share and access the uploaded files.
                > To know more read the docs at: [Documentation](https://huggingface.co/spaces/abhicodes/file-sharing-system/blob/main/README.md)
                ''')
    with gr.Row():
        with gr.Column():
            gr.Markdown('''<h1 style="text-align:center;">Uploader</h1>''')
            input = gr.File(label="Input", file_count="multiple")
            input.upload(fn=upload, inputs=input)
        with gr.Column():
            gr.Markdown('''<h1 style="text-align:center;">Downloader</h1>''')
            secret_key = gr.Text(placeholder="Enter your Secret Key")
            uploads = gr.Markdown(label="Uploads")
            get_upload_button = gr.Button("Get Uploads", variant='primary')
            get_upload_button.click(fn=get_uploads, inputs=secret_key, outputs=uploads)

demo.launch(debug=True)