from flask import Flask, render_template, request, session
from functools import wraps
from datetime import datetime, date, timedelta
import json

folder_config = {}

def require_secret(f):
    @wraps(f)
    def decorated_function(*args, **kwargs):
        global folder_config 
        if not folder_config:
            with open('folder_config.json') as file:
                folder_config = json.load(file)

        def is_valid(config_item, provided_secret):
            """
            Checks if today's date is <= validity date
            AND if the provided secret matches config_item['secret'].
            """
            folder_validity = config_item['validity']
            # Convert string to a date if necessary:
            if isinstance(folder_validity, str):
                folder_validity = datetime.strptime(folder_validity, '%d.%m.%Y').date()

            # Return whether it's still valid and secrets match:
            return (
                date.today() <= folder_validity and
                provided_secret == config_item['secret']
            )

        # 1) Get secret from query params (if any)
        args_secret = request.args.get('secret')

        # 2) Initialize 'allowed_secrets' in the session if missing
        if 'allowed_secrets' not in session:
            session['allowed_secrets'] = []

        # 3) If a new secret is provided, check if it’s valid, and add to session if so
        if args_secret:
            for config_item in folder_config:
                if is_valid(config_item, args_secret):
                    if args_secret not in session['allowed_secrets']:
                        session['allowed_secrets'].append(args_secret)
                    session.permanent = True  # Make the session permanent

        # 4) Re-check validity of each secret in session['allowed_secrets']
        #    If a secret is no longer valid (or not in config), remove it.
        for secret_in_session in session['allowed_secrets'][:]:
            # Find the current config item with matching secret
            config_item = next(
                (c for c in folder_config if c['secret'] == secret_in_session),
                None
            )
            # If the config item doesn’t exist or is invalid, remove secret
            if config_item is None or not is_valid(config_item, secret_in_session):
                session['allowed_secrets'].remove(secret_in_session)

        # 5) Build session['folders'] fresh from the valid secrets
        session['folders'] = {}
        for secret_in_session in session.get('allowed_secrets', []):
            config_item = next(
                (c for c in folder_config if c['secret'] == secret_in_session),
                None
            )
            if config_item:
                for folder_info in config_item['folders']:
                    session['folders'][folder_info['foldername']] = folder_info['folderpath']

        # 6) If we have folders, proceed; otherwise show index
        if session['folders']:
            return f(*args, **kwargs)
        else:
            return render_template('index.html')

    return decorated_function