first commit
This commit is contained in:
@@ -0,0 +1,176 @@
|
||||
import { app, BrowserWindow, shell, ipcMain, dialog } from 'electron'
|
||||
import { fileURLToPath } from 'node:url'
|
||||
import path from 'node:path'
|
||||
import fs from 'node:fs'
|
||||
import os from 'node:os'
|
||||
import * as pbService from './pocketbase.js'
|
||||
import { parseVaccinationRecordPdf } from './parseVaccinationRecordPdf.js'
|
||||
|
||||
const __dirname = path.dirname(fileURLToPath(import.meta.url))
|
||||
|
||||
process.env.APP_ROOT = path.join(__dirname, '../..')
|
||||
|
||||
export const MAIN_DIST = path.join(process.env.APP_ROOT, 'dist-electron')
|
||||
export const RENDERER_DIST = path.join(process.env.APP_ROOT, 'dist')
|
||||
export const VITE_DEV_SERVER_URL = process.env.VITE_DEV_SERVER_URL
|
||||
|
||||
process.env.VITE_PUBLIC = VITE_DEV_SERVER_URL
|
||||
? path.join(process.env.APP_ROOT, 'public')
|
||||
: RENDERER_DIST
|
||||
|
||||
let win = null
|
||||
const preload = path.join(__dirname, '../preload/index.mjs')
|
||||
const indexHtml = path.join(RENDERER_DIST, 'index.html')
|
||||
|
||||
async function createWindow() {
|
||||
win = new BrowserWindow({
|
||||
title: 'CTM Document Builder',
|
||||
icon: path.join(process.env.APP_ROOT, 'assets', 'icon.ico'),
|
||||
width: 900,
|
||||
height: 700,
|
||||
webPreferences: {
|
||||
preload,
|
||||
contextIsolation: true,
|
||||
nodeIntegration: false,
|
||||
},
|
||||
})
|
||||
|
||||
if (VITE_DEV_SERVER_URL) {
|
||||
await win.loadURL(VITE_DEV_SERVER_URL)
|
||||
win.webContents.openDevTools()
|
||||
} else {
|
||||
await win.loadFile(indexHtml)
|
||||
}
|
||||
|
||||
win.webContents.setWindowOpenHandler(({ url }) => {
|
||||
if (url.startsWith('https:')) shell.openExternal(url)
|
||||
return { action: 'deny' }
|
||||
})
|
||||
}
|
||||
|
||||
app.whenReady().then(async () => {
|
||||
await createWindow()
|
||||
})
|
||||
|
||||
app.on('window-all-closed', () => {
|
||||
win = null
|
||||
if (process.platform !== 'darwin') app.quit()
|
||||
})
|
||||
|
||||
app.on('activate', () => {
|
||||
if (BrowserWindow.getAllWindows().length === 0) {
|
||||
createWindow()
|
||||
} else {
|
||||
BrowserWindow.getAllWindows()[0].focus()
|
||||
}
|
||||
})
|
||||
|
||||
ipcMain.handle('pdf:getLogo', () => {
|
||||
const logoPath = app.isPackaged
|
||||
? path.join(process.resourcesPath, 'assets', 'logo.png')
|
||||
: path.join(process.env.APP_ROOT, 'assets', 'logo.png')
|
||||
try {
|
||||
if (!fs.existsSync(logoPath)) return { base64: null }
|
||||
const buf = fs.readFileSync(logoPath)
|
||||
return { base64: buf.toString('base64') }
|
||||
} catch {
|
||||
return { base64: null }
|
||||
}
|
||||
})
|
||||
|
||||
const PDF_FOOTER_TEMPLATE = `
|
||||
<div style="width: 100%; font-size: 10px; color: #707070; padding: 0 0.67in; box-sizing: border-box;
|
||||
display: flex; justify-content: space-between; align-items: center; font-family: Helvetica, Arial, sans-serif;">
|
||||
<span class="date"></span>
|
||||
<span>Page <span class="pageNumber"></span> of <span class="totalPages"></span></span>
|
||||
</div>
|
||||
`.trim()
|
||||
|
||||
async function generatePdfFromHtml(html, { landscape = false, footer = true } = {}) {
|
||||
const printWin = new BrowserWindow({
|
||||
show: false,
|
||||
webPreferences: {
|
||||
nodeIntegration: false,
|
||||
contextIsolation: true,
|
||||
},
|
||||
})
|
||||
|
||||
const tempPath = path.join(os.tmpdir(), `ctm-print-${Date.now()}.html`)
|
||||
try {
|
||||
fs.writeFileSync(tempPath, html, 'utf8')
|
||||
await printWin.loadFile(tempPath)
|
||||
await printWin.webContents.executeJavaScript(`
|
||||
(window.__pdfLayoutReady ?? Promise.all([
|
||||
document.fonts.ready,
|
||||
...Array.from(document.images).map((img) =>
|
||||
img.complete
|
||||
? Promise.resolve()
|
||||
: new Promise((resolve) => {
|
||||
img.onload = resolve
|
||||
img.onerror = resolve
|
||||
})
|
||||
),
|
||||
]))
|
||||
`)
|
||||
|
||||
const pdfBuffer = await printWin.webContents.printToPDF({
|
||||
printBackground: true,
|
||||
preferCSSPageSize: true,
|
||||
landscape,
|
||||
displayHeaderFooter: true,
|
||||
headerTemplate: '<div></div>',
|
||||
footerTemplate: footer ? PDF_FOOTER_TEMPLATE : '<div></div>',
|
||||
margins: { marginType: 'none' },
|
||||
})
|
||||
|
||||
return pdfBuffer
|
||||
} finally {
|
||||
printWin.destroy()
|
||||
try {
|
||||
fs.unlinkSync(tempPath)
|
||||
} catch {
|
||||
// ignore cleanup errors
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
ipcMain.handle('pdf:generate', async (_, { html, landscape = false, footer = true }) => {
|
||||
if (!html || typeof html !== 'string') {
|
||||
throw new Error('PDF export requires HTML content')
|
||||
}
|
||||
const pdfBuffer = await generatePdfFromHtml(html, { landscape, footer })
|
||||
return { base64: pdfBuffer.toString('base64') }
|
||||
})
|
||||
|
||||
ipcMain.handle('pdf:save', async (_, { base64, defaultName = 'document.pdf' }) => {
|
||||
const w = BrowserWindow.getFocusedWindow() ?? win
|
||||
const { canceled, filePath } = await dialog.showSaveDialog(w, {
|
||||
defaultPath: defaultName,
|
||||
filters: [{ name: 'PDF', extensions: ['pdf'] }],
|
||||
})
|
||||
if (canceled || !filePath) return { ok: false }
|
||||
const buf = Buffer.from(base64, 'base64')
|
||||
fs.writeFileSync(filePath, buf)
|
||||
return { ok: true, filePath }
|
||||
})
|
||||
|
||||
ipcMain.handle('vaccinationRecord:importPdf', async () => {
|
||||
const w = BrowserWindow.getFocusedWindow() ?? win
|
||||
const { canceled, filePaths } = await dialog.showOpenDialog(w, {
|
||||
filters: [{ name: 'PDF', extensions: ['pdf'] }],
|
||||
properties: ['openFile'],
|
||||
})
|
||||
if (canceled || !filePaths?.[0]) return { ok: false, canceled: true }
|
||||
try {
|
||||
const data = await parseVaccinationRecordPdf(filePaths[0])
|
||||
return { ok: true, ...data }
|
||||
} catch (e) {
|
||||
const msg = e instanceof Error ? e.message : String(e)
|
||||
return { ok: false, error: msg || 'Failed to parse PDF' }
|
||||
}
|
||||
})
|
||||
|
||||
ipcMain.handle('pb:fetchEntries', () => pbService.fetchEntries())
|
||||
ipcMain.handle('pb:createEntry', (_, data) => pbService.createEntry(data))
|
||||
ipcMain.handle('pb:updateEntry', (_, payload) => pbService.updateEntry(payload.id, payload))
|
||||
ipcMain.handle('pb:deleteEntry', (_, { id }) => pbService.deleteEntry(id))
|
||||
Reference in New Issue
Block a user