custom markdown

Author: willmcgugan

CHANGELOG.md

@@ -0,0 +1,18 @@
+# Changelog
+
+All notable changes to this project will be documented in this file.
+
+The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.0.0/),
+and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0.html).
+
+## [Unreleased]
+
+### Added
+
+- Added space key to page down to pager
+
+### Changed
+
+- Change how code blocks in markdown are rendered (remove border, adding padding)
+
+[unreleased]: https://github.com/olivierlacan/keep-a-changelog/compare/v1.5.1...HEAD

src/rich_cli/markdown.py

@@ -0,0 +1,32 @@
+from typing import Any
+
+from rich.console import Console, ConsoleOptions, RenderResult
+from rich.markdown import Markdown, TextElement
+from rich.padding import Padding
+from rich.syntax import Syntax
+
+
+class CodeBlock(TextElement):
+    """A code block with syntax highlighting."""
+
+    style_name = "markdown.code_block"
+
+    @classmethod
+    def create(cls, markdown: "Markdown", node: Any) -> "CodeBlock":
+        node_info = node.info or ""
+        lexer_name = node_info.partition(" ")[0]
+        return cls(lexer_name or "default", markdown.code_theme)
+
+    def __init__(self, lexer_name: str, theme: str) -> None:
+        self.lexer_name = lexer_name
+        self.theme = theme
+
+    def __rich_console__(
+        self, console: Console, options: ConsoleOptions
+    ) -> RenderResult:
+        code = str(self.text).rstrip()
+        syntax = Syntax(code, self.lexer_name, theme=self.theme, word_wrap=True)
+        yield Padding(syntax, (0, 4))
+
+
+Markdown.elements["code_block"] = CodeBlock