Sha256: 8d994391dab6f5ab96460a740e1e5371ef52ac763aec2b8dc1eacbbbfedb5ffb

Contents?: true

Size: 1.5 KB

Versions: 6

Compression:

Stored size: 1.5 KB

Contents

require_relative 'ec2/security_groups'
require_relative 'provider/json'
require_relative 'provider/ec2'
require_relative 'renderer/graphviz'
require_relative 'renderer/json'
require_relative 'graph'
require_relative 'exclusions'
require_relative 'debug_graph'
require_relative 'color_picker'
require_relative 'aws_config'

class VisualizeAws
  def initialize(config, options={})
    @options = options
    @config = config
    provider = options[:source_file].nil? ? Ec2Provider.new(options) : JsonProvider.new(options)
    @security_groups = SecurityGroups.new(provider, config)
  end

  def unleash(output_file)
    g = build
    if output_file.end_with?('json')
      g.output(Renderer::Json.new(output_file, @config))
      FileUtils.copy(File.expand_path('../export/html/view.html', __FILE__),
                     File.expand_path('../view.html', output_file))
    else
      g.output(Renderer::GraphViz.new(output_file, @config))
    end
  end

  def build
    g = @config.obfuscate? ? DebugGraph.new(@config) : Graph.new(@config)
    @security_groups.each_with_index { |group, index|
      picker = ColorPicker.new(@options[:color])
      g.add_node(group.name)
      group.traffic.each { |traffic|
        if traffic.ingress
          g.add_edge(traffic.from, traffic.to, :color => picker.color(index, traffic.ingress), :label => traffic.port_range)
        else
          g.add_edge(traffic.to, traffic.from, :color => picker.color(index, traffic.ingress), :label => traffic.port_range)
        end
      }
    }
    g
  end

end

Version data entries

6 entries across 6 versions & 1 rubygems

Version Path
aws_security_viz-0.1.4.pre.alpha.pre.81 lib/aws_security_viz.rb
aws_security_viz-0.1.4.pre.alpha.pre.77 lib/aws_security_viz.rb
aws_security_viz-0.1.4.pre.alpha.pre.75 lib/aws_security_viz.rb
aws_security_viz-0.1.4.pre.alpha.pre.73 lib/aws_security_viz.rb
aws_security_viz-0.1.3 lib/aws_security_viz.rb
aws_security_viz-0.1.3.pre.alpha.pre.71 lib/aws_security_viz.rb