import os
import random
from flask import Flask, send_from_directory, session

app = Flask(__name__)

# Implement count to load specific page on Nth load
# Using session['requests']
app.secret_key = '420-69'

# Script must run from root dir containing all websites dirs 
# OR change the ROOT_DIR path below :)
ROOT_DIR = os.path.dirname(os.path.abspath(__file__))
WEBSITE_DIRS = [name for name in os.listdir(ROOT_DIR) if not name.startswith('.') and os.path.isdir(os.path.join(ROOT_DIR, name))]

# Site choosing logic
def current_website_dir():
    session.pop('website_dir', None)
    match session['requests']:
        case 5:
            session['website_dir'] = (os.path.join(ROOT_DIR, 'Marvel'))
        case 6:
            session['website_dir'] = (os.path.join(ROOT_DIR, 'Escape'))
            session.pop('requests', None)
        case _:
            try:
                WEBSITE_DIRS.remove(os.path.join(ROOT_DIR, 'Escape'))
                WEBSITE_DIRS.remove(os.path.join(ROOT_DIR, 'Marvel'))
            except (ValueError, IndexError) as e:
                pass
            session['website_dir'] = random.choice(WEBSITE_DIRS)

# Make static files available
@app.route('/<path:filename>', methods=['GET'])
def static_proxy(filename):
    return send_from_directory(session['website_dir'], filename)

# Serve site index.html
@app.route('/', methods=['GET'])
def index():
    if 'requests' in session:
        session['requests'] += 1
    else:
        session['requests'] = 1
    current_website_dir()
    return send_from_directory(session['website_dir'], 'index.html')

if __name__ == "__main__":
    app.run()