abp-next-admin/vueJs/src/store/modules/user.ts

import { VuexModule, Module, Action, Mutation, getModule } from 'vuex-module-decorators'
import UserApiService, { UserLoginData, UserLoginPhoneData } from '@/api/users'
import TenantService from '@/api/tenant'
import { getItem, setItem, removeItem } from '@/utils/localStorage'
import { resetRouter } from '@/router'
import { TagsViewModule } from './tags-view'
import { removeTenant, setTenant } from '@/utils/sessions'
import { PermissionModule } from '@/store/modules/permission'
import { AbpModule } from '@/store/modules/abp'
import store from '@/store'

export interface IUserState {
  token: string
  refreshToken: string
  id: string
  name: string
  roles: string[]
  email: string
}

const scopeKey = 'vue_typescript_admin_scope'
const tokenKey = 'vue_typescript_admin_token'
const refreshTokenKey = 'vue_typescript_admin_refresh_token'

@Module({ dynamic: true, store, name: 'user' })
class User extends VuexModule implements IUserState {
  public token = getItem(tokenKey)
  public refreshToken = getItem(refreshTokenKey)
  public id = ''
  public name = ''
  public roles: string[] = []
  public email = ''

  @Mutation
  private SET_TOKEN(token: string) {
    this.token = token
    setItem(tokenKey, token)
  }

  @Mutation
  private SET_REFRESHTOKEN(token: string) {
    this.refreshToken = token
    setItem(refreshTokenKey, token)
  }

  @Mutation
  private SET_ID(id: string) {
    this.id = id
  }

  @Mutation
  private SET_NAME(name: string) {
    this.name = name
  }

  @Mutation
  private SET_ROLES(roles: string[]) {
    this.roles = roles
  }

  @Mutation
  private SET_EMAIL(email: string) {
    this.email = email
  }

  @Action({ rawError: true })
  public async Login(userInfo: { tenantName: string | undefined, username: string, password: string}) {
    if (userInfo.tenantName) {
      await this.PreLogin(userInfo.tenantName)
    }
    const userLoginData = new UserLoginData()
    userLoginData.userName = userInfo.username
    userLoginData.password = userInfo.password
    const loginResult = await UserApiService.userLogin(userLoginData)
    const token = loginResult.token_type + ' ' + loginResult.access_token
    this.SET_TOKEN(token)
    this.SET_REFRESHTOKEN(loginResult.refresh_token)
    await this.PostLogin()
  }

  @Action({ rawError: true })
  public async PhoneLogin(userInfo: { tenantName: string | undefined, phoneNumber: string, verifyCode: string}) {
    if (userInfo.tenantName) {
      await this.PreLogin(userInfo.tenantName)
    }
    const userLoginData = new UserLoginPhoneData()
    userLoginData.phoneNumber = userInfo.phoneNumber
    userLoginData.verifyCode = userInfo.verifyCode
    const loginResult = await UserApiService.userLoginWithPhone(userLoginData)
    const token = loginResult.token_type + ' ' + loginResult.access_token
    this.SET_TOKEN(token)
    this.SET_REFRESHTOKEN(loginResult.refresh_token)
    await this.PostLogin()
  }

  @Action
  public ResetToken() {
    removeItem(tokenKey)
    removeTenant()
    this.SET_TOKEN('')
    this.SET_ROLES([])
  }

  @Action
  public async GetUserInfo() {
    if (this.token === '') {
      throw Error('GetUserInfo: token is undefined!')
    }
    const userInfo = await UserApiService.getUserInfo()
    this.SET_ID(userInfo.sub)
    this.SET_NAME(userInfo.name)
    this.SET_EMAIL(userInfo.email)
    return userInfo
  }

  @Action
  public async LogOut() {
    if (this.token === '') {
      throw Error('LogOut: token is undefined!')
    }
    const token = getItem(refreshTokenKey)
    if (token) {
      await UserApiService.userLogout(token)
    }
    this.SET_TOKEN('')
    this.SET_REFRESHTOKEN('')
    this.SET_ROLES([])
    removeItem(tokenKey)
    removeItem(refreshTokenKey)
    removeTenant()
    resetRouter()
    // Reset visited views and cached views
    TagsViewModule.delAllViews()
    PermissionModule.ResetPermissions()
    PermissionModule.ResetRoutes()
  }

  @Action
  public RefreshSession() {
    return new Promise((resolve, reject) => {
      const refreshToken = getItem(refreshTokenKey)
      const token = getItem(tokenKey)
      if (refreshToken) {
        UserApiService.refreshToken(token, refreshToken).then(result => {
          const token = result.token_type + ' ' + result.access_token
          this.SET_TOKEN(token)
          this.SET_REFRESHTOKEN(result.refresh_token)
          return resolve(token)
        }).catch(error => {
          return reject(error)
        })
      } else {
        return resolve('')
      }
    })
  }

  @Action
  private async PreLogin(tenantName: string) {
    const tenantResult = await TenantService.getTenantByName(tenantName)
    setTenant(tenantResult.tenantId)
  }

  @Action
  private async PostLogin() {
    await AbpModule.GetAbpConfiguration()
  }
}

export const UserModule = getModule(User)