From bd0780c344ab2f16821f7c8a85677e60819c84fe Mon Sep 17 00:00:00 2001
From: JoeanAmier <yonglelolu@gmail.com>
Date: Mon, 3 Jun 2024 22:26:11 +0800
Subject: [PATCH] =?UTF-8?q?=E6=96=B0=E5=A2=9E=20Web=20API=20=E6=A8=A1?=
 =?UTF-8?q?=E5=BC=8F?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 README.md                       |  61 +++++++++++++++++++++++++++++---
 locale/en_GB/LC_MESSAGES/xhs.mo | Bin 10971 -> 11461 bytes
 locale/en_GB/LC_MESSAGES/xhs.po |  15 ++++++++
 locale/zh_CN/LC_MESSAGES/xhs.po |  15 ++++++++
 main.py                         |  16 ++++++---
 source/TUI/setting.py           |   2 +-
 source/__init__.py              |   3 +-
 source/application/app.py       |  56 +++++++++++++++++++++++++++--
 source/application/video.py     |   2 +-
 source/module/manager.py        |   2 ++
 source/module/settings.py       |   6 ++--
 static/XHS-Downloader.js        |   6 ++--
 12 files changed, 166 insertions(+), 18 deletions(-)

diff --git a/README.md b/README.md
index bf05a0c..5317690 100644
--- a/README.md
+++ b/README.md
@@ -28,7 +28,7 @@
 <li>✅ 支持命令行下载作品文件</li>
 <li>✅ 从浏览器读取 Cookie</li> 
 <li>✅ 自定义文件名称格式</li> 
-<li>☑️ 支持 API 调用功能</li>
+<li>✅ 支持 API 调用功能</li>
 </ul>
 <ul><b>脚本功能</b>
 <li>✅ 下载小红书无水印作品文件</li>
@@ -70,14 +70,66 @@
 <li>运行 <code>main.py</code> 即可使用</li>
 </ol>
 <h1>🛠 命令行模式</h1>
-<p>项目支持命令行运行模式，若想要下载图文作品的部分图片，可以使用此模式传入需要下载的图片序号！</p>
-<p>可以使用命令行从浏览器读取 Cookie 并写入配置文件！注意需要关闭对应浏览器才能读取数据！</p>
-<p><code>bool</code> 类型参数支持使用 <code>true</code>、<code>false</code>、<code>1</code>、<code>0</code>、<code>yes</code>、<code>no</code>、<code>on</code> 或 <code>off</code>（不区分大小写）来设置。</p>
+<p>项目支持命令行运行模式，若想要下载图文作品的部分图片，可以使用此模式设置需要下载的图片序号！</p>
+<p>可以使用命令行<b>从浏览器读取 Cookie 并写入配置文件</b>！注意需要关闭浏览器才能读取数据！</p>
 <p>命令示例：<code>python .\main.py --browser_cookie Chrome --update_settings</code></p>
+<p><code>bool</code> 类型参数支持使用 <code>true</code>、<code>false</code>、<code>1</code>、<code>0</code>、<code>yes</code>、<code>no</code>、<code>on</code> 或 <code>off</code>（不区分大小写）来设置。</p>
 <hr>
 <img src="static/screenshot/命令行模式截图1.png" alt="">
 <hr>
 <img src="static/screenshot/命令行模式截图2.png" alt="">
+<h1>🖥 服务器模式</h1>
+<p><b>启动：</b>运行命令：<code>python .\main.py server</code></p>
+<p><b>关闭：</b>按下 <code>Ctrl</code> + <code>C</code> 关闭服务器</p>
+<p><b>请求接口：</b><code>/xhs/</code></p>
+<p><b>请求类型：</b><code>POST</code></p>
+<p><b>请求参数：</b></p>
+<table>
+<thead>
+<tr>
+<th align="center">参数</th>
+<th align="center">类型</th>
+<th align="center">含义</th>
+<th align="center">默认值</th>
+</tr>
+</thead>
+<tbody>
+<tr>
+<td align="center">url</td>
+<td align="center">str</td>
+<td align="center">小红书作品链接，自动提取，不支持多链接</td>
+<td align="center">无</td>
+</tr>
+<tr>
+<td align="center">download</td>
+<td align="center">bool</td>
+<td align="center">是否下载作品文件；设置为 <code>true</code> 将会耗费更多时间</td>
+<td align="center">false</td>
+</tr>
+<tr>
+<td align="center">index</td>
+<td align="center">str</td>
+<td align="center">下载指定序号的图片文件，仅对图文作品生效；多个序号之间使用空格分隔；<code>download</code> 参数设置为 <code>false</code> 时不生效</td>
+<td align="center">null</td>
+</tr>
+<tr>
+<td align="center">skip</td>
+<td align="center">bool</td>
+<td align="center">是否跳过存在下载记录的作品；设置为 <code>true</code> 将不会返回存在下载记录的作品数据</td>
+<td align="center">false</td>
+</tr>
+</tbody>
+</table>
+<p><b>代码示例：</b></p>
+<pre>
+def api_demo():
+    server = "http://127.0.0.1:8080"
+    data = {
+        "url": "https://www.xiaohongshu.com/explore/123456789",
+    }
+    response = requests.post(server, data=data)
+    print(response.json())
+</pre>
 <h1>🕹 用户脚本</h1>
 <img src="static/screenshot/用户脚本截图1.png" alt="">
 <hr>
@@ -235,6 +287,7 @@ async def example():
 <h1>🌐 Cookie</h1>
 <ol>
 <li>打开浏览器（可选无痕模式启动），访问 <code>https://www.xiaohongshu.com/explore</code></li>
+<li>登录小红书账号（可跳过）</li>
 <li>按下 <code>F12</code> 打开开发人员工具</li>
 <li>选择 <code>网络</code> 选项卡</li>
 <li>勾选 <code>保留日志</code></li>
diff --git a/locale/en_GB/LC_MESSAGES/xhs.mo b/locale/en_GB/LC_MESSAGES/xhs.mo
index 366c6548f01c5deaada6a70001072b84363e2468..8a5b173a6c5048b301733386ab7d610be8b4b9ac 100644
GIT binary patch
delta 2313
zcmaLYZ%oxy9LMqF4MqHegr*@@KS%<Ngj5KrG*J<ylr%unnHRVSi|%#YOU9txxypz|
zxWddvHBrN@l_oh%r5=>C9@Knbn=Y!g{9Q5EAA8hR!``2J4?L>Q_T_cX@6Y-3J?D4%
zw)|{W`1|Czy@par%pop#j0xk*<M~5rk2mHve978}Gijg2bR4qn8<<9W=0sy^umIO#
z52oTZoQClU#{7)4Fk3@;CPgbI2XE!Z7MzRCs1Dz>*Uz9j`o?-4r_i1{nZVh27iQrt
zxZdhR_1lVzu^W@|OU%_$hp1@4sr2?ZE<|<Ei79vxvv3d#F=9<))Z1xq!RP?UEz^yf
z=mF~>=Fq-`8Ydw!wy*_Qr&O0w(Tf8(3qQw99Kt0SN8U8RQdCANQCrf2>Tn-w>&~E_
zyM*fR4}1LvYC&_7V$Ux`E}2RUpQEy!ic*qubL>VDs)G&IC#;Ql2iIHB!TqQSow5#L
zF6|N2L~`g<{g&|OUfhiLU?*yUr<2M5-Bf<!LOxEL9-BZhs)GvDL;|Rkw%PU@s1+Q>
z=kbKSUO2-T#x|?50IN`k^i@=5PvC6)3Nvtc2KiUViR5Pi&c_$A3b)~psQX16c@D7I
zib><h80z_F=v6E3LuKY7YGr@od~{fa&QK}RmGPm@QY*4%b0|#ZJUtI!9v9llr(XC3
z^}t2c1g_zI7|(Ll(K1|$<#;D{<2F2rdOwAW%+V}DrG71zVjHf)FHm1gIFS>indc!{
zH08J$8&EI4jhpa17UMkfrt3Sf4tr5skw9-cgej=Z<l}PmVK(-mCU(L4D>8?$nMl60
z!fa#=Q-a#Vr%<Onj4a-~g-YG0sEH2a2>y+HS>`Wp>b-<{v8_r)&b66~x}S$6$E-jG
zHg%Y-@4thJRy2T0=~dK(ve-a%xDxZR0<}e*$Rp-6EW%%`xtwHeSr9eh-8cb{pnfNg
zq9*h;s^1^*7~`AkRFuMA@}v%q<2ck{4l!CPsPJfXyVjwH&;Sn*$_N=9E05ZmCaR(Z
z>7W)6C4`D5ukp=BLWfIfQW;OIj@Dv(W~*Z)Qd?nLE61+JKC#iJwsN6uX!Yg9gM?DQ
zfw+mFM8}VwXl;Q?88InZi~Up58PLzJ_I@p)OzB7DA%e3P-J12Nl&k3bR(eNEx~)va
zVq4cXtE|%a4^vUvRhAH?#1u7biN_p*T``XkafD9#8e+8Y^+xx0wEdW^Bv|z+E+_ci
zF_Q^>VVqlYzf!-9P|;_@Su%^%sPt7k%gfd{k)BSs<AA&Wh&yoHeWCyAzW0ASv+LjQ
z?m6mq9N}*N?-K^%Q|)V)Po9qSoN~K5uDo~D?K)WKwB|nBe<(4-Q}oJ!+x5D8tn13D
z-pe1qdwH<OeQj4{@3Bbd2kxN{hd(-OhX4DCNc$eQ<DLF<NqaqoqcaWqnx6JGIrZM4
zQ|<HlonU=)$f*tN^rwwY*FF*qd7DDM+O+=7)6<hvHrn?B!I0zi*E%}_O`+H{+cKW>
zY%TUSH2P|tP{0XPhrA7br`8+tIsw14slgkl5BRq;PG->A8EASkXf`(2)cAtIy5`2l
R7W+K!{?8lzBbncM{sAq1O>qDK

delta 1826
zcmX}sSx8h-9LMp$<1(YB>15g7sBN0om}yyNnmMi7LQ&Z!SqYI;4-vEtv|zmzQ6%Jp
zL?}@d41qor5m^$nJzCL231tu%Xw*yg{oOlunE!mvy?5?e{^y+O-u_`r_>({Gno%|p
zlZm??vnDKy=Zi9sU^WhaVmA7`?)4nZ;(9qY;6ALy9-M@kKC{tSgx|3W^KovX>kb@m
z7PgC2rt_c;)!=jYfqg?Y6qDpS1xImRiYXYvnYbCp;>oC2kcO=t7vK~0;}8bXm+bae
zigomFyQ!#wE}VdGa5nbiV)WCitih>x5uFZ@k3Hc_1AP^>ALnr$&nPvqV${Uy@Bp4d
zz4rw(>E8yZ%tBu(v%?^&gM+9S&!HN+jas6Y(dS=L4JpsMpMsi59_sfM$jA2c^#YHf
zGO;wxy}tp&YM`EqY>IjUr*Ykk0c^)0_Mis#2i1^2-K-F2U;%DLP2>X3!FyPOy{Lht
z@si#Tp!zHIlYgZy6uq$*^~1w>9>dZ5abwJA!_sjP=HNu!i8Jsxs-0Gxi(S}^->@E!
zWVp}YA^XsNU|NjXuMG097rbn7&9DSD^G4K6nlTUWqxQyoREh^tdn75-%|sEtrR5M7
zb3ZfS{=Np8gEgWCcmfyVbyPdA!&J(t^kEL>u;BH$64hY`(w9BOLhQx0n9NzM#;vGR
z(25%PGi1`%huZzWQSTMd_%_^wA?!rm4-Zq(t}JB7DD^i`o2dhpnQpAaLCnVz@~we2
zAg|hK)XdtDpV%YRQhmlLIEakd{ESi=TYwtiF8r(Wf0PObz>bh_y?6$-MwgL&W-X}a
zcaX_gC#vHQsLTzc22f7xdcO`!@DOS#ZecF=;&P1TLWWTL&n{8X2ybEnKE|oojT%57
zs^KAgfL=C)GS!J%+a8R;k)rJBJr$h-os2{xL}=|*7_(Ex(7#hOVioP~ctSI(C6vOE
zq79<~s*qLZPbSs5a=+Nr&YF@?r)W812`!fftU?lO1)+hfgq6ZIL=CZlm``ZkRRY8^
zRVYgdrJX-TC#_!88CF?B=pRzcq)d4Tl`29@ww_2Llw}odR2|<{TK`Q{d_*-7M`+h_
zimZ&NBsLN%D+&F1sA&Bvh}DFOHfE4mN9c6u#IP#PGQ$7D3R)3SI&$Y^<hl2uH_|dX
a+Y=d1zv_v+%E<CZe6!AaBK^4^J^ujs3ZQ`i

diff --git a/locale/en_GB/LC_MESSAGES/xhs.po b/locale/en_GB/LC_MESSAGES/xhs.po
index a65a1f2..83950ca 100644
--- a/locale/en_GB/LC_MESSAGES/xhs.po
+++ b/locale/en_GB/LC_MESSAGES/xhs.po
@@ -255,3 +255,18 @@ msgstr "Format of works file name"
 
 msgid "邀请链接："
 msgstr "Invitation link: "
+
+msgid "获取小红书作品数据成功"
+msgstr "Successfully obtained data on Xiaohongshu's works"
+
+msgid "获取小红书作品数据失败"
+msgstr "Failed to obtain data on Xiaohongshu's works"
+
+msgid "Web API 服务器已启动！"
+msgstr "Web API server started!"
+
+msgid "Web API 服务器已关闭！"
+msgstr "Web API server has been shut down!"
+
+msgid "服务器主机及端口: {0}"
+msgstr "Server host and port: {0}"
diff --git a/locale/zh_CN/LC_MESSAGES/xhs.po b/locale/zh_CN/LC_MESSAGES/xhs.po
index f7de05c..4ab8049 100644
--- a/locale/zh_CN/LC_MESSAGES/xhs.po
+++ b/locale/zh_CN/LC_MESSAGES/xhs.po
@@ -255,3 +255,18 @@ msgstr ""
 
 msgid "邀请链接："
 msgstr ""
+
+msgid "获取小红书作品数据成功"
+msgstr ""
+
+msgid "获取小红书作品数据失败"
+msgstr ""
+
+msgid "Web API 服务器已启动！"
+msgstr ""
+
+msgid "Web API 服务器已关闭！"
+msgstr ""
+
+msgid "服务器主机及端口: {0}"
+msgstr ""
diff --git a/main.py b/main.py
index c34611e..6c15b48 100644
--- a/main.py
+++ b/main.py
@@ -1,6 +1,7 @@
 from asyncio import run
 from sys import argv
 
+from source import Settings
 from source import XHS
 from source import XHSDownloader
 from source import cli
@@ -47,13 +48,20 @@ async def example():
         print(await xhs.extract(multiple_links, download, ))
 
 
-async def main():
+async def app():
     async with XHSDownloader() as xhs:
         await xhs.run_async()
 
 
+async def server():
+    async with XHS(**Settings().run()) as xhs:
+        await xhs.run_server()
+
+
 if __name__ == '__main__':
-    if len(argv) > 1:
-        cli()
+    if len(argv) == 1:
+        run(app())
+    elif argv[1] == "server":
+        run(server())
     else:
-        run(main())
+        cli()
diff --git a/source/TUI/setting.py b/source/TUI/setting.py
index c28d324..ddff758 100644
--- a/source/TUI/setting.py
+++ b/source/TUI/setting.py
@@ -107,7 +107,7 @@ class Setting(Screen):
             "language": self.query_one("#language").value,
             "image_download": self.query_one("#image_download").value,
             "video_download": self.query_one("#video_download").value,
-            "server": False,
+            # "server": False,
         })
 
     @on(Button.Pressed, "#abandon")
diff --git a/source/__init__.py b/source/__init__.py
index 6551c0e..3818bec 100644
--- a/source/__init__.py
+++ b/source/__init__.py
@@ -1,5 +1,6 @@
 from .CLI import cli
 from .TUI import XHSDownloader
 from .application import XHS
+from .module import Settings
 
-__all__ = ['XHS', 'XHSDownloader', 'cli']
+__all__ = ['XHS', 'XHSDownloader', 'cli', 'Settings', ]
diff --git a/source/application/app.py b/source/application/app.py
index 5800aa3..c512105 100644
--- a/source/application/app.py
+++ b/source/application/app.py
@@ -1,3 +1,4 @@
+from asyncio import CancelledError
 from asyncio import Event
 from asyncio import Queue
 from asyncio import QueueEmpty
@@ -9,6 +10,7 @@ from re import compile
 from typing import Callable
 from urllib.parse import urlparse
 
+from aiohttp import web
 from pyperclip import paste
 
 from source.expansion import BrowserCookie
@@ -22,6 +24,7 @@ from source.module import (
     ERROR,
     WARNING,
     MASTER,
+    REPOSITORY,
 )
 from source.module import Translate
 from source.module import logging
@@ -62,7 +65,7 @@ class XHS:
             video_download=True,
             folder_mode=False,
             language="zh_CN",
-            server=False,
+            # server=False,
             transition: Callable[[str], str] = None,
             read_cookie: int | str = None,
             *args,
@@ -85,6 +88,7 @@ class XHS:
             image_download,
             video_download,
             folder_mode,
+            # server,
             self.message,
         )
         self.html = Html(self.manager)
@@ -98,7 +102,8 @@ class XHS:
         self.clipboard_cache: str = ""
         self.queue = Queue()
         self.event = Event()
-        self.server = server
+        self.runner = self.init_server()
+        self.site = None
 
     def __extract_image(self, container: dict, data: Namespace):
         container["下载地址"] = self.image.get_image_link(
@@ -279,3 +284,50 @@ class XHS:
     def read_browser_cookie(value: str | int) -> str:
         return BrowserCookie.get(
             value, domain="xiaohongshu.com") if value else ""
+
+    @staticmethod
+    async def index(request):
+        return web.HTTPFound(REPOSITORY)
+
+    async def handle(self, request):
+        data = await request.post()
+        url = data.get("url")
+        download = data.get("download", False)
+        index = data.get("index")
+        skip = data.get("skip", False)
+        url = await self.__extract_links(url, None)
+        if not url:
+            msg = self.message("提取小红书作品链接失败")
+            data = None
+        else:
+            if data := await self.__deal_extract(url[0], download, index, None, None, not skip, ):
+                msg = self.message("获取小红书作品数据成功")
+            else:
+                msg = self.message("获取小红书作品数据失败")
+                data = None
+        return web.json_response(dict(message=msg, url=url[0], data=data))
+
+    def init_server(self, ):
+        app = web.Application(debug=True)
+        app.router.add_get('/', self.index)
+        app.router.add_post('/xhs/', self.handle)
+        return web.AppRunner(app)
+
+    async def run_server(self, log=None, ):
+        try:
+            await self.start_server(log)
+            while True:
+                await sleep(3600)  # 保持服务器运行
+        except (CancelledError, KeyboardInterrupt):
+            await self.close_server(log)
+
+    async def start_server(self, log=None, ):
+        await self.runner.setup()
+        self.site = web.TCPSite(self.runner, "0.0.0.0")
+        await self.site.start()
+        logging(log, self.message("Web API 服务器已启动！"))
+        logging(log, self.message("服务器主机及端口: {0}".format(self.site.name, )))
+
+    async def close_server(self, log=None, ):
+        await self.runner.cleanup()
+        logging(log, self.message("Web API 服务器已关闭！"))
diff --git a/source/application/video.py b/source/application/video.py
index 89cf042..e8367ce 100644
--- a/source/application/video.py
+++ b/source/application/video.py
@@ -13,5 +13,5 @@ class Video:
 
     @classmethod
     def get_video_link(cls, data: Namespace) -> list:
-        return [Html.format_url(f"https://sns-video-hw.xhscdn.com/{t}")] if (
+        return [Html.format_url(f"https://sns-video-bd.xhscdn.com/{t}")] if (
             t := data.safe_extract(".".join(cls.VIDEO_LINK))) else []
diff --git a/source/module/manager.py b/source/module/manager.py
index 6ed4cdf..94fd878 100644
--- a/source/module/manager.py
+++ b/source/module/manager.py
@@ -50,6 +50,7 @@ class Manager:
             image_download: bool,
             video_download: bool,
             folder_mode: bool,
+            # server: bool,
             transition: Callable[[str], str],
     ):
         self.root = root
@@ -77,6 +78,7 @@ class Manager:
         self.message = transition
         self.image_download = self.check_bool(image_download, True)
         self.video_download = self.check_bool(video_download, True)
+        # self.server = self.check_bool(server, False)
 
     def __check_path(self, path: str) -> Path:
         if not path:
diff --git a/source/module/settings.py b/source/module/settings.py
index c4a10a1..0c0580e 100644
--- a/source/module/settings.py
+++ b/source/module/settings.py
@@ -3,6 +3,8 @@ from json import load
 from pathlib import Path
 from platform import system
 
+from .static import ROOT
+
 __all__ = ['Settings']
 
 
@@ -23,11 +25,11 @@ class Settings:
         "video_download": True,
         "folder_mode": False,
         "language": "zh_CN",
-        "server": False,
+        # "server": False,
     }
     encode = "UTF-8-SIG" if system() == "Windows" else "UTF-8"
 
-    def __init__(self, root: Path):
+    def __init__(self, root: Path = ROOT):
         self.file = root.joinpath("./settings.json")
 
     def run(self):
diff --git a/static/XHS-Downloader.js b/static/XHS-Downloader.js
index 020f956..a1d9c63 100644
--- a/static/XHS-Downloader.js
+++ b/static/XHS-Downloader.js
@@ -1,7 +1,7 @@
 // ==UserScript==
 // @name         XHS-Downloader
 // @namespace    https://github.com/JoeanAmier/XHS-Downloader
-// @version      1.4.3
+// @version      1.4.4
 // @description  提取小红书作品/用户链接，下载小红书无水印图文/视频作品文件
 // @author       JoeanAmier
 // @match        http*://xhslink.com/*
@@ -42,7 +42,7 @@
 2. 提取账号发布、收藏、点赞作品链接时，脚本会尝试自动滚动屏幕直至加载全部作品，滚动检测间隔：2.5 秒
 3. 提取搜索结果作品、用户链接时，脚本会自动滚动屏幕以尝试加载更多内容，滚动屏幕次数：10 次
 4. 可以修改滚动检测间隔、滚动屏幕次数，修改后立即生效；亦可关闭自动滚动屏幕功能，手动滚动屏幕加载内容
-5. XHS-Downloader 用户脚本仅实现可见即可得的数据采集功能，无任何破解功能
+5. XHS-Downloader 用户脚本仅实现可见即可得的数据采集功能，无任何收费功能和破解功能
 
 项目开源地址：https://github.com/JoeanAmier/XHS-Downloader
 `
@@ -132,7 +132,7 @@
 
     const generateVideoUrl = note => {
         try {
-            return [`https://sns-video-hw.xhscdn.com/${note.video.consumer.originVideoKey}`];
+            return [`https://sns-video-bd.xhscdn.com/${note.video.consumer.originVideoKey}`];
         } catch (error) {
             console.error("Error generating video URL:", error);
             return [];