old-kivi/backend/server.js

import express from 'express'
import bodyParser from 'body-parser';
import distanceInWordsToNow from 'date-fns/distance_in_words_to_now';
var hr = require('date-fns/locale/hr');

var MongoClient = require('mongodb').MongoClient;
var ObjectID = require('mongodb').ObjectID;

var url = 'mongodb://localhost:27017/kivi';

require("babel-polyfill");

const router = express.Router({mergeParams: true})

const PORT = process.env.PORT || 3001;
const AGENTURA_KEY = process.env.AGENTURA_KEY || '1somethingverysecret';

let db;

router.get('/search/listings/:id', async (req, res, next) => {
  try {
    const id = req.params.id;

    const listings = db.collection('listings');
    const listing = await listings.findOne({_id: new ObjectID(id)});
    if (listing) {
      res.json(listing);
    } else {
      res.status(404);
    }

    res.end();
  } catch (e) {
    console.log('error:', e);
    next(e);
  }
});

router.get('/search/listings', async (req, res, next) => {
  try {
    const bounds       = req.query.bounds || '';
    const minPrice     = req.query.minPrice;
    const maxPrice     = req.query.maxPrice;
    const minSize      = req.query.minSize;
    const maxSize      = req.query.maxSize;
    const rooms        = req.query.rooms;
    const adType       = req.query.adType;
    const category     = req.query.category;
    const sort         = req.query.sort;
    const page         = req.query.page || 0;
    const pins         = req.query.pins || false;

    const properties = db.collection('listings');
    let query        = {};

    if (bounds) {
      const [lat1, lng1, lat2, lng2] = bounds.split(',').map(parseFloat)
      const box                      = [[lat1, lng1], [lat2, lng2]];

      query = Object.assign(query, {
        loc: {
          "$geoWithin": {
            "$box": box
          }
        }
      });
    }

    if (adType) {
      query = Object.assign(query, {
        adType: parseInt(adType)
      });
    }

    if (minPrice || maxPrice) {
      const price = {}
      if (minPrice) {
        price["$gte"] = parseFloat(minPrice);
      }

      if (maxPrice) {
        price["$lte"] = parseFloat(maxPrice);
      }

      query = Object.assign(query, {
        price
      });
    }

    const and = [];
    if (rooms) {
      const allRooms = rooms.split(',');
      const or = allRooms.map(val => {
        if (val === '4+') {
          return {
            rooms: {
              "$gte": 4
            }
          }
        }
        return {
          rooms: parseFloat(val)
        };
      });

      and.push({ "$or": or });
    }

    if (minSize || maxSize) {
      const size = {}
      if (minSize) {
        size["$gte"] = parseFloat(minSize);
      }

      if (maxSize) {
        size["$lte"] = parseFloat(maxSize);
      }

      query = Object.assign(query, {
        size
      });
    }

    if (category) {
      const allCategories = category.split(',');
      const or = allCategories.map(val => {
        return {
          category: parseInt(val)
        };
      });

      and.push({ "$or": or });
    }

    if (and.length > 0) {
      query = Object.assign(query, {
        "$and": and
      });
    }

    console.log('QUERY: ', query);
    const cnt = await properties.find(query).count();

    res.header('X-Total-Count', cnt);

    const getSort = () => {
      if (sort === 'price-min') {
        return [['price', 'asc']];
      } else if (sort === 'price-max') {
        return [['price', 'desc']];
      } else if (sort === 'newest') {
        return [['_id', 'desc']];
      } else if (sort === 'relevance') {
        // TODO: figure out what the relevance is
        return [];
      }
    };

    let all = properties.find(query, {
      //"sort": [['field1','asc'], ['field2','desc']]
      "sort": getSort()
    });

    const isPins = pins === "true";

    if (!isPins) {
      all = await all.skip(20 * page).limit(20).toArray();
    } else {
      all = await all.toArray();
    }

    if (all.length > 0) {
      res.header('X-Last-Record-Id', [...all].pop()._id);
    }

    if (isPins) {
      res.json(all.map(val => {
        return {
          _id: val._id,
          loc: val.loc
        }
      }));
    } else {
      res.json(all.map(({_id,
                       address,
                       images,
                       price,
                       rooms,
                       size,
                       time
      }) => ({
        _id,
        address,
        images: [images[0]],
        price,
        rooms,
        size,
        time: distanceInWordsToNow(
          new Date(time),
          {locale: hr}
        )
      })));
    }

    res.end();
  } catch (e) {
    console.log('error:', e);
    next(e);
  }
});


const app = express()
app.use(bodyParser.json());

app.use(function(req, res, next) {
  res.header("Access-Control-Allow-Origin", "*");
  res.header("Access-Control-Allow-Headers", "Origin, X-Requested-With, Content-Type, Accept, X-Last-Record-Id, X-Total-Count");
  res.header("Access-Control-Expose-Headers", "X-Last-Record-Id, X-Total-Count");
  res.header("Access-Control-Allow-Methods", "GET, POST, OPTIONS");
  res.header('Access-Control-Allow-Credentials', 'true');
  next();
});

app.use('/api', router);

MongoClient.connect(url).then((database) => {
  db = database;
  db.collection('listings').createIndex({loc: "2d"});
  app.listen(PORT, () => console.log('Express server running at localhost: ' + PORT));
});