//  Copyright (c) 2005-2008 Andre Merzky
// 
//  Distributed under the Boost Software License, Version 1.0. (See accompanying 
//  file LICENSE_1_0.txt or copy at http://www.boost.org/LICENSE_1_0.txt)

#include "saga-shell.hpp"
#include <boost/lexical_cast.hpp>

int shell::c_run2 (std::vector <std::string> & args,
                   props & p)
{
  // sanity checks
  if ( args.empty () )
  {
    carp ("format: run <exe> [args] ...", p);
    return FAILURE;
  }

  std::string exe = args[0];
  args.erase (args.begin());

  bool bg = false;

  // check if we run in background
  if ( !args.empty() && args.back () == "&" )
  {
    bg = true;
    args.erase ((++args.rbegin()).base());    // erase last element
  }

  // create io streams for job io
  saga::job::ostream in;
  saga::job::istream out;
  saga::job::istream err;

  saga::job::job job;
  // run job
  // Use 'host_' as target host.
  // use 'js_'   as saga::job_service
  try {
    saga::job::description jd;

    jd.set_attribute (saga::job::attributes::description_executable, exe);

    if ( ! bg )
    {
        jd.set_attribute (saga::job::attributes::description_interactive, 
            saga::attributes::common_true);
    }

    jd.set_vector_attribute (saga::job::attributes::description_arguments, 
                             args);

    job = js_.create_job (jd);

    if ( ! bg ) 
    {
        in  = job.get_stdin  ();
        out = job.get_stdout ();
        err = job.get_stderr ();
    }
    job.run ();
  }

  catch (std::exception const & e)
  {
    p.out += "run failed: ";
    p.out += e.what();
    p.out += "\n";
    return FAILURE;
  }

  // get job state
  saga::job::state state = job.get_state ();

  // check if that worked
  if ( state != saga::job::Running &&
       state != saga::job::Done    )
  {
    carp ("run failed: " + exe, p);
    return FAILURE;
  }

  if ( bg )
  {
    // store background jobs in process table
    std::string jobid (job.get_job_id());
    int pid = jobs_.add (jobid, exe, job);

    // output
    p.out += "  [";
    p.out += boost::lexical_cast<std::string>(pid);
    p.out += "] ";
    p.out += exe;
    p.out += "\n";
    p.out += "  ";
    p.out += jobid;
    p.out += "\n";
  }
  else
  {
    // a foreground job is pulled for its stdout
    while ( true ) 
    {
      char buffer[255];

      // get stdout
      out.read (buffer, sizeof (buffer));
      if ( out.gcount () > 0 )
      {
        p.out += std::string (buffer, out.gcount ());
      }


      // get stderr
      err.read (buffer, sizeof (buffer));
      if ( err.gcount () > 0 )
      {
        p.out += std::string (buffer, err.gcount ());
      }


      if ( out.fail () ||
           err.fail () ) 
      {
        break;
      }
    }

    p.out += "\n";
  }

  return SUCCESS;
}