Rocket.Chat.ReactNative/app/sagas/init.js

import { AsyncStorage } from 'react-native';
import { put, takeLatest, all } from 'redux-saga/effects';
import SplashScreen from 'react-native-splash-screen';
import RNUserDefaults from 'rn-user-defaults';
import { sanitizedRaw } from '@nozbe/watermelondb/RawRecord';

import * as actions from '../actions';
import { selectServerRequest } from '../actions/server';
import { setAllPreferences } from '../actions/sortPreferences';
import { toggleMarkdown } from '../actions/markdown';
import { toggleCrashReport } from '../actions/crashReport';
import { APP } from '../actions/actionsTypes';
import RocketChat from '../lib/rocketchat';
import log from '../utils/log';
import Navigation from '../lib/Navigation';
import {
	SERVERS, SERVER_ICON, SERVER_NAME, SERVER_URL, TOKEN, USER_ID
} from '../constants/userDefaults';
import { isIOS } from '../utils/deviceInfo';
import database from '../lib/database';
import protectedFunction from '../lib/methods/helpers/protectedFunction';

const restore = function* restore() {
	try {
		let hasMigration;
		if (isIOS) {
			hasMigration = yield AsyncStorage.getItem('hasMigration');
		}

		let { token, server } = yield all({
			token: RNUserDefaults.get(RocketChat.TOKEN_KEY),
			server: RNUserDefaults.get('currentServer')
		});

		if (!hasMigration && isIOS) {
			let servers = yield RNUserDefaults.objectForKey(SERVERS);
			// if not have current
			if (servers && servers.length !== 0 && (!token || !server)) {
				server = servers[0][SERVER_URL];
				token = servers[0][TOKEN];
			}

			// get native credentials
			if (servers) {
				try {
					// parse servers
					servers = yield Promise.all(servers.map(async(s) => {
						await RNUserDefaults.set(`${ RocketChat.TOKEN_KEY }-${ s[SERVER_URL] }`, s[USER_ID]);
						return ({ id: s[SERVER_URL], name: s[SERVER_NAME], iconURL: s[SERVER_ICON] });
					}));
					const serversDB = database.servers;
					yield serversDB.action(async() => {
						const serversCollection = serversDB.collections.get('servers');
						const allServerRecords = await serversCollection.query().fetch();

						// filter servers
						let serversToCreate = servers.filter(i1 => !allServerRecords.find(i2 => i1.id === i2.id));

						// Create
						serversToCreate = serversToCreate.map(record => serversCollection.prepareCreate(protectedFunction((s) => {
							s._raw = sanitizedRaw({ id: record.id }, serversCollection.schema);
							Object.assign(s, record);
						})));

						const allRecords = serversToCreate;

						try {
							await serversDB.batch(...allRecords);
						} catch (e) {
							log(e);
						}
						return allRecords.length;
					});
				} catch (e) {
					log(e);
				}
			}

			try {
				yield AsyncStorage.setItem('hasMigration', '1');
			} catch (e) {
				log(e);
			}
		}

		const sortPreferences = yield RocketChat.getSortPreferences();
		yield put(setAllPreferences(sortPreferences));

		const useMarkdown = yield RocketChat.getUseMarkdown();
		yield put(toggleMarkdown(useMarkdown));

		const allowCrashReport = yield RocketChat.getAllowCrashReport();
		yield put(toggleCrashReport(allowCrashReport));

		if (!token || !server) {
			yield all([
				RNUserDefaults.clear(RocketChat.TOKEN_KEY),
				RNUserDefaults.clear('currentServer')
			]);
			yield put(actions.appStart('outside'));
		} else {
			const serversDB = database.servers;
			const serverCollections = serversDB.collections.get('servers');
			const serverObj = yield serverCollections.find(server);
			yield put(selectServerRequest(server, serverObj && serverObj.version));
		}

		yield put(actions.appReady({}));
	} catch (e) {
		log(e);
		yield put(actions.appStart('outside'));
	}
};

const start = function* start({ root }) {
	if (root === 'inside') {
		yield Navigation.navigate('InsideStack');
	} else if (root === 'setUsername') {
		yield Navigation.navigate('SetUsernameView');
	} else if (root === 'outside') {
		yield Navigation.navigate('OutsideStack');
	}
	SplashScreen.hide();
};

const root = function* root() {
	yield takeLatest(APP.INIT, restore);
	yield takeLatest(APP.START, start);
};
export default root;