diff --git a/src/main/main.js b/src/main/main.js index a0a5be1..234490d 100644 --- a/src/main/main.js +++ b/src/main/main.js @@ -1,4 +1,4 @@ -const { app, BrowserWindow, ipcMain, session } = require('electron') +const { app, BrowserWindow, ipcMain, session, protocol } = require('electron') const path = require('path') const fs = require('fs') const isDev = require('electron-is-dev') @@ -48,11 +48,15 @@ app.on('ready', async () => { await session.defaultSession.loadExtension( 'C:/Users/timot/AppData/Local/Google/Chrome/User Data/Profile 2/Extensions/fmkadmapgofadopljbjfkapdkoienihi/4.25.0_0' ) + protocol.registerFileProtocol('atom', (request, callback) => { + const url = request.url.substr(7) + callback({ path: path.normalize(`${__dirname}/${url}`) }) + }) session.defaultSession.webRequest.onHeadersReceived((details, callback) => { callback({ responseHeaders: { ...details.responseHeaders, - 'Content-Security-Policy': ['default-src \'self\' \'unsafe-inline\' filesystem:; object-src \'self\' filesystem:;'] + 'Content-Security-Policy': ['default-src \'self\' \'unsafe-inline\' atom:; object-src \'self\' atom:;'] } }) }) diff --git a/src/main/pdfGenerator/pdfGenerator.js b/src/main/pdfGenerator/pdfGenerator.js index c16e42e..9e797d7 100644 --- a/src/main/pdfGenerator/pdfGenerator.js +++ b/src/main/pdfGenerator/pdfGenerator.js @@ -32,7 +32,7 @@ const pdfGeneratorHandler = async (event, args) => { return new Promise((resolve, reject) => { pdf.on('error', (err) => reject(err)) - pdf.on('finish', resolve(path.join(app.getPath("userData"), 'output_files/pdf_files', `output${args.id}.pdf`))) + pdf.on('finish', resolve(`atom://${app.getPath("userData")}\\input${args.id}.pdf`)) }) } diff --git a/src/renderer/components/PdfDisplay.js b/src/renderer/components/PdfDisplay.js index 199cec9..8604d7f 100644 --- a/src/renderer/components/PdfDisplay.js +++ b/src/renderer/components/PdfDisplay.js @@ -5,15 +5,16 @@ import React from 'react' const PdfDisplay = ({ url }) => { return (