File size: 2,239 Bytes
b5a1b3b | 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 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 | import { createContext, useContext, useState, useCallback, useEffect } from "react";
const AuthContext = createContext();
const TOKEN_KEY = "diffweave_token";
const USER_KEY = "diffweave_user";
export function AuthProvider({ children }) {
const [token, setToken] = useState(() => localStorage.getItem(TOKEN_KEY) || localStorage.getItem("token"));
const [user, setUser] = useState(() => {
try {
const u = localStorage.getItem(USER_KEY) || localStorage.getItem("user");
return u ? JSON.parse(u) : null;
} catch {
return null;
}
});
// Validate token with DocWeave auth engine on mount
useEffect(() => {
if (token && !user) {
fetch("/api/auth/me", {
headers: { Authorization: `Bearer ${token}` },
})
.then((r) => (r.ok ? r.json() : null))
.then((data) => {
if (data && !data.detail) {
setUser(data);
localStorage.setItem(USER_KEY, JSON.stringify(data));
} else {
// Invalid token - clear
localStorage.removeItem(TOKEN_KEY);
localStorage.removeItem(USER_KEY);
localStorage.removeItem("token");
setToken(null);
setUser(null);
}
})
.catch(() => {});
}
}, [token]); // eslint-disable-line
const login = useCallback((accessToken, userData = null) => {
localStorage.setItem(TOKEN_KEY, accessToken);
localStorage.setItem("token", accessToken);
setToken(accessToken);
if (userData) {
localStorage.setItem(USER_KEY, JSON.stringify(userData));
setUser(userData);
}
}, []);
const logout = useCallback(() => {
// Notify server to clear CLI credentials
fetch("/api/auth/logout", { method: "POST" }).catch(() => {});
localStorage.removeItem(TOKEN_KEY);
localStorage.removeItem(USER_KEY);
localStorage.removeItem("token");
setToken(null);
setUser(null);
}, []);
return (
<AuthContext.Provider
value={{
token,
user,
setUser,
isAuthenticated: Boolean(token),
login,
logout,
}}
>
{children}
</AuthContext.Provider>
);
}
export const useAuth = () => useContext(AuthContext);
|