db.js

var mongo = require('mongodb').MongoClient;
var airports = require('./json/airports.json');
var routes  = require('./json/routes.json');
var moment = require('moment');
var DB = null;
var dbURL = process.env.MONGODB_URI  ||
            'mongodb://localhost:27017/alaska';
var ObjectId = require('mongodb').ObjectId;

exports.connect = function(cb) {
  return mongo.connect(dbURL, function(err, db) {
    if (err) return cb(err);
    console.log('connected to db');
    DB = db;
    cb(null, db);
  });
}

exports.db = function() {
  if (DB === null) throw Error('DB Object has not yet been initialized');
  return DB;
};
exports.contact= function(contact,cb){
  DB.collection('contactUs').insert(contact, function(err){
    if(err)
    cb(err);
  });
}
function seedFlights(cb){
  DB.collection('flights').find().toArray(function (err, docs) {
    if (err) return cb(err);
    if (docs.length > 0)
    cb(false);
    else{
      outGoingFlightsSeed(function(){
        returnFlightSeed(function(){
          cb();
        })
      });
    }
  });
}

function outGoingFlightsSeed(cb){
  seedAFlight(false,function(err){
    if (err) return err;
    cb();
  });
}

function returnFlightSeed(cb){
  seedAFlight(true,function(err){
    if (err) return err;
    cb();
  });
}

function flightNumberGenerator (min, max) {
  return Math.floor(Math.random() * (max - min + 1) + min);
}

function seedAFlight (reverseFlag, cb){

  var allFlights = [];
  for(var i=0 ;i<routes.length;i++){
    var flight = routes[i];
    if(reverseFlag) {
      var temp = routes[i].origin;
      flight = routes[i];
      flight.origin = flight.destination;
      flight.destination = temp;
    }
    for (var k=1 ; k<=45 ; k++){
      var flightToBeInserted = {};
      var seatmap = [];
      flightToBeInserted.flightNumber = flight.flightNumber+(k * flightNumberGenerator(k,(routes.length*45)));
      flightToBeInserted.aircraftModel = flight.aircraftModel;
      flightToBeInserted.aircraftType = flight.aircraftType;
      flightToBeInserted.capacity = flight.capacity;
      flightToBeInserted.duration = flight.duration;
      flightToBeInserted.origin = flight.origin;
      flightToBeInserted.destination = flight.destination;
      flightToBeInserted.economyCost = flight.economyCost;
      flightToBeInserted.businessCost = flight.businessCost;
      flightToBeInserted.availableSeats = flight.availableSeats;
      flightToBeInserted.availableEconomySeats = 40;
      flightToBeInserted.availableBusinessSeats = 20;
      // flightToBeInserted.status = flight.status;
      flightToBeInserted.departureDateTime = moment('2016-04-13 12:25 AM', 'YYYY-MM-DD hh:mm A').add(k, 'days').format('YYYY-MM-DD hh:mm A');
      flightToBeInserted.arrivalDateTime = moment('2016-04-13 15:25 AM', 'YYYY-MM-DD hh:mm A').add(k, 'days').format('YYYY-MM-DD hh:mm A');
      for(var j=1 ; j<11 ; j++){
        var seat = {};
        seat.seatNumber = "A"+j
        seat.type = 'window'
        seat.cabin = 'business'
        seat.cost  = flight["businessCost"]
        seat.reservationID = null
        seatmap.push(seat);
      }
      for(var j=1 ; j<11 ; j++){
        var seat = {};
        seat.seatNumber = "B"+j
        seat.type = 'window'
        seat.cabin = 'business'
        seat.cost  = flight["businessCost"]
        seat.reservationID = null
        seatmap.push(seat);
      }
      for(var j =1 ; j<11 ; j++){
        var seat = {};
        seat.seatNumber = "C"+j
        seat.type = 'aisle'
        seat.cabin = 'economy'
        seat.cost  = flight["economyCost"]
        seat.reservationID = null
        seatmap.push(seat);
      }
      for(var j =1 ; j<11 ; j++){
        var seat = {};
        seat.seatNumber = "D"+j
        seat.type = 'aisle'
        seat.cabin = 'economy'
        seat.cost  = flight["economyCost"]
        seat.reservationID = null
        seatmap.push(seat);
      }
      for(var j =1 ; j<11 ; j++){
        var seat = {};
        seat.seatNumber = "E"+j
        seat.type = 'aisle'
        seat.cabin = 'economy'
        seat.cost  = flight["economyCost"]
        seat.reservationID = null
        seatmap.push(seat);
      }
      for(var j =1 ; j<11 ; j++){
        var seat = {};
        seat.seatNumber = "F"+j
        seat.type = 'aisle'
        seat.cabin = 'economy'
        seat.cost  = flight["economyCost"]
        seat.reservationID = null
        seatmap.push(seat);
      }
      flightToBeInserted.seatmap = seatmap;
      allFlights.push(flightToBeInserted);
    }
  }
  DB.collection('flights').insert(allFlights, function (err) {
    if (err) return cb(err);
    cb();
  });
}



