75 lines
2.2 KiB
TypeScript
75 lines
2.2 KiB
TypeScript
import { createContext, useContext, useEffect, useState, type Dispatch, type SetStateAction, type ReactNode } from 'react';
|
|
import { api } from '@/api';
|
|
import type { User } from '@/types';
|
|
|
|
export type { User }; // re-export so existing `@/hooks/useAuth` importers keep working
|
|
|
|
interface MeResponse {
|
|
user?: User | null;
|
|
single_user_mode?: boolean;
|
|
has_new_version?: boolean;
|
|
}
|
|
|
|
interface AuthContextValue {
|
|
user: User | null | undefined; // undefined = loading
|
|
singleUserMode: boolean;
|
|
setUser: Dispatch<SetStateAction<User | null | undefined>>;
|
|
logout: () => void | Promise<void>;
|
|
refresh: () => void;
|
|
hasNewVersion: boolean;
|
|
setHasNewVersion: (v: boolean) => void;
|
|
}
|
|
|
|
const AuthContext = createContext<AuthContextValue | null>(null);
|
|
|
|
export function AuthProvider({ children }: { children: ReactNode }) {
|
|
const [user, setUser] = useState<User | null | undefined>(undefined); // undefined = loading
|
|
const [singleUserMode, setSUM] = useState(false);
|
|
const [hasNewVersion, setHasNewVersion] = useState(false);
|
|
|
|
function applyMeResponse(raw: unknown) {
|
|
const d = raw as MeResponse;
|
|
setUser(d.user);
|
|
setSUM(d.single_user_mode || false);
|
|
setHasNewVersion(!!d.has_new_version);
|
|
}
|
|
|
|
useEffect(() => {
|
|
api.authMode().then(d => {
|
|
if ((d as { auth_mode?: string }).auth_mode === 'single') setSUM(true);
|
|
}).catch(err => console.error('[useAuth] authMode check failed', err));
|
|
|
|
api.me().then(applyMeResponse).catch(() => setUser(null));
|
|
}, []);
|
|
|
|
const logout = async () => {
|
|
await api.logout();
|
|
setUser(null);
|
|
setSUM(false);
|
|
setHasNewVersion(false);
|
|
};
|
|
|
|
const refresh = () => {
|
|
api.me().then(applyMeResponse).catch(() => {
|
|
setUser(null);
|
|
setSUM(false);
|
|
});
|
|
};
|
|
|
|
return (
|
|
<AuthContext.Provider value={{
|
|
user, singleUserMode, setUser, logout, refresh,
|
|
hasNewVersion, setHasNewVersion,
|
|
}}>
|
|
{children}
|
|
</AuthContext.Provider>
|
|
);
|
|
}
|
|
|
|
export function useAuth(): AuthContextValue {
|
|
return useContext(AuthContext) || {
|
|
user: null, setUser: () => {}, logout: () => {}, refresh: () => {},
|
|
singleUserMode: false, hasNewVersion: false, setHasNewVersion: () => {},
|
|
};
|
|
}
|