1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
| const puppeteer = require('puppeteer');
const express = require('express');
const { readFile } = require('node:fs/promises');
const app = express();
const PORT = process.env.PORT || 3000;
app.use(express.json({ limit: '10mb' }));
app.use(express.urlencoded({ extended: true, limit: '10mb' }));
app.get('/health', (req, res) => {
res.json({
status: 'healthy',
timestamp: new Date().toISOString(),
node: process.version,
memory: process.memoryUsage()
});
});
app.get('/', async (req, res) => {
let content_url = encodeURIComponent('https://www.psvmc.cn/article/2026-01-07-android-audio-m4a-recorder.html');
let header_url = encodeURIComponent('http://49.4.11.108:8924/static/paperPDF/header.html');
let finish_selector = encodeURIComponent('.footer');
let top = 25;
let bottom = 10;
let timeout_sec = 180;
let url_all = `/html2pdf?header_url=${header_url}&content_url=${content_url}&finish_selector=${finish_selector}&top=${top}&bottom=${bottom}&timeout_sec=${timeout_sec}`;
let htmlAll = ""
try {
htmlAll = await readFile('./home.html', 'utf8');
} catch (err) {
console.error('读取文件失败:', err.message);
}
htmlAll = htmlAll.replace("{{url_all}}", url_all);
res.send(htmlAll);
});
app.get('/test', async (req, res) => {
try {
const browser = await puppeteer.launch({
headless: "new",
timeout: 1000 * 300,
args: [
"--no-sandbox",
"--disable-setuid-sandbox",
"--disable-dev-shm-usage",
],
});
const page = await browser.newPage();
await page.setContent('<p>服务运行正常！</p>');
const pdfBuffer = await page.pdf({ format: 'A4' });
await browser.close();
res.set({
'Content-Type': 'application/pdf',
'Content-Disposition': 'inline; filename="test.pdf"'
});
res.send(pdfBuffer);
} catch (error) {
res.status(500).json({ code: 1, msg: error.message });
}
});
app.get('/html2pdf', async (req, res) => {
try {
let { timeout_sec = 120, header_url, content_url, finish_selector, top, bottom, left, right } = req.query;
let timeout = Number(timeout_sec) * 1000;
top = top || 5;
bottom = bottom || 5;
left = left || 5;
right = right || 5;
if (!content_url) {
return res.status(500).json({
code: 1,
msg: '内容地址不能为空'
});
}
const browser = await puppeteer.launch({
headless: "new",
timeout: Number(timeout),
args: [
"--no-sandbox",
"--disable-setuid-sandbox",
"--disable-dev-shm-usage",
],
});
const page = await browser.newPage();
await page.goto(content_url, {
waitUntil: "networkidle2",
timeout: Number(timeout)
});
if (finish_selector) {
await page.waitForSelector(finish_selector, {
timeout: Number(timeout)
});
}
await new Promise((resolve) => setTimeout(resolve, 1000));
let headerHtml = '';
if (header_url) {
try {
const response = await fetch(header_url);
if (response.ok) {
headerHtml = await response.text();
}
} catch (err) {
console.error("获取远程页眉模板失败：", err);
}
}
let pdfOptions = {
format: "A4",
printBackground: true,
displayHeaderFooter: true,
headerTemplate: headerHtml,
preferCSSPageSize: true,
footerTemplate: `
<div style="position: absolute;bottom:0;z-index:999;font-size: 12px; width: 100%; height:40px;line-height:40px; text-align: center;">
<span class="pageNumber"></span> / <span class="totalPages"></span>
</div>`,
margin: {
top: `${top}mm`,
left: `${left}mm`,
right: `${right}mm`,
bottom: `${bottom}mm`,
},
}
const pdfBuffer = await page.pdf(pdfOptions);
await browser.close();
res.set({
'Content-Type': 'application/pdf',
'Content-Length': pdfBuffer.length,
'Content-Disposition': 'inline; filename="document.pdf"'
});
res.send(pdfBuffer);
} catch (error) {
res.status(500).json({ code: 1, msg: error.message });
}
});
app.listen(PORT, () => {
console.log(`PDF服务运行在 http://localhost:${PORT}`);
console.log(`环境: ${process.env.NODE_ENV || 'development'}`);
console.log(`服务启动时间: ${new Date().toISOString()}`);
});
|