@gylautorun/dev-proxy-cookie 1.0.2 → 1.0.4

package/src/proxy/vue-proxy-config.ts

@@ -10,6 +10,7 @@ import type { IncomingMessage } from 'http';
 import * as path from 'path';
 import { CookieReader, watchCookieFile, shouldEnableWatch } from '../utils';
 import { applyDevCookieHeader } from './apply-dev-cookie-header';
+import { applyDevAuthentications, type AuthenticationItem } from './apply-dev-authentications';
 
 /** Options for {@link createFileCookieGetter}. */
 export interface CreateFileCookieGetterOptions {
@@ -50,6 +51,19 @@ export interface VueProxyConfigOptions {
   getCookie?: () => string;
   /** 是否输出调试日志 */
   debug?: boolean;
+  /** 
+   * 是否使用 Cookie 文件中的 Cookie 注入到请求中
+   * - true: 使用文件中的 Cookie（默认）
+   * - false: 不注入 Cookie，使用浏览器发送的 Cookie
+   * 
+   * 当使用账号密码登录时，设置为 false，避免覆盖浏览器的登录 Cookie
+   */
+  useCookie?: boolean;
+  /** 
+   * 自定义鉴权信息数组，每个元素是一个键值对对象，会被注入到请求头中
+   * 例如: [{ 'ticket': 'xxxx' }, { 'X-Custom-Token': 'yyyy' }]
+   */
+  authentications?: AuthenticationItem[];
   /** 自定义请求头 */
   headers?: Record<string, string>;
   /** 是否启用 WebSocket 代理 */
@@ -100,6 +114,8 @@ export function createVueProxyConfig(
   const {
     getCookie,
     debug = false,
+    useCookie = true,
+    authentications = [],
     headers = {},
     ws = false,
     changeOrigin = true,
@@ -114,13 +130,25 @@ export function createVueProxyConfig(
     secure,
     headers,
     onProxyReq: (proxyReq: any, req: IncomingMessage) => {
-      const cookie = getCookie ? getCookie() : '';
-      if (cookie) {
-        applyDevCookieHeader(proxyReq, cookie);
+      const reqPath = req.url || '/';
+      
+      if (useCookie) {
+        const cookie = getCookie ? getCookie() : '';
+        if (cookie) {
+          applyDevCookieHeader(proxyReq, cookie);
+        }
+        if (debug) {
+          console.log('[Proxy Request]', reqPath, req.method, cookie ? '(with cookie)' : '(no cookie)');
+        }
+      } else if (debug) {
+        console.log('[Proxy Request]', reqPath, req.method, '(useCookie is false, skipping cookie injection)');
       }
-      if (debug) {
-        const reqPath = req.url || '/';
-        console.log('[Proxy Request]', reqPath, req.method, cookie ? '(with cookie)' : '(no cookie)');
+
+      if (authentications && authentications.length > 0) {
+        applyDevAuthentications(proxyReq, authentications);
+        if (debug) {
+          console.log('[Proxy Request]', reqPath, req.method, '(with authentications)');
+        }
       }
     },
     onError: customOnError || ((err: Error) => {
@@ -209,13 +237,13 @@ export interface AutoProxyConfigOptions extends VueProxyConfigOptions {
  * @returns Vue CLI 代理配置对象映射
  */
 export function createAutoProxyConfig(options: AutoProxyConfigOptions): Record<string, ProxyConfig> {
-  const { target, ignorePaths = [], includePaths = [], additionalProxies = {}, getCookie, debug, headers } = options;
+  const { target, ignorePaths = [], includePaths = [], additionalProxies = {}, getCookie, debug, headers, useCookie = true, authentications = [] } = options;
 
   const result: Record<string, ProxyConfig> = {};
 
   if (includePaths.length > 0) {
     for (const proxyPath of includePaths) {
-      result[proxyPath] = createVueProxyConfig(target, { getCookie, debug, headers });
+      result[proxyPath] = createVueProxyConfig(target, { getCookie, debug, headers, useCookie, authentications });
     }
   } else {
     const defaultProxy: ProxyConfig = {
@@ -231,12 +259,23 @@ export function createAutoProxyConfig(options: AutoProxyConfigOptions): Record<s
           return;
         }
 
-        const cookie = getCookie ? getCookie() : '';
-        if (cookie) {
-          applyDevCookieHeader(proxyReq, cookie);
+        if (useCookie) {
+          const cookie = getCookie ? getCookie() : '';
+          if (cookie) {
+            applyDevCookieHeader(proxyReq, cookie);
+          }
+          if (debug) {
+            console.log('[Proxy Request]', reqPath, req.method, cookie ? '(with cookie)' : '(no cookie)');
+          }
+        } else if (debug) {
+          console.log('[Proxy Request]', reqPath, req.method, '(useCookie is false, skipping cookie injection)');
         }
-        if (debug) {
-          console.log('[Proxy Request]', reqPath, req.method, cookie ? '(with cookie)' : '(no cookie)');
+
+        if (authentications && authentications.length > 0) {
+          applyDevAuthentications(proxyReq, authentications);
+          if (debug) {
+            console.log('[Proxy Request]', reqPath, req.method, '(with authentications)');
+          }
         }
       },
       onError: (err: Error) => {
@@ -247,7 +286,7 @@ export function createAutoProxyConfig(options: AutoProxyConfigOptions): Record<s
   }
 
   for (const [proxyPath, proxyTarget] of Object.entries(additionalProxies)) {
-    result[proxyPath] = createVueProxyConfig(proxyTarget, { getCookie, debug, headers });
+    result[proxyPath] = createVueProxyConfig(proxyTarget, { getCookie, debug, headers, useCookie, authentications });
   }
 
   return result;