Krishya | 6bf199c | 2025-06-06 21:14:23 +0800 | [diff] [blame] | 1 | |
22301009 | 207e2db | 2025-06-09 00:27:28 +0800 | [diff] [blame] | 2 | // import React, { createContext, useContext, useState, useEffect } from 'react'; |
| 3 | |
| 4 | // const avatarBaseUrl = process.env.REACT_APP_AVATAR_BASE_URL || ''; |
| 5 | |
| 6 | // export const UserContext = createContext(); |
| 7 | |
| 8 | // export const UserProvider = ({ children }) => { |
| 9 | // const [user, setUser] = useState(null); |
| 10 | // const [loading, setLoading] = useState(true); |
| 11 | |
| 12 | // const formatUser = (raw) => { |
| 13 | // if (!raw) return null; |
| 14 | // return { |
| 15 | // userId: raw.userId || raw.id || null, |
| 16 | // username: raw.username || '匿名用户', |
| 17 | // avatarUrl: raw.avatarUrl ? `${avatarBaseUrl}${raw.avatarUrl}` : null, |
| 18 | // ...raw, |
| 19 | // }; |
| 20 | // }; |
| 21 | |
| 22 | // useEffect(() => { |
| 23 | // const storedUser = localStorage.getItem('user'); |
| 24 | // if (storedUser) { |
| 25 | // try { |
| 26 | // const parsed = JSON.parse(storedUser); |
| 27 | // setUser(formatUser(parsed)); |
| 28 | // } catch (err) { |
| 29 | // console.error('本地用户信息解析失败:', err); |
| 30 | // localStorage.removeItem('user'); |
| 31 | // setUser(null); |
| 32 | // } |
| 33 | // } else { |
| 34 | // // 可以默认不自动登录,退出就是空 |
| 35 | // setUser(null); |
| 36 | // } |
| 37 | // setLoading(false); |
| 38 | // }, []); |
| 39 | |
| 40 | // const saveUser = (userData) => { |
| 41 | // if (userData && (userData.userId || userData.id)) { |
| 42 | // localStorage.setItem('user', JSON.stringify(userData)); |
| 43 | // setUser(formatUser(userData)); |
| 44 | // } else { |
| 45 | // console.error('无效的用户数据:', userData); |
| 46 | // } |
| 47 | // }; |
| 48 | |
| 49 | // const logout = () => { |
| 50 | // localStorage.removeItem('user'); |
| 51 | // setUser(null); |
| 52 | // }; |
| 53 | |
| 54 | // return ( |
| 55 | // <UserContext.Provider value={{ user, saveUser, logout, loading }}> |
| 56 | // {children} |
| 57 | // </UserContext.Provider> |
| 58 | // ); |
| 59 | // }; |
| 60 | |
| 61 | // export const useUser = () => { |
| 62 | // const context = useContext(UserContext); |
| 63 | // if (!context) { |
| 64 | // throw new Error('useUser 必须在 UserProvider 内使用'); |
| 65 | // } |
| 66 | // return context; |
| 67 | // }; |
Krishya | 75e43c0 | 2025-04-05 21:16:30 +0800 | [diff] [blame] | 68 | import React, { createContext, useContext, useState, useEffect } from 'react'; |
| 69 | |
22301009 | df48f96 | 2025-06-05 13:40:44 +0800 | [diff] [blame] | 70 | const avatarBaseUrl = process.env.REACT_APP_AVATAR_BASE_URL || ''; |
| 71 | |
Krishya | 8f2fec8 | 2025-06-04 21:54:46 +0800 | [diff] [blame] | 72 | export const UserContext = createContext(); |
Krishya | 75e43c0 | 2025-04-05 21:16:30 +0800 | [diff] [blame] | 73 | |
| 74 | export const UserProvider = ({ children }) => { |
| 75 | const [user, setUser] = useState(null); |
| 76 | const [loading, setLoading] = useState(true); |
| 77 | |
22301009 | df48f96 | 2025-06-05 13:40:44 +0800 | [diff] [blame] | 78 | const formatUser = (raw) => { |
| 79 | if (!raw) return null; |
| 80 | return { |
| 81 | userId: raw.userId || raw.id || null, |
| 82 | username: raw.username || '匿名用户', |
22301009 | 207e2db | 2025-06-09 00:27:28 +0800 | [diff] [blame] | 83 | avatarUrl: raw.avatarUrl |
| 84 | ? raw.avatarUrl.startsWith('http') |
| 85 | ? raw.avatarUrl |
| 86 | : `${avatarBaseUrl}${raw.avatarUrl}` |
| 87 | : null, |
Krishya | 6bf199c | 2025-06-06 21:14:23 +0800 | [diff] [blame] | 88 | ...raw, |
22301009 | df48f96 | 2025-06-05 13:40:44 +0800 | [diff] [blame] | 89 | }; |
| 90 | }; |
| 91 | |
Krishya | 75e43c0 | 2025-04-05 21:16:30 +0800 | [diff] [blame] | 92 | useEffect(() => { |
| 93 | const storedUser = localStorage.getItem('user'); |
| 94 | if (storedUser) { |
22301009 | df48f96 | 2025-06-05 13:40:44 +0800 | [diff] [blame] | 95 | try { |
| 96 | const parsed = JSON.parse(storedUser); |
| 97 | setUser(formatUser(parsed)); |
| 98 | } catch (err) { |
| 99 | console.error('本地用户信息解析失败:', err); |
| 100 | localStorage.removeItem('user'); |
| 101 | setUser(null); |
| 102 | } |
22301009 | 37ebec1 | 2025-06-03 21:17:04 +0800 | [diff] [blame] | 103 | } else { |
Krishya | 6bf199c | 2025-06-06 21:14:23 +0800 | [diff] [blame] | 104 | setUser(null); |
Krishya | 75e43c0 | 2025-04-05 21:16:30 +0800 | [diff] [blame] | 105 | } |
| 106 | setLoading(false); |
| 107 | }, []); |
| 108 | |
Krishya | 75e43c0 | 2025-04-05 21:16:30 +0800 | [diff] [blame] | 109 | const saveUser = (userData) => { |
22301009 | df48f96 | 2025-06-05 13:40:44 +0800 | [diff] [blame] | 110 | if (userData && (userData.userId || userData.id)) { |
22301009 | 207e2db | 2025-06-09 00:27:28 +0800 | [diff] [blame] | 111 | const formatted = formatUser(userData); |
| 112 | localStorage.setItem('user', JSON.stringify(formatted)); |
| 113 | setUser(formatted); |
Krishya | 8f2fec8 | 2025-06-04 21:54:46 +0800 | [diff] [blame] | 114 | } else { |
| 115 | console.error('无效的用户数据:', userData); |
| 116 | } |
Krishya | 75e43c0 | 2025-04-05 21:16:30 +0800 | [diff] [blame] | 117 | }; |
| 118 | |
Krishya | 75e43c0 | 2025-04-05 21:16:30 +0800 | [diff] [blame] | 119 | const logout = () => { |
| 120 | localStorage.removeItem('user'); |
| 121 | setUser(null); |
| 122 | }; |
| 123 | |
| 124 | return ( |
Krishya | 8f2fec8 | 2025-06-04 21:54:46 +0800 | [diff] [blame] | 125 | <UserContext.Provider value={{ user, saveUser, logout, loading }}> |
Krishya | 75e43c0 | 2025-04-05 21:16:30 +0800 | [diff] [blame] | 126 | {children} |
| 127 | </UserContext.Provider> |
| 128 | ); |
| 129 | }; |
| 130 | |
Krishya | 8f2fec8 | 2025-06-04 21:54:46 +0800 | [diff] [blame] | 131 | export const useUser = () => { |
| 132 | const context = useContext(UserContext); |
| 133 | if (!context) { |
Krishya | 6bf199c | 2025-06-06 21:14:23 +0800 | [diff] [blame] | 134 | throw new Error('useUser 必须在 UserProvider 内使用'); |
Krishya | 8f2fec8 | 2025-06-04 21:54:46 +0800 | [diff] [blame] | 135 | } |
| 136 | return context; |
22301009 | df48f96 | 2025-06-05 13:40:44 +0800 | [diff] [blame] | 137 | }; |