Link to source code file on GitHub in code boxes with filenames (#166)

[GitHub/WoltLab/woltlab.github.io.git] / main.py

import materialx.emoji
from markdown import markdown

def define_env(env):
    @env.macro
    def codebox(title = None, language = "", filepath = None, contents = ""):
        if title is not None:
            if filepath is not None:
                editLink = f"""{env.variables['config']['repo_url']}tree/{env.variables['config']['edit_uri'].split("/")[1]}/snippets/{filepath}"""
                icon = markdown(':material-link:',
                    extensions=['pymdownx.emoji'],
                    extension_configs={
                        'pymdownx.emoji': {
                            'emoji_index': materialx.emoji.twemoji,
                            'emoji_generator': materialx.emoji.to_svg
                        }
                    }
                ).replace('<p>', '').replace('</p>', '')
                
                return f"""
<div class="titledCodeBox">
    <div class="codeBoxTitle">
        <code>{title}</code>
        <a class="codeBoxTitleGitHubLink" href="{editLink}" title="View on GitHub">{icon}</a>
    </div>
    ```{language}
    --8<-- "{filepath}"
    ```
</div>
"""
            else:
                return f"""
<div class="titledCodeBox">
    <div class="codeBoxTitle"><code>{title}</code></div>
```{language}
{contents}
```
</div>
"""
        else:
            if filepath is not None:
                return f"""
```{language}
--8<-- "{filepath}"
```
"""
            else:
                return f"""
```{language}
{contents}
```
"""