chore: update ext info plugin

Author: yanglbme

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'
+difficulty: 中等
 rating: 1345
 ---
 

docs/stylesheets/extra.css

@@ -0,0 +1,12 @@
+.rating {
+    display: flex;
+    gap: .5em;
+}
+
+.rating-item {
+    float: left;
+    padding: 8px 12px;
+    border-radius: 2.4rem;
+    font-size: min(.8em,.64rem);
+    background-color: #00000012;
+}

hooks/ext_info.py

@@ -1,24 +1,39 @@
 from mkdocs import plugins
 
+from bs4 import BeautifulSoup
+
 
 # https://www.mkdocs.org/dev-guide/plugins/#events
 
+colors = {
+    "简单": "#46c6c2",
+    "中等": "#fac31d",
+    "困难": "#f8615c",
+    "Easy": "#46c6c2",
+    "Medium": "#fac31d",
+    "Hard": "#f8615c",
+}
+
 
 @plugins.event_priority(90)
 def on_post_page(output, page, config):
-    difficulty = page.meta.get('difficulty')
-    rating = page.meta.get('rating')
-
-    # 在 h1 标题之后插入难度和评分信息及其样式
+    is_cn_problem_page = (page.edit_url or "").endswith("README.md")
+    difficulty = page.meta.get("difficulty")
+    rating = page.meta.get("rating")
+    if not difficulty and not rating:
+        return output
+    if rating:
+        rating = f"难度分 {rating}" if is_cn_problem_page else f"Rating {rating}"
     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
+    soup = BeautifulSoup(html, "html.parser")
+    div = soup.new_tag("div", attrs={"class": "rating"})
+    soup.select_one("h1").insert_after(div)
+    color = colors.get(difficulty) or "black"
+    for s in (difficulty, rating):
+        if not s:
+            continue
+        sub = soup.new_tag("div", attrs={"class": "rating-item"})
+        sub.string = str(s)
+        sub["style"] = f"color: {color};"
+        div.append(sub)
+    return str(soup)

mkdocs.yml

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