2 files changed, 90 insertions, 0 deletions
diff --git a/src/function/read_directory.cc b/src/function/read_directory.cc
new file mode 100644
index 0000000..7b73526
--- /dev/null
+++ b/src/function/read_directory.cc
@@ -0,0 +1,49 @@
+#include "read_directory.h"
+
+#include <iostream>
+
+namespace InputXSLT {
+
+FunctionReadDirectory::FunctionReadDirectory(const FilesystemContext& context):
+	fs_context_(context) { }
+
+xalan::XObjectPtr FunctionReadDirectory::execute(
+	xalan::XPathExecutionContext&                executionContext,
+	xalan::XalanNode*                            context,
+	const xalan::Function::XObjectArgVectorType& arguments,
+	const xalan::Locator*                        locator
+) const {
+	if ( arguments.size() != 1 ) {
+		xalan::XPathExecutionContext::GetAndReleaseCachedString guard(
+			executionContext
+		);
+
+		this->generalError(executionContext, context, locator);
+	}
+
+	std::string files;
+
+	this->fs_context_.iterate(
+		arguments[0]->str(),
+		[&files](const boost::filesystem::path& p) {
+		files += p.string() + "\n";
+	});
+
+	return executionContext.getXObjectFactory().createString(
+		xalan::XalanDOMString(files.data())
+	);
+}
+
+FunctionReadDirectory* FunctionReadDirectory::clone(
+	xalan::MemoryManager& manager) const {
+	return xalan::XalanCopyConstruct(manager, *this);
+}
+
+const xalan::XalanDOMString& FunctionReadDirectory::getError(
+	xalan::XalanDOMString& result) const {
+	result.assign("The read-directory() function expects one argument.");
+
+	return result;
+}
+
+}
diff --git a/src/function/read_directory.h b/src/function/read_directory.h
new file mode 100644
index 0000000..4a865f2
--- /dev/null
+++ b/src/function/read_directory.h
@@ -0,0 +1,41 @@
+#ifndef INPUTXSLT_SRC_FUNCTION_READ_DIRECTORY_H_
+#define INPUTXSLT_SRC_FUNCTION_READ_DIRECTORY_H_
+
+#include <xalanc/XalanTransformer/XalanTransformer.hpp>
+#include <xalanc/XPath/XObjectFactory.hpp>
+#include <xalanc/XPath/Function.hpp>
+#include <xalanc/XPath/XObject.hpp>
+
+#include <string>
+
+#include "common.h"
+#include "support/filesystem_context.h"
+
+namespace InputXSLT {
+
+class FunctionReadDirectory : public xalan::Function {
+	public:
+		FunctionReadDirectory(const FilesystemContext&);
+
+		virtual xalan::XObjectPtr execute(
+			xalan::XPathExecutionContext&,
+			xalan::XalanNode*,
+			const xalan::Function::XObjectArgVectorType&,
+			const xalan::Locator*
+		) const;
+
+		virtual FunctionReadDirectory* clone(xalan::MemoryManager&) const;
+
+		FunctionReadDirectory& operator=(const FunctionReadDirectory&) = delete;
+		bool operator==(const FunctionReadDirectory&) const            = delete;
+
+	private:
+		const FilesystemContext& fs_context_;
+
+		const xalan::XalanDOMString& getError(xalan::XalanDOMString&) const;
+
+};
+
+}
+
+#endif  // INPUTXSLT_SRC_FUNCTION_READ_DIRECTORY_H_