import { mkdirp } from "mkdirp"; import path from "path"; import fs from "fs"; import { getTeaPath } from "./tea-dir"; import { app } from "electron"; import * as log from "electron-log"; import axios from "axios"; const sessionFilePath = path.join(getTeaPath(), "tea.xyz/gui/tmp.dat"); const sessionFolder = path.join(getTeaPath(), "tea.xyz/gui"); export interface Session { device_id?: string; key?: string; user?: any; locale?: string; } let sessionMemory: Session = { device_id: "", locale: "en" }; const initialized: Promise = new Promise((resolve, reject) => { try { log.info("initializing GUI session folder"); createInitialSessionFile().then((newSession) => { resolve(newSession); }); } catch (error) { log.error(error); reject(error); } }); async function addEmptySessionFile(): Promise { const locale = app.getLocale(); await mkdirp(sessionFolder); const data = { device_id: await getDeviceId(), locale }; await writeSessionData(data); log.info("new session file created"); return data; } export async function createInitialSessionFile(): Promise { // TODO: this looks nasty, refactor this // the app is too dependent that this function succeeds let session = { ...sessionMemory }; const locale = app.getLocale(); try { if (fs.existsSync(sessionFilePath)) { log.info("session file exists!"); const sessionBuffer = await fs.readFileSync(sessionFilePath); const sessionData = JSON.parse(sessionBuffer.toString()) as Session; if (!sessionData?.device_id) { throw new Error("device_id is empty!"); } else { session = sessionData; session.locale = locale; } } } catch (error) { log.error(error); } if (!session?.device_id) { try { const newSession = await addEmptySessionFile(); if (newSession) { session = newSession; session.locale = locale; } } catch (error) { log.error(error); } } sessionMemory = session; return session; } let deviceIdRetryCount = 0; async function getDeviceId() { let deviceId = ""; try { const req = await axios.get<{ deviceId: string }>("https://api.tea.xyz/v1/auth/registerDevice"); deviceId = req.data.deviceId; } catch (error) { log.error(error); } if (deviceIdRetryCount < 3 && !deviceId) { deviceIdRetryCount++; deviceId = await getDeviceId(); } return deviceId; } export async function readSessionData(): Promise { log.info("read session data."); const data = await initialized; log.info( "initialized session device_id:", data?.device_id, "developer_id:", data?.user?.developer_id ); if (sessionMemory?.device_id) { log.debug("use session cache"); return sessionMemory; } try { log.info("re-reading session data"); const locale = app.getLocale(); const sessionBuffer = await fs.readFileSync(sessionFilePath); const session = JSON.parse(sessionBuffer.toString()) as Session; if (!session?.device_id) throw new Error("device_id is empty!"); session.locale = locale; sessionMemory = session; log.info("re-read session data done"); } catch (error) { sessionMemory = await createInitialSessionFile(); log.error(error); } return sessionMemory; } export async function writeSessionData(data: Session) { try { sessionMemory = { ...sessionMemory, ...data }; if (!sessionMemory.device_id) throw new Error("writing without device_id is not allowed!"); log.info("creating:", sessionFolder); await mkdirp(sessionFolder); log.info("writing session data:", data); // rm this await fs.writeFileSync(sessionFilePath, JSON.stringify(data), { encoding: "utf-8" }); } catch (error) { log.error(error); } }