#!/usr/bin/env python3
from fastapi.templating import Jinja2Templates
from pathlib import Path
from jinja2 import Environment, FileSystemLoader
from starlette.templating import _TemplateResponse
from datetime import datetime
from typing import Any

BASE_DIR = Path(__file__).resolve().parent

# Create a Jinja2 Environment with the 'jinja2.ext.do' extension
class MyJinjaTemplates(Jinja2Templates):
    def __init__(self, directory: str, **kwargs: Any):
        super().__init__(directory=directory, **kwargs)
        self.env.add_extension('jinja2.ext.do')
        self.env.globals['now'] = datetime.now  # Add 'now' function

    def TemplateResponse(self, *args: Any, **kwargs: Any) -> _TemplateResponse:
        """Support existing pre-Starlette 0.29 TemplateResponse calls."""
        if args and isinstance(args[0], str):
            name = args[0]
            context = args[1] if len(args) > 1 else kwargs.pop("context", {})
            request = context.get("request") if isinstance(context, dict) else None
            if request is None:
                raise ValueError("TemplateResponse context must include request")
            return super().TemplateResponse(request, name, context, *args[2:], **kwargs)

        return super().TemplateResponse(*args, **kwargs)

# Create templates object that can be imported elsewhere
templates = MyJinjaTemplates(directory=str(BASE_DIR / "templates"))

# Register a context processor to add current_user to all templates
templates.env.globals["get_current_user"] = lambda: None  # Will be overridden at runtime
templates.env.globals["current_user"] = None