lib/hammer_cli/options/normalizers.rb in hammer_cli-0.0.18 vs lib/hammer_cli/options/normalizers.rb in hammer_cli-0.1.0

@@ -1,5 +1,7 @@
+require 'json'
+
 module HammerCLI
   module Options
     module Normalizers
 
 
@@ -19,31 +21,31 @@
 
 
       class KeyValueList < AbstractNormalizer
 
         def description
-          "Comma-separated list of key=value."
+          _("Comma-separated list of key=value.")
         end
 
         def format(val)
           return {} unless val.is_a?(String)
 
           val.split(",").inject({}) do |result, item|
             parts = item.split("=")
             if parts.size != 2
-              raise ArgumentError, "value must be defined as a comma-separated list of key=value"
+              raise ArgumentError, _("value must be defined as a comma-separated list of key=value")
             end
             result.update(parts[0] => parts[1])
           end
         end
       end
 
 
       class List < AbstractNormalizer
 
         def description
-          "Comma separated list of values."
+          _("Comma separated list of values.")
         end
 
         def format(val)
           val.is_a?(String) ? val.split(",") : []
         end
@@ -51,21 +53,21 @@
 
 
       class Bool < AbstractNormalizer
 
         def description
-          "One of true/false, yes/no, 1/0."
+          _("One of true/false, yes/no, 1/0.")
         end
 
         def format(bool)
           bool = bool.to_s
           if bool.downcase.match(/^(true|t|yes|y|1)$/i)
             return true
           elsif bool.downcase.match(/^(false|f|no|n|0)$/i)
             return false
           else
-            raise ArgumentError, "value must be one of true/false, yes/no, 1/0"
+            raise ArgumentError, _("value must be one of true/false, yes/no, 1/0")
           end
         end
 
         def complete(value)
           ["yes ", "no "]
@@ -88,26 +90,43 @@
             end
           end
         end
       end
 
+      class JSONInput < File
 
+        def format(val)
+          # The JSON input can be either the path to a file whose contents are
+          # JSON or a JSON string.  For example:
+          #   /my/path/to/file.json
+          # or
+          #   '{ "units":[ { "name":"zip", "version":"9.0", "inclusion":"false" } ] }')
+          json_string = ::File.exist?(::File.expand_path(val)) ? super(val) : val
+          ::JSON.parse(json_string)
+
+        rescue ::JSON::ParserError => e
+          raise ArgumentError, _("Unable to parse JSON input")
+        end
+
+      end
+
+
       class Enum < AbstractNormalizer
 
         def initialize(allowed_values)
           @allowed_values = allowed_values
         end
 
         def description
-          "One of %s" % quoted_values
+          _("One of %s") % quoted_values
         end
 
         def format(value)
           if @allowed_values.include? value
             value
           else
-            raise ArgumentError, "value must be one of '%s'" % quoted_values
+            raise ArgumentError, _("value must be one of '%s'") % quoted_values
           end
         end
 
         def complete(value)
           Completer::finalize_completions(@allowed_values)
@@ -117,9 +136,25 @@
 
         def quoted_values
           @allowed_values.map { |v| "'#{v}'" }.join(', ')
         end
       end
+
+
+      class DateTime < AbstractNormalizer
+
+        def description
+          _("Date and time in YYYY-MM-DD HH:MM:SS or ISO 8601 format")
+        end
+
+        def format(date)
+          raise ArgumentError unless date
+          ::DateTime.parse(date).to_s
+        rescue ArgumentError
+          raise ArgumentError, _("'%s' is not a valid date") % date
+        end
+      end
+
 
     end
   end
 end