bros

#!/usr/bin/env node

var argv = require('yargs').argv,
  log = require('cli-color'),
  db = require('./db/db'),
  os = require('os'),
  menu = require('./modules/menu'),
  secondaryMenu = require("./modules/secondaryMenu"),
  check = require('./modules/inputChecks'),
  simpleWeb = require('./modules/webserver/simple.js'),
  simpleFtp = require('./modules/webserver/simpleFtp.js'),
  interfaces = require('./modules/interfaces'),
  settings = require('./modules/settings'),
  child_process = require('child_process'),
  utilities = require('./modules/utilities'),
  log = require('./modules/log.js'),
  enigma = require('./modules/enigma'),
  initialEntry = true,
  initialData = "",
  sshd = require('./modules/sshd.js');

var firstArgument = argv._[0];
var secondArgument = argv._[1];
var thirdArgument = argv._[2];
var fourthArgument = argv._[3];

function getIt(){
  if (initialEntry){
    
    var chunk = process.stdin.read()
    // _readableState:
    initialEntry = false;
  }
  if (chunk){
    initialData = chunk;
  }
}

function getFirstArgValue(arg) {
  switch (arg) {
    case 1:
      return secondaryMenu.infoGathering;
      break;
    case 2:
      return secondaryMenu.linux;
      break;
    case 3:
      return secondaryMenu.windows;
      break;
    case 4:
      return secondaryMenu.webAttacks;
    case 5:
      return secondaryMenu.misc;
    case 6:
      return secondaryMenu.aux;
  }
}

function parseArgs() {
  var oldSecondary;

  if (firstArgument >= 1 && firstArgument <= 6) {
    secondaryMenu = getFirstArgValue(firstArgument);

    if (!secondArgument) {
      secondaryMenu();
    } else if (!thirdArgument) {
      secondaryMenu(secondArgument);
    } else {
      secondaryMenu(secondArgument, thirdArgument);
    }

  } else if (firstArgument > 10) {

    // If input is entered as "bros 111" split up the input and handle them individually
    var splitInput = firstArgument.toString(10).split("").map(function(t) {
      return parseInt(t)
    })
    var inputLength = splitInput.length;
    secondaryMenu = getFirstArgValue(splitInput[0]);
    
    switch (inputLength) {
      case 2:
        setTimeout(function(){
          secondaryMenu(splitInput[1]);
        },15)
        break;
      case 3:
        setTimeout(function(){
          secondaryMenu(splitInput[1], splitInput[2])
        },15)
        break;
      case 4:
        var a = new String(splitInput[2]);
        var b = new String(splitInput[3]);
        var c = a+b
        secondaryMenu(splitInput[1], parseInt(c))
        break;
    }
  } else if (typeof(firstArgument) === "string") {
    firstArgument = firstArgument.toUpperCase();
    try {
      oldSecondary = secondArgument;
      secondArgument = secondArgument.toUpperCase();
    } catch(err){

    }
    // If input is bros http, start a webserver using expressjs, pass all args
    // to the simple web server module
    if (firstArgument === "HTTP") {
      simpleWeb.http(argv);
    } else if (firstArgument === "HTTPS") {
      simpleWeb.https(argv);
    } else if (firstArgument === "FTP"){
      simpleFtp.ftp(argv);
    } else if (firstArgument === "FTPS"){
      console.log("Sorry, FTPs support is planned but currently not available")
      //simpleFtp.ftps(argv);
    } else if (firstArgument === "SSHD"){
      sshd.logger(argv);
    } 
      else if (firstArgument === "UPDATE"){
      utilities.update();
    } else if (firstArgument === "ENCODE" || firstArgument === "DECODE" || firstArgument === "ENC" || firstArgument === "DEC"){
      process.stdin.setEncoding('utf8');
      process.stdin.once('readable', function () {
        getIt();
      });
      utilities.delay(10, function(){
        if (initialData){
              if(argv._.length === 2){
                enigma(argv._,initialData);  
              }  
          } else {
                  if (secondArgument && !thirdArgument){

                    utilities.encoder(oldSecondary);    
                  } else if (thirdArgument){
                    enigma(argv._);
                  } else {
                    utilities.encoder();
                  }
          }    
      })


    } else if (firstArgument+" "+secondArgument === "SET LHOST" && !thirdArgument){
      interfaces.setlhost();

    } else if (firstArgument != "CONFIG") {
      if (secondArgument && thirdArgument) {
        check.allInputChecks(firstArgument + " " + secondArgument + " " + thirdArgument, console.log, menu.mainMenu)
      } else if (firstArgument === "HELP") {
        check.allInputChecks(firstArgument, menu.mainMenu, menu.mainMenu);
      } else if (firstArgument === "CLEAN") {

        var currentOS = os.type();
        var cmd = "rm";
        var dir = settings.dbPath;
        if (currentOS.match("Windows")){
            cmd = "del";
        } else {

        }

        child_process.execFile(cmd, [dir], function(error, stdout, stderr){
        });
        console.log("The Brosec database located at %s has been wiped.", settings.dbPath)
      }
      else {
        check.allInputChecks(firstArgument, console.log, console.log)
      }
    } else {
      console.log(log.yellow("\n[*] Sorry, that isn't valid input from the command line"))
      menu.mainMenu();
    }

  } else {
    menu.mainMenu();
  }
}

if (!firstArgument) {

  menu.mainMenu();
} else {
  parseArgs();
}