X-Git-Url: http://git.tdb.fi/?a=blobdiff_plain;f=source%2Fexternaltask.h;h=00465f3af4f3ba501952f1a7bf29ce149d04865a;hb=74ea8208bb1aa1d9afc9657a4cdfac6714241887;hp=1b330ff745afd235d31638c87e63ff23adf0bae3;hpb=b3851916ce08424fb5a28a57d70945e805d8d931;p=builder.git diff --git a/source/externaltask.h b/source/externaltask.h index 1b330ff..00465f3 100644 --- a/source/externaltask.h +++ b/source/externaltask.h @@ -3,43 +3,76 @@ #include #include +#include #include #include #include "task.h" +/** +Runs an external command. A zero exit status is translated to a SUCCESS status +for the task, and anything else is treated as an error. Output can optionally +be captured. +*/ class ExternalTask: public Task { public: - enum Destination + enum StreamAction { - PASSTHROUGH, - CAPTURE, - IGNORE + PASSTHROUGH, //< Do not touch the stream + CAPTURE, //< Capture the stream + REDIRECT, //< Redirect the stream to/from a file + IGNORE //< Redirect the stream to oblivion }; - typedef std::vector Arguments; + typedef Msp::Process::Arguments Arguments; private: Arguments argv; Msp::FS::Path work_dir; - int pid; + Msp::Process *process; int exit_code; - Destination stdout_dest; - Destination stderr_dest; + Msp::FS::Path stdin_file; + StreamAction stdout_action; + Msp::FS::Path stdout_file; + StreamAction stderr_action; Msp::IO::Pipe *capture_pipe; std::string output; public: + /** Creates an ExternalTask with an argument array and an optional working + directory. The first element of the argument array should be the command + name. If the working directory is not specified, no chdir is done. */ ExternalTask(const Arguments &, const Msp::FS::Path & = Msp::FS::Path()); + virtual ~ExternalTask(); virtual std::string get_command() const; virtual void start(); virtual Status check(); + virtual Status wait(); +private: + Status do_wait(bool); + +public: + /// Redirect stdin from a file. Has no effect after the task is started. + void set_stdin(const Msp::FS::Path &); - void set_stdout(Destination); - void set_stderr(Destination); + /// Sets destination for stdout. Has no effect after the task is started. + void set_stdout(StreamAction); + + /// Redirect stdout to a file. Has no effect after the task is started. + void set_stdout(const Msp::FS::Path &); + + /// Sets destination for stderr. Has no effect after the task is started. + void set_stderr(StreamAction); + + /** Returns captured output, if any. This may be called while the task is + still running, but it will always return all output. */ const std::string &get_output() const { return output; } + + /** Executes a command and captures its output. Stderr is ignored, but if + the command exits with a nonzero status, an exception is thrown. */ + static std::string run_and_capture_output(const Arguments &, const Msp::FS::Path & = Msp::FS::Path()); }; #endif