ProcExec.cpp

Go to the documentation of this file.

#include <sstream>
#include <iterator>
#include <Poco/Pipe.h>
#include <Poco/PipeStream.h>

#include "ProcExec.hpp"

//-----------------------------------------------------------------------------
ProcExec::ProcExec(const std::string& cmd_)
:inherited(cmd_), args(), env()
{
  std::stringstream istr(cmd), ostr;
  std::istream_iterator<std::string> eos;
  std::istream_iterator<std::string> iit(istr);

  ostr << *iit;
  cmd = ostr.str();
  while(++iit!=eos)
  {
    args.push_back(*iit);
  }
}
//-----------------------------------------------------------------------------
ProcExec::ProcExec(const std::string& cmd_, const Poco::Process::Args& args_)
:inherited(cmd_), args(args_), env()
{
}
//-----------------------------------------------------------------------------
ProcExec::ProcExec(const std::string& cmd_, const Poco::Process::Args& args_, const Poco::Process::Env& env_)
:inherited(cmd_), args(args_), env(env_)
{
}
//-----------------------------------------------------------------------------
bool ProcExec::exec(std::ostream& os, std::ostream& es)
{
  _IsError = false;
  std::stringstream outMsg, errMsg;
  Poco::Pipe outPipe, errPipe;
  Poco::ProcessHandle ph(Poco::Process::launch(cmd, args, 0, &outPipe, &errPipe, env));
  Poco::PipeInputStream istr(outPipe);
  Poco::PipeInputStream estr(errPipe);

  outMsg << istr.rdbuf();
  errMsg << estr.rdbuf();

  if (ph.wait() == 0)
  {
    os << outMsg.str();
    es << errMsg.str();
  }
  else
  {
    es << outMsg.str() << " " << errMsg.str();
    _IsError = true;
  }

  return !_IsError;
}
//-----------------------------------------------------------------------------
std::ostream& operator << (std::ostream& os, const ProcExec& procExec)
{
  const_cast<ProcExec&>(procExec).exec(os, os);
  return os;
}
//-----------------------------------------------------------------------------
std::string& operator << (std::string& str, const ProcExec& procExec)
{
  std::stringstream outMsg;
  outMsg << procExec;
  if (!procExec.isError())
    str = outMsg.str();
  return str;
}
//-----------------------------------------------------------------------------
//-----------------------------------------------------------------------------