require 'fileutils'

class TestRequest < Task

  set_table_name "test_requests"

  belongs_to :user
  belongs_to :problem
  belongs_to :submission

  def self.get_inqueue_and_change_status(status)
    # since there will be only one grader grading TestRequest
    # we do not need locking (hopefully)
    
    task = Task.find(:first, 
                     :order => "created_at", 
                     :conditions => {:status=> Task::STATUS_INQUEUE})
    if task!=nil
      task.status = status
      task.save!
    end
    
    task
  end

  # interfacing with form
  def self.new_from_form_params(user,params)
    test_request = TestRequest.new
    test_request.user = user
    problem = Problem.find(params[:problem_id])
    test_request.problem = problem
    test_request.submission = 
      Submission.find_by_user_problem_number(user.id,
                                             problem.id,
                                             params[:submission_number])
    test_request.input_file_name = save_input_file(params[:input_file], user, problem)
    test_request.submitted_at = Time.new
    test_request.status_inqueue
    test_request
  end

  protected
  def self.input_file_name(user,problem)
    begin
      tmpname = UPLOADED_INPUT_FILE_DIR + "/#{user.login}/#{problem.name}/#{rand(10000)}"
    end while File.exists?(tmpname)
    tmpname
  end

  def self.save_input_file(tempfile, user, problem)
    new_file_name = input_file_name(user,problem)
    dirname = File.dirname(new_file_name)
    FileUtils.mkdir_p(File.dirname(new_file_name)) if !File.exists?(dirname)
    if tempfile.instance_of?(Tempfile)
      tempfile.close
      FileUtils.move(tempfile.path,new_file_name)
    else      
      File.open(new_file_name, "wb") do |f| 
        f.write(tempfile.read) 
      end
    end
    new_file_name
  end
end