Add builds backend

src/main/main.ts

@@ -13,31 +13,39 @@ import { app, BrowserWindow, shell, ipcMain, globalShortcut } from 'electron';
 import MenuBuilder from './menu';
 import { resolveHtmlPath } from './util';
 import { loadTevefData } from './load';
-import { Prisma, PrismaClient } from '@prisma/client'
+import { Prisma, PrismaClient } from '@prisma/client';
 import { executeCommand } from './dirt/keyboard';
-import { armFishing, stopFishingAndUnregisterHotkeys } from './services/fishing';
+import {
+  armFishing,
+  stopFishingAndUnregisterHotkeys,
+} from './services/fishing';
 import { parseLastRun } from './lastrun';
 import { createClassesService } from './services/classes';
 import { createItemsService } from './services/items';
 import { createSettingsService } from './services/settings';
+import { createBuildsService } from './services/builds';
 import { createDamageService } from './services/damage';
 
 let mainWindow: BrowserWindow | null = null;
 
-const DB_PATH = (process.env.NODE_ENV === 'development') ? 'file:./dev.db' : 'file:' + path.join(app.getAppPath(), "../prisma/dev.db");
+const DB_PATH =
+  process.env.NODE_ENV === 'development'
+    ? 'file:./dev.db'
+    : 'file:' + path.join(app.getAppPath(), '../prisma/dev.db');
 
 export const prismaClient = new PrismaClient({
   datasources: {
     db: {
-      url: DB_PATH
-    }
-  }
-})
+      url: DB_PATH,
+    },
+  },
+});
 
-const settingsService = createSettingsService(app.getPath.bind(app))
+const settingsService = createSettingsService(app.getPath.bind(app));
 const damageService = createDamageService(settingsService);
-const classesService = createClassesService(prismaClient)
-const itemsService = createItemsService(prismaClient)
+const classesService = createClassesService(prismaClient);
+const itemsService = createItemsService(prismaClient);
+const buildsService = createBuildsService(app.getPath.bind(app));
 
 ipcMain.on('loadData', async (event, arg) => {
   const data = await loadTevefData(arg);
@@ -51,8 +59,8 @@ ipcMain.on('load', async (event, arg) => {
     globalShortcut.unregister('A');
     // eslint-disable-next-line no-restricted-syntax
     for (const command of arg) {
-        // eslint-disable-next-line no-await-in-loop
-        await executeCommand(command);
+      // eslint-disable-next-line no-await-in-loop
+      await executeCommand(command);
     }
   });
 });
@@ -66,35 +74,60 @@ ipcMain.on('fishing_disarm', async () => {
 });
 
 ipcMain.on('settings_read', async (event) => {
-  event.reply('settings_read', await settingsService.getSettings())
+  event.reply('settings_read', await settingsService.getSettings());
 });
 
