import type { PathLike } from 'node:fs';
import { BrowserWindow, dialog, ipcMain } from 'electron';
import { download } from 'electron-dl';
import { writeFileSync } from 'fs-extra';

const debug = require('../../preload-safe-debug')('Ferdium:ipcApi:download');

function decodeBase64Image(dataString: string) {
  const matches = dataString.match(/^data:([+/A-Za-z-]+);base64,(.+)$/);

  if (matches?.length !== 3) {
    return new Error('Invalid input string');
  }

  return Buffer.from(matches[2], 'base64');
}

export default (params: { mainWindow: BrowserWindow }) => {
  ipcMain.on(
    'download-file',
    async (_event, { url, content, fileOptions = {} }) => {
      const win = BrowserWindow.getFocusedWindow();

      try {
        if (content) {
          try {
            const saveDialog = await dialog.showSaveDialog(params.mainWindow, {
              defaultPath: fileOptions.name,
            });

            if (saveDialog.canceled) return;

            const binaryImage = decodeBase64Image(content);
            writeFileSync(
              saveDialog.filePath as PathLike,
              binaryImage as unknown as string,
              'binary',
            );

            debug('File blob saved to', saveDialog.filePath);
          } catch (error) {
            console.error(error);
          }
        } else {
          const dl = await download(win!, url, {
            saveAs: true,
          });
          debug('File saved to', dl.savePath);
        }
      } catch (error) {
        console.error(error);
      }
    },
  );

  ipcMain.handle('download-folder-select', async () => {
    const result = await dialog.showOpenDialog(params.mainWindow, {
      properties: ['openDirectory'],
    });

    if (result.canceled) return null;

    return result.filePaths[0];
  });
};