function filteringClass(flights,i){
  if(i.class === "economy") {
    var filteredFlights = flights.filter(function(flight){
      return flight.availableEconomySeats >= i.seats;
    });
    for(var x=0;x<filteredFlights.length;x++)
    {
      filteredFlights[x].cost=filteredFlights[x].economyCost;
    }
  }
  else {
    var filteredFlights = flights.filter(function(flight){
      return flight.availableBusinessSeats >= i.seats;
    });
    for(var x=0;x<filteredFlights.length;x++)
    {
      filteredFlights[x].cost=filteredFlights[x].bussinessCost;
    }
  }

  for(var x=0;x<filteredFlights.length;x++)
  {
    filteredFlights[x].class=i.class;
  }

  for(var x=0;x<filteredFlights.length;x++)
  {
    filteredFlights[x].Airline="Alaska";
  }

  for(var x=0;x<filteredFlights.length;x++)
  {
    filteredFlights[x].currency="USD";
  }

  for(var x=0;x<filteredFlights.length;x++)
  {
    filteredFlights[x].flightId=filteredFlights[x]._id;
  }
  var filteredFlights2 = filteredFlights.filter(function(flight){
    return moment(flight.departureDateTime).format('YYYY-MM-DD')==moment(parseInt(i.departureDate)).format('YYYY-MM-DD');
  });
  for(var x=0;x<filteredFlights2.length;x++)
  {
    filteredFlights2[x].departureDateTime=moment(moment(filteredFlights2[x].departureDateTime).format('YYYY-MM-DD')).toDate().getTime();
  }

  var result={ "outgoingFlights" : filteredFlights2};
  return result;

}


function searchFlightsOneWay(i,cb) {
  DB.collection('flights').find({ destination:i.destination , origin:i.origin }).toArray(function(err,flights) {
    if (err) return cb(err);
    cb(null,filteringClass(flights,i));
  });
}



function seedAirports (cb) {
  DB.collection('airports').find().toArray(function (err, docs) {
    if (err) return cb(err);
    if (docs.length > 0)
    cb(null, false);
    else {
      DB.collection('airports').insertMany(airports, function (err) {
        if (err) return cb(err);
        cb(null, true);
      });
    }
  });
}


function searchRoundTripFlight (requiredFlight, cb){
  var chosenFlights = {}
  var filteredFlights = [];
  var classString = firstToUpperCase(requiredFlight.class);

  var query = {
    origin:requiredFlight.origin,
    destination:requiredFlight.destination,
  };
  if(requiredFlight.class==="economy"){
    query.availableEconomySeats={$gte: parseInt(requiredFlight.seats)};
  }
  else{
    query.availableBusinessSeats={$gte: parseInt(requiredFlight.seats)};
  }
  DB.collection('flights').find(query).toArray(function(err,flights) {
    if (err) return cb(err);
    for(var i =0;i<flights.length;i++){
      var temp = flights[i];
      var date = moment(parseInt(requiredFlight.departingDate)).format('YYYY-MM-DD');
      var currdate=moment(temp.departureDateTime).format('YYYY-MM-DD');
      if (date == currdate)
      filteredFlights.push(temp);
    }
    chosenFlights.outgoingFlights = filteredFlights;
    searchOtherWayAround(classString,chosenFlights, requiredFlight, function(){
      cb(chosenFlights);
    });
  });
}

