package org.broadinstitute.sting.queue.engine

import org.broadinstitute.sting.queue.function.CommandLineFunction
import org.broadinstitute.sting.queue.util.{JobExitException, Logging, ShellJob}

/**
 * Runs jobs one at a time locally
 */
class ShellJobRunner(val function: CommandLineFunction) extends CommandLineJobRunner with Logging {
  private var runStatus: RunnerStatus.Value = _

  /**
   * Runs the function on the local shell.
   * @param function Command to run.
   */
  def start() = {
    try {
      val job = new ShellJob

      job.workingDir = function.commandDirectory
      job.outputFile = function.jobOutputFile
      job.errorFile = function.jobErrorFile

      writeExec()
      job.shellScript = exec

      // Allow advanced users to update the job.
      updateJobRun(job)

      if (logger.isDebugEnabled) {
        logger.debug("Starting: " + function.commandDirectory + " > " + function.commandLine)
      } else {
        logger.info("Starting: " + function.commandLine)
      }

      logger.info("Output written to " + function.jobOutputFile)
      if (function.jobErrorFile != null)
        logger.info("Errors written to " + function.jobErrorFile)

      function.deleteLogs()
      function.deleteOutputs()
      function.mkOutputDirectories()
      runStatus = RunnerStatus.RUNNING
      job.run()
      removeTemporaryFiles()
      function.doneOutputs.foreach(_.createNewFile())
      runStatus = RunnerStatus.DONE
      logger.info("Done: " + function.commandLine)
    } catch {
      case jee: JobExitException =>
        runStatus = RunnerStatus.FAILED
        try {
          removeTemporaryFiles()
          function.failOutputs.foreach(_.createNewFile())
          writeError(jee.getMessage)
        } catch {
          case _ => /* ignore errors in the exception handler */
        }
        logger.error("Error: " + function.commandLine)
        logger.error(jee.stdErr)
      case e =>
        runStatus = RunnerStatus.FAILED
        try {
          removeTemporaryFiles()
          function.failOutputs.foreach(_.createNewFile())
          writeStackTrace(e)
        } catch {
          case _ => /* ignore errors in the exception handler */
        }
        logger.error("Error: " + function.commandLine, e)
    }
  }

  def status = runStatus
}