78 lines
2.8 KiB
JavaScript
78 lines
2.8 KiB
JavaScript
import { onError } from "apollo-link-error";
|
|
import { Observable } from "apollo-link";
|
|
import { auth } from "../firebase/firebase.utils";
|
|
//https://stackoverflow.com/questions/57163454/refreshing-a-token-with-apollo-client-firebase-auth
|
|
const errorLink = onError(
|
|
({ graphQLErrors, networkError, operation, forward }) => {
|
|
let access_token = window.localStorage.getItem("token");
|
|
// console.log("graphQLErrors", graphQLErrors);
|
|
// console.log("networkError", networkError);
|
|
// console.log("operation", operation);
|
|
// console.log("forward", forward);
|
|
|
|
let expired = false;
|
|
|
|
if (graphQLErrors) {
|
|
if (graphQLErrors[0].message.includes("JWTExpired")) {
|
|
expired = true;
|
|
}
|
|
}
|
|
if (networkError) {
|
|
if (networkError?.message.includes("JWTExpired")) {
|
|
expired = true;
|
|
}
|
|
}
|
|
if (expired) {
|
|
//User access token has expired
|
|
//props.history.push("/network-error");
|
|
console.log("We need a new token!");
|
|
if (access_token && access_token !== "undefined") {
|
|
// Let's refresh token through async request
|
|
return new Observable(observer => {
|
|
const unsubscribeFromAuth = auth.onAuthStateChanged(async user => {
|
|
console.log("Auth change in error handling.");
|
|
if (user) {
|
|
user
|
|
.getIdToken(true)
|
|
.then(function(idToken) {
|
|
if (!idToken) {
|
|
window.localStorage.removeItem("token");
|
|
return console.log("Refresh token has expired");
|
|
}
|
|
console.log("Got a new token", idToken);
|
|
window.localStorage.setItem("token", idToken);
|
|
|
|
// reset the headers
|
|
operation.setContext(({ headers = {} }) => ({
|
|
headers: {
|
|
// Re-add old headers
|
|
...headers,
|
|
// Switch out old access token for new one
|
|
authorization: idToken ? `Bearer ${idToken}` : ""
|
|
}
|
|
}));
|
|
|
|
const subscriber = {
|
|
next: observer.next.bind(observer),
|
|
error: observer.error.bind(observer),
|
|
complete: observer.complete.bind(observer)
|
|
};
|
|
console.log("About to resend the request.");
|
|
// Retry last failed request
|
|
forward(operation).subscribe(subscriber);
|
|
})
|
|
.catch(error => {
|
|
// No refresh or client token available, we force user to login
|
|
console.log("Hit an error.");
|
|
observer.error(error);
|
|
});
|
|
}
|
|
});
|
|
});
|
|
}
|
|
}
|
|
}
|
|
);
|
|
|
|
export default errorLink;
|