update:增加本地代理功能

Author: hjdhnx

README.md

@@ -8,10 +8,9 @@ nodejs作为服务端的drpy实现。全面升级异步写法
 
 ## 更新记录
 
-### 20241204
+### 20241205
 
-1. 引入crypto-js-wasm.js和使用文档
-2. 增加docs接口可以查看文档md文件的html页面
+1. 完善本地代理功能
 
 [点此查看完整更新记录](docs/updateRecord.md)
 

controllers/api.js

@@ -13,13 +13,21 @@ export default (fastify, options, done) => {
             reply.status(404).send({error: `Module ${moduleName} not found`});
             return;
         }
-
+        const protocol = request.protocol;
+        const hostname = request.hostname;
+        const proxyUrl = `${protocol}://${hostname}${request.url}`.split('?')[0].replace('/api/', '/proxy/') + '/?do=js';
+        // console.log(`proxyUrl:${proxyUrl}`);
+        const env = {
+            proxyUrl, getProxyUrl: function () {
+                return proxyUrl
+            }
+        };
         const pg = Number(query.pg) || 1;
         try {
             // 根据 query 参数决定执行逻辑
             if ('play' in query) {
                 // 处理播放逻辑
-                const result = await drpy.play(modulePath, query.flag, query.play);
+                const result = await drpy.play(modulePath, env, query.flag, query.play);
                 return reply.send(result);
             }
 
@@ -33,35 +41,35 @@ export default (fastify, options, done) => {
                     }
                 }
                 // 分类逻辑
-                const result = await drpy.cate(modulePath, query.t, pg, 1, extend);
+                const result = await drpy.cate(modulePath, env, query.t, pg, 1, extend);
                 return reply.send(result);
             }
 
             if ('ac' in query && 'ids' in query) {
                 // 详情逻辑
-                const result = await drpy.detail(modulePath, query.ids.split(','));
+                const result = await drpy.detail(modulePath, env, query.ids.split(','));
                 return reply.send(result);
             }
 
             if ('wd' in query) {
                 // 搜索逻辑
                 const quick = 'quick' in query ? query.quick : 0;
-                const result = await drpy.search(modulePath, query.wd, quick, pg);
+                const result = await drpy.search(modulePath, env, query.wd, quick, pg);
                 return reply.send(result);
             }
 
             if ('refresh' in query) {
                 // 强制刷新初始化逻辑
-                const refreshedObject = await drpy.init(modulePath, true);
+                const refreshedObject = await drpy.init(modulePath, env, true);
                 return reply.send(refreshedObject);
             }
             if (!('filter' in query)) {
                 query.filter = 1
             }
             // 默认逻辑，返回 home + homeVod 接口
             const filter = 'filter' in query ? query.filter : 1;
-            const resultHome = await drpy.home(modulePath, filter);
-            const resultHomeVod = await drpy.homeVod(modulePath);
+            const resultHome = await drpy.home(modulePath, env, filter);
+            const resultHomeVod = await drpy.homeVod(modulePath, env);
             const result = {
                 ...resultHome,
                 list: resultHomeVod,
@@ -77,5 +85,76 @@ export default (fastify, options, done) => {
             reply.status(500).send({error: `Failed to process module ${moduleName}: ${error.message}`});
         }
     });
+
+    fastify.get('/proxy/:module/*', async (request, reply) => {
+        const moduleName = request.params.module;
+        const query = request.query; // 获取 query 参数
+        const modulePath = path.join(options.jsDir, `${moduleName}.js`);
+        if (!existsSync(modulePath)) {
+            reply.status(404).send({error: `Module ${moduleName} not found`});
+            return;
+        }
+        const proxy_url = request.params['*']; // 捕获整个路径
+        fastify.log.info(`try proxy for ${moduleName} -> ${proxy_url}:`);
+        const protocol = request.protocol;
+        const hostname = request.hostname;
+        const proxyUrl = `${protocol}://${hostname}${request.url}`.split('?')[0].replace(proxy_url, '') + '?do=js';
+        // console.log(`proxyUrl:${proxyUrl}`);
+        const env = {
+            proxyUrl, getProxyUrl: function () {
+                return proxyUrl
+            }
+        };
+        try {
+            const backRespList = await drpy.proxy(modulePath, env, query);
+            const statusCode = backRespList[0];
+            const mediaType = backRespList[1];
+            let content = backRespList[2];
+            const headers = backRespList.length > 3 ? backRespList[3] : null;
+            const toBytes = backRespList.length > 4 ? backRespList[4] : null;
+            // 如果需要转换为字节内容
+            if (toBytes) {
+                try {
+                    if (content.includes('base64,')) {
+                        content = unescape(content.split("base64,")[1]);
+                    }
+                    content = Buffer.from(content, 'base64');
+                } catch (e) {
+                    fastify.log.error(`Local Proxy toBytes error: ${e}`);
+                }
+            }
+
+            // 根据媒体类型来决定如何设置字符编码
+            if (typeof content === 'string') {
+                // 如果返回的是文本内容（例如 JSON 或字符串）
+                if (mediaType.includes('text') || mediaType === 'application/json') {
+                    // 对于文本类型，设置 UTF-8 编码
+                    reply
+                        .code(statusCode)
+                        .type(`${mediaType}; charset=utf-8`)  // 设置编码为 UTF-8
+                        .headers(headers || {})  // 如果有headers, 则加上
+                        .send(content);
+                } else {
+                    // 对于其他类型的文本（例如 XML），直接返回，不指定 UTF-8 编码
+                    reply
+                        .code(statusCode)
+                        .type(mediaType)
+                        .headers(headers || {})
+                        .send(content);
+                }
+            } else {
+                // 如果返回的是二进制内容（例如图片或其他文件）
+                reply
+                    .code(statusCode)
+                    .type(mediaType)  // 使用合适的媒体类型，如 image/png
+                    .headers(headers || {})
+                    .send(content);
+            }
+
+        } catch (error) {
+            fastify.log.error(`Error proxy module ${moduleName}:`, error);
+            reply.status(500).send({error: `Failed to proxy module ${moduleName}: ${error.message}`});
+        }
+    });
     done();
 };

