ieplus-app/src/hooks/useTransferPage.ts

import { Transfer } from "@/types";
import { onLoad, onUnload } from "@dcloudio/uni-app";
import { routes } from "@/common/routes";
export const useTransferPage = <T1 = any, T2 = any>() => {
  const funcMap = {
    redirectTo: uni.redirectTo,
    reLaunch: uni.reLaunch,
    switchTab: uni.switchTab,
    // navigateTo: uni.navigateTo,
    // navigateBack: uni.navigateBack
  } as const;
  let prevData = ref<T1>({} as T1);
  let transferResolver: ((value: unknown) => void) | null = null;
  let eventChannel: UniApp.EventChannel | null = null;
  const eventId = ref<string | null>(null);
  // onMounted(() => {
  // 使用instance获取getOpenerEventChannel在非页面级组件中会报错
  // const instance = getCurrentInstance().proxy
  // if (instance) {
  //   eventChannel = instance.getOpenerEventChannel();
  // }
  // });
  type Callback = (() => void) | null;
  type TransferOptions = {
    data?: T2 | null,
    type?: Transfer.TransferType,
    bigData?: Record<string, any> | null,
    callback?: Callback,
    crossTrigger?: boolean // 当出现A navTo C redirectTo B又需要把参数回传给 A 时，可设置此参数为 true 实现跨页面传递
  }

  const getEventChannel = () => {
    const pages = getCurrentPages();
    const page = pages[pages.length - 1];
    return (page as any).getOpenerEventChannel();
  }

  onLoad((opt) => {
    eventChannel = getEventChannel();
    emit('onPageOpen');
    for (const key in opt) {
      if (opt.hasOwnProperty(key) && !['data', 'storageKey'].includes(key)) {
        prevData.value[key] = decodeURIComponent(opt[key]);
      }
    }
    if (!opt) return;
    let originData = opt.data;
    if (opt.data) {
      if (opt.data.startsWith('%257B')) {
        originData = decodeURIComponent(opt.data);
      }
      const data = JSON.parse(decodeURIComponent(originData || '{}'));
      Object.assign(prevData.value, data);
    }
    const storageKey = opt.storageKey;
    if (storageKey) {
      const storageData = uni.getStorageSync(storageKey);
      Object.assign(prevData.value, storageData);
    }
  });
  onUnload(() => {
    eventId.value = null;
  });
  function getTransferUrl(url: string, data: any, bigData: any) {
    const urlParams = {
      data: data ? encodeURIComponent(JSON.stringify(data)) : null,
      storageKey: bigData ? 'transferBigData' : null
    };

    // 处理URL中已存在的参数
    const [baseUrl, existingQuery] = url.split('?');
    const existingParams = {};

    // 解析现有参数
    if (existingQuery) {
      existingQuery.split('&').forEach(param => {
        const [key, value] = param.split('=');
        if (key) {
          // 为 existingParams 显式声明类型为 Record<string, string>
          (existingParams as Record<string, string>)[key] = value;
        }
      });
    }

    // 合并新参数
    Object.entries(urlParams)
      .filter(([_, value]) => value)
      .forEach(([key, value]) => {
        (existingParams as Record<string, string>)[key] = value as string;
      });

    // 构建查询字符串
    const queryString = Object.entries(existingParams)
      .map(([key, value]) => `${key}=${value}`)
      .join('&');

    return queryString ? `${baseUrl}?${queryString}` : baseUrl;
  }

  function transferTo(url: string, { data = null, type = 'navigateTo', bigData = null, callback = null, crossTrigger = false }: TransferOptions = {}) {
    return new Promise((resolve, reject) => {
      transferResolver = resolve;
      if (bigData) {
        uni.setStorageSync('transferBigData', bigData);
      } else {
        // clear big data by force
        uni.removeStorageSync('transferBigData');
      }
      let nextUrl = getTransferUrl(url, data, bigData);
      // 单独处理，函数签名不一样
      if (type === 'navigateTo') {
        if (crossTrigger) {
          eventId.value = Date.now().toString();
          uni.$once(`${eventId.value}_TransferEnd`, (data: any) => {
            transferResolver?.(data);
          });
          nextUrl += `&crossPageEventId=${eventId.value}`;
        }
        uni.navigateTo({
          url: nextUrl,
          events: {
            transferEnd: (data: any) => {
              transferResolver?.(data);
            },
            onPageOpen: () => {
              if (callback) {
                callback?.();
              }
            }
          },
          fail: (err) => {
            eventId.value = null;
            console.log('transferTo fail', err);
          }
        });
      } else {
        const routeFunc = funcMap[type as keyof typeof funcMap];
        if (prevData.value.crossPageEventId) {
          nextUrl += `&crossPageEventId=${prevData.value.crossPageEventId}`;
        }
        routeFunc({
          url: nextUrl,
          fail: (err) => {
            console.log('transferTo fail', err);
            reject(err);
          }
        });
      }


    });
  }
  function transferBack(data: any = null, delta = 1) {
    emit('transferEnd', data);
    if (prevData.value.crossPageEventId) {
      uni.$emit(`${prevData.value.crossPageEventId}_TransferEnd`, data);
    }
    // force clear big data
    uni.removeStorageSync('transferBigData');
    const pages = getCurrentPages();
    if (pages.length === 1) {
      switchTab('/pagesMain/pages/splash/splash');
    } else {
      uni.navigateBack({
        delta: delta,
        success: () => { },
        fail: (err) => {
          switchTab('/pagesMain/pages/splash/splash');
        }
      });
    }
  }
  function switchTab(url: string) {
    uni.switchTab({
      url: url,
      fail: (err) => {
        console.log(err);
      }
    });
  }

  function emit(event: string, data = null) {
    if (eventChannel?.emit) {
      eventChannel.emit(event, data);
    } else {
      console.warn('上级页面丢失，无法触发transferEnd');
    }
  }

  return {
    prevData,
    transferTo,
    transferBack,
    routes,
    eventChannel
  }
};