abp-vnext-pro/vben28/src/store/modules/user.ts

import type { UserInfo } from '/#/store';
import type { ErrorMessageMode } from '/#/axios';
import { defineStore } from 'pinia';
import { store } from '/@/store';
import { RoleEnum } from '/@/enums/roleEnum';
import { PageEnum } from '/@/enums/pageEnum';
import {
  ROLES_KEY,
  TOKEN_KEY,
  USER_INFO_KEY,
  ABP_LOCALE_KEY,
  ABP_TETANT_KEY,
} from '/@/enums/cacheEnum';
import { getAuthCache, setAuthCache } from '/@/utils/auth';
import {  LoginParams } from '/@/api/sys/model/userModel';
import { login, getAbpApplicationConfiguration, id4, github } from '/@/api/sys/user';
import { useI18n } from '/@/hooks/web/useI18n';
import { useMessage } from '/@/hooks/web/useMessage';
import { router } from '/@/router';
import { usePermissionStore } from '/@/store/modules/permission';
import { LoginInput } from '/@/services/ServiceProxies';
import jwt_decode from 'jwt-decode';
import { useSignalR } from '/@/hooks/web/useSignalR';
import { useOidcLogout } from '/@/views/sys/login/useLogin';
interface UserState {
  userInfo: Nullable<UserInfo>;
  token?: string;
  id_token?: string;
  roleList: RoleEnum[];
  sessionTimeout?: boolean;
  lastUpdateTime: number;
  language: string;
  tenantId: string;
}

export const useUserStore = defineStore({
  id: 'app-user',
  state: (): UserState => ({
    // user info
    userInfo: null,
    // token
    token: undefined,
    id_token: undefined,
    // roleList
    roleList: [],
    // Whether the login expired
    sessionTimeout: false,
    // Last fetch time
    lastUpdateTime: 0,
    language: '',
    tenantId: '',
  }),
  getters: {
    getUserInfo(): UserInfo {
      return this.userInfo || getAuthCache<UserInfo>(USER_INFO_KEY) || {};
    },
    getToken(): string {
      return this.token || getAuthCache<string>(TOKEN_KEY);
    },

    getRoleList(): RoleEnum[] {
      return this.roleList.length > 0 ? this.roleList : getAuthCache<RoleEnum[]>(ROLES_KEY);
    },
    getSessionTimeout(): boolean {
      return !!this.sessionTimeout;
    },
    getLastUpdateTime(): number {
      return this.lastUpdateTime;
    },
    getLanguage(): string {
      return this.language || getAuthCache<string>(ABP_LOCALE_KEY);
    },
    getTenant(): string {
      return this.tenantId || getAuthCache<string>(ABP_TETANT_KEY);
    },
    checkUserLoginExpire(): boolean {
      try {
        const userStore = useUserStoreWithOut();
        const token = userStore.getToken;
        if (!token) return true;
        const decoded: any = jwt_decode(token);
        // 获取当前时间戳
        let currentTimeStamp = new Date().getTime() / 1000;
        if (currentTimeStamp >= decoded.exp) {
          return true;
        } else {
          return false;
        }
      } catch (error) {
        return true;
      }
    },
  },
  actions: {
    setToken(info: string | undefined) {
      this.token = info;
      setAuthCache(TOKEN_KEY, info);
    },

    setRoleList(roleList: RoleEnum[]) {
      this.roleList = roleList;
      setAuthCache(ROLES_KEY, roleList);
    },
    setUserInfo(info: UserInfo) {
      this.userInfo = info;
      this.lastUpdateTime = new Date().getTime();
      setAuthCache(USER_INFO_KEY, info);
    },
    setTenant(tenantId: string) {
      this.tenantId = tenantId;
      setAuthCache(ABP_TETANT_KEY, tenantId);
    },
    setSessionTimeout(flag: boolean) {
      this.sessionTimeout = flag;
    },
    resetState() {
      this.userInfo = null;
      this.roleList = [];
      this.sessionTimeout = false;
      this.tenantId = '';
      this.setToken(undefined);
    },

    async login(
      params: LoginParams & {
        goHome?: boolean;
        mode?: ErrorMessageMode;
      }
    ) {
      try {
        const { goHome = true } = params;
        const request = new LoginInput();
        request.name = params.username;
        request.password = params.password;
        const data = await login(request);
        this.setToken(data.token as string);
        this.setUserInfo({
          userId: data.id as string,
          username: data.userName as string,
          realName: data.name as string,
          roles: data.roles as [],
          avatar: '',
          isSts: false,
          idToken: '',
        });

        await this.getAbpApplicationConfigurationAsync();
        goHome && (await router.replace(PageEnum.BASE_HOME));
        return null;
      } catch (error) {
        router.replace(PageEnum.BASE_LOGIN);
        return null;
      }
    },
    async getAbpApplicationConfigurationAsync() {
      const application = await getAbpApplicationConfiguration();
      const permissionStore = usePermissionStore();

      const grantPolicy = Object.keys(application.auth?.grantedPolicies as object);
      if (grantPolicy.length == 0) {
        router.replace(PageEnum.BASE_LOGIN);
        return;
      }
      permissionStore.setPermCodeList(grantPolicy);
    },

    async id4Login(token: string, id_token: string) {
      console.log(token, id_token);
      // try {
      //   // 获取token中的租户信息
      //   const decoded: any = jwt_decode(token);
      //   if (decoded.tenantid == undefined) {
      //     this.setTenant('');
      //   } else {
      //     this.setTenant(decoded.tenantid);
      //   }

      //   const data = await id4(token);
      //   this.setToken(data.token as string);

      //   this.setUserInfo({
      //     userId: decoded.sub as string,
      //     username: data.userName as string,
      //     realName: data.name as string,
      //     roles: data.roles as [],
      //     avatar: '',
      //     isSts: true,
      //     idToken: id_token,
      //   });
      //   await this.getAbpApplicationConfigurationAsync();
      //   await router.replace(PageEnum.BASE_HOME);
      // } catch (error) {
      //   this.setTenant('');
      //   router.replace(PageEnum.BASE_LOGIN);
      // }
    },

    async githubLogin(code: string) {
      // try {
      //   const data = await github(code);
      //   this.setToken(data.token as string);
      //   this.setTenant('');
      //   this.setUserInfo({
      //     userId: data.id as string,
      //     username: data.userName as string,
      //     realName: data.name as string,
      //     roles: data.roles as [],
      //     avatar: '',
      //     isSts: false,
      //     idToken: '',
      //   });
      //   await this.getAbpApplicationConfigurationAsync();
      //   await router.replace(PageEnum.BASE_HOME);
      // } catch (error) {
      //   this.setTenant('');
      //   router.replace(PageEnum.BASE_LOGIN);
      // }
      console.log(code);
    },

  
    /**
     * @description: logout
     */
    async logout(goLogin = false) {
      try {
        if (this.userInfo?.isSts) {
          await useOidcLogout();
        } else {
          this.resetState();
          goLogin && router.push(PageEnum.BASE_LOGIN);
        }
      } catch (ex) {
        console.log(ex);
      }
    },

    /**
     * @description: Confirm before logging out
     */
    confirmLoginOut() {
      const { createConfirm } = useMessage();
      const { t } = useI18n();
      createConfirm({
        iconType: 'warning',
        title: t('sys.app.logoutTip'),
        content: t('sys.app.logoutMessage'),
        onOk: async () => {
          await this.logout(true);
          const { closeConnect } = useSignalR();
          closeConnect();
        },
      });
    },
  },
});

// Need to be used outside the setup
export function useUserStoreWithOut() {
  return useUserStore(store);
}