const stripAnsi = require('strip-ansi')

function normalizeCols (rows, max) {
  function padEnd (string, maxLength, fillString) {
    while ((stripAnsi(string) + '').length < maxLength) {
      string = string + fillString
    }

    return string
  }

  var widths = []
  max = max || rows[0].length

  for (var i = 0; i < rows.length; i++) {
    for (var j = 0; j < rows[i].length && j < max; j++) {
      if (!widths[j] || widths[j] < (stripAnsi(rows[i][j]) + '').length) {
        widths[j] = (stripAnsi(rows[i][j] + '') + '').length
      }
    }
  }

  for (var i = 0; i < rows.length; i++) {
    for (var j = 0; j < rows[i].length && j < max; j++) {
      if (rows[i][j] == '-') {
        rows[i][j] = padEnd(rows[i][j], widths[j], '-')
      } else {
        rows[i][j] = padEnd(rows[i][j], widths[j], ' ')
      }
    }
  }

  return rows
}

function keyValueArray(columns, keys, obj) {
  return keys.map(el => columns[el](obj))
}

module.exports.format = function (data, options) {
    // Keys for each column
    var keys = []

    // Separators for each column
    var separators = []
    
    // Add to collection of keys
    for(var key in options.columns) {
      keys.push(key)
      separators.push('-')
    }

    // Create the rows
    var items = [
        keys,
        separators,
        ...data.map(data => keyValueArray(options.columns, keys, data))
    ]

    // Normalize column widths.
    items = normalizeCols(items).map(el => {
      return el.join(' | ').replace(/\n/g, '')
    }).join('\n')

    

    if(options.program) {

      // Disable color output
      if (!options.program.color) { items = stripAnsi(items) }

      // If reporting output is defined, ignore console log here.
      if (options.program.reportOutput === undefined) {
        console.log(items)
      }
    } else {
      console.log(items)
    }
    
    return items
}

const fs = require('fs-extra')
const path = require('path')

module.exports.finalReport = async function(reports, program) {
    if (program.reportOutput === undefined) {
      return
    }
    
    // Ensure the output directory exists.
    fs.ensureDirSync(program.reportOutput)

    // Write each report to the disk
    for(var report of reports) {
      var outPath = path.join(program.reportOutput, report.name + '.txt')
      console.log('saving', outPath)
      fs.writeFileSync(outPath, report.contents, 'utf8')
    }
  }