app.js

import express from 'express';

const svg2img = require('svg2img');
const aws = require('./utils/aws');

var crypto = require('crypto');

// React Components
import React from 'react';
import RDS from 'react-dom/server';
import Avataaars from 'avataaars';

const app = express();

app.get('/', async (req, res) => {
  if (req.query.facialHairType === 'BeardMagestic') {
    req.query.facialHairType = 'BeardMajestic';
  }
  const appString = RDS.renderToString(<Avataaars {...req.query} />);

  res.writeHead(200, {
    'Content-Type': 'image/svg+xml',
  });
  res.end(appString);
});

const getHash = (req) => {
  return crypto.createHash('md5').update(req.path + "-" + JSON.stringify(req.query)).digest('hex');
};

app.get('/png/:width?', async (req, res) => {
  if (process.env.DISABLE_PNG) {
    return res.send('PNG is turned off due to abuse, but you can use SVGs still! https://github.com/gkoberger/avataaars/issues/16');
  }

  // You'll have to add this back to the package.json
  const { convert } = require('convert-svg-to-png');

  const hash = getHash(req);
  const fileName = `${getHash(req)}.png`;

  res.set('Content-Type', 'image/png');

  aws.getObject(fileName, async (err, data) => {
    if (data) {
      console.log('Existing avatar found');
      return res.end(data.Body);
    }

    const appString = RDS.renderToString(<Avataaars {...req.query} />);

    const png = await convert(appString, {
      width: parseInt(req.params.width || 500, 10),
      puppeteer: {
        args: ['--no-sandbox', '--disable-setuid-sandbox'],
      },
    });

    console.log('Generating new avatar');
    aws.uploadFile(fileName, png, () => {
      res.end(png);
    });
  });
});

// catch 404 and forward to error handler
app.use(function(req, res, next) {
  let err = new Error('Not Found');
  err.status = 404;
  next(err);
});

// error handlers

// development error handler
// will print stacktrace
if (app.get('env') === 'development') {
  app.use(function(err, req, res, next) {
    res.status(err.status || 500);
    res.render('error', {
      message: err.message,
      error: err,
    });
  });
}

module.exports = app;