aboutsummaryrefslogtreecommitdiff
path: root/src/function/external_command.cc
diff options
context:
space:
mode:
Diffstat (limited to 'src/function/external_command.cc')
-rw-r--r--src/function/external_command.cc99
1 files changed, 99 insertions, 0 deletions
diff --git a/src/function/external_command.cc b/src/function/external_command.cc
new file mode 100644
index 0000000..bd2bf84
--- /dev/null
+++ b/src/function/external_command.cc
@@ -0,0 +1,99 @@
+#include "external_command.h"
+
+#include <xalanc/XSLT/XSLTInputSource.hpp>
+
+#include <xercesc/parsers/XercesDOMParser.hpp>
+
+#include <boost/process.hpp>
+
+#include <sstream>
+
+#include "support/xerces_string_guard.h"
+#include "support/dom/result_node_facade.h"
+
+namespace {
+
+using InputXSLT::XercesStringGuard;
+
+inline xercesc::DOMNode* importDocumentElement(
+ boost::process::pistream& outputStream,
+ xercesc::DOMDocument* const domDocument
+) {
+ std::stringstream xmlStream(
+ "<output>" + std::string(
+ (std::istreambuf_iterator<char>(outputStream)),
+ (std::istreambuf_iterator<char>())
+ ) + "</output>"
+ );
+
+ xercesc::XercesDOMParser parser;
+ parser.parse(xalan::XSLTInputSource(xmlStream));
+
+ return domDocument->importNode(
+ parser.getDocument()->getDocumentElement(),
+ true
+ );
+}
+
+}
+
+namespace InputXSLT {
+
+DomDocumentCache::document_ptr FunctionExternalCommand::constructDocument(
+ std::string command,
+ boost::optional<std::string> stdinText
+) const {
+ DomDocumentCache::document_ptr domDocument(
+ DomDocumentCache::createDocument("content")
+ );
+
+ boost::process::context context;
+ context.environment = boost::process::self::get_environment();
+ context.stdout_behavior = boost::process::capture_stream();
+ context.stdin_behavior = boost::process::capture_stream();
+
+ boost::process::child commandProcess(
+ boost::process::launch_shell(command, context)
+ );
+
+ if ( stdinText ) {
+ boost::process::postream& inputStream = commandProcess.get_stdin();
+ inputStream << *stdinText;
+ inputStream.close();
+ }
+
+ boost::process::pistream& outputStream = commandProcess.get_stdout();
+ boost::process::status status = commandProcess.wait();
+
+ ResultNodeFacade result(domDocument.get(), "output");
+ result.setAttribute("command", command);
+ result.setAttribute("code", std::to_string(status.exit_status()));
+
+ if ( status.exited() ) {
+ try {
+ result.setContent(
+ importDocumentElement(
+ outputStream,
+ domDocument.get()
+ )->getChildNodes()
+ );
+
+ result.setAttribute("result", "success");
+ }
+ catch ( const xercesc::DOMException& exception ) {
+ result.setAttribute("result", "error");
+
+ result.setValueNode(
+ "error",
+ *XercesStringGuard<char>(exception.msg)
+ );
+ }
+ } else {
+ result.setAttribute("result", "error");
+ }
+
+
+ return domDocument;
+}
+
+}