projects/command/src/concerns/domain_mappers.rb in foobara-0.0.2 vs projects/command/src/concerns/domain_mappers.rb in foobara-0.0.3

@@ -2,25 +2,38 @@
   class Command
     module Concerns
       module DomainMappers
         include Concern
 
-        def run_mapped_subcommand!(subcommand_class, unmapped_inputs)
+        def run_mapped_subcommand!(subcommand_class, *args)
+          unmapped_inputs, has_result_type, result_type =
+            case args.size
+            when 1
+              [args.first]
+            when 2
+              [args[1], true, args[0]]
+            else
+              # :nocov:
+              raise ArgumentError,
+                    "Wrong number of arguments: (#{args.size}. Expected 2 or 3 argument."
+              # :nocov:
+            end
+
           inputs = domain_map!(unmapped_inputs, to: subcommand_class, strict: true)
 
           result = run_subcommand!(subcommand_class, inputs)
 
-          result_mapper = self.class.domain.foobara_domain_mapper_registry.lookup(
-            from: result,
-            to: result_type,
-            strict: true
-          )
+          if has_result_type
+            result_mapper = self.class.domain.foobara_domain_mapper_registry.lookup!(
+              from: result,
+              to: result_type,
+              strict: true
+            )
 
-          if result_mapper
-            result_mapper.map(result)
-          else
-            result
+            result = result_mapper.map(result)
           end
+
+          result
         end
 
         def domain_map(...)
           self.class.domain.foobara_domain_map(...)
         end