pyama86/pec

module Pec::Handler
  class UserData::Nic
    extend Pec::Core
    autoload :Base,   "pec/handler/user_data/nic/base"
    autoload :Rhel,   "pec/handler/user_data/nic/rhel"
    autoload :Ubuntu, "pec/handler/user_data/nic/ubuntu"
    self.kind = 'networks'

    class << self
      def post_build(config, attribute)
        nic = if config.os_type
          os = Pec::Handler::UserData::Nic.constants.find do |c|
            Object.const_get("Pec::Handler::UserData::Nic::#{c}").os_type.include?(config.os_type)
          end
          Object.const_get("Pec::Handler::UserData::Nic::#{os}")
        else
          Pec::Handler::UserData::Nic::Rhel
        end

        attribute.deep_merge(
          {
            user_data: {
              "write_files" => nic.gen_user_data(config.networks, ports(attribute))
            }
          }
        )
      end

      def ports(attribute)
        port_ids(attribute).map do |id|
          Yao::Port.get(id)
        end
      end

      def port_ids(attribute)
        attribute[:networks].map {|n|n[:port]}
      end

    end
  end
end