feat: rename in import

Author: sylingd

docs/docs/zh-CN/guide/task.md

@@ -71,7 +71,7 @@ return res.json();
 
 注意：
 * 可以在规则的 重定向至、请求/响应头内容、响应体中使用此语法。
-* 如果响应类型配置为“文本”，则任何 `{$TASK.task1.*}` 均会获取到完整的响应文本。
+* 如果响应类型配置为“文本”，则任何 `{$TASK.task1.*}` 均会获取到完整的响应文本，但不可以直接使用 `{$TASK.task1}`。
 * 当请求失败或尚未完成请求时，该语法会被替换为空字符串。
 
 ## 工具函数

src/pages/options/sections/import-and-export/import-drawer/index.tsx

@@ -90,10 +90,11 @@ export default class ImportDrawer extends React.Component<
     }
   }
 
-  handleConfirm() {
+  async handleConfirm() {
     // 确认导入
-    const queue: any[] = [];
+    const queue: Promise<void>[] = [];
     const tasks = new Set<string>();
+    const rules: BasicRule[] = [];
     this.state.list.forEach((e: any) => {
       // 不导入
       if (e.importAction === 3) {
@@ -113,19 +114,56 @@ export default class ImportDrawer extends React.Component<
         e.enable = true;
       }
       getRuleUsedTasks(e).forEach(task => tasks.add(task));
-      queue.push(Api.saveRule(e));
+      rules.push(e);
     });
     // 处理 task 导入
+    const taskKeyAlias = new Map<string, string>();
     if (tasks.size > 0) {
+      const allTasks = await Api.getTasks();
+      const taskKeys = allTasks.map(x => x.key);
       Array.from(tasks)
         .map(x => this.tasks?.[x])
-        .forEach(t => queue.push(t ? Api.saveTask(t) : Promise.resolve()));
+        .forEach(t => {
+          if (!t) {
+            return Promise.resolve();
+          }
+          while (taskKeys.includes(t.key)) {
+            const newKey = `${t.key}${Math.random().toString(36).substring(1)}`;
+            taskKeyAlias.set(t.key, newKey);
+            t.key = newKey;
+            taskKeys.push(newKey);
+          }
+          queue.push(Api.saveTask(t));
+        });
     }
-    Promise.all(queue).then(() => {
-      // this.imports.status = 0;
-      Toast.success(t('import_success'));
-      this.props.onSuccess?.();
+
+    const replaceTaskKeys = (s: string) => {
+      let res = s;
+      taskKeyAlias.forEach((value, key) => {
+        res = res.replaceAll(`{\$TASK.${key}.}`, `{\$TASK.${value}.}`);
+      });
+      return res;
+    };
+
+    // 处理 rule 导入
+    rules.forEach(e => {
+      if (e.headers) {
+        Object.keys(e.headers).forEach(key => {
+          e.headers![key] = replaceTaskKeys(e.headers![key]);
+        });
+      }
+      if (e.to) {
+        e.to = replaceTaskKeys(e.to);
+      }
+      if (e.body?.value) {
+        e.body.value = replaceTaskKeys(e.body.value);
+      }
+      queue.push(Api.saveRule(e));
     });
+    await Promise.all(queue);
+    // this.imports.status = 0;
+    Toast.success(t('import_success'));
+    this.props.onSuccess?.();
     this.setState({
       list: [],
       visible: false,

tests/e2e/task.spec.mjs

@@ -1,7 +1,12 @@
 import assert from 'node:assert';
 import { setTimeout as sleep } from 'node:timers/promises';
 import { getTask, runTask, saveRule, saveTask } from './scripts/api.mjs';
-import { getPageValue, runInBrowsers, runTest } from './scripts/browser.mjs';
+import {
+  getHeader,
+  getPageValue,
+  runInBrowsers,
+  runTest,
+} from './scripts/browser.mjs';
 import { randStr, testServer } from './scripts/utils.mjs';
 
 describe('Fetch task', () => {
@@ -64,4 +69,31 @@ describe('Fetch task', () => {
         }
       },
     ));
+
+  describe('Use in headers', () =>
+    runTest(
+      ['edge_v2', 'chrome_v3', 'firefox_v2', 'firefox_v3'],
+      async browser => {
+        const { remove } = await saveRule(browser.popup, {
+          name: 'test modify request header',
+          ruleType: 'modifySendHeader',
+          condition: {
+            urlPrefix: testServer,
+          },
+          isFunction: false,
+          enable: true,
+          action: {
+            name: 'X-Test-Header',
+            value: `X-{\$TASK.${key}.value}-Y`,
+          },
+        });
+
+        try {
+          const header = await getHeader(browser.browser);
+          assert.strictEqual(header.X_TEST_HEADER, `X-${value}-Y`);
+        } finally {
+          await remove();
+        }
+      },
+    ));
 });