Fix accessToken hook

overthq · Jun 22, 2020 · 9078747 · 9078747
1 parent b59344a
commit 9078747
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 9 deletions.
diff --git a/app/App.tsx b/app/App.tsx
@@ -184,6 +184,7 @@ const App = () => {
 		})
 	});
 
+	// Keep the splash screen open until loading becomes true.
 	return loading ? (
 		<View>
 			<ActivityIndicator />

diff --git a/app/src/hooks/useAccessToken.ts b/app/src/hooks/useAccessToken.ts
@@ -8,9 +8,10 @@ const useAccessToken = () => {
 	React.useEffect(() => {
 		(async () => {
 			try {
-				const savedAccessToken = await AsyncStorage.getItem('accessToken');
-				if (!savedAccessToken) return setAccessToken(null);
-				setAccessToken(savedAccessToken);
+				const savedToken = await AsyncStorage.getItem('accessToken');
+				setAccessToken(savedToken);
+			} catch (error) {
+				setAccessToken(null);
 			} finally {
 				setLoading(false);
 			}
@@ -25,12 +26,9 @@ const useAccessToken = () => {
 		AsyncStorage.setItem('accessToken', accessToken);
 	}, [accessToken]);
 
-	const saveAccessToken = (token: string) =>
-		AsyncStorage.setItem('accessToken', token);
-
 	const logOut = () => setAccessToken(null);
 
-	return { loading, accessToken, saveAccessToken, logOut };
+	return { loading, accessToken, setAccessToken, logOut };
 };
 
 export default useAccessToken;
diff --git a/app/src/screens/VerifyAuthentication.tsx b/app/src/screens/VerifyAuthentication.tsx
@@ -17,7 +17,7 @@ const VerifyAuthentication = () => {
 		RouteProp<AuthStackParamList, 'VerifyAuthentication'>
 	>();
 	const { navigate } = useNavigation();
-	const { saveAccessToken } = useAccessToken();
+	const { setAccessToken } = useAccessToken();
 	const [code, setCode] = React.useState('');
 	const [
 		{ data, fetching },
@@ -27,7 +27,7 @@ const VerifyAuthentication = () => {
 
 	React.useEffect(() => {
 		if (data?.verifyAuthentication) {
-			saveAccessToken(data.verifyAuthentication);
+			setAccessToken(data.verifyAuthentication);
 			navigate('Main');
 		}
 	}, [data]);