"use strict";

import * as _ from "lodash";

// @todo should be moved as middleware
export function customResponses(req, res, next){

    res.badRequest = function(data){
        if(_.isString(data)) {
            data = {message: data};
        }
        data.data = data.data || {};
        if (data.errors) {
            data.data.errors = data.errors;
        }
        var errResponse = {
            status: data.status || "error",
            code: data.code || "badRequest",
            message: data.message || "",
            data: data.data || {}
        };

        return res.status(400).send(errResponse);
    };

    res.created = function(data){
        return res.status(201).send(data);
    };

    res.ok = function(data){
        return res.status(200).send(data);
    };

    res.notFound = function(data){
        var errResponse: any = {};
        errResponse.status = "error";
        errResponse.code = "notFound";
        errResponse.message = data;
        errResponse.data = {};
        return res.status(404).send(errResponse);
    };

    res.updated = function(data){
        return res.status(200).send(data);
    };

    res.serverError = function(err: any){
        let errResponse: any = {};
        errResponse.status = "error";
        errResponse.code = "serverError";
        errResponse.message = "An internal error occured";
        errResponse.data = {};

        // Handle Error object
        if(err instanceof Error) {
            let error: any = err; // workaround for 'code' property
            errResponse = _.merge(errResponse, {message: error.message, data: {stack: error.stack, code: error.code}});
        }

        if(_.isString(err)) {
            errResponse.message = err;
        }

        return res.status(500).send(errResponse)
    };

    return next();
}