mirror of
https://github.com/NohamR/pdfy.git
synced 2026-10-10 18:29:55 +00:00
Add server mode
This commit is contained in:
9
.dockerignore
Normal file
9
.dockerignore
Normal file
@@ -0,0 +1,9 @@
|
|||||||
|
node_modules
|
||||||
|
output
|
||||||
|
extension/reader-view
|
||||||
|
extension/*.crx
|
||||||
|
extension/*.zip
|
||||||
|
.git
|
||||||
|
.gitignore
|
||||||
|
.DS_Store
|
||||||
|
*~
|
||||||
33
Dockerfile
Normal file
33
Dockerfile
Normal file
@@ -0,0 +1,33 @@
|
|||||||
|
FROM node:22-bookworm-slim
|
||||||
|
|
||||||
|
# Install Chromium
|
||||||
|
RUN apt-get update && apt-get install -y \
|
||||||
|
chromium \
|
||||||
|
unzip \
|
||||||
|
ca-certificates \
|
||||||
|
--no-install-recommends \
|
||||||
|
&& rm -rf /var/lib/apt/lists/*
|
||||||
|
|
||||||
|
# Tell Puppeteer to use the system Chromium instead of downloading its own
|
||||||
|
ENV PUPPETEER_SKIP_BROWSER_DOWNLOAD=true
|
||||||
|
ENV PUPPETEER_EXECUTABLE_PATH=/usr/bin/chromium
|
||||||
|
|
||||||
|
WORKDIR /app
|
||||||
|
|
||||||
|
COPY package.json package-lock.json ./
|
||||||
|
RUN npm ci --omit=dev
|
||||||
|
|
||||||
|
COPY . .
|
||||||
|
|
||||||
|
# Pre-download the Reader View extension so it works offline at runtime
|
||||||
|
RUN node src/download-extension.js
|
||||||
|
|
||||||
|
RUN useradd -m appuser && chown -R appuser:appuser /app
|
||||||
|
USER appuser
|
||||||
|
|
||||||
|
EXPOSE 8080
|
||||||
|
|
||||||
|
ENV PORT=8080
|
||||||
|
ENV LOG_LEVEL=info
|
||||||
|
|
||||||
|
CMD ["node", "src/server.js"]
|
||||||
37
README.md
37
README.md
@@ -144,6 +144,43 @@ try {
|
|||||||
**`BotProtectionError`** — thrown when the target page appears to be behind bot protection.
|
**`BotProtectionError`** — thrown when the target page appears to be behind bot protection.
|
||||||
|
|
||||||
|
|
||||||
|
## HTTP Server
|
||||||
|
|
||||||
|
Run as a headless service via Docker or directly with Node:
|
||||||
|
|
||||||
|
```bash
|
||||||
|
# Start server
|
||||||
|
npm run start:server
|
||||||
|
|
||||||
|
# Or via Docker
|
||||||
|
docker compose up -d
|
||||||
|
```
|
||||||
|
|
||||||
|
### Endpoints
|
||||||
|
|
||||||
|
```
|
||||||
|
POST /convert Generate a PDF from a URL
|
||||||
|
GET /health Health check (browser status, uptime)
|
||||||
|
GET / Help page
|
||||||
|
```
|
||||||
|
|
||||||
|
### Usage
|
||||||
|
|
||||||
|
```bash
|
||||||
|
curl -X POST http://localhost:8080/convert \
|
||||||
|
-H "Content-Type: application/json" \
|
||||||
|
-d '{"url":"https://example.com/article","theme":"dark"}' \
|
||||||
|
-o article.pdf
|
||||||
|
```
|
||||||
|
|
||||||
|
| Body field | Type | Description |
|
||||||
|
|------------|------|-------------|
|
||||||
|
| `url` | string | Article URL (required unless `html` is provided) |
|
||||||
|
| `theme` | string | Reader View theme |
|
||||||
|
| `css` | string | Custom CSS string |
|
||||||
|
| `html` | string | Pre-fetched HTML (bypasses URL fetch) |
|
||||||
|
| `prefs` | object | Extension preferences |
|
||||||
|
|
||||||
## Credits
|
## Credits
|
||||||
|
|
||||||
- [Reader View](https://chromewebstore.google.com/detail/reader-view/ecabifbgmdmgdllomnfinbmaellmclnh) Chrome extension
|
- [Reader View](https://chromewebstore.google.com/detail/reader-view/ecabifbgmdmgdllomnfinbmaellmclnh) Chrome extension
|
||||||
|
|||||||
10
docker-compose.yml
Normal file
10
docker-compose.yml
Normal file
@@ -0,0 +1,10 @@
|
|||||||
|
services:
|
||||||
|
pdfy:
|
||||||
|
build: .
|
||||||
|
ports:
|
||||||
|
- "8080:8080"
|
||||||
|
environment:
|
||||||
|
- PORT=8080
|
||||||
|
- LOG_LEVEL=info
|
||||||
|
restart: unless-stopped
|
||||||
|
init: true
|
||||||
@@ -10,6 +10,7 @@
|
|||||||
},
|
},
|
||||||
"scripts": {
|
"scripts": {
|
||||||
"start": "node src/index.js",
|
"start": "node src/index.js",
|
||||||
|
"start:server": "node src/server.js",
|
||||||
"test": "node src/test/test.js"
|
"test": "node src/test/test.js"
|
||||||
},
|
},
|
||||||
"standard": {
|
"standard": {
|
||||||
|
|||||||
128
src/server.js
Normal file
128
src/server.js
Normal file
@@ -0,0 +1,128 @@
|
|||||||
|
import { createServer } from 'node:http'
|
||||||
|
import process from 'node:process'
|
||||||
|
import { convert, launchBrowser, BotProtectionError, VALID_THEMES } from './pdfy.js'
|
||||||
|
import logger from './logger.js'
|
||||||
|
|
||||||
|
const PORT = parseInt(process.env.PORT || '8080', 10)
|
||||||
|
const HOST = process.env.HOST || '0.0.0.0'
|
||||||
|
|
||||||
|
let browser
|
||||||
|
|
||||||
|
async function parseBody (req) {
|
||||||
|
const chunks = []
|
||||||
|
for await (const chunk of req) {
|
||||||
|
chunks.push(chunk)
|
||||||
|
}
|
||||||
|
return JSON.parse(Buffer.concat(chunks).toString())
|
||||||
|
}
|
||||||
|
|
||||||
|
function sendJson (res, status, data) {
|
||||||
|
res.writeHead(status, { 'Content-Type': 'application/json' })
|
||||||
|
res.end(JSON.stringify(data))
|
||||||
|
}
|
||||||
|
|
||||||
|
async function handleConvert (req, res) {
|
||||||
|
let body
|
||||||
|
try {
|
||||||
|
body = await parseBody(req)
|
||||||
|
} catch {
|
||||||
|
return sendJson(res, 400, { error: 'Invalid JSON body' })
|
||||||
|
}
|
||||||
|
|
||||||
|
const { url, theme, css, prefs, html } = body
|
||||||
|
|
||||||
|
if (!url && !html) {
|
||||||
|
return sendJson(res, 400, { error: 'Missing required field: url (or html for direct content)' })
|
||||||
|
}
|
||||||
|
|
||||||
|
if (theme && !VALID_THEMES.includes(theme)) {
|
||||||
|
return sendJson(res, 400, {
|
||||||
|
error: `Invalid theme. Valid themes: ${VALID_THEMES.join(', ')}`
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
try {
|
||||||
|
const result = await convert(url || 'about:blank', {
|
||||||
|
browser,
|
||||||
|
html: html || null,
|
||||||
|
theme: theme || 'light',
|
||||||
|
css: css || null,
|
||||||
|
prefs: prefs || {}
|
||||||
|
})
|
||||||
|
|
||||||
|
const sizeKb = (result.pdfBuffer.length / 1024).toFixed(0)
|
||||||
|
logger.info(`Converted: "${result.title}" (${sizeKb} KB)`)
|
||||||
|
|
||||||
|
res.writeHead(200, {
|
||||||
|
'Content-Type': 'application/pdf',
|
||||||
|
'Content-Disposition': `attachment; filename="${encodeURIComponent(result.title)}.pdf"`,
|
||||||
|
'Content-Length': result.pdfBuffer.length
|
||||||
|
})
|
||||||
|
res.end(result.pdfBuffer)
|
||||||
|
} catch (err) {
|
||||||
|
if (err instanceof BotProtectionError) {
|
||||||
|
return sendJson(res, 400, {
|
||||||
|
error: 'Bot protection detected.',
|
||||||
|
hint: 'Pass the page HTML directly via the "html" field.'
|
||||||
|
})
|
||||||
|
}
|
||||||
|
logger.error(`Conversion failed: ${err.message}`)
|
||||||
|
sendJson(res, 500, { error: err.message })
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
const server = createServer((req, res) => {
|
||||||
|
const { method, url } = req
|
||||||
|
|
||||||
|
if (method === 'POST' && url === '/convert') {
|
||||||
|
return handleConvert(req, res)
|
||||||
|
}
|
||||||
|
|
||||||
|
if (method === 'GET' && url === '/health') {
|
||||||
|
return sendJson(res, 200, {
|
||||||
|
status: 'ok',
|
||||||
|
browser: browser ? 'connected' : 'disconnected',
|
||||||
|
uptime: process.uptime()
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
if (method === 'GET' && url === '/') {
|
||||||
|
res.writeHead(200, { 'Content-Type': 'text/plain' })
|
||||||
|
res.end(`
|
||||||
|
pdfy server
|
||||||
|
|
||||||
|
POST /convert Convert a URL to PDF
|
||||||
|
Body: { "url": "...", "theme": "...", "css": "...", "html": "..." }
|
||||||
|
|
||||||
|
GET /health Health check
|
||||||
|
GET / This help
|
||||||
|
`)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
sendJson(res, 404, { error: 'Not found' })
|
||||||
|
})
|
||||||
|
|
||||||
|
async function start () {
|
||||||
|
logger.info('Launching browser...')
|
||||||
|
browser = await launchBrowser()
|
||||||
|
logger.info('Browser ready')
|
||||||
|
|
||||||
|
server.listen(PORT, HOST, () => {
|
||||||
|
logger.info(`Server listening on http://${HOST}:${PORT}`)
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
async function shutdown () {
|
||||||
|
logger.info('Shutting down...')
|
||||||
|
if (browser) await browser.close()
|
||||||
|
server.close(() => process.exit(0))
|
||||||
|
}
|
||||||
|
|
||||||
|
process.on('SIGINT', shutdown)
|
||||||
|
process.on('SIGTERM', shutdown)
|
||||||
|
|
||||||
|
start().catch((err) => {
|
||||||
|
logger.error(`Failed to start: ${err.message}`)
|
||||||
|
process.exit(1)
|
||||||
|
})
|
||||||
Reference in New Issue
Block a user