keyrack 0.3.0.pre → 0.3.0

data/lib/keyrack/site.rb

@@ -0,0 +1,92 @@
+module Keyrack
+  class Site < Hash
+    def initialize(*args)
+      if args[0].is_a?(Hash)
+        hash = args[0]
+        if !hash.has_key?('name')
+          raise ArgumentError, "hash is missing the 'name' key"
+        end
+        if !hash['name'].is_a?(String)
+          raise ArgumentError, "name is not a String"
+        end
+        if !hash.has_key?('username')
+          raise ArgumentError, "hash is missing the 'username' key"
+        end
+        if !hash['username'].is_a?(String)
+          raise ArgumentError, "name is not a String"
+        end
+        if !hash.has_key?('password')
+          raise ArgumentError, "hash is missing the 'password' key"
+        end
+        if !hash['password'].is_a?(String)
+          raise ArgumentError, "name is not a String"
+        end
+        self.update(hash)
+      else
+        self['name'] = args[0]
+        self['username'] = args[1]
+        self['password'] = args[2]
+      end
+
+      @event_hooks = []
+    end
+
+    def change_attribute(name, value)
+      event = Event.new(self, 'change')
+      event.attribute_name = name
+      event.previous_value = self[name]
+      event.new_value = value
+
+      self[name] = value
+      trigger(event)
+    end
+
+    def name
+      self['name']
+    end
+
+    def name=(name)
+      change_attribute('name', name)
+    end
+
+    def username
+      self['username']
+    end
+
+    def username=(username)
+      change_attribute('username', username)
+    end
+
+    def password
+      self['password']
+    end
+
+    def password=(password)
+      change_attribute('password', password)
+    end
+
+    def after_event(&block)
+      @event_hooks << block
+    end
+
+    def encode_with(coder)
+      coder.represent_map(nil, self)
+    end
+
+    def ==(other)
+      if other.instance_of?(Site)
+        other.name == name && other.username == username
+      else
+        super
+      end
+    end
+
+    private
+
+    def trigger(event)
+      @event_hooks.each do |block|
+        block.call(event)
+      end
+    end
+  end
+end

data/lib/keyrack/ui/console.rb

@@ -1,7 +1,7 @@
 module Keyrack
   module UI
     class Console
-      attr_accessor :database, :mode
+      attr_accessor :mode
 
       def initialize
         @highline = HighLine.new
@@ -9,64 +9,51 @@ module Keyrack
       end
 
       def get_password
-        @highline.ask("Keyrack password: ") { |q| q.echo = false }
+        @highline.ask("Keyrack password: ") { |q| q.echo = false }.to_s
       end
 
-      def menu(options = {})
+      def menu(options)
+        current_group = options[:group]
+        dirty = options[:dirty]
+        at_top = options[:at_top]
+
         choices = {'n' => :new, 'q' => :quit, 'm' => :mode}
-        index = 1
-
-        sites = @database.sites(options)
-        count = sites.length
-        count += @database.groups.length  if !options[:group]
-        width = count / 10
-
-        if !options[:group]
-          # Can't have subgroups (yet?).
-          @highline.say("=== #{@highline.color("Keyrack Main Menu", :yellow)} ===")
-          @database.groups.each do |group|
-            choices[index.to_s] = {:group => group}
-            @highline.say(" %#{width}d. %s" % [index, @highline.color(group, :green)])
-            index += 1
-          end
-        else
-          @highline.say("===== #{@highline.color(options[:group], :green)} =====")
-        end
+        entry_choices = print_entries(current_group,
+          at_top ? "Keyrack Main Menu" : current_group.name)
+        choices.update(entry_choices)
 
-        sites.each do |site|
-          site_entry = @database.get(site, options)
-          site_entry = [site_entry] unless site_entry.is_a?(Array)
-          site_entry.each do |entry|
-            choices[index.to_s] = entry
-            @highline.say(" %#{width}d. %s [%s]" % [index, site, entry[:username]])
-            index += 1
-          end
+        @highline.say("Mode: #{@mode}")
+        commands = "Commands: [n]ew"
+        if !current_group.sites.empty?
+          choices['e'] = :edit
+          commands << " [e]dit"
         end
 
-        @highline.say("Mode: #{mode}")
-        commands = "Commands: [n]ew"
-        if !sites.empty?
-          choices['d'] = :delete
-          commands << " [d]elete"
+        choices['g'] = :new_group
+        commands << " [g]roup"
+
+        if options[:enable_up]
+          choices['u'] = :up
+          commands << " [u]p"
         end
-        if !options[:group]
-          choices['g'] = :new_group
-          commands << " [g]roup"
-        else
+
+        if !at_top
           choices['t'] = :top
           commands << " [t]op"
         end
-        if @database.dirty?
+
+        if dirty
           choices['s'] = :save
           commands << " [s]ave"
         end
         commands << " [m]ode [q]uit"
         @highline.say(commands)
-        answer = @highline.ask(" ? ") { |q| q.in = choices.keys }
+
+        answer = @highline.ask("? ") { |q| q.in = choices.keys }.to_s
         result = choices[answer]
         case result
         when Symbol
-          if result == :quit && @database.dirty? && !@highline.agree("Really quit?  You have unsaved changes! [yn] ")
+          if result == :quit && dirty && !@highline.agree("Really quit?  You have unsaved changes! [yn] ")
             nil
           elsif result == :mode
             @mode = @mode == :copy ? :print : :copy
@@ -76,13 +63,15 @@ module Keyrack
           end
         when Hash
           if result.has_key?(:group)
-            result
+            {:group => current_group.group(result[:group])}
           else
-            if mode == :copy
-              Copier(result[:password])
+            password = result[:site].password
+
+            if @mode == :copy
+              Clipboard.copy(password)
               @highline.say("The password has been copied to your clipboard.")
-            elsif mode == :print
-              password = @highline.color(result[:password], :cyan)
+            elsif @mode == :print
+              password = @highline.color(password, :cyan)
               @highline.ask("Here you go: #{password}. Done? ") do |question|
                 question.echo = false
                 if HighLine::SystemExtensions::CHARACTER_MODE != 'stty'
@@ -96,50 +85,31 @@ module Keyrack
         end
       end
 
-      def get_new_group
-        group = @highline.ask("Group: ") { |q| q.validate = /^\w[\w\s]*$/ }
-        {:group => group}
+      def get_new_group(options = {})
+        @highline.ask("Group: ") { |q| q.validate = /^\w[\w\s]*$/ }.to_s
       end
 
       def get_new_entry
         result = {}
-        result[:site]     = @highline.ask("Label: ")
-        result[:username] = @highline.ask("Username: ")
-        if @highline.agree("Generate password? [yn] ")
-          loop do
-            password = Utils.generate_password
-            if @highline.agree("Generated #{@highline.color(password, :cyan)}.  Sound good? [yn] ")
-              result[:password] = password
-              break
-            end
-          end
-        else
-          loop do
-            password = @highline.ask("Password: ") { |q| q.echo = false }
-            confirmation = @highline.ask("Password (again): ") { |q| q.echo = false }
-            if password == confirmation
-              result[:password] = password
-              break
-            end
-            @highline.say("Passwords didn't match. Try again!")
-          end
-        end
-        result
+        result[:site]     = @highline.ask("Label: ").to_s
+        result[:username] = @highline.ask("Username: ").to_s
+        result[:password] = get_new_password
+        result[:password].nil? ? nil : result
       end
 
       def display_first_time_notice
         @highline.say("This looks like your first time using Keyrack.  I'll need to ask you a few questions first.")
       end
 
-      def rsa_setup
+      def password_setup
         password = confirmation = nil
         loop do
-          password = @highline.ask("New passphrase: ") { |q| q.echo = false }
-          confirmation = @highline.ask("Confirm passphrase: ") { |q| q.echo = false }
+          password = @highline.ask("New passphrase: ") { |q| q.echo = false }.to_s
+          confirmation = @highline.ask("Confirm passphrase: ") { |q| q.echo = false }.to_s
           break if password == confirmation
           @highline.say("Passphrases didn't match.")
         end
-        { 'password' => password, 'path' => 'rsa' }
+        password
       end
 
       def store_setup
@@ -153,35 +123,204 @@ module Keyrack
         when 'filesystem'
           result['path'] = 'database'
         when 'ssh'
-          result['host'] = @highline.ask("Host: ")
-          result['user'] = @highline.ask("User: ")
-          result['path'] = @highline.ask("Remote path: ")
+          result['host'] = @highline.ask("Host: ").to_s
+          result['user'] = @highline.ask("User: ").to_s
+          result['path'] = @highline.ask("Remote path: ").to_s
         end
 
         result
       end
 
-      def delete_entry(options = {})
+      def choose_entry_to_edit(group)
         choices = {'c' => :cancel}
-        index = 1
-        @highline.say("Choose entry to delete:")
-        @database.sites(options).each do |site|
-          site_entry = @database.get(site, options)
-          site_entry = [site_entry] unless site_entry.is_a?(Array)
-          site_entry.each do |entry|
-            choices[index.to_s] = {:site => site, :username => entry[:username]}
-            @highline.say("% 2d. %s [%s]" % [index, site, entry[:username]])
-            index += 1
+        entry_choices = print_entries(group, "Choose entry")
+        choices.update(entry_choices)
+
+        @highline.say("c. Cancel")
+
+        answer = @highline.ask("? ") { |q| q.in = choices.keys }.to_s
+        result = choices[answer]
+        if result == :cancel
+          nil
+        else
+          result
+        end
+      end
+
+      def edit_entry(site)
+        colored_entry = @highline.color("#{site.name} [#{site.username}]", :cyan)
+        @highline.say("Editing entry: #{colored_entry}")
+        @highline.say("u. Change username")
+        @highline.say("p. Change password")
+        @highline.say("d. Delete")
+        @highline.say("c. Cancel")
+
+        case @highline.ask("? ") { |q| q.in = %w{u p d c} }.to_s
+        when "u"
+          :change_username
+        when "p"
+          :change_password
+        when "d"
+          :delete
+        when "c"
+          nil
+        end
+      end
+
+      def change_username(old_username)
+        colored_old_username = @highline.color(old_username, :cyan)
+        @highline.say("Current username: #{colored_old_username}")
+        @highline.ask("New username (blank to cancel): ") { |q| q.validate = /\S/ }.to_s
+      end
+
+      def confirm_overwrite_entry(site)
+        entry_name = @highline.color("#{site.name} [#{site.username}]", :cyan)
+        @highline.agree("There's already an entry for: #{entry_name}. Do you want to overwrite it? [yn] ")
+      end
+
+      def confirm_delete_entry(site)
+        entry_name = @highline.color("#{site.name} [#{site.username}]", :red)
+        @highline.agree("You're about to delete #{entry_name}. Are you sure? [yn] ")
+      end
+
+      def display_invalid_password_notice
+        @highline.say("Invalid password.")
+      end
+
+      def get_new_password
+        result = nil
+        case @highline.ask("Generate password? [ync] ") { |q| q.in = %w{y n c} }.to_s
+        when "y"
+          result = get_generated_password
+          if result.nil?
+            result = get_manual_password
           end
+        when "n"
+          result = get_manual_password
         end
-        @highline.say(" c. Cancel")
+        result
+      end
 
-        answer = @highline.ask(" ?  ") { |q| q.in = choices.keys }
-        result = choices[answer]
-        if result != :cancel
-          entry = @highline.color("#{result[:site]} [#{result[:username]}]", :red)
-          if @highline.agree("You're about to delete #{entry}.  Are you sure? [yn] ")
-            @database.delete(result[:site], result[:username], options)
+      def get_generated_password
+        password = nil
+        loop do
+          password = Utils.generate_password
+          colored_password = @highline.color(password, :cyan)
+          case @highline.ask("Generated #{colored_password}.  Sound good? [ync] ") { |q| q.in = %w{y n c} }.to_s
+          when "y"
+            break
+          when "c"
+            password = nil
+            break
+          end
+        end
+        password
+      end
+
+      def get_manual_password
+        password = nil
+        loop do
+          password = @highline.ask("Password: ") { |q| q.echo = false }.to_s
+          confirmation = @highline.ask("Password (again): ") { |q| q.echo = false }.to_s
+          if password == confirmation
+            break
+          end
+          @highline.say("Passwords didn't match. Try again!")
+        end
+        password
+      end
+
+      private
+
+      def print_entries(group, title)
+        selections = []
+        max_width = 0
+        choices = {}
+        selection_index = 1
+        group.group_names.each do |group_name|
+          choices[selection_index.to_s] = {:group => group_name}
+
+          text = @highline.color(group_name, :green)
+          width = group_name.length
+          selections.push({:width => width, :text => text})
+
+          max_width = width if width > max_width
+          selection_index += 1
+        end
+
+        group.sites.each do |site|
+          choices[selection_index.to_s] = {:site => site}
+
+          text = "%s [%s]" % [site.name, site.username]
+          width = text.length
+          selections.push({:width => width, :text => text})
+
+          max_width = width if width > max_width
+          selection_index += 1
+        end
+
+        title = {
+          :text => @highline.color(title, :yellow),
+          :width => title.length
+        }
+
+        columnize_menu(selections, max_width, title)
+        choices
+      end
+
+      def columnize_menu(selections, max_width, title = nil)
+        terminal_size = HighLine::SystemExtensions.terminal_size
+
+        if selections.empty?
+          if title
+            @highline.say("=== #{title[:text]} ===")
+          end
+          return
+        end
+
+        # add in width for numbers
+        number_width = Math.log10(selections.count).floor + 1
+        max_width += number_width + 2
+
+        multiples = max_width == 0 ? 1 : terminal_size[0] / max_width
+        num_columns =
+          if multiples > 1
+            if (terminal_size[0] - (multiples * max_width)) < (multiples - 1)
+              # If there aren't sufficient spaces, decrease column count
+              multiples - 1
+            else
+              multiples
+            end
+          else
+            1
+          end
+        #puts "Terminal width: %d; Max width: %d; Multiples: %d; Columns: %d" %
+          #[ terminal_size[0], max_width, multiples, num_columns ]
+        total_width = num_columns * max_width + (num_columns - 1)
+
+        if title
+          padding_total = total_width - title[:width] - 2
+          padding_left = [padding_total / 2, 3].max
+          padding_right = [padding_total - padding_left, 3].max
+          @highline.say(("=" * padding_left) + " #{title[:text]} " + ("=" * padding_right))
+        end
+
+        selection_index = 0
+        catch(:stop) do
+          loop do
+            num_columns.downto(1) do |i|
+              selection = selections[selection_index]
+              throw(:stop) if selection.nil?
+
+              label = "%#{number_width}d. " % (selection_index + 1)
+              if i == 1 || selection_index == (selections.count - 1)
+                @highline.say(label + selection[:text])
+              else
+                spaces = max_width - (selection[:width] + number_width + 2) + 1
+                @highline.say(label + selection[:text] + (" " * spaces))
+              end
+              selection_index += 1
+            end
           end
         end
       end