require 'optparse'
require './asset_generator'
require './cve_info'
require './kdi_jsonify'
require 'pry'

asset_generator = KennaKdi::AssetGenerator.new('./data/cve')

ARGV << '-h' if ARGV.empty?

params = {}

OptionParser.new do |opts|
  opts.banner = "Usage: kenna_kdi_importer.rb [OPTIONS]"

  opts.on('-a NUM', '--assets', Integer, "Number of assets to create") { |a| params[:assets] = a }
  opts.on('-v NUM', '--max_vulns', Integer, "Max number of vulns that may be created per asset") {|v| params[:max_vulns] = v }

  opts.on_tail("-h", "--help", "Show this message") do
    puts opts
    exit
  end
end.parse!(into: params)

raise OptionParser::MissingArgument if params[:assets].nil?
raise OptionParser::MissingArgument if params[:max_vulns].nil?

num_of_assets = params[:assets]
max_vulns = params[:max_vulns]

puts "Now creating #{num_of_assets} assets. Each asset with a random set of vulnerabilities between 1 and #{max_vulns}"
puts "\n"

while true do
  hash = asset_generator.create_assets(num_of_assets, max_vulns)
  if hash
    break
  end
end

json = hash.to_json
puts JSON.pretty_generate(JSON.parse(json))