CTM-Document-Builder/electron/main/index.js

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))