require 'chef/knife/cloud/list_resource_command'
require 'chef/knife/openstack_helpers'
require 'chef/knife/cloud/openstack_service_options'

class Chef
  class Knife
    class Cloud
      class OpenstackGroupList < ResourceListCommand
        include OpenstackHelpers
        include OpenstackServiceOptions

        banner "knife openstack group list (options)"

        def query_resource
          begin
            @service.connection.security_groups
          rescue Excon::Errors::BadRequest => e
            response = Chef::JSONCompat.from_json(e.response.body)
            ui.fatal("Unknown server error (#{response['badRequest']['code']}): #{response['badRequest']['message']}")
            raise e
          end
        end

        def list(security_groups)
          group_list = [
            ui.color('Name', :bold),
            ui.color('Protocol', :bold),
            ui.color('From', :bold),
            ui.color('To', :bold),
            ui.color('CIDR', :bold),
            ui.color('Description', :bold),
          ]
          security_groups.sort_by(&:name).each do |group|
            group.security_group_rules.each do |rule|
              unless rule.ip_protocol.nil?
                group_list << group.name
                group_list << rule.ip_protocol
                group_list << rule.from_port.to_s
                group_list << rule.to_port.to_s
                group_list << rule.ip_range['cidr']
                group_list << group.description
              end
            end
          end
          puts ui.list(group_list, :uneven_columns_across, 6)
        end
      end
    end
  end
end