Installation et configuration¶

from setuptools import setup, find_packages

setup(
    name="mkdocs-classification-filter",
    version="0.1.0",
    packages=find_packages(),
    entry_points={
        "mkdocs.plugins": [
            "classification-filter = mkdocs_classification_filter.plugin:ClassificationFilterPlugin",
        ]
    },
    install_requires=["mkdocs>=1.5"],
)

import os
import logging
from mkdocs.plugins import BasePlugin

log = logging.getLogger("mkdocs.plugins.classification-filter")

LEVELS = {"public": 0, "interne": 1, "confidentiel": 2, "restreint": 3}


class ClassificationFilterPlugin(BasePlugin):

    def on_files(self, files, config):
        target = os.environ.get("MKDOCS_CLASSIFICATION", "public").lower()
        if target not in LEVELS:
            raise ValueError(
                f"MKDOCS_CLASSIFICATION='{target}' invalide. "
                f"Valeurs autorisees : {', '.join(LEVELS.keys())}"
            )
        target_level = LEVELS[target]
        log.info(f"Classification filter: build niveau '{target}' (<= {target_level})")

        to_remove = []
        for file in files:
            if not file.src_path.endswith(".md"):
                continue
            page_level = self._get_level(file, config)
            if page_level is not None and page_level > target_level:
                log.info(f"  Exclusion: {file.src_path}")
                to_remove.append(file)

        for file in to_remove:
            files.remove(file)

        return files

    def on_nav(self, nav, config, files):
        self._clean_nav(nav.items)
        return nav

    def on_env(self, env, config, files):
        target = os.environ.get("MKDOCS_CLASSIFICATION", "public").lower()
        env.globals["classification_level"] = target
        return env

    def _get_level(self, file, config):
        import yaml

        full_path = os.path.join(config["docs_dir"], file.src_path)
        try:
            with open(full_path, "r", encoding="utf-8") as f:
                content = f.read()
        except (OSError, IOError):
            return None

        if not content.startswith("---"):
            return LEVELS["public"]

        parts = content.split("---", 2)
        if len(parts) < 3:
            return LEVELS["public"]

        try:
            meta = yaml.safe_load(parts[1])
        except yaml.YAMLError:
            return LEVELS["public"]

        if not meta or "classification" not in meta:
            return LEVELS["public"]

        classification = str(meta["classification"]).lower()
        if classification not in LEVELS:
            raise ValueError(
                f"classification '{classification}' invalide dans {file.src_path}. "
                f"Valeurs autorisees : {', '.join(LEVELS.keys())}"
            )
        return LEVELS[classification]

    def _clean_nav(self, items):
        to_remove = []
        for item in items:
            if hasattr(item, "children") and item.children:
                self._clean_nav(item.children)
                if not item.children:
                    to_remove.append(item)
            elif hasattr(item, "file") and item.file is None:
                to_remove.append(item)
        for item in to_remove:
            items.remove(item)