lib/capitate/plugins/prompt.rb in capitate-0.2.3 vs lib/capitate/plugins/prompt.rb in capitate-0.2.5

@@ -1,26 +1,69 @@
+require 'md5'
 
 module Capitate::Plugins::Prompt
   
   def ask(label, &block)
     Capistrano::CLI.ui.ask(label, &block)
   end
   
-  def password(label, verify = false, lazy = true)
+  # Prompt for password.
+  #
+  # ==== Options
+  # +label+:: Label
+  # +options+:: Options
+  # - +verify+:: If true, prompt twice and verify
+  # - +lazy+:: If true, returns a block. _Defaults to true_
+  # - +check_hash+:: If present, checks that md5 is same as password md5
+  #
+  def password(label, options = {})
+    
+    verify = options[:verify]
+    lazy = options[:lazy].nil? ? true : options[:lazy]
+    check_hash = options[:check_hash]
+    
     # Lazy
     password_prompt = Proc.new { 
-      password = Capistrano::CLI.password_prompt(label)
+      
+      max_attempts = 2
+      attempts = 0
+      password = nil
+      success = true
+
+      loop { 
+        password = Capistrano::CLI.password_prompt(label)
+        attempts += 1
     
-      if verify
-        password_verify = Capistrano::CLI.password_prompt("[VERIFY] #{label}")
-        raise "Passwords do not match" if password != password_verify
-      end
+        if verify
+          password_verify = Capistrano::CLI.password_prompt("[VERIFY] #{label}")
+          if password != password_verify
+            logger.important "Passwords do not match" 
+            success = false
+          end
+        end
+      
+        if check_hash
+          if MD5.md5(password).hexdigest != check_hash
+            logger.important "Invalid password, try again." 
+            success = false
+          end         
+        end
+        
+        break if success
+        break if attempts >= max_attempts
+      }
+      
+      raise "Invalid password, too many tries" unless success
     
       password
     }
     
     return password_prompt if lazy
     password_prompt.call
+  end  
+  
+  def check_password_hash(password, hash)
+    MD5.md5()
   end
   
 end
 
 Capistrano.plugin :prompt, Capitate::Plugins::Prompt