var crypto = require('crypto'); var Step = require('step'); var fs = require('fs'); var spawn = require('child_process').spawn; var ogr = require('./ogr'); function shp() { } shp.prototype = new ogr('shp'); var p = shp.prototype; p._contentType = "application/zip; charset=utf-8"; p._fileExtension = "zip"; p._needSRS = true; p.generate = function(options, callback) { var o = options; this.toSHP(o.database, o.user_id, o.gn, o.sql, o.skipfields, o.filename, callback); }; p.toSHP = function (dbname, user_id, gcol, sql, skipfields, filename, callback) { var fmtObj = this; var zip = 'zip'; // FIXME: make configurable var tmpdir = global.settings.tmpDir || '/tmp'; var reqKey = [ 'shp', dbname, user_id, gcol, this.generateMD5(sql) ].concat(skipfields).join(':'); var outdirpath = tmpdir + '/sqlapi-' + process.pid + '-' + reqKey; var zipfile = outdirpath + '.zip'; var shapefile = outdirpath + '/' + filename + '.shp'; // TODO: following tests: // - fetch query with no "the_geom" column Step ( function createOutDir() { fs.mkdir(outdirpath, 0777, this); }, function spawnDumper(err) { if ( err ) throw err; fmtObj.toOGR(dbname, user_id, gcol, sql, skipfields, 'ESRI Shapefile', shapefile, this); }, function doZip(err) { if ( err ) throw err; var next = this; var child = spawn(zip, ['-qrj', zipfile, outdirpath ]); child.on('exit', function(code) { //console.log("Zip complete, zip return code was " + code); if (code) { next(new Error("Zip command return code " + code)); //res.statusCode = 500; } next(null); }); }, function cleanupDir(topError) { var next = this; //console.log("Cleaning up " + outdirpath); // Unlink the dir content var unlinkall = function(dir, files, finish) { var f = files.shift(); if ( ! f ) { finish(null); return; } var fn = dir + '/' + f; fs.unlink(fn, function(err) { if ( err ) { console.log("Unlinking " + fn + ": " + err); finish(err); } else unlinkall(dir, files, finish) }); } fs.readdir(outdirpath, function(err, files) { if ( err ) { if ( err.code != 'ENOENT' ) { next(new Error([topError, err].join('\n'))); } else { next(topError); } } else { unlinkall(outdirpath, files, function(err) { fs.rmdir(outdirpath, function(err) { if ( err ) console.log("Removing dir " + path + ": " + err); next(topError, zipfile); }); }); } }); }, function finalStep(err, zipfile) { callback(err, zipfile); } ); }; module.exports = shp;