lib/rturk/operations/create_hit.rb in rturk-2.0.3 vs lib/rturk/operations/create_hit.rb in rturk-2.0.4

@@ -1,28 +1,50 @@
 require File.join(File.dirname(__FILE__), 'register_hit_type')
 
 module RTurk
   class CreateHIT < RegisterHITType
-    attr_accessor :hit_type_id
-
+    attr_accessor :hit_type_id, :assignments, :lifetime, :note
+    
     def parse(response)
       RTurk::CreateHITResponse.new(response)
     end
 
+    # Gives us access to a question builder attached to this HIT
+    #
+    # @param [String, Hash] URL Params, if none is passed, simply returns the question
+    # @return [RTurk::Question] The question if instantiated or nil
+    def question(*args)
+      unless args.empty?
+        @question ||= RTurk::Question.new(*args)
+      else
+        @question
+      end
+    end
+
     def to_params
-      super.merge('HITTypeId' => hit_type_id)
+      super.merge(
+        'HITTypeId'           => hit_type_id, 
+        'MaxAssignments'      => (assignments || 1),
+        'Question'            => question.to_params,
+        'LifetimeInSeconds'   => (lifetime || 3600),
+        'RequesterAnnotation' => note
+      )
     end
 
     # More complicated validation run before request
     #
     def validate
       if hit_type_id
-        unless question && lifetime
-          raise RTurk::MissingParameters, "When you specify a HitTypeID, you must incude a question and lifetime length"
+        unless question
+          raise RTurk::MissingParameters, "When you specify a HitTypeID, you must incude a question"
         end
       else
         super # validate as RegisterHitType
       end
+    end
+    
+    def required_fields
+      super << :question
     end
   end
 
   def self.CreateHIT(*args, &blk)
     RTurk::CreateHIT.create(*args, &blk)