function searchOtherWayAround(classString,chosenFlights, requiredFlight, cb) {
  var filteredFlights = []
  var query = {
    origin:requiredFlight.destination,
    destination:requiredFlight.origin
  };
  if(requiredFlight.class==="economy"){
    query.availableEconomySeats={$gte: parseInt(requiredFlight.seats)};
  }
  else{
    query.availableBusinessSeats={$gte: parseInt(requiredFlight.seats)};
  }
  DB.collection('flights').find(query).toArray(function(err,flights) {
    if (err) return err;

    for(var i =0;i<flights.length;i++){
      var temp = flights[i];
      var date = moment(parseInt(requiredFlight.returningDate)).format('YYYY-MM-DD');
      var currdate=moment(temp.arrivalDateTime).format('YYYY-MM-DD');
      if (date == currdate)
      filteredFlights.push(temp);
    }
    chosenFlights.returnFlights = filteredFlights;
    cb();
  });
}

function firstToLowerCase(string) {
  return string.substr(0, 1).toLowerCase() + string.substr(1);
}

function formatData(beforeFormattingData,reqClass,cb) {
  var formattedData = {};
  var formattedOutgoing = [];
  var formattedReturn = [];
  var classCost = reqClass+"Cost";
  var costOfClass = firstToLowerCase(classCost);

  for(var i =0; i < beforeFormattingData.outgoingFlights.length ; i++){
    var temp = {};
    temp.flightNumber=beforeFormattingData.outgoingFlights[i].flightNumber
    temp.aircraftType=beforeFormattingData.outgoingFlights[i].aircraftType
    temp.aircraftModel=beforeFormattingData.outgoingFlights[i].aircraftModel
    temp.departureDateTime=moment(moment(beforeFormattingData.outgoingFlights[i].departureDateTime).format('YYYY-MM-DD hh:mm A')).toDate().getTime();
    temp.arrivalDateTime=moment(moment(beforeFormattingData.outgoingFlights[i].arrivalDateTime).format('YYYY-MM-DD hh:mm A')).toDate().getTime();
    temp.origin=beforeFormattingData.outgoingFlights[i].origin
    temp.destination=beforeFormattingData.outgoingFlights[i].destination
    temp.cost=beforeFormattingData.outgoingFlights[i][costOfClass]
    temp.flightId=beforeFormattingData.outgoingFlights[i]._id
    temp.currency="USD";
    temp.class=firstToLowerCase(reqClass);
    temp.Airline="Alaska";
    formattedOutgoing.push(temp)
  }
  for(var i =0; i < beforeFormattingData.returnFlights.length ; i++){
    var temp = {};

    temp.flightNumber=beforeFormattingData.returnFlights[i].flightNumber
    temp.aircraftType=beforeFormattingData.returnFlights[i].aircraftType
    temp.aircraftModel=beforeFormattingData.returnFlights[i].aircraftModel
    temp.departureDateTime=moment(moment(beforeFormattingData.returnFlights[i].departureDateTime).format('YYYY-MM-DD hh:mm A')).toDate().getTime();
    temp.arrivalDateTime=moment(moment(beforeFormattingData.returnFlights[i].arrivalDateTime).format('YYYY-MM-DD hh:mm A')).toDate().getTime();
    temp.origin=beforeFormattingData.returnFlights[i].origin
    temp.destination=beforeFormattingData.returnFlights[i].destination
    temp.cost=beforeFormattingData.returnFlights[i][costOfClass]
    temp.flightId=beforeFormattingData.returnFlights[i]._id
    temp.currency="USD"
    temp.class=firstToLowerCase(reqClass)
    temp.Airline="Alaska"
    formattedReturn.push(temp)
  }
  formattedData.outgoingFlights = formattedOutgoing;
  formattedData.returnFlights = formattedReturn;
  cb(formattedData);
}

exports.seed= function (cb) {
  seedFlights(function(seeded){
    if(!seeded){
      seedAirports(function(err,seeded){
        if(!err) cb();
      });
    }
    else {
      console.log("flights already seeded");
    }
  });
}



//Find flight from DB when given flight number
exports.searchFlight = function(flightId,cb){
  var o_id = new ObjectId(flightId);
  DB.collection('flights').find({ _id : o_id},function(err,cursor){
    cursor.toArray(cb);
  });
};

exports.searchFlightNumber = function(flightNo,cb){

  DB.collection('flights').find({ flightNumber : flightNo},function(err,cursor){
    cursor.toArray(cb);
  });
};

//Find booking from DB when given booking reference number
exports.searchBooking = function(bookingRef,cb){
  DB.collection('bookings').find({"bookingRefNo":bookingRef},function(err,cursor){
    cursor.toArray(cb);
  });
};


