From 7142544d43b431df44d34921b0f3012fa1e0137d Mon Sep 17 00:00:00 2001 From: Adrian Kummerländer Date: Mon, 21 Apr 2014 21:47:20 +0200 Subject: Implemented basic external directory traversal function * _read-directory_ lists all files in a given directory ** currently text-only output, xml planned * improved FilesystemContext path resolution (relative path is fully resolved by boost::filesystem) --- src/function/read_directory.cc | 49 ++++++++++++++++++++++++++++++++++++++++++ src/function/read_directory.h | 41 +++++++++++++++++++++++++++++++++++ 2 files changed, 90 insertions(+) create mode 100644 src/function/read_directory.cc create mode 100644 src/function/read_directory.h (limited to 'src/function') 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 + +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 +#include +#include +#include + +#include + +#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_ -- cgit v1.2.3