-
Notifications
You must be signed in to change notification settings - Fork 1
/
Copy pathweb.js
68 lines (54 loc) · 2.09 KB
/
web.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
var fs=require('fs');
var http = require('http');
var phantom = require('node-phantom');
var Router = require('node-simple-router');
var router = Router();
router.get("/", function(request, response) {
var headers = {
'Content-type': 'image/png'
};
if (request.get.download) {
headers['Content-Disposition'] = 'attachment; filename="image.png"';
}
phantom.create(function (err,ph) {
ph.createPage(function (err,page) {
page.set('settings.customHeaders', {'Referer': request.get.url}, function(err) {
page.set('settings.userAgent', 'Mozilla/5.0 (Macintosh; Intel Mac OS X 10_9_2) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/37.0.1995.2 Safari/537.36', function(err) {
page.setViewport({ width: request.get.width || 1024, height: 768 }, function(err){
var resourceCount = 0;
var timeout;
page.onResourceRequested = function (ev) {
resourceCount++;
clearTimeout(timeout);
// console.log('request', resourceCount);
}
page.onResourceReceived = function (ev) {
if (ev.stage == 'end') {
resourceCount--;
// console.log('receive', resourceCount);
if (resourceCount <= 0) {
clearTimeout(timeout);
timeout = setTimeout(function() {
console.log('Rasterising...')
page.renderBase64('png', function(err, imagedata){
response.end(new Buffer(imagedata, 'base64'));
console.log("Finished rendering: ", request.get.url);
ph.exit();
});
}, 1000);
}
}
};
console.log("Started rendering: ", request.get.url);
page.open(request.get.url, function (err, status) {
// handler error here
response.writeHead(200, headers);
});
});
});
});
});
});
});
var server = http.createServer(router);
server.listen(process.env.PORT);