exports.addBooking=function(i,cb){
  var resIDToBe;
  var refNum ;
  var numberOfSeats = i.passengerDetails.length;

  DB.collection('bookings').insert({

    "passengerDetails":i.passengerDetails ,
    "paymentToken":i.paymentToken,
    "bookingRefNo":null,
    "reservationID":null,
    "outgoingFlightId":i.outgoingFlightId,
    "returnFlightId":i.returnFlightId

  },function (err){
    if (err) return err;
    DB.collection('bookings').find().sort({_id:-1}).toArray(function (err,doc){
      if (err) return err;
      resIDToBe = ""+doc[0]._id;
      var bookRef = resIDToBe.substr(0, 7);
      DB.collection('bookings').update({ _id: doc[0]._id }, {$set: { reservationID: resIDToBe , bookingRefNo: bookRef }}, function (err) {
        if (err) return err;
        DB.collection('bookings').find({ _id: doc[0]._id }).toArray(function (err,returnedBooking){
          refNum = returnedBooking[0].bookingRefNo;
          updateSeatmapRec(0,i,resIDToBe,numberOfSeats,function(){
            cb(null,refNum);
          })
        });
      });
    });
  });
}

function updateSeatmapRec(x,i,resIDToBe,numberOfSeats,cb){
  if(x<numberOfSeats){
    updateSeatmap(i,resIDToBe,function(){
      if(x===numberOfSeats-1){
        cb();
      }
      else{
        updateSeatmapRec(x+1,i,resIDToBe,numberOfSeats,cb)
      }
    });
  }
}


function updateSeatmap (i,resIDToBe,cb){
  if(i.class === "economy"){

    var out_id = new ObjectId(i.outgoingFlightId);


    DB.collection('flights').update(
      {
        _id:out_id ,
        seatmap:{
          $elemMatch : {
            cabin : "economy",
            reservationID : null }
          }
        },
        {
          $set:{"seatmap.$.reservationID":resIDToBe},
          $inc:{availableSeats:-1,availableEconomySeats:-1}
        },function(err,results){
          if(err) return err;
          var ret_id = new ObjectId(i.returnFlightId);
          DB.collection('flights').update(
            {
              _id:ret_id ,
              seatmap:{
                $elemMatch : {
                  cabin : "economy",
                  reservationID : null }
                }
              },
              {
                $set:{"seatmap.$.reservationID":resIDToBe},
                $inc:{availableSeats:-1,availableEconomySeats:-1}
              },function(err,results){
                if(err) return err;
                cb();
              }
            );
          }
        );
      }
      else{
        if(i.class=="business"){
          var out_id = new ObjectId(i.outgoingFlightId);

          DB.collection('flights').update(
            {
              _id:out_id ,
              seatmap:{
                $elemMatch : {
                  cabin : "business",
                  reservationID : null }
                }
              },
              {
                $set:{"seatmap.$.reservationID":resIDToBe},
                $inc:{availableSeats:-1,availableBusinessSeats:-1}
              },function(err,results){
                if(err) return err;
                var ret_id = new ObjectId(i.returnFlightId);
                DB.collection('flights').update(
                  {
                    _id:ret_id ,
                    seatmap:{
                      $elemMatch : {
                        cabin : "business",
                        reservationID : null }
                      }
                    },
                    {
                      $set:{"seatmap.$.reservationID":resIDToBe},
                      $inc:{availableSeats:-1,availableBusinessSeats:-1}
                    },function(err,results){
                      if(err) return err;
                      cb();
                    }
                  );
                }
              );
            }
          }
        }

        exports.getAirports = function(cb){
          //fixed Db typo
          DB.collection('airports').find().toArray(function(err,airports){
            cb(err,airports);
          });
        };



        exports.clearDB=function (done) {
          DB.listCollections().toArray().then(function (collections) {
            collections.forEach(function (c) {
              DB.collection(c.name).removeMany();
            });
            done();
          }).catch(done);
        };

        function firstToUpperCase(string) {
          return string.substr(0, 1).toUpperCase() + string.substr(1);
        }
        exports.seedAirports = seedAirports;
        exports.seedFlights = seedFlights;
        exports.searchRoundTripFlight = searchRoundTripFlight;
        exports.formatData = formatData;
        exports.searchFlightsOneWay=searchFlightsOneWay;
        exports.filteringClass=filteringClass;