-ipcMain.on(
-  'get_latest_damage_by_type',
-   async (event) => {
-      const res = await damageService.getLatestRunPerDamageType();
-      event.reply('get_latest_damage_by_type', res);
-   });
+ipcMain.on('get_latest_damage_by_type', async (event) => {
+  const res = await damageService.getLatestRunPerDamageType();
+  event.reply('get_latest_damage_by_type', res);
+});
 
 ipcMain.on('get_all_classes', async (event) => {
-  event.reply('get_all_classes', await classesService.getAllClasses())
-})
+  event.reply('get_all_classes', await classesService.getAllClasses());
+});
 
 ipcMain.on('get_all_items', async (event) => {
   event.reply('get_all_items', await itemsService.getAllItemsDict());
-})
+});
 
 ipcMain.on('request_last_run', async (event, arg) => {
   const data = await parseLastRun(arg);
   event.reply('last_run_info', data);
 });
 
 ipcMain.on('settings_write', async (event, arg) => {
-    const res = await settingsService.writeSettings(arg);
-    // we expect boolean on the other side, true means success.
-    // should change it probably
-    event.reply('settings_write', !res);
-})
+  const res = await settingsService.writeSettings(arg);
+  // we expect boolean on the other side, true means success.
+  // should change it probably
+  event.reply('settings_write', !res);
+});
+
+ipcMain.on('get_all_builds', async (event) => {
+  event.reply('get_all_builds', await buildsService.getBuilds());
+});
+
+ipcMain.on('save_build', async (event, arg) => {
+  await buildsService.saveBuild(arg);
+  event.reply('get_all_builds', await buildsService.getBuilds());
+});
+
+ipcMain.on('delete_build', async (event, arg) => {
+  await buildsService.deleteBuild(arg);
+  event.reply('get_all_builds', await buildsService.getBuilds());
+});
+
+ipcMain.on('get_selected_builds', async (event) => {
+  event.reply('get_selected_builds', await buildsService.getSelectedBuilds());
+});
+
+ipcMain.on('save_selected_build', async (event, arg) => {
+  await buildsService.saveSelectedBuild(
+    arg.playerName,
+    arg.playerClass,
+    arg.buildId,
+  );
+  event.reply('get_selected_builds', await buildsService.getSelectedBuilds());
+});
 
 if (process.env.NODE_ENV === 'production') {
   const sourceMapSupport = require('source-map-support');

src/main/services/builds.ts

@@ -0,0 +1,98 @@
+import path from 'path';
+import fs from 'fs/promises';
+
+export function createBuildsService(
+  getPath: (type: 'userData' | 'documents') => string,
+) {
+  const getDefaultBuilds = () => ({});
+  const getDefaultSelectedBuilds = () => ({});
+
+  const getBuildsFilePath = () => path.join(getPath('userData'), 'builds.json');
+  const getSelectedBuildsFilePath = () =>
+    path.join(getPath('userData'), 'selectedBuilds.json');
+  const getBuildIdFilePath = () =>
+    path.join(getPath('userData'), 'buildId.json');
+
+  const readFile = async (filePath: string, defaultValue: any) => {
+    try {
+      const data = await fs.readFile(filePath, 'utf-8');
+      return JSON.parse(data);
+    } catch (e) {
+      console.log(e);
+      return defaultValue;
+    }
+  };
+
+  const writeFile = async (filePath: string, data: any) => {
+    try {
+      await fs.writeFile(filePath, JSON.stringify(data));
+    } catch (e) {
+      console.log(e);
+    }
+  };
+
+  const readNextBuildId = async () => {
+    const buildIdData = await readFile(getBuildIdFilePath(), { nextId: 1 });
+    return buildIdData.nextId;
+  };
+
+  const incrementBuildId = async () => {
+    const buildIdData = await readFile(getBuildIdFilePath(), { nextId: 1 });
+    buildIdData.nextId += 1;
+    await writeFile(getBuildIdFilePath(), buildIdData);
+  };
+
+  return {
+    async getBuilds() {
+      return await readFile(getBuildsFilePath(), getDefaultBuilds());
+    },
+    async saveBuild(build: any) {
+      try {
+        const builds = await this.getBuilds();
+        if (!build.id) {
+          build.id = await readNextBuildId();
+          await incrementBuildId();
+        }
+        builds[build.id] = build;
+        await writeFile(getBuildsFilePath(), builds);
+        return 0;
+      } catch (e) {
+        console.log(e);
+        return 1;
+      }
+    },
+    async deleteBuild(buildId: number) {
+      try {
+        const builds = await this.getBuilds();
+        delete builds[buildId];
+        await writeFile(getBuildsFilePath(), builds);
+        return 0;
+      } catch (e) {
+        console.log(e);
+        return 1;
+      }
+    },
+    async getSelectedBuilds() {
+      return await readFile(
+        getSelectedBuildsFilePath(),
+        getDefaultSelectedBuilds(),
+      );
+    },
+    async saveSelectedBuild(
+      playerName: string,
+      playerClass: string,
+      buildId: number,
+    ) {
+      try {
+        const selectedBuilds = (await this.getSelectedBuilds()) ?? {};
+        selectedBuilds[playerName] = selectedBuilds[playerName] ?? {};
+        selectedBuilds[playerName][playerClass] = buildId;
+        await writeFile(getSelectedBuildsFilePath(), selectedBuilds);
+        return 0;
+      } catch (e) {
+        console.log(e);
+        return 1;
+      }
+    },
+  };
+}