const
  UserTypes = require('../helpers/constants').UserTypes,
  User = require('./user'),
  Customer = require('./customer'),
  Client = require('./client'),
  Pilot = require('./pilot'),
  Vehicle = require('./vehicle');

function create(userType) {
  switch (userType) {
    case UserTypes.APP:
      return Customer;
    case UserTypes.CLIENT:
      return Client;
    case UserTypes.PILOT:
      return Pilot;
    case UserTypes.DEVICE:
      return Vehicle;

    case UserTypes.ADMIN:
    case UserTypes.OFFICER:
    case UserTypes.INSPECTOR:
    default:
      return User;
  }
}

module.exports = {
  create
}