Unify Array and Proc AnswerConverter convertions

1 files changed, 11 insertions, 5 deletions

diff --git a/lib/highline/question/answer_converter.rb b/lib/highline/question/answer_converter.rb
index 0aa81c8..3f73f9f 100644
--- a/lib/highline/question/answer_converter.rb
+++ b/lib/highline/question/answer_converter.rb
@@ -25,11 +25,8 @@ class HighLine
           answer_type.parse(answer)
         elsif answer_type.is_a? Class
           send(answer_type.name)
-        elsif answer_type.is_a?(Array)
-          self.answer = choices_complete(answer)
-          answer.last
-        elsif answer_type.is_a?(Proc)
-          answer_type.call(answer)
+        else
+          send(answer_type.class.name)
         end
 
         check_range
@@ -72,6 +69,15 @@ class HighLine
         self.answer = choices_complete(answer)
         Pathname.new(File.join(directory.to_s, answer.last))
       end
+
+      def Array
+        self.answer = choices_complete(answer)
+        answer.last
+      end
+
+      def Proc
+        answer_type.call(answer)
+      end
     end
   end
 end
\ No newline at end of file