server.js

const express = require('express')
const app = express()
const bodyParser = require('body-parser');
const mongoose = require('mongoose');
const punycode = require('punycode');
const subdomain = require('subdomain');
const check = require('ch3ck');
const emoji = require('./emoji');
const parser = require('ua-parser-js');
const http = require('http').Server(app);
const io = require('socket.io')(http);
const port = process.env.PORT || 3000;

app.use(subdomain({ base : '🌍✌🏼', removeWWW : true }));
app.use(bodyParser.json());
app.use(bodyParser.urlencoded({
  extended: true
}));
app.use(express.static('public'));

var Schema = new mongoose.Schema({
  key: { type: String, unique: true, required: true },
  url: { type: String, unique: true, required: true },
  stats: []
});

var mongolabUri = process.env.MONGODB_URI;
mongoose.connect(mongolabUri, err => console.log(err ? err : 'Mongo connected.'));

var Keys = mongoose.model('Sites', Schema);

const find = (key) => {
  return new Promise((resolve, reject) => {
    console.log(key);
    Keys.findOne({ 'key': key }, (err, key) => {
      console.log(key);
      if (key && key.url.length > 0) {
        resolve(key);
      } else {
        reject(true);
      }
    });
  });
}

app.get('/check/:key', async (req, res) => {
  try {
    let response = await find(req.params.key);
      response.message = `Aww, ${res.url} grabbed it first!`
      res.json(response);
  } catch (e) {
      res.json('Available for purchase! Haha, It\'s joke. But contributions is welcome.')
  }
});

app.get('/subdomain/:key/', async (req, res) => {
  try {
    req.params.key = punycode.toUnicode(req.params.key);
    const response = await find(req.params.key);
    console.log('REDIRECT', response);
    res.redirect(301, response.url);
  } catch (e) {
    res.json({message:'Available for purchase! Haha, It\'s joke. But contributions is welcome.'});
  }
});

app.get('/', (req, res) => res.sendFile('index.html'));

app.get('/:key', async (req, res) => {

  Keys.findOne({ 'key': req.params.key }, (err, key) => {
    if (key && key.url.length > 0) {
      let ua = parser(req.headers['user-agent']);
      let country = req.headers['cf-ipcountry'];
      let ip = req.headers['cf-connecting-ip'];
      ua = {
         browser: ua.browser,
         os: ua.os,
       }
      key.stats.push({ip,country,ua});
      key.save();
      res.redirect(301, key.url);
    } else {
      res.json({message:'Available for purchase! Haha, It\'s joke. But contributions is welcome.'});
    }
  });
})

app.get('/stats/:key', (req, res) => res.sendFile(__dirname + '/public/traffic.html'));

app.post('/', async (req, res) => {
  if (check(req.body.url) !== false) {
    let url = check(req.body.url);
    console.log(url);
    let emoji = req.body.emoji;

      let keys = new Keys({
       url,
       key: emoji,
       stats: [],
     });

     keys.save((err, doc) => {
       if (err) {
         res.json({status:false, message: 'Somebody grabbed your emoji or you already shortened your url.'})
       } else {
         io.sockets.emit('new', `${emoji} grabbed by ${url} now.`)
         res.json({status:true, url: `http://${punycode.toUnicode(req.headers.host)}/${emoji}`, subdomain: `http://${emoji}.${punycode.toUnicode(req.headers.host)}`, friendly: `http://coool.ws/${emoji}`, stats:`http://${punycode.toUnicode(req.headers.host)}/stats/${emoji}`})
       }
     });
 } else {
   res.json({status:false, message: 'Gimme valid url.'})
 }
});

io.on('connection', (socket) => {
  socket.on('url', async (msg) => {
    if (check(msg) !== false) {
      let url = check(msg);
      console.log(url);
      let response = await emoji(url);
      console.log(response, url);
      io.to(socket.id).emit('emojis', {status:true, response})
   } else {
     io.to(socket.id).emit('emojis', {status:false, response:'Gimme valid url.'})
   }
  });
});

http.listen(process.env.PORT || 3000, () => console.log(`Shitty emoji shortener listening on port ${process.env.PORT || 3000}!`));