everyone-can-use-english/enjoy/src/main/db/handlers/speeches-handler.ts

import { ipcMain, IpcMainEvent } from "electron";
import { Speech } from "@main/db/models";
import fs from "fs-extra";
import path from "path";
import settings from "@main/settings";
import { hashFile } from "@main/utils";
import { Attributes, WhereOptions } from "sequelize";

class SpeechesHandler {
  private async findOne(
    _event: IpcMainEvent,
    where: WhereOptions<Attributes<Speech>>
  ) {
    const speech = await Speech.findOne({ where });
    if (!speech) {
      return null;
    }

    return speech.toJSON();
  }

  private async create(
    event: IpcMainEvent,
    params: {
      sourceId: string;
      sourceType: string;
      text: string;
      configuration: {
        engine: string;
        model: string;
        voice: string;
      };
    },
    blob: {
      type: string;
      arrayBuffer: ArrayBuffer;
    }
  ) {
    const format = blob.type.split("/")[1];
    const filename = `${Date.now()}.${format}`;
    const file = path.join(settings.userDataPath(), "speeches", filename);
    await fs.outputFile(file, Buffer.from(blob.arrayBuffer));
    const md5 = await hashFile(file, { algo: "md5" });
    fs.renameSync(file, path.join(path.dirname(file), `${md5}.${format}`));

    return Speech.create({ ...params, extname: `.${format}`, md5 })
      .then((speech) => {
        return speech.toJSON();
      })
      .catch((err) => {
        event.sender.send("on-notification", {
          type: "error",
          message: err.message,
        });
      });
  }

  register() {
    ipcMain.handle("speeches-find-one", this.findOne);
    ipcMain.handle("speeches-create", this.create);
  }

  unregister() {
    ipcMain.removeHandler("speeches-find-one");
    ipcMain.removeHandler("speeches-create");
  }
}

export const speechesHandler = new SpeechesHandler();