-
Notifications
You must be signed in to change notification settings - Fork 0
/
server.js
97 lines (86 loc) · 2.2 KB
/
server.js
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
const PROTO_PATH = __dirname + "/renderer.proto";
var fs = require("fs");
const grpc = require("grpc");
const { Chromeless } = require("chromeless");
const renderer_proto = grpc.load(PROTO_PATH).renderer;
const agent =
"Mozilla/5.0 (Macintosh; Intel Mac OS X 10_13_3) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/65.0.3325.181 Safari/537.36";
async function runRender(url) {
const chromeless = new Chromeless();
let html = await chromeless
.setUserAgent(agent)
.goto(url)
.wait(1500)
.html();
await chromeless.end();
return html;
}
async function runPrint(url) {
const chromeless = new Chromeless();
let src = await chromeless
.setUserAgent(agent)
.goto(url)
.wait(1500)
.pdf();
await chromeless.end();
var bmp = fs.readFileSync(src);
return new Buffer(bmp).toString("base64");
}
async function runScreenshot(url) {
const chromeless = new Chromeless();
let src = await chromeless
.setUserAgent(agent)
.goto(url)
.wait(1500)
.screenshot();
await chromeless.end();
var bmp = fs.readFileSync(src);
return new Buffer(bmp).toString("base64");
}
/**
* Implements the CheckHealth RPC method.
*/
function checkHealth(empty, callback) {
// TODO - can we validate that chromeless is working?
callback(null, {});
}
/**
* Implements the Render RPC method.
*/
function render(req, callback) {
runRender(req.request.url)
.then(data => callback(null, { data }))
.catch(err => callback(err));
}
/**
* Implements the Print RPC method.
*/
function print(req, callback) {
runPrint(req.request.url)
.then(data => callback(null, { data }))
.catch(err => callback(err));
}
/**
* Implements the Screenshot RPC method.
*/
function screenshot(req, callback) {
runScreenshot(req.request.url)
.then(data => callback(null, { data }))
.catch(err => callback(err));
}
/**
* Starts an RPC server that receives requests for the Greeter service at the
* sample server port
*/
function main() {
var server = new grpc.Server();
server.addService(renderer_proto.RendererService.service, {
checkHealth,
render,
screenshot,
print
});
server.bind("0.0.0.0:3000", grpc.ServerCredentials.createInsecure());
server.start();
}
main();