ykt-team-wxapp/store/account.js

import { ref, watch } from "vue";
import { defineStore } from "pinia";
import api from '@/utils/api';
import { toast } from '@/utils/widget';
import { initGlobalTIM, globalTimChatManager } from "@/utils/tim-chat.js";
import { globalUnreadListenerManager } from "@/utils/global-unread-listener.js";
import { isSh6CorpId, normalizeCorpId, normalizeCorpIdFields, resolveApiContext } from "@/utils/api-base-config";
import { setMiniappSessionRefresher } from "@/utils/http";
const env = __VITE_ENV__;

export default defineStore("accountStore", () => {
  const appid = env.MP_WX_APP_ID;
  const account = ref(null);
  const loading = ref(false)
  const isIMInitialized = ref(false);
  const openid = ref("");
  const externalUserId = ref('');
  const externalUserIdMap = ref({});
  const teams = ref([]);
  const teamsPromise = ref(null);

  function hasValidMiniappSession() {
    if (!isSh6CorpId(resolveApiContext().corpId)) return true;
    const token = uni.getStorageSync("miniappSessionToken");
    const expiresAt = Number(uni.getStorageSync("miniappSessionExpiresAt"));
    return Boolean(token && (!expiresAt || expiresAt > Date.now() + 30 * 1000));
  }

  async function login(phoneCode = '', options = {}) {
    if (loading.value) return;
    if (account.value && account.value.mobile && !options.forceVerify && hasValidMiniappSession()) {
      return account.value
    }
    loading.value = true;
    try {
      const { code } = await uni.login({
        appid,
        provider: "weixin",
        scope: "snsapi_base",
      });
      if (code) {
        const res = await api('wxAppLogin', {
          appId: appid,
          phoneCode,
          code
        });
        loading.value = false
        if (res.success && res.data) {
          const { sessionToken, sessionExpiresAt, sessionKey, ...safeAccount } = res.data;
          if (sessionToken) {
            uni.setStorageSync("miniappSessionToken", sessionToken);
            uni.setStorageSync("miniappSessionExpiresAt", sessionExpiresAt || 0);
          }
          const normalizedAccount = normalizeCorpIdFields(safeAccount);
          account.value = normalizedAccount;
          openid.value = normalizedAccount.openid;

          // 保存账户信息和 openId 到本地存储
          uni.setStorageSync('account', normalizedAccount);
          uni.setStorageSync('openid', normalizedAccount.openid);

          // initIMAfterLogin(openid.value)
          return normalizedAccount
        } else {
          account.value = null;
          openid.value = '';
          uni.removeStorageSync('account');
          uni.removeStorageSync('openid');
          return
        }
      }
      if (!options.silent) toast('登录失败，请重新登录');
    } catch (e) {
      if (!options.silent) toast('登录失败，请重新登录');
    }
    loading.value = false
  }

  async function initIMAfterLogin() {
    if (isIMInitialized.value) {
      return true;
    }
    try {
      // 使用 openid 作为 userID 初始化 IM
      const userID = openid.value || uni.getStorageSync('openid');
      if (!userID) {
        console.log('未获取到有效的 userId，跳过 IM 初始化');
        return false;
      }

      console.log('开始初始化 IM，userID:', userID);
      const success = await initGlobalTIM(userID);

      if (!success) {
        console.log('initGlobalTIM 返回失败，跳过 IM 初始化');
        return false;
      }

      // 验证 TIM 实例是否真正创建成功
      if (!globalTimChatManager || !globalTimChatManager.tim) {
        console.log('IM 初始化后 TIM 实例不存在，跳过 IM 初始化');
        return false;
      }

      isIMInitialized.value = true;
      console.log('IM 初始化成功');

      // IM 初始化成功后，设置全局未读消息监听
      // globalUnreadListenerManager.setup();

      return true;
    } catch (error) {
      console.log('IM初始化异常，跳过 IM 初始化:', error.message);
      return false;
    }
  }
  // 退出登录
  async function logout() {
    try {
      // 退出腾讯IM
      if (globalTimChatManager && globalTimChatManager.tim) {
        console.log('开始退出腾讯IM');
        await globalTimChatManager.destroy();
        console.log('腾讯IM退出成功');
      }

      // 清除全局未读监听
      if (globalUnreadListenerManager.isInitialized) {
        // globalUnreadListenerManager.destroy();
      }
    } catch (error) {
      console.error('退出腾讯IM失败:', error);
    }

    // 清空账户信息
    account.value = null;
    uni.removeStorageSync("miniappSessionToken");
    uni.removeStorageSync("miniappSessionExpiresAt");
    openid.value = "";
    isIMInitialized.value = false;

    // 清除本地存储
    uni.removeStorageSync('account');
    uni.removeStorageSync('openid');

    // 清除 tabBar 徽章
    uni.removeTabBarBadge({
      index: 1
    });
  }

  async function searchTeams() {
    const res = await api('getWxappRelateTeams', { openid: account.value.openid });
    teams.value = res && Array.isArray(res.data) ? normalizeCorpIdFields(res.data) : [];
    return teams.value;
  }

  async function getTeams() {
    if (!teamsPromise.value) {
      teamsPromise.value = searchTeams();
    }
    await teamsPromise.value;
    teamsPromise.value = null;
    return teams.value;
  }

  async function getExternalUserId(corpId) {
    const normalizedCorpId = normalizeCorpId(corpId);
    const unionid = account.value?.unionid;
    const openid = account.value?.openid;
    const miniAppId = account.value?.openid;
    if (!(normalizedCorpId && unionid && openid)) {
      externalUserId.value = '';
      return
    };
    const key = `${normalizedCorpId}_${unionid}_${openid}_${miniAppId}`;
    if (externalUserIdMap.value[key]) {
      return externalUserIdMap.value[key]
    }
    const res = await api('getUnionidToExternalUserid', { unionid, openid, corpId: normalizedCorpId, miniAppId }, false);
    const id = res && res.success && typeof res.data === 'string' && res.data.trim() ? res.data.trim() : '';
    externalUserId.value = id;
    if (id) {
      externalUserIdMap.value[key] = id;
    }
    return id;
  }

  watch(openid, (n, o) => {
    if (n && !o) {
      getTeams();
    }
  }, { immediate: true })

  setMiniappSessionRefresher(() => login('', { forceVerify: true, silent: true }));

  return { account, teams, login, hasValidMiniappSession, initIMAfterLogin, logout, openid, isIMInitialized, externalUserId, getExternalUserId, getTeams }
})