Contents

module Elzar
  module SshKeyLocator

    class << self
      def default_key_file_paths
        %w[id_dsa.pub id_ecdsa.pub id_rsa.pub].map do |base_filename|
          File.expand_path("~/.ssh/#{base_filename}")
        end
      end

      def find_keys(candidate_key_file_paths = default_key_file_paths)
        local_keys = find_local_keys(candidate_key_file_paths)
        local_keys.empty? ? find_agent_keys : local_keys
      end

      def find_local_keys(candidate_key_file_paths = default_key_file_paths)
        first_existing_file = candidate_key_file_paths.find { |p| File.exist?(p) }
        return [] unless first_existing_file

        file_content = File.read(first_existing_file)
        split_keys(file_content)
      end

      def find_agent_keys
        keys = split_keys(`ssh-add -L`)
        $?.success? ? keys : []
      end

      private

      def split_keys(s)
        s.split("\n").reject { |k| k.strip.empty? }
      end
    end

  end
end

Version data entries

1 entries across 1 versions & 1 rubygems

Version	Path
elzar-0.2.0	lib/elzar/ssh_key_locator.rb