116 lines
2.9 KiB
TypeScript
116 lines
2.9 KiB
TypeScript
import type { ApiSessionData } from '../api/types';
|
|
|
|
import {
|
|
DEBUG, GLOBAL_STATE_CACHE_KEY, SESSION_USER_KEY,
|
|
} from '../config';
|
|
|
|
const DC_IDS = [1, 2, 3, 4, 5];
|
|
|
|
export function hasStoredSession() {
|
|
if (checkSessionLocked()) {
|
|
return true;
|
|
}
|
|
|
|
const userAuthJson = localStorage.getItem(SESSION_USER_KEY);
|
|
if (!userAuthJson) {
|
|
return false;
|
|
}
|
|
|
|
try {
|
|
const userAuth = JSON.parse(userAuthJson);
|
|
return Boolean(userAuth && userAuth.id && userAuth.dcID);
|
|
} catch (err) {
|
|
// Do nothing.
|
|
return false;
|
|
}
|
|
}
|
|
|
|
export function storeSession(sessionData: ApiSessionData, currentUserId?: string) {
|
|
const { mainDcId, keys, hashes } = sessionData;
|
|
|
|
localStorage.setItem(SESSION_USER_KEY, JSON.stringify({ dcID: mainDcId, id: currentUserId }));
|
|
localStorage.setItem('dc', String(mainDcId));
|
|
Object.keys(keys).map(Number).forEach((dcId) => {
|
|
localStorage.setItem(`dc${dcId}_auth_key`, JSON.stringify(keys[dcId]));
|
|
});
|
|
|
|
if (hashes) {
|
|
Object.keys(hashes).map(Number).forEach((dcId) => {
|
|
localStorage.setItem(`dc${dcId}_hash`, JSON.stringify(hashes[dcId]));
|
|
});
|
|
}
|
|
}
|
|
|
|
export function clearStoredSession() {
|
|
[
|
|
SESSION_USER_KEY,
|
|
'dc',
|
|
...DC_IDS.map((dcId) => `dc${dcId}_auth_key`),
|
|
...DC_IDS.map((dcId) => `dc${dcId}_hash`),
|
|
...DC_IDS.map((dcId) => `dc${dcId}_server_salt`),
|
|
].forEach((key) => {
|
|
localStorage.removeItem(key);
|
|
});
|
|
}
|
|
|
|
export function loadStoredSession(): ApiSessionData | undefined {
|
|
if (!hasStoredSession()) {
|
|
return undefined;
|
|
}
|
|
|
|
const userAuth = JSON.parse(localStorage.getItem(SESSION_USER_KEY)!);
|
|
if (!userAuth) {
|
|
return undefined;
|
|
}
|
|
const mainDcId = Number(userAuth.dcID);
|
|
const keys: Record<number, string> = {};
|
|
const hashes: Record<number, string> = {};
|
|
|
|
DC_IDS.forEach((dcId) => {
|
|
try {
|
|
const key = localStorage.getItem(`dc${dcId}_auth_key`);
|
|
if (key) {
|
|
keys[dcId] = JSON.parse(key);
|
|
}
|
|
|
|
const hash = localStorage.getItem(`dc${dcId}_hash`);
|
|
if (hash) {
|
|
hashes[dcId] = JSON.parse(hash);
|
|
}
|
|
} catch (err) {
|
|
if (DEBUG) {
|
|
// eslint-disable-next-line no-console
|
|
console.warn('Failed to load stored session', err);
|
|
}
|
|
// Do nothing.
|
|
}
|
|
});
|
|
|
|
if (!Object.keys(keys).length) return undefined;
|
|
|
|
return {
|
|
mainDcId,
|
|
keys,
|
|
hashes,
|
|
};
|
|
}
|
|
|
|
export function importTestSession() {
|
|
const sessionJson = process.env.TEST_SESSION!;
|
|
try {
|
|
const sessionData = JSON.parse(sessionJson) as ApiSessionData & { userId: string };
|
|
storeSession(sessionData, sessionData.userId);
|
|
} catch (err) {
|
|
if (DEBUG) {
|
|
// eslint-disable-next-line no-console
|
|
console.warn('Failed to load test session', err);
|
|
}
|
|
}
|
|
}
|
|
|
|
function checkSessionLocked() {
|
|
const stateFromCache = JSON.parse(localStorage.getItem(GLOBAL_STATE_CACHE_KEY) || '{}');
|
|
|
|
return Boolean(stateFromCache?.passcode?.isScreenLocked);
|
|
}
|