index.js.v2

/**
 * Copyright 2017, Google, Inc.
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *    http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 **/
var path = require('path');
var fs = require('fs');

var express = require('express');
var multer = require('multer');
var request = require('request');

var app = express();
var port = process.env.port || 8080;

var storage = multer.memoryStorage();
var upload = multer({ storage: storage }).single('userPhoto');

// From http://expressjs.com/en/guide/error-handling.html
function errorHandler(err, req, res, next) {
    res.status(500).send(JSON.stringify({ error: err }));
    next();
}

function sendPhoto(req, res, next) {
    request.post({
        url: 'http://flipper-service',
        encoding: null,
        formData: {
            file: {
                value: req.file.buffer,
                options: {
                    filename: req.file.originalname,
                    contentType: 'image/jpg'
                }
            }
        }
    }, (err, httpResponse, body) => {
        if (err) return errorHandler(err, req, res, next);
        res.contentType('jpeg')
        res.send(body);
        next();
    });
}

app.get('/', (req, res, next) => {
    res.sendFile(path.join(__dirname, 'index.html'), next);
});

app.get('/api/version', (req, res, next) => {
    fs.readFile('version.txt', 'utf8', function(err, data) {
        res.contentType('text/html')
        if (err) {
            res.send("N/A");
        }
        res.send(data);
    });
});

app.post('/api/photo', (req, res, next) => {
    upload(req, res, function(err) {
        if (err || !req.file) return errorHandler(err, req, res, next);
        sendPhoto(req, res, next);
    });
});

app.listen(port, _ => {
    console.log(`Listening on port ${port}`);
});