1 files changed, 117 insertions, 8 deletions
diff --git a/src/util.cc b/src/util.cc
index 92560b6..76d380b 100644
--- a/src/util.cc
+++ b/src/util.cc
@@ -5,6 +5,14 @@
 
 #include <iostream>
 #include <fstream>
+#include <algorithm>
+
+std::string to_lower(const std::string& str)
+{
+	std::string out{str};
+	std::transform(out.begin(), out.end(), out.begin(), ::tolower);
+	return out;
+}
 
 std::string readFile(const std::string& fileName)
 {
@@ -80,30 +88,37 @@ std::vector<std::string> readDeps(const std::string& depFile)
 	return output;
 }
 
-Language languageFromExtension(const std::filesystem::path& file)
+ctor::language languageFromExtension(const std::filesystem::path& file)
 {
 	auto ext = file.extension().string();
+
+	// First a few case sensitive comparisons
 	if(ext == ".c")
 	{
-		return Language::C;
+		return ctor::language::c;
+	}
+
+	if(ext == ".C")
+	{
+		return ctor::language::cpp;
 	}
 
-	if(ext == ".C" ||
-	   ext == ".cc" ||
+	// The rest are compared in lowercase
+	ext = to_lower(ext);
+
+	if(ext == ".cc" ||
 	   ext == ".cpp" ||
-	   ext == ".CPP" ||
 	   ext == ".c++" ||
 	   ext == ".cp" ||
 	   ext == ".cxx")
 	{
-		return Language::Cpp;
+		return ctor::language::cpp;
 	}
 
 	if(ext == ".s" ||
-	   ext == ".S" ||
 	   ext == ".asm")
 	{
-		return Language::Asm;
+		return ctor::language::assembler;
 	}
 
 	std::cerr << "Could not deduce language from " << file.string() << "\n";
@@ -135,3 +150,97 @@ std::string cleanUp(const std::string& path)
 	}
 	return cleaned;
 }
+
+std::string esc(const std::string& in)
+{
+	std::string out;
+	for(auto c : in)
+	{
+		switch(c)
+		{
+		case '\\': out += "\\\\"; break;
+		case '"': out += "\\\""; break;
+		default:
+			out += c;
+			break;
+		}
+	}
+	return out;
+}
+
+std::vector<std::string> get_paths(const std::string& path_env)
+{
+	std::vector<std::string> paths;
+
+#ifdef _WIN32
+	const char sep{';'};
+#else
+	const char sep{':'};
+#endif
+
+	std::stringstream ss(path_env);
+	std::string path;
+	while (std::getline(ss, path, sep))
+	{
+		paths.push_back(path);
+	}
+
+	return paths;
+}
+
+bool check_executable(const std::filesystem::path& prog)
+{
+	auto perms = std::filesystem::status(prog).permissions();
+
+	if((perms & std::filesystem::perms::owner_exec) != std::filesystem::perms::none)
+	{
+		return true;
+	}
+
+	if((perms & std::filesystem::perms::group_exec) != std::filesystem::perms::none)
+	{
+		return true;
+	}
+
+	if((perms & std::filesystem::perms::others_exec) != std::filesystem::perms::none)
+	{
+		return true;
+	}
+
+	return false;
+}
+
+std::string locate(const std::string& prog,
+                   const std::vector<std::string>& paths,
+                   const std::string& arch)
+{
+	std::string program = prog;
+	if(!arch.empty())
+	{
+		program = arch + "-" + prog;
+	}
+
+	// first check if arch contains an absolute path to prog
+	if(std::filesystem::exists(program))
+	{
+		if(check_executable(program))
+		{
+			return program;
+		}
+	}
+
+	for(const auto& path_str : paths)
+	{
+		std::filesystem::path path(path_str);
+		auto prog_path = path / program;
+		if(std::filesystem::exists(prog_path))
+		{
+			if(check_executable(prog_path))
+			{
+				return prog_path.string();
+			}
+		}
+	}
+
+	return {};
+}