require 'sshkey'

module AcquiaToolbelt
  class CLI
    class SSH < AcquiaToolbelt::Thor
      # Public: List all SSH users.
      #
      # Returns a list of users and their SSH keys truncated.
      desc 'list', 'List all SSH users.'
      def list
        if options[:subscription]
          subscription = options[:subscription]
        else
          subscription = AcquiaToolbelt::CLI::API.default_subscription
        end

        users = AcquiaToolbelt::CLI::API.request "sites/#{subscription}/sshkeys"
        users.each do |user|
          say
          say "> ID: #{user['id']}"
          say "> Name: #{user['nickname']}"
          say "> Fingerprint: #{SSHKey.fingerprint user['ssh_pub_key']}"
        end
      end

      # Public: Delete a SSH key.
      #
      # Returns a status message.
      desc 'delete', 'Delete a SSH key.'
      method_option :id, :type => :string, :aliases => %w(-i),
        :required => true, :desc => 'User ID to delete from SSH users.'
      def delete
        if options[:subscription]
          subscription = options[:subscription]
        else
          subscription = AcquiaToolbelt::CLI::API.default_subscription
        end

        id = options[:id]

        delete_ssh_request = AcquiaToolbelt::CLI::API.request "sites/#{subscription}/sshkeys/#{id}", 'DELETE'

        if delete_ssh_request['id']
          ui.success "SSH key #{id} has been successfully deleted."
        else
          ui.fail AcquiaToolbelt::CLI::API.display_error(delete_ssh_key)
        end
      end

      # Public: Add a SSH key to a subscription.
      #
      # Returns a status message.
      desc 'add', 'Add a SSH key.'
      method_option :nickname, :type => :string, :aliases => %w(-n),
        :required => true, :desc => 'Nickname to assign to the key.'
      method_option :key, :type => :string, :aliases => %w(-k),
        :required => true, :desc => 'Public SSH key to add.'
      def add
        if options[:subscription]
          subscription = options[:subscription]
        else
          subscription = AcquiaToolbelt::CLI::API.default_subscription
        end

        key      = options[:key]
        nickname = options[:nickname]
        data     = {
          :key => 'nickname',
          :value => "#{nickname}",
          :ssh_pub_key => "#{key}"
        }

        add_ssh_key = AcquiaToolbelt::CLI::API.request "sites/#{subscription}/sshkeys", 'QUERY-STRING-POST', data
        if add_ssh_key['id']
          ui.success "SSH key '#{nickname}' has been successfully added."
        else
          ui.fail AcquiaToolbelt::CLI::API.display_error(add_ssh_key)
        end
      end
    end
  end
end