mrajagopal/nodejs-ssllabs-api

#!/usr/bin/env node


var sslLabsApi = require('../ssllabs-api');
var SegfaultHandler = require('segfault-handler');
var fs = require("fs");
var program = require('commander');
var validator = require('validator');
var progressDotTimer = undefined;

// With no argument, SegfaultHandler will generate a generic log file name
SegfaultHandler.registerHandler("crash.log"); 

function getVersion(){
  var sslApi = sslLabsApi('', false);
  return sslApi.version();
};

function printProgressDot(){
  process.stdout.write(".");
}

program.version(getVersion());
program.description('Command line tool for SSL Labs API using Node.js');
program.option('--verbosity', 'Enable verbose logging');
program.option('--host <hostname>', 'Hostname to analyze');
program.option('--grade', 'Output only the hostname: grade');
program.option('--info', 'API command info');
program.option('--codes', 'Get API status codes');
program.option('-u, --usecache', 'Accept cached results (if available), else force live scan');
program.parse(process.argv);

if(!process.argv.slice(2).length){
  program.help();
}

if(program.info){
  var sslApi = sslLabsApi('', program.verbosity);
  sslApi.info(); 
  process.stdout.write("Working.");
}
else if(program.codes){
  var sslApi = sslLabsApi('', program.verbosity);
  sslApi.getStatusCodes(); 
}
else if(program.host && !validator.isFQDN(program.host)){
  console.log('  Error: invalid host string');
  program.help();
  process.exit(0);
}

// Assume some defaults if only hostname is provided
if(program.host){
  var sslApi = sslLabsApi(program.host, program.verbosity);
  if(program.usecache){
    progressDotTimer = setInterval(printProgressDot, 5000);
    sslApi.analyzeHostCached();
    process.stdout.write("Working.");
  }
  else{
    progressDotTimer = setInterval(printProgressDot, 5000);
    sslApi.analyzeHostNew();
    process.stdout.write("Working.");
  }
}

function printGrade(data){
  clearInterval(progressDotTimer);
  data = JSON.parse(data);
  console.log('\r\nhost: ' + data.host + 
              '\r\nFound ' + data.endpoints.length + ' endpoints' + 
              '\r\nGrades for each:');
  for(var i = 0; i < data.endpoints.length; i++){
    console.log(data.endpoints[i].ipAddress + ': ' + data.endpoints[i].grade);
  }
}

function printAllAnalyzeData(data){
  clearInterval(progressDotTimer);
  console.log(data);
}

//Register for events
if(program.grade)
  sslApi.on('analyzeData', printGrade);  
else
  sslApi.on('analyzeData', printAllAnalyzeData);

sslApi.on('infoResponse', function(data){
  clearInterval(progressDotTimer);
  console.log('\r\nReceived info data: "' + JSON.stringify(data, null, 4) + '"');
});

sslApi.on('endpointData', function(data){
  clearInterval(progressDotTimer);
  if (program.grade){
    if(data.grade && (data.grade !='')){
      console.log('\r\n' + data.ipAddress + ': ' + data.grade + '\r\n');
    }
  }
  else{
    console.log('\r\nReceived endpoints data: "' + JSON.stringify(data) + '"');
  }
});

sslApi.on('statusCodesData', function(data){
  clearInterval(progressDotTimer);
  console.log('\r\nReceived status-codes data: "' + JSON.stringify(data) + '"');
});

sslApi.on('error', function(data){
  clearInterval(progressDotTimer);
  console.log('\r\nReceived error event: ', JSON.stringify(data));
});