const db = require("../models/index");

const { currentRERequest } = require("../helpers/url");
const { realEstateTypes, getRealEstateTypeEnum } = require("../helpers/enums");

const getRealEstateTypes = (req, res) => {
  const title = "Koju nekretninu tražite?";
  res.render("realEstateType", { realEstateTypes, title });
};

const postRealEstateTypes = async (req, res) => {
  const request = await currentRERequest(req);

  const nextStepPage = req.query.nextStep || "grad";

  if (request && request.uniqueId) {
    const nextStepUrl = `/${nextStepPage}/${request.uniqueId}`;
    request.realEstateType = req.body.realestatetype;
    if (!getRealEstateTypeEnum(request.realEstateType).hasGardenSize) {
      request.gardenSize = null;
    }
    await request.save();

    res.redirect(nextStepUrl);
  } else {
    db.RealEstateRequest.create({
      realEstateType: req.body.realestatetype
    })
      .then(result => {
        const nextStepUrl = `/${nextStepPage}/${result.uniqueId}`;
        res.redirect(nextStepUrl);
      })
      .catch(e => {
        res.send(e);
      });
  }
};

module.exports = {
  getRealEstateTypes,
  postRealEstateTypes
};