feat: add hooks script

Author: yanglbme

.gitignore

@@ -3,4 +3,5 @@ node_modules
 .cache
 .vscode
 solution
-.DS_Store
+.DS_Store
+__pycache__

docs-en/lc/1.md

@@ -1,3 +1,12 @@
+---
+comments: true
+tags:
+  - Array
+  - Hash Table
+edit_url: 'https://github.com/doocs/leetcode/edit/main/solution/0000-0099/0001.Two%20Sum/README_EN.md'
+---
+
+
 # [1. Two Sum](https://leetcode.com/problems/two-sum)
 
 

docs/lc/1.md

@@ -1,5 +1,13 @@
 ---
 comments: true
+tags:
+  - 数组
+  - 哈希表
+  - 排序
+  - 暴力
+  - 模拟
+  - 枚举
+  - 双指针
 edit_url: 'https://github.com/doocs/leetcode/edit/main/solution/0000-0099/0001.Two%20Sum/README.md'
 ---
 

docs/lc/2.md

@@ -1,5 +1,8 @@
 ---
 comments: true
+tags:
+  - 模拟
+  - 堆（优先队列）
 edit_url: 'https://github.com/doocs/leetcode/edit/main/solution/0000-0099/0002.Add%20Two%20Numbers/README.md'
 ---
 

hooks.py

@@ -1,6 +1,60 @@
-# https://www.mkdocs.org/dev-guide/plugins/#events
+from bs4 import BeautifulSoup
 
 
+# https://www.mkdocs.org/dev-guide/plugins/#events
 def on_page_markdown(markdown, page, config, files):
     page_edit_url = page.meta.get("edit_url")
     page.edit_url = str(page_edit_url) if page_edit_url else None
+
+
+def on_page_content(html, page, config, files):
+    # 修改 tags 页每个 tag 的 id 属性值
+    is_cn_tag_page = page.abs_url == '/leetcode/tags/'
+    if is_cn_tag_page:
+        soup = BeautifulSoup(html, "html.parser")
+        h2_tags = soup.find_all("h2")
+        for tag in h2_tags:
+            span_tag = tag.find("span", class_="md-tag")
+            if span_tag:
+                tag["id"] = span_tag.text.strip()
+        # 重新排序每个 tag 下的问题列表
+        ul_tags = soup.find_all("ul")
+        for ul_tag in ul_tags:
+            li_tags = ul_tag.find_all("li")
+            li_tags.sort(key=lambda x: int(x.text.split('.')[0].strip()))
+            ul_tag.clear()
+            for li_tag in li_tags:
+                ul_tag.append(li_tag)
+        html = str(soup)
+        
+    return html
+
+
+def on_page_context(context, page, config, nav):
+    # 修改每个问题顶部 tags 的跳转链接
+    is_cn_problem_page = (page.edit_url or '').endswith('README.md')
+    has_tag = 'tags' in context
+    if is_cn_problem_page and has_tag:
+        for tag in context['tags']:
+            tag['url'] = f'tags/#{tag["name"]}'
+    return context
+
+
+def on_post_page(output, page, config):
+    # 修改 tags 页的左侧导航链接
+    is_cn_tag_page = page.abs_url == '/leetcode/tags/'
+    if is_cn_tag_page:
+        soup = BeautifulSoup(output, "html.parser")
+        nav_tag = soup.find("nav", class_="md-nav md-nav--secondary")
+        if nav_tag:
+            ul_tag = nav_tag.find("ul", class_="md-nav__list")
+            if ul_tag:
+                for li_tag in ul_tag.find_all("li", class_="md-nav__item"):
+                    a_tag = li_tag.find("a", class_="md-nav__link")
+                    if a_tag:
+                        span_tag = a_tag.find("span", class_="md-ellipsis")
+                        if span_tag:
+                            tag_name = span_tag.text.strip()
+                            a_tag["href"] = f"#{tag_name}"
+        output = str(soup)
+    return output

mkdocs.yml

@@ -431,3 +431,5 @@ nav:
     - 17.24. 最大子矩阵: lcci/17.24.md
     - 17.25. 单词矩阵: lcci/17.25.md
     - 17.26. 稀疏相似度: lcci/17.26.md
+
+  - 专项训练: tags.md

requirements.txt

@@ -5,4 +5,5 @@ urllib3==1.26.18
 mkdocs-git-committers-plugin-2==2.3.0
 mkdocs-material==9.5.13
 mkdocs-glightbox==0.3.7
-jieba==0.42.1
+jieba==0.42.1
+beautifulsoup4==4.12.0