country-data/data/country_csv_to_json.js

Data about countries - like their ISO codes and currencies

'use strict';

// Take the csv and convert to json and tidy it up so that it is consistent.

var path = require('path');
var _ = require('underscore');
var csv = require('csv');
var canonicalJSON = require('canonical-json');
var fs = require('fs');

var output = [];
var countriesFilename = 'countries.csv';
var deletedCountriesFilename = 'deleted_countries.csv';


function readFile(filename) {
  return new Promise(function (resolve) {
    var csvFile = path.join( __dirname, filename );
    var parser = csv.parse({"columns": true});

    parser.on('readable', function () {
      var record = null;
      while(record = parser.read()){
        output.push(record);
      }
    });

    parser.on('finish', function(){
      resolve(output);
    });

    fs.createReadStream(csvFile).pipe(parser);
  });
}

var countriesPromise = readFile(countriesFilename);
var deletedCountriesPromise = readFile(deletedCountriesFilename);

Promise.all([countriesPromise, deletedCountriesPromise])
  .then(function (resulsts){
    output = _.sortBy(output, function (i) { return i.alpha2;} );

    // strip out fields that are not ready yet
    _.each(output, function (country) {
      delete country.ccTLD;
    });

    // change the appropriate fields to be an array
    _.each(['currencies', 'countryCallingCodes', 'languages'], function(key) {
      _.each(output, function (country) {
        country[key] = country[key] ? country[key].split(',') : [];
      });
    });

    // print out results to stdout
    console.log( canonicalJSON( output, null, 2 ));
  });