require 'aws-sdk'

module Awspec::Type
  class Base
    attr_reader :id, :ec2_client

    def initialize(id = nil)
      @tmp = id
      @ec2_client = Aws::EC2::Client.new
    end

    def find_vpc(id)
      res = @ec2_client.describe_vpcs({
                                        filters: [{ name: 'vpc-id', values: [id] }]
                                      })
      return res[:vpcs][0] if res[:vpcs].count == 1
      res = @ec2_client.describe_vpcs({
                                        filters: [{ name: 'tag:Name', values: [id] }]
                                      })
      return res[:vpcs][0] if res[:vpcs].count == 1
    end

    def find_route_table(id)
      res = @ec2_client.describe_route_tables({
                                                filters: [{ name: 'route-table-id', values: [id] }]
                                              })
      return res[:route_tables][0] if res[:route_tables].count == 1
      res = @ec2_client.describe_route_tables({
                                                filters: [{ name: 'tag:Name', values: [id] }]
                                              })
      return res[:route_tables][0] if res[:route_tables].count == 1
    end

    def find_network_acl(id)
      res = @ec2_client.describe_network_acls({
                                                filters: [{ name: 'network-acl-id', values: [id] }]
                                              })
      return res[:network_acls][0] if res[:network_acls].count == 1
      res = @ec2_client.describe_network_acls({
                                                filters: [{ name: 'tag:Name', values: [id] }]
                                              })
      return res[:network_acls][0] if res[:network_acls].count == 1
    end

    def find_security_group(id)
      res = @ec2_client.describe_security_groups({
                                                   filters: [{ name: 'group-id', values: [id] }]
                                                 })

      return res[:security_groups][0] if res[:security_groups].count == 1
      res = @ec2_client.describe_security_groups({
                                                   filters: [{ name: 'group-name', values: [id] }]
                                                 })

      return res[:security_groups][0] if res[:security_groups].count == 1
      res = @ec2_client.describe_security_groups({
                                                   filters: [{ name: 'tag:Name', values: [id] }]
                                                 })

      return res[:security_groups][0] if res[:security_groups].count == 1
    end
  end
end