Sha256: 745da41e33bb168f89f31851cdeb001b445560ed3ace2719ba415da0dc65a95a
Contents?: true
Size: 1.46 KB
Versions: 81
Compression:
Stored size: 1.46 KB
Contents
#!/usr/bin/env ruby
require 'aws_security_viz'
require 'optimist'
opts = Optimist::options do
opt :access_key, 'AWS access key', :default => ENV['AWS_ACCESS_KEY'] || ENV['AWS_ACCESS_KEY_ID'], :type => :string
opt :secret_key, 'AWS secret key', :default => ENV['AWS_SECRET_KEY'] || ENV['AWS_SECRET_ACCESS_KEY'], :type => :string
opt :session_token, 'AWS session token', :default => ENV['AWS_SESSION_TOKEN'] || nil, :type => :string
opt :region, 'AWS region to query', :default => 'us-east-1', :type => :string
opt :vpc_id, 'AWS VPC id to show', :type => :string
opt :source_file, 'JSON source file containing security groups', :type => :string
opt :filename, 'Output file name', :type => :string, :default => 'aws-security-viz.png'
opt :config, 'Config file (opts.yml)', :type => :string, :default => 'opts.yml'
opt :color, 'Colored node edges', :default => false
opt :renderer, "Renderer (#{Renderer.all.join('|')})", :default => 'graphviz'
opt :source_filter, 'Source filter', :default => nil, :type => :string
opt :target_filter, 'Target filter', :default => nil, :type => :string
end
cmd = ARGV.shift
if cmd=="setup"
AwsConfig.write(opts[:config])
puts "#{opts[:config]} created in current directory."
exit
end
config = AwsConfig.load(opts[:config]).merge(obfuscate: ENV['OBFUSCATE'], debug: ENV['DEBUG'])
begin
VisualizeAws.new(config, opts).unleash(opts[:filename])
rescue Exception => e
puts "[ERROR] #{e.message}"
raise e if config.debug?
exit 1
end
Version data entries
81 entries across 81 versions & 1 rubygems