Source code for mindroot.lib.parent_templates

"""Parent template environment with single template per filename."""
import os
from jinja2 import Environment, FileSystemLoader, ChoiceLoader
from .plugins import list_enabled, get_plugin_path



class FirstMatchLoader(ChoiceLoader):
    """Custom loader that only returns the first matching template."""

    def __init__(self, loaders):
        super().__init__(loaders)
        self._template_cache = {}



    def get_source(self, environment, template):
        """Get template source, returning only first match for each template name."""
        if template in self._template_cache:
            return self._template_cache[template]
        for loader in self.loaders:
            try:
                source = loader.get_source(environment, template)
                self._template_cache[template] = source
                return source
            except Exception:
                continue
        raise TemplateNotFound(template)





def get_parent_templates_env(plugins=None):
    """Set up Jinja2 environment that only loads first matching template.
    
    Args:
        plugins (list, optional): List of plugin names. Defaults to enabled plugins.
    
    Returns:
        Environment: Configured Jinja2 environment with FirstMatchLoader
    """
    if plugins is None:
        plugins = list_enabled(False)
    template_paths = []
    for plugin in plugins:
        plugin_path = get_plugin_path(plugin)
        if not plugin_path:
            continue
        template_dir = os.path.join(plugin_path, 'templates')
        if os.path.exists(template_dir):
            template_paths.append(template_dir)
        if os.path.dirname(plugin_path) not in template_paths:
            template_paths.append(os.path.dirname(plugin_path))
        if os.path.dirname(os.path.dirname(plugin_path)) not in template_paths:
            template_paths.append(os.path.dirname(os.path.dirname(plugin_path)))
    template_paths.append('templates')
    loaders = [FileSystemLoader(path) for path in template_paths]
    env = Environment(loader=FirstMatchLoader(loaders))
    return env