暂存

src/module/index.ts

@@ -8,9 +8,11 @@ import { useConfig } from '@abearxiong/use-config';
 import { redis } from './redis/redis.ts';
 import { getContentType } from './get-content-type.ts';
 import { sleep } from '@/utils/sleep.ts';
+import { handleProxyRequest } from './proxy.ts';
 const { api, domain, allowedOrigins } = useConfig<{
   api: {
     host: string;
+    port?: number;
   };
   domain: string;
   allowedOrigins: string[];
@@ -19,6 +21,60 @@ const { api, domain, allowedOrigins } = useConfig<{
 const fileStore = useFileStore('upload');
 const noProxyUrl = ['/', '/favicon.ico'];
 export const handleRequest = async (req: http.IncomingMessage, res: http.ServerResponse) => {
+  if (req.url === '/favicon.ico') {
+    return;
+  }
+  if (req.url.startsWith('/api/router')) {
+    // 代理到 http://codeflow.xiongxiao.me/api
+    const _u = new URL(req.url, `http://${api.host}`);
+    // 设置代理请求的目标 URL 和请求头
+    let header: any = {};
+    if (req.headers?.['Authroization']) {
+      header.Authorization = req.headers?.['Authroization'];
+    }
+    if (req.headers?.['Content-Type']) {
+      header['Content-Type'] = req.headers?.['Content-Type'];
+    }
+    const options = {
+      host: _u.hostname,
+      path: req.url,
+      method: req.method,
+      headers: {
+        ...header,
+      },
+    };
+    if (_u.port) {
+      // @ts-ignore
+      options.port = _u.port;
+    }
+    // 创建代理请求
+    const proxyReq = http.request(options, (proxyRes) => {
+      // 将代理服务器的响应头和状态码返回给客户端
+      res.writeHead(proxyRes.statusCode, proxyRes.headers);
+      // 将代理响应流写入客户端响应
+      proxyRes.pipe(res, { end: true });
+    });
+    // 处理代理请求的错误事件
+    proxyReq.on('error', (err) => {
+      console.error(`Proxy request error: ${err.message}`);
+      res.writeHead(500, { 'Content-Type': 'text/plain' });
+      res.write(`Proxy request error: ${err.message}`);
+    });
+    // 处理 POST 请求的请求体（传递数据到目标服务器）
+    req.pipe(proxyReq, { end: true });
+    return;
+  }
+  if (req.url.startsWith('/api/proxy')) {
+    return;
+  }
+  if (req.url.startsWith('/api')) {
+    res.end('not catch api');
+    return;
+  }
+  if (req.url.startsWith('/test')) {
+    handleProxyRequest(req, res);
+    return;
+  }
   const dns = getDNS(req);
   // 配置可以跨域
   // 配置可以访问的域名 localhost, xiongxiao.me
@@ -87,42 +143,7 @@ export const handleRequest = async (req: http.IncomingMessage, res: http.ServerR
     user = _user;
     app = _app;
   }
-  const [_, _api] = req.url.split('/');
-  if (_api === 'api') {
-    // 代理到 http://codeflow.xiongxiao.me/api
-    // 设置代理请求的目标 URL 和请求头
-    let header: any = {};
-    if (req.headers?.['Authroization']) {
-      header.Authorization = req.headers?.['Authroization'];
-    }
-    if (req.headers?.['Content-Type']) {
-      header['Content-Type'] = req.headers?.['Content-Type'];
-    }
-    const options = {
-      host: api.host,
-      path: req.url,
-      method: 'POST',
-      headers: {
-        ...header,
-      },
-    };
-    // 创建代理请求
-    const proxyReq = http.request(options, (proxyRes) => {
-      // 将代理服务器的响应头和状态码返回给客户端
-      res.writeHead(proxyRes.statusCode, proxyRes.headers);
-      // 将代理响应流写入客户端响应
-      proxyRes.pipe(res, { end: true });
-    });
-    // 处理代理请求的错误事件
-    proxyReq.on('error', (err) => {
-      console.error(`Proxy request error: ${err.message}`);
-      res.writeHead(500, { 'Content-Type': 'text/plain' });
-      res.write(`Proxy request error: ${err.message}`);
-    });
-    // 处理 POST 请求的请求体（传递数据到目标服务器）
-    req.pipe(proxyReq, { end: true });
-    return;
-  }
+
   const userApp = new UserApp({ user, app });
   let isExist = await userApp.getExist();
   if (!isExist) {