2 files changed, 100 insertions, 0 deletions
diff --git a/src/support/include_entity_resolver.cc b/src/support/include_entity_resolver.cc
new file mode 100644
index 0000000..79631a5
--- /dev/null
+++ b/src/support/include_entity_resolver.cc
@@ -0,0 +1,70 @@
+#include "include_entity_resolver.h"
+
+#include <xercesc/framework/LocalFileInputSource.hpp>
+
+#include "boost/filesystem.hpp"
+
+#include "support/xerces_string_guard.h"
+
+namespace InputXSLT {
+
+IncludeEntityResolver::IncludeEntityResolver(
+	const std::vector<std::string>& path):
+	path_() {
+	this->path_.reserve(path.size());
+
+	std::transform(
+		path.begin(),
+		path.end(),
+		std::back_inserter(this->path_),
+		[](const std::string& path) -> FilesystemContext {
+			return FilesystemContext(path);
+		}
+	);
+}
+
+xercesc::InputSource* IncludeEntityResolver::resolveEntity(
+	const XMLCh* const,
+	const XMLCh* const systemId
+) {
+	if ( systemId != nullptr ) {
+		const std::string rawPath(
+			*XercesStringGuard<char>(systemId)
+		);
+
+		const std::size_t leadingDelimiter = rawPath.find_first_of('[');
+		const std::size_t closingDelimiter = rawPath.find_last_of(']');
+
+		if ( leadingDelimiter != std::string::npos &&
+		     closingDelimiter != std::string::npos &&
+		     leadingDelimiter <  closingDelimiter ) {
+			const std::string filePath(
+				rawPath.substr(
+					leadingDelimiter + 1,
+					closingDelimiter - leadingDelimiter - 1
+				)
+			);
+
+			for ( auto&& context : this->path_ ) {
+				const boost::filesystem::path resolvedPath(
+					context.resolve(filePath)
+				);
+
+				if ( boost::filesystem::exists(resolvedPath) &&
+					 boost::filesystem::is_regular_file(resolvedPath) ) {
+					return new xercesc::LocalFileInputSource(
+						*XercesStringGuard<XMLCh>(resolvedPath.string())
+					);
+				}
+			}
+
+			return nullptr;
+		} else {
+			return nullptr;
+		}
+	} else {
+		return nullptr;
+	}
+}
+
+}
diff --git a/src/support/include_entity_resolver.h b/src/support/include_entity_resolver.h
new file mode 100644
index 0000000..390cfe1
--- /dev/null
+++ b/src/support/include_entity_resolver.h
@@ -0,0 +1,30 @@
+#ifndef INPUTXSLT_SRC_SUPPORT_INCLUDE_ENTITY_RESOLVER_H_
+#define INPUTXSLT_SRC_SUPPORT_INCLUDE_ENTITY_RESOLVER_H_
+
+#include <xercesc/sax/EntityResolver.hpp>
+#include <xercesc/sax/InputSource.hpp>
+
+#include <string>
+#include <vector>
+
+#include "filesystem_context.h"
+
+namespace InputXSLT {
+
+class IncludeEntityResolver : public xercesc::EntityResolver {
+	public:
+		IncludeEntityResolver(const std::vector<std::string>&);
+
+		xercesc::InputSource* resolveEntity(
+			const XMLCh* const,
+			const XMLCh* const
+		);
+
+	private:
+		std::vector<FilesystemContext> path_;
+
+};
+
+}
+
+#endif  // INPUTXSLT_SRC_SUPPORT_INCLUDE_ENTITY_RESOLVER_H_