__doc__ = """Example of a mdbook preprocessor"""

import json
import sys
from libs.utils import debug


# TODO: This needs to be rewritten

def modify_content(content: str) -> str | None:
    ############## MODIFY 'content' HERE ##############
    """Alter the contents of each chapter

    Args:
        content (str):  The contents of a chapter received. Is in markdown format.

    Returns:
        str | None: The chapter contents modified.
        If `None`, the original content will be used instead
    """
    return content
    ###################################################


#
#
#
#
#
#
#
#
#
#
#
#

if __name__ == "__main__":
    if len(sys.argv) > 1:
        if sys.argv[1] == "supports":
            sys.exit(0)
    context, book = json.load(sys.stdin)
    book: dict[str, list]
    context: dict
    debug(f"context: {context}")

    sections = book["sections"]

    for i, section in enumerate(sections):
        section: dict
        if not section.get("Chapter"):
            continue
        for chapter in section.values():
            chapter: dict
            debug(chapter)
            content: str = chapter["content"]

            if type(content) is str and (res := modify_content(content)):
                content = res

            chapter.update({"content": content})
            debug(book)
    print(json.dumps(book))