require 'vagrant-vultr/helpers/client'

module VagrantPlugins
  module Vultr
    module Action
      class SetupSSHKey
        include Helpers::Client

        def initialize(app, env)
          @app = app
          @machine = env[:machine]
          @name = @machine.provider_config.ssh_key
          @client = client
          @logger = Log4r::Logger.new('vagrant::vultr::setup_ssh_key')
        end

        def call(env)
          ssh_key_id = @client.ssh_key_id(@name)
          unless ssh_key_id
            @logger.info 'SSH key does not exist. Creating new one...'
            key_path = File.expand_path("#{@machine.config.ssh.private_key_path.first}.pub")
            ssh_key_id = @client.create_ssh_key(@name, File.read(key_path))
          end
          @logger.info "Using SSH key: #{ssh_key_id}."

          @app.call(env)
        end
      end
    end
  end
end