Blame view

Yi.Vben5.Vue3/apps/backend-mock/utils/jwt-utils.ts 1.55 KB
515fceeb   “wangming”   框架初始化
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
  import type { EventHandlerRequest, H3Event } from 'h3';
  
  import jwt from 'jsonwebtoken';
  
  import { UserInfo } from './mock-data';
  
  // TODO: Replace with your own secret key
  const ACCESS_TOKEN_SECRET = 'access_token_secret';
  const REFRESH_TOKEN_SECRET = 'refresh_token_secret';
  
  export interface UserPayload extends UserInfo {
    iat: number;
    exp: number;
  }
  
  export function generateAccessToken(user: UserInfo) {
    return jwt.sign(user, ACCESS_TOKEN_SECRET, { expiresIn: '7d' });
  }
  
  export function generateRefreshToken(user: UserInfo) {
    return jwt.sign(user, REFRESH_TOKEN_SECRET, {
      expiresIn: '30d',
    });
  }
  
  export function verifyAccessToken(
    event: H3Event<EventHandlerRequest>,
  ): null | Omit<UserInfo, 'password'> {
    const authHeader = getHeader(event, 'Authorization');
    if (!authHeader?.startsWith('Bearer')) {
      return null;
    }
  
    const token = authHeader.split(' ')[1];
    try {
      const decoded = jwt.verify(token, ACCESS_TOKEN_SECRET) as UserPayload;
  
      const username = decoded.username;
      const user = MOCK_USERS.find((item) => item.username === username);
      const { password: _pwd, ...userinfo } = user;
      return userinfo;
    } catch {
      return null;
    }
  }
  
  export function verifyRefreshToken(
    token: string,
  ): null | Omit<UserInfo, 'password'> {
    try {
      const decoded = jwt.verify(token, REFRESH_TOKEN_SECRET) as UserPayload;
      const username = decoded.username;
      const user = MOCK_USERS.find((item) => item.username === username);
      const { password: _pwd, ...userinfo } = user;
      return userinfo;
    } catch {
      return null;
    }
  }