controllers/root.js

@@ -1,5 +1,5 @@
 import path from 'path';
-import {readdirSync, readFileSync} from 'fs';
+import {readdirSync, readFileSync, existsSync} from 'fs';
 import '../utils/marked.min.js';
 
 export default (fastify, options, done) => {
@@ -41,5 +41,22 @@ export default (fastify, options, done) => {
                 </html>
             `);
     });
+
+    // 新增 /favicon.ico 路由
+    fastify.get('/favicon.ico', async (request, reply) => {
+        try {
+            // 设置文件路径
+            const faviconPath = path.join(options.rootDir, 'public', 'favicon.ico');
+
+            // 如果文件存在，返回图片
+            if (existsSync(faviconPath)) {
+                return reply.sendFile('favicon.ico', path.join(options.rootDir, 'public')); // 直接返回图片
+            } else {
+                reply.status(404).send({error: 'Favicon not found'}); // 如果文件不存在，返回 404 错误
+            }
+        } catch (error) {
+            reply.status(500).send({error: 'Failed to fetch favicon', details: error.message});
+        }
+    });
     done();
 };

docs/updateRecord.md

@@ -1,12 +1,42 @@
 # drpyS更新记录
 
+### 20241205
+
+更新至V1.0.2
+
+1. 增加本地代理功能,示例参考_qq.js 用法:在源的各个js函数里(http://192.168.31.49:5757/api/_qq)
+
+```javascript
+let {getProxyUrl} = this;
+let vod_url = getProxyUrl() + '&url=' + 'https://hls09.cntv.myhwcdn.cn/asp/hls/2000/0303000a/3/default/d9b0eaa065934f25abd193d391f731b6/2000.m3u8';
+```
+
 ### 20241204
+
 更新至V1.0.1
+
 1. 引入crypto-js-wasm.js和使用文档
 2. 增加docs接口可以查看文档md文件的html页面
 3. 完成index.js接口剥离，保持主文件的干净。同时导出start和stop方法
 4. 改进本地配置接口，增加外网可用配置。
 5. 支持puppeteer,仅pc可用。如需使用请手动安装puppeteer库，然后drpyS的源里支持使用puppeteerHelper对象。
+6. 添加favicon.ico
+7. 引入全局CryptoJSW对象(海阔暂时会报错无法使用)
+8. 增加本地代理功能，示例(跳转百度):
+
+```javascript
+{
+    proxy_rule: async function (params) {
+        // log(this);
+        let {input, MY_URL} = this;
+        log(`params:`, params);
+        log(`input:${input}`);
+        log(`MY_URL::${MY_URL}`);
+        // return [404, 'text/plain', 'Not Found']
+        return [302, 'text/html', '', {location: 'http://www.baidu.com'}]
+    }
+}
+```
 
 ### 20241203
 

js/_qq.js

@@ -5,7 +5,7 @@ var rule = {
     class_parse: async () => {
         // await CryptoJSW.MD5.loadWasm();
         const rstMD5 = CryptoJSW.MD5('message').toString();
-        console.log('rstMD5:',rstMD5);
+        console.log('rstMD5:', rstMD5);
         console.log('执行了分类获取')
         return [
             {type_id: '1', type_name: '电影'},
@@ -20,8 +20,12 @@ var rule = {
         await local.set(rule.title, 'cookie', 'qwer1234')
 
     },
-    推荐: async () => {
-        console.log('执行了推荐函数')
+    推荐: async function (...args) {
+        console.log('执行了推荐函数');
+        // log(typeof (getProxyUrl));
+        // log(getProxyUrl());
+        let {getProxyUrl} = this;
+        log(getProxyUrl());
         console.log(typeof (jsp))
         console.log(typeof (pdfh))
         console.log('pako:', typeof (pako))
@@ -35,7 +39,12 @@ var rule = {
         // console.log(template.getMubans())
         console.log(typeof template.getMubans)
         return [
-            {vod_name: '测试电影1', vod_pic: '1.png', vod_remarks: '测试描述1', vod_id: 'http://www.1.com'},
+            {
+                vod_name: '测试电影1',
+                vod_pic: '1.png',
+                vod_remarks: '测试描述1',
+                vod_id: getProxyUrl() + '&url=' + 'https://hls09.cntv.myhwcdn.cn/asp/hls/2000/0303000a/3/default/d9b0eaa065934f25abd193d391f731b6/2000.m3u8'
+            },
             {vod_name: '测试电影2', vod_pic: '2.png', vod_remarks: '测试描述2', vod_id: 'http://www.2.com'},
         ]
     },
@@ -61,4 +70,13 @@ var rule = {
     lazy: async () => {
         return template.getMubans()
     },
+    proxy_rule: async function (params) {
+        // log(this);
+        let {input, MY_URL} = this;
+        log(`params:`, params);
+        log(`input:${input}`);
+        log(`MY_URL:${MY_URL}`);
+        // return [404, 'text/plain', 'Not Found']
+        return [302, 'text/html', '', {location: input || 'http://www.baidu.com'}]
+    }
 };