CartoDB-SQL-API/lib/models/formats/pg/topojson.js

'use strict';

var Pg = require('./../pg');
var _ = require('underscore');
var geojson = require('./geojson');
var TopoJSON = require('topojson');

function TopoJsonFormat () {
    this.features = [];
}

TopoJsonFormat.prototype = new Pg('topojson');

TopoJsonFormat.prototype.getQuery = function (sql, options) {
    return geojson.prototype.getQuery(sql, options) + ' where ' + options.gn + ' is not null';
};

TopoJsonFormat.prototype.handleQueryRow = function (row) {
    var _geojson = {
        type: 'Feature'
    };
    _geojson.geometry = JSON.parse(row[this.opts.gn]);
    delete row[this.opts.gn];
    delete row.the_geom_webmercator;
    _geojson.properties = row;
    this.features.push(_geojson);
};

TopoJsonFormat.prototype.handleQueryEnd = function () {
    if (this.error) {
        this.callback(this.error);
        return;
    }

    if (this.opts.profiler) {
        this.opts.profiler.done('gotRows');
    }

    var topology = TopoJSON.topology(this.features, {
        quantization: 1e4,
        'force-clockwise': true,
        'property-filter': function (d) {
            return d;
        }
    });

    this.features = [];

    var stream = this.opts.sink;
    var jsonpCallback = this.opts.callback;
    var bufferedRows = this.opts.bufferedRows;
    var buffer = '';

    var immediately = global.setImmediate || process.nextTick;

    /* eslint-disable */
    function streamObjectSubtree (obj, key, done) {
        buffer += '"' + key + '":';

        var isObject = _.isObject(obj[key]);
        var isArray = _.isArray(obj[key]);
        var isIterable = isArray || isObject;

        if (isIterable) {
            buffer += isArray ? '[' : '{';
            var subtreeKeys = Object.keys(obj[key]);
            var pos = 0;
            function streamNext () {
                immediately(function () {
                    var subtreeKey = subtreeKeys.shift();
                    if (!isArray) {
                        buffer += '"' + subtreeKey + '":';
                    }
                    buffer += JSON.stringify(obj[key][subtreeKey]);

                    if (pos++ % (bufferedRows || 1000)) {
                        stream.write(buffer);
                        buffer = '';
                    }

                    if (subtreeKeys.length > 0) {
                        delete obj[key][subtreeKey];
                        buffer += ',';
                        streamNext();
                    } else {
                        buffer += isArray ? ']' : '}';
                        stream.write(buffer);
                        buffer = '';
                        done();
                    }
                });
            }
            streamNext();
        } else {
            buffer += JSON.stringify(obj[key]);
            done();
        }
    }
    /* eslint-enable */

    if (jsonpCallback) {
        buffer += jsonpCallback + '(';
    }
    buffer += '{';
    var keys = Object.keys(topology);
    function sendResponse () {
        immediately(function () {
            var key = keys.shift();
            function done () {
                if (keys.length > 0) {
                    delete topology[key];
                    buffer += ',';
                    sendResponse();
                } else {
                    buffer += '}';
                    if (jsonpCallback) {
                        buffer += ')';
                    }
                    stream.write(buffer);
                    stream.end();
                    topology = null;
                }
            }
            streamObjectSubtree(topology, key, done);
        });
    }
    sendResponse();

    this.callback();
};

TopoJsonFormat.prototype.cancel = function () {
    if (this.queryCanceller) {
        this.queryCanceller.call();
    }
};

module.exports = TopoJsonFormat;