abp-next-admin2/apps/vben5/packages/@abp/account/src/api/useTokenApi.ts

import type {
  OAuthTokenRefreshModel,
  OAuthTokenResult,
  PasswordTokenRequestModel,
  TokenResult,
} from '../types';

import { useAppConfig } from '@vben/hooks';

import { useRequest } from '@abp/request';

export function useTokenApi() {
  const { cancel, request } = useRequest();
  /**
   * 用户登录
   * @param input 参数
   * @returns 用户token
   */
  async function loginApi(
    input: PasswordTokenRequestModel,
  ): Promise<TokenResult> {
    const { audience, clientId, clientSecret } = useAppConfig(
      import.meta.env,
      import.meta.env.PROD,
    );
    const result = await request<OAuthTokenResult>('/connect/token', {
      data: {
        client_id: clientId,
        client_secret: clientSecret,
        grant_type: 'password',
        scope: audience,
        ...input,
      },
      headers: {
        'Content-Type': 'application/x-www-form-urlencoded',
      },
      method: 'POST',
    });
    return {
      accessToken: result.access_token,
      expiresIn: result.expires_in,
      refreshToken: result.refresh_token,
      tokenType: result.token_type,
    };
  }

  /**
   * 刷新令牌
   * @param input 参数
   * @returns 用户token
   */
  async function refreshTokenApi(input: OAuthTokenRefreshModel) {
    const { audience, clientId, clientSecret } = useAppConfig(
      import.meta.env,
      import.meta.env.PROD,
    );
    const result = await request<OAuthTokenResult>('/connect/token', {
      data: {
        client_id: clientId,
        client_secret: clientSecret,
        grant_type: 'refresh_token',
        refresh_token: input.refreshToken,
        scope: audience,
      },
      headers: {
        'Content-Type': 'application/x-www-form-urlencoded',
      },
      method: 'POST',
    });
    return {
      accessToken: result.access_token,
      expiresIn: result.expires_in,
      refreshToken: result.refresh_token,
      tokenType: result.token_type,
    };
  }

  return {
    cancel,
    loginApi,
    refreshTokenApi,
  };
}