1 files changed, 91 insertions, 0 deletions
diff --git a/src/plugins/filesystem/FileLocator.cpp b/src/plugins/filesystem/FileLocator.cpp
new file mode 100644
index 0000000..9ab398e
--- /dev/null
+++ b/src/plugins/filesystem/FileLocator.cpp
@@ -0,0 +1,91 @@
+/*
+    Ousía
+    Copyright (C) 2014, 2015  Benjamin Paaßen, Andreas Stöckel
+
+    This program is free software: you can redistribute it and/or modify
+    it under the terms of the GNU General Public License as published by
+    the Free Software Foundation, either version 3 of the License, or
+    (at your option) any later version.
+
+    This program is distributed in the hope that it will be useful,
+    but WITHOUT ANY WARRANTY; without even the implied warranty of
+    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+    GNU General Public License for more details.
+
+    You should have received a copy of the GNU General Public License
+    along with this program.  If not, see <http://www.gnu.org/licenses/>.
+*/
+
+#include "FileLocator.hpp"
+
+#include <boost/filesystem.hpp>
+
+#include <fstream>
+
+namespace ousia {
+
+void FileLocator::addSearchPath(const std::string &path,
+                                std::set<ResourceType> types)
+{
+	// Canonicalize the given path
+	std::string canonicalPath =
+	    boost::filesystem::canonical(path).generic_string();
+
+	// Insert the path for all given types.
+	for (auto &type : types) {
+		auto it = searchPaths.find(type);
+		if (it != searchPaths.end()) {
+			it->second.push_back(canonicalPath);
+		} else {
+			searchPaths.insert({type, {canonicalPath}});
+		}
+	}
+}
+
+bool FileLocator::doLocate(Resource &resource, const std::string &path,
+                           const ResourceType type,
+                           const std::string &relativeTo) const
+{
+	boost::filesystem::path base(relativeTo);
+	if (boost::filesystem::exists(base)) {
+		// Look if 'relativeTo' is a directory already.
+		if (!boost::filesystem::is_directory(base)) {
+			// If not we use the parent directory.
+			base = base.parent_path();
+		}
+
+		// Use the / operator to append the path.
+		base /= path;
+
+		// If we already found a fitting resource there, use that.
+		if (boost::filesystem::exists(base)) {
+			std::string location =
+			    boost::filesystem::canonical(base).generic_string();
+			resource = Resource(true, *this, type, location);
+			return true;
+		}
+	}
+
+	// Otherwise look in the search paths.
+	auto it = searchPaths.find(type);
+	if (it != searchPaths.end()) {
+		for (boost::filesystem::path p : it->second) {
+			p /= path;
+			if (boost::filesystem::exists(p)) {
+				std::string location =
+				    boost::filesystem::canonical(p).generic_string();
+				resource = Resource(true, *this, type, location);
+				return true;
+			}
+		}
+	}
+	return false;
+}
+
+std::unique_ptr<std::istream> FileLocator::doStream(
+    const std::string &location) const
+{
+	std::unique_ptr<std::istream> ifs{new std::ifstream(location)};
+	return std::move(ifs);
+}
+}