👏更新docker镜像中处理裂图

Author: zkep

Dockerfile

@@ -6,7 +6,7 @@ WORKDIR /app
 
 COPY . ./
 
-RUN pip install -r requirements.txt
+RUN pip install -r requirements.txt  -i https://pypi.tuna.tsinghua.edu.cn/simple
 
 RUN python mkdocs.py
 
@@ -16,8 +16,11 @@ FROM python:3.11-slim  AS production
 
 WORKDIR /app
 
+RUN pip install fastapi uvicorn requests  -i https://pypi.tuna.tsinghua.edu.cn/simple
+
 COPY --from=builder-base /app/site ./site
+ADD  app.py .
 
 EXPOSE 8091
 
-CMD python -m http.server 8091 --directory /app/site
+CMD python app.py

README.md

@@ -13,15 +13,17 @@
 
 ###  markdown 在线文档
 
- * [github](https://uaxe.github.io/geektime-docs/) （可能有裂图问题，可以在仓库里直接看markdown）
- * [netlify](https://geektime-docs.netlify.app/)   (可能有裂图问题，可以在仓库里直接看markdown)
-
+ * [github](https://uaxe.github.io/geektime-docs/) 
+ * [netlify](https://geektime-docs.netlify.app/)  
 
 > tips: 在线文档支持 PC 浏览器，也支持移动端浏览器
+> 
+> 第三方托管在线文档可能有裂图问题，可以在仓库里直接看markdown，或使用docker方式本地部署
 
 ### 本地部署
 
 #### docker方式
+> 该方式实现了服务端代理请求图片，解决裂图，放心使用
 ```shell
 docker run -d -p 8091:8091 --restart always  --name geektime-docs  zkep/geektime-docs
 ```
@@ -50,7 +52,7 @@ mkdocs serve
 
 方案1： 直接看pdf吧 [geektime-pdfs](https://github.com/uaxe/geektime-pdfs)
 
-方案2： VIP用户，部署[my-geektime](https://github.com/zkep/my-geektime)服务，缓存对应的VIP课程 
+方案2： VIP用户，部署[my-geektime](https://github.com/zkep/my-geektime)服务，缓存对应的VIP课程, 支持播放下载音视频 
 
 方案3： 推荐本地使用中间代理人服务，拦截请求，改写 http 请求的 Referer 的思路
 

app.py

@@ -0,0 +1,31 @@
+import requests
+from fastapi import FastAPI, Query, HTTPException
+from fastapi.staticfiles import StaticFiles
+from fastapi.responses import StreamingResponse
+
+app = FastAPI()
+
+
+@app.get("/proxy")
+async def proxy(
+    url: str = Query(None),
+):
+    headers = {
+        "User-Agent": "Mozilla/5.0",
+        "Referer": url,
+    }
+    try:
+        with requests.get(url, timeout=60, headers=headers) as resp:
+            if not resp or resp.status_code != 200:
+                raise HTTPException(status_code=resp.status_code)
+            content_type = resp.headers.get("Content-Type")
+            return StreamingResponse(resp, headers=resp.headers, media_type=content_type)
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=str(e))
+
+
+app.mount("/", StaticFiles(html=True, directory="./site"),  name="static")
+
+if __name__ == "__main__":
+    import uvicorn
+    uvicorn.run("app:app", workers=6, host="0.0.0.0", port=8091)

mkdocs.py

@@ -1,19 +1,20 @@
 import os
+import re
 import shutil
 from pathlib import Path
-
 import yaml
 
 
 def _main():
-    dirs = [
-        'AI-大数据',
-        '产品-运营',
-        '前端-移动',
-        '后端-架构',
-        '管理-成长',
-        '计算机基础',
-        '运维-测试',
+    dirs = ['AI-大数据','产品-运营','前端-移动','后端-架构','管理-成长','计算机基础','运维-测试']
+    patterns = [
+        re.compile(r'!\[\]\((https?://\S+?)\)'),
+        re.compile(r'!$.*?$$(https?://[^\s$]+)'),
+    ]
+    proxy_url = "http://127.0.0.1:8091/proxy?url={url}"
+    proxy_urls = [
+        "https://static001.geekbang.org/resource/image",
+        "https://static001.geekbang.org/resource/avatar",
     ]
     all = []
     docs_dir = Path(__file__).parent.joinpath('dist')
@@ -32,7 +33,20 @@ def _main():
                         real_nav_path = os.path.join(docs_dir, nav_path)
                         src_nav_path = os.path.join(item_dir, "docs", nav)
                         os.makedirs(os.path.dirname(real_nav_path), exist_ok=True)
-                        shutil.copyfile(src_nav_path, real_nav_path)
+                        dst_raw = ''
+                        with open(src_nav_path, 'r') as ff:
+                            for line in ff.readlines():
+                                line = line.strip()
+                                for pattern in patterns:
+                                    for uri in pattern.findall(line):
+                                        for purl in proxy_urls:
+                                            if purl in uri:
+                                                dst_url = proxy_url.format(url=uri)
+                                                line = line.replace(uri, dst_url)
+                                dst_raw += line
+                                dst_raw += "\n"
+                        with open(real_nav_path, 'w') as fi:
+                            fi.write(dst_raw)
                         items.append(nav_path)
                     index_text += f'\n[{os.path.basename(item_dir)}]({os.path.basename(item_dir)}/{os.path.basename(item_dir)})\n'
                 navs.append({sub_dir: items})

requirements.txt

@@ -8,4 +8,7 @@ pyyaml
 webdriver-manager
 requests
 pypdf
-pillow
+pillow
+fastapi
+uvicorn
+markdown==3.6