notesnook/apps/mobile/src/components/Auth/sessionexpired.js

import React, { useEffect, useRef, useState } from 'react';
import { Modal, View } from 'react-native';
import { useTracked } from '../../provider';
import { useUserStore } from '../../provider/stores';
import BiometricService from '../../services/BiometricService';
import {
  eSendEvent,
  eSubscribeEvent,
  eUnSubscribeEvent,
  presentSheet,
  ToastEvent
} from '../../services/EventManager';
import { clearMessage } from '../../services/Message';
import PremiumService from '../../services/PremiumService';
import Sync from '../../services/Sync';
import { db } from '../../utils/database';
import { MMKV } from '../../utils/mmkv';
import { SIZE } from '../../utils/SizeUtils';
import { sleep } from '../../utils/TimeUtils';
import { ActionIcon } from '../ActionIcon';
import { Button } from '../Button';
import { Dialog } from '../Dialog';
import { presentDialog } from '../Dialog/functions';
import Input from '../Input';
import { Toast } from '../Toast';
import Heading from '../Typography/Heading';
import Paragraph from '../Typography/Paragraph';

function getEmail(email) {
  if (!email) return null;
  return email.replace(/(.{2})(.*)(?=@)/, function (gp1, gp2, gp3) {
    for (let i = 0; i < gp3.length; i++) {
      gp2 += '*';
    }
    return gp2;
  });
}

export const SessionExpired = () => {
  const [state, dispatch] = useTracked();
  const colors = state.colors;
  const email = useRef();
  const emailInputRef = useRef();
  const passwordInputRef = useRef();
  const password = useRef();
  const [visible, setVisible] = useState(false);
  const [focused, setFocused] = useState(false);

  const setUser = useUserStore(state => state.setUser);

  const [error, setError] = useState(false);
  const [loading, setLoading] = useState(false);

  const logout = async () => {
    try {
      await db.user.logout();
      await BiometricService.resetCredentials();
      await Storage.write('introCompleted', 'true');
      setVisible(false);
    } catch (e) {
      ToastEvent.show({
        heading: e.message,
        type: 'error',
        context: 'local'
      });
    }
  };

  useEffect(() => {
    eSubscribeEvent('session_expired', open);
    return () => {
      eUnSubscribeEvent('session_expired', open);
      setFocused(false);
    };
  }, [visible]);

  const open = async () => {
    try {
      console.log('REQUESTING NEW TOKEN');
      let res = await db.user.tokenManager.getToken();
      if (!res) throw new Error('no token found');
      if (db.user.tokenManager._isTokenExpired(res)) throw new Error('token expired');
      if (!(await Sync.run())) throw new Error('e');
      await MMKV.removeItem('loginSessionHasExpired');
      setVisible(false);
    } catch (e) {
      console.log(e);
      let user = await db.user.getUser();
      if (!user) return;
      email.current = user.email;
      setVisible(true);
    }
  };

  const login = async () => {
    if (error) return;
    setLoading(true);
    let user;
    try {
      await db.user.login(email.current.toLowerCase(), password.current);
      user = await db.user.getUser();
      if (!user) throw new Error('Email or password incorrect!');
      PremiumService.setPremiumStatus();
      setUser(user);
      clearMessage();
      ToastEvent.show({
        heading: 'Login successful',
        message: `Logged in as ${user.email}`,
        type: 'success',
        context: 'global'
      });
      setVisible(false);
      await MMKV.removeItem('loginSessionHasExpired');
      eSendEvent('userLoggedIn', true);
      await sleep(500);
      presentSheet({
        title: 'Syncing your data',
        paragraph: 'Please wait while we sync all your data.',
        progress: true
      });
    } catch (e) {
      setLoading(false);
      ToastEvent.show({
        heading: user ? 'Failed to sync' : 'Login failed',
        message: e.message,
        type: 'error',
        context: 'local'
      });
    }
  };

  return (
    visible && (
      <Modal
        onShow={async () => {
          await sleep(300);
          passwordInputRef.current?.focus();
          setFocused(true);
        }}
        visible={true}
      >
        <View
          style={{
            width: focused ? '100%' : '99.9%',
            padding: 12,
            justifyContent: 'center',
            flex: 1
          }}
        >
          <View
            style={{
              justifyContent: 'center',
              alignItems: 'center',
              width: '100%',
              marginBottom: 20,
              borderRadius: 10,
              paddingVertical: 20
            }}
          >
            <ActionIcon
              customStyle={{
                width: 60,
                height: 60
              }}
              name="alert"
              color={colors.errorText}
              size={50}
            />
            <Heading size={SIZE.xxxl} color={colors.heading}>
              Session expired
            </Heading>
            <Paragraph
              style={{
                textAlign: 'center'
              }}
            >
              Your session on this device has expired. Please enter password for{' '}
              {getEmail(email.current)} to continue.
            </Paragraph>
          </View>

          <Input
            fwdRef={passwordInputRef}
            onChangeText={value => {
              password.current = value;
            }}
            returnKeyLabel="Next"
            returnKeyType="next"
            secureTextEntry
            autoCompleteType="password"
            autoCapitalize="none"
            autoCorrect={false}
            placeholder="Password"
            onSubmit={login}
          />

          <Button
            style={{
              marginTop: 10,
              width: '100%'
            }}
            loading={loading}
            onPress={login}
            type="accent"
            title={loading ? null : 'Login'}
          />

          <Button
            style={{
              marginTop: 10,
              width: '100%'
            }}
            onPress={() => {
              presentDialog({
                context: 'session_expiry',
                title: 'Logout',
                paragraph:
                  'Are you sure you want to logout from this device? Any unsynced changes will be lost.',
                positiveText: 'Logout',
                positiveType: 'errorShade',
                positivePress: logout
              });
            }}
            type="errorShade"
            title={loading ? null : 'Logout from this device'}
          />
        </View>
        <Toast context="local" />
        <Dialog context="session_expiry" />
      </Modal>
    )
  );
};