From 11cd17cbf48e2643d822c78a4f36d9337115ded1 Mon Sep 17 00:00:00 2001
From: sleepwithoutbz <zy14074573@outlook.com>
Date: Fri, 26 Sep 2025 01:11:55 +0800
Subject: [PATCH] =?UTF-8?q?feat(WIP):=20=E6=B7=BB=E5=8A=A0=E9=A1=B9?=
 =?UTF-8?q?=E7=9B=AE=E6=9E=B6=E6=9E=84=E5=92=8Ccallback=E6=8E=A5=E5=8F=A3?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .gitignore              |  6 ++--
 deploy.sh               |  9 ++++++
 kds/__init__.py         |  0
 kds/main.py             |  8 +++++
 kds/routers/__init__.py |  0
 kds/routers/callback.py | 67 +++++++++++++++++++++++++++++++++++++++++
 kds/routers/root.py     |  8 +++++
 main.py                 | 14 ---------
 8 files changed, 96 insertions(+), 16 deletions(-)
 create mode 100755 deploy.sh
 create mode 100644 kds/__init__.py
 create mode 100644 kds/main.py
 create mode 100644 kds/routers/__init__.py
 create mode 100644 kds/routers/callback.py
 create mode 100644 kds/routers/root.py
 delete mode 100644 main.py

diff --git a/.gitignore b/.gitignore
index 728e1e9..16b8ecc 100644
--- a/.gitignore
+++ b/.gitignore
@@ -1,2 +1,4 @@
-__pycache__
-.kds_venv
+__pycache__/
+.kds_venv/
+
+.env
diff --git a/deploy.sh b/deploy.sh
new file mode 100755
index 0000000..0b09932
--- /dev/null
+++ b/deploy.sh
@@ -0,0 +1,9 @@
+#! /usr/bin/bash
+
+echo "Deploying..."
+
+git push
+# 连接ssh服务器：tencent，进入特定目录，执行指令
+ssh tencent "cd /home/ubuntu/kds_project && git pull && sudo systemctl restart kds"
+
+echo "Deploy finished!"
diff --git a/kds/__init__.py b/kds/__init__.py
new file mode 100644
index 0000000..e69de29
diff --git a/kds/main.py b/kds/main.py
new file mode 100644
index 0000000..e91acb3
--- /dev/null
+++ b/kds/main.py
@@ -0,0 +1,8 @@
+from typing import Union
+from fastapi import FastAPI
+from .routers import callback, root
+
+app = FastAPI()
+
+app.include_router(callback.router)
+app.include_router(root.router)
diff --git a/kds/routers/__init__.py b/kds/routers/__init__.py
new file mode 100644
index 0000000..e69de29
diff --git a/kds/routers/callback.py b/kds/routers/callback.py
new file mode 100644
index 0000000..317663a
--- /dev/null
+++ b/kds/routers/callback.py
@@ -0,0 +1,67 @@
+from fastapi import FastAPI, Request, HTTPException, Query, APIRouter
+from fastapi.responses import JSONResponse, PlainTextResponse
+import hashlib
+import xml.etree.ElementTree as ET
+
+router = APIRouter()
+
+
+APP_ID = "wx33726a2d409faa94"
+APP_SECRET = "59b7919d45d4631ab55bf418abb3e8bf"
+
+# TODO: 接收消息推送使用安全模式
+TOKEN = "sleepwithoutbz"
+ENCODING_AES_KEY = "njMce1oYKW7V11yug9Qjm689DYzXQceFedaeRDHSOsL"
+
+
+@router.get("/callback", response_class=PlainTextResponse)
+async def wechat_callback(
+    signature: str = Query(..., description="微信加密签名"),
+    timestamp: str = Query(..., description="时间戳"),
+    nonce: str = Query(..., description="随机数"),
+    echostr: str = Query(..., description="验证字符串"),
+):
+    """
+    微信验证回调接口
+    1. 将token/timestamp/nonce按字典序排序
+    2. 拼接后sha1加密
+    3. 与signature比对验证
+    """
+    # 1. 参数排序
+    params = sorted([TOKEN, timestamp, nonce])
+
+    # 2. 拼接字符串并加密
+    sign_str = "".join(params).encode("utf-8")
+    calculated_signature = hashlib.sha1(sign_str).hexdigest()
+
+    # 3. 验证签名
+    if calculated_signature == signature:
+        # 验证成功返回原始echostr
+        return echostr
+    else:
+        # 验证失败返回错误
+        raise HTTPException(status_code=403, detail="Invalid signature")
+
+
+@router.post("/callback")
+async def wechat_callback_post(request: Request):
+    """
+    微信消息推送接口
+    """
+    # 1. 获取请求数据
+    xml_data = await request.body()
+
+    # 2. 解析xml数据
+    root = ET.fromstring(xml_data)
+
+    # 3. 获取消息类型
+    msg_type = root.find("MsgType").text
+
+    # 4. 处理消息
+    if msg_type == "text":
+        # 4.1. 处理文本消息
+        content = root.find("Content").text
+        return JSONResponse(content={"Content": content})
+    else:
+        # 4.2. 处理其他类型消息
+        return JSONResponse(content={"Content": ""})
diff --git a/kds/routers/root.py b/kds/routers/root.py
new file mode 100644
index 0000000..58da68d
--- /dev/null
+++ b/kds/routers/root.py
@@ -0,0 +1,8 @@
+from fastapi import APIRouter
+
+router = APIRouter()
+
+
+@router.get("/")
+async def home():
+    return {"Hello": "World"}
diff --git a/main.py b/main.py
deleted file mode 100644
index d4ca7f1..0000000
--- a/main.py
+++ /dev/null
@@ -1,14 +0,0 @@
-from typing import Union
-from fastapi import FastAPI
-
-app = FastAPI()
-
-
-@app.get("/")
-async def read_root():
-    return {"Hello": "World"}
-
-
-@app.get("/items/{item_id}")
-async def read_item(item_id: int, q: Union[str, None] = None):
-    return {"item_id": item_id, "q": q}
\ No newline at end of file