tools.js

'use strict';

const domain = 'https://dgd-oipa.blue4you.be';
let request = require('request-promise');
let Promise = require('bluebird');
let cacheProvider = require('./cache-provider');
let Url = require('url');

module.exports = {
    getOptions: function (url) {
        return {
            method: 'GET',
            url: url,
            json: true,
            headers: {
                'Connection': 'keep-alive',
                'Content-Type': 'application/json',
                'Content-Language': 'en',
                'Access-Control-Allow-Origin': '*',
                'Access-Control-Allow-Methods': 'GET, POST, OPTIONS',
                'Access-Control-Allow-Headers': 'Origin, X-Requested-With, Content-Type, Content-Language, Accept',
                'Accept-Encoding': '',
                'Accept-Language': 'en-US,en;q=0.8'
            }
        }
    },
    query: function (endpoint, type, output) {
        return request(module.exports.getOptions(endpoint)).then(function (data) {
            switch (type) {
                case "pager":
                    if (!output) output = data.results;
                    else output = output.concat(data.results);
                    break;

                case "sectors":
                    return Promise.map(data.results, function (result) {
                        let url_parts = Url.parse(result.url);
                        let sector_url = domain + url_parts.pathname;
                        return Promise.resolve(module.exports.query(sector_url).catch(function (err) {
                            return module.exports.errorHandler(err, sector_url);
                        }));
                    }, {concurrency: 5}).then(function (data) {
                        return data;
                    });

                case "documents":
                    if (!output) output = [];
                    data.results.map(function (docs) {
                        docs.document_links.map(function (doc) {
                            output.push(doc);
                        });
                    });
                    break;

                case "countries":
                    if (!output) output = {};
                    let cc = {},
                        url = domain + "/api/transactions/aggregations/?format=json" +
                            "&group_by=recipient_country" +
                            "&aggregations=activity_count,disbursement_expenditure" +
                            "&order_by=recipient_country" +
                            "&page_size=400" +
                            "&transaction_date_year=" + (new Date()).getFullYear();
                    if (!output["results"] && !output["country_data"]) {
                        output["results"] = [];
                        output["country_data"] = {};
                    }
                    data.results.map(function (result) {
                        output["results"].push(result);
                        // Get the countries at activity level and build a array.
                        // This is used to determine the polygon for valid locations.
                        if (result.recipient_countries.length > 0) {
                            result.recipient_countries.map(function (country) {
                                if (output["country_data"].length <= 0 || output["country_data"][country.country.code] === undefined) {
                                    let budget_url = url + "&recipient_country=" + country.country.code;
                                    output["country_data"][country.country.code] = {
                                        "country": country.country.name,
                                        "id": country.country.code,
                                        "budget": new Promise(function (resolve, reject) {
                                            return module.exports.query(budget_url).then(function (budget_data) {
                                                if (budget_data.results === undefined || budget_data.results.length <= 0)
                                                    return reject("No data for - " + budget_url);

                                                return resolve({
                                                    "budget": budget_data.results[0].disbursement_expenditure,
                                                    "activity_count": budget_data.results[0].activity_count
                                                });
                                            });
                                        }),
                                        "flag": "https://cdnjs.cloudflare.com/ajax/libs/flag-icon-css/3.1.0/flags/1x1/" + country.country.code.toLowerCase() + ".svg"
                                    };
                                }
                            });
                        }
                    });
                    break;

                default:
                    return data;
            }

            if (data.next !== null) {
                return module.exports.query(data.next, type, output);
            }

            return output;
        }).catch(function (err) {
            return module.exports.errorHandler(err, endpoint);
        });
    },
    setCache: function (key, obj) {
        const CACHE_DURATION = 86400;
        cacheProvider.instance().set(key, obj, CACHE_DURATION, function (err, success) {
            if (!err && success) {
                console.log("Cache entry on " + key + " has been set.");
            }
        });
    },
    errorHandler: function (err, url) {
        switch (err.message) {
            case '400 - {"detail":"Bad Request Error."}':
                console.log('HTTP to HTTPS not supported: ' + url);
                throw err;

            case '404 - {"detail":"Not found."}':
                console.log('Detail not found on request: ' + url);
                return;

            default:
                console.log('Error on request: ' + url);
                throw err;
        }
    }
};