html2image/server.js at master · bomee/html2image · GitHub

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
var express = require('express');
var app = express();
var path = require('path');
var execFile = require('child_process').execFile;
var phantomjs = require('phantomjs-prebuilt');
var binPath = phantomjs.path;
var fs = require('fs');
var querystring = require('querystring');
var bodyParser = require('body-parser');

const hostname = '127.0.0.1';
const port = process.env.PORT || 5050;

var tempFile = function(ext){
  return `temp_${timestamp}.${ext}`;
}

app.get('/echart2png', function (req, res) {
  var fileName = tempFile('png');
  var url = `http://${hostname}:${port}/echart?` + querystring.stringify(req.query);
  execFile(binPath, ['phantomjs-url2png.js', url, fileName], function(err, stdout, stderr) {
    !err && res.sendFile(fileName, {root:__dirname}, function (err) {
      if (err) {
        console.log(err);
        res.status(500).end();
      }
      fs.unlink(fileName);
    });
  });
});

app.get('/highcharts2png', function (req, res) {
  var fileName = tempFile('png');
  var url = `http://${hostname}:${port}/highcharts?` + querystring.stringify(req.query);
  execFile(binPath, ['phantomjs-url2png.js', url, fileName], function(err, stdout, stderr) {
    !err && res.sendFile(fileName, {root:__dirname}, function (err) {
      if (err) {
        console.log(err);
        res.status(500).end();
      }
      fs.unlink(fileName);
    });
  });
});

app.all('/html2png', function (req, res) {
  var fileName = tempFile('png');
  execFile(binPath, ['phantomjs-html2png.js', req.body || req.query.html, fileName, req.query.w, req.query.h], function(err, stdout, stderr) {
    !err && res.sendFile(fileName, {root:__dirname}, function (err) {
      if (err) {
        console.log(err);
        res.status(500).end();
      }
      fs.unlink(fileName);
    });
  });
});

app.all('/html2pdf', function (req, res) {
  var fileName = tempFile('pdf');
  execFile(binPath, ['phantomjs-html2pdf.js', req.body || req.query.html, fileName], function(err, stdout, stderr) {
    !err && res.sendFile(fileName, {root:__dirname}, function (err) {
      if (err) {
        console.log(err);
        res.status(500).end();
      }
      fs.unlink(fileName);
    });
  });
});

app.get('/echart', function(req, res){
  res.send(`
    <!DOCTYPE html>
    <html>
      <head><script src="http://echarts.baidu.com/dist/echarts.min.js"></script></head>
      <body>
        <div id="echart" style="width:${req.query.w}px; height:${req.query.h}px;"></div>
      </body>
      <script>echarts.init(document.getElementById("echart")).setOption(${req.query.opts});</script>
    </html>`
  );
});

app.get('/highcharts', function(req, res){
  res.send(`
    <!DOCTYPE html>
    <html>
      <head>
        <script src="http://cdn.hcharts.cn/jquery/jquery-1.8.3.min.js"></script>
        <script src="http://cdn.hcharts.cn/highcharts/highcharts.js" ></script>
        <script src="http://cdn.hcharts.cn/highcharts/highcharts-more.js" ></script>
      </head>
      <body>
        <div id="container" style="width:${req.query.w}px; height:${req.query.h}px;"></div>
      </body>
      <script>
        Highcharts.setOptions({
            plotOptions: {
                series: {
                    animation: false
                }
            }
        });
        $("#container").highcharts(${req.query.opts});
    </script>
    </html>`
  );
});

console.log(`html to image server running at http://${hostname}:${port}`);
app.listen(port);