feat: add ext info plugin

Author: yanglbme

docs/lc/1.md

@@ -5,6 +5,7 @@ edit_url: https://github.com/doocs/leetcode/edit/main/solution/0000-0099/0001.Tw
 tags:
     - 数组
     - 哈希表
+rating: 1234
 ---
 
 # [1. 两数之和](https://leetcode.cn/problems/two-sum)

docs/lc/2.md

@@ -4,6 +4,7 @@ tags:
   - 模拟
   - 堆（优先队列）
 edit_url: 'https://github.com/doocs/leetcode/edit/main/solution/0000-0099/0002.Add%20Two%20Numbers/README.md'
+rating: 1345
 ---
 
 # [2. 两数相加](https://leetcode.cn/problems/add-two-numbers)

hooks/ext_info.py

@@ -0,0 +1,24 @@
+from mkdocs import plugins
+
+
+# https://www.mkdocs.org/dev-guide/plugins/#events
+
+
+@plugins.event_priority(90)
+def on_post_page(output, page, config):
+    difficulty = page.meta.get('difficulty')
+    rating = page.meta.get('rating')
+
+    # 在 h1 标题之后插入难度和评分信息及其样式
+    html = str(output)
+    # if difficulty:
+    #     # 找到 h1 标签的结束位置
+    #     h1_end = html.find('</h1>') + 5
+    #     # 在 h1 标签之后插入难度信息
+    #     html = html[:h1_end] + f'<p>Difficulty: {difficulty}</p>' + html[h1_end:]
+    # if rating:
+    #     # 找到 h1 标签的结束位置
+    #     h1_end = html.find('</h1>') + 5
+    #     # 在 h1 标签之后插入评分信息
+    #     html = html[:h1_end] + f'<p>Rating: {rating}</p>' + html[h1_end:]
+    return html

mkdocs.yml

@@ -86,6 +86,7 @@ hooks:
   - hooks/tags.py
   - hooks/committer.py
   - hooks/fix_markdown_block.py
+  - hooks/ext_info.py
 
 markdown_extensions:
   - admonition