aboutsummaryrefslogtreecommitdiff
path: root/src/function/read_directory.cc
blob: 245f311d9182ae4367c124ab8bc3d7e0d9ae315e (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
#include "read_directory.h"

#include <xercesc/dom/DOMDocument.hpp>
#include <xercesc/dom/DOMImplementation.hpp>
#include <xercesc/dom/DOMElement.hpp>
#include <xercesc/dom/DOMText.hpp>
#include <xercesc/util/XMLString.hpp>

#include "support/xerces_string_guard.h"

namespace InputXSLT {

FunctionReadDirectory::FunctionReadDirectory(const FilesystemContext& context):
	fs_context_(context),
	documents_(std::make_shared<std::stack<DomDocumentGuard>>()) { }

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);
	}

	this->documents_->emplace("content");
	DomDocumentGuard& domDocument = this->documents_->top();

	xercesc::DOMNode* const rootNode = domDocument->getDocumentElement();

	this->fs_context_.iterate(
		arguments[0]->str(),
		[&domDocument, &rootNode](const boost::filesystem::path& p) {
		xercesc::DOMElement* const itemNode(
			domDocument->createElement(*XercesStringGuard("item"))
		);

		if ( boost::filesystem::is_regular_file(p) ) {
			itemNode->setAttribute(
				*XercesStringGuard("type"),
				*XercesStringGuard("file")
			);
		} else if ( boost::filesystem::is_directory(p) ) {
			itemNode->setAttribute(
				*XercesStringGuard("type"),
				*XercesStringGuard("directory")
			);
		}

		xercesc::DOMText* const textNode(
			domDocument->createTextNode(
				*XercesStringGuard(p.filename().string())
			)
		);

		itemNode->appendChild(textNode);
		rootNode->appendChild(itemNode);
	});

	return executionContext.getXObjectFactory().createNodeSet(
		domDocument.finalize()
	);
}

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;
}

}