lib/eff/verifier.rb in eff-0.0.1 vs lib/eff/verifier.rb in eff-0.0.2

@@ -1,10 +1,18 @@
 module Eff
   class Verifier
     class << self
-      def check(file, checksum, algo = Digest::SHA1)
-        file_digest = algo.file(File.expand_path(file)).hexdigest
+      def check(file, checksum, hash_function)
+        return unless File.exists?(file)
+        function    = function_const_get(hash_function)
+        file_digest = function.file(file).hexdigest
         file_digest == checksum
+      end
+
+    private
+      def function_const_get(value)
+        hash_function = value.to_s.upcase
+        Digest.const_get(hash_function)
       end
     end
   end
 end