Move path related functions into PathUtil

Common/PathUtil.cpp

@@ -0,0 +1,150 @@
+/*
+  Copyright (C) 2013 Rainmeter Team
+
+  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 2
+  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, write to the Free Software
+  Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.
+*/
+
+#include "PathUtil.h"
+#include <Shlobj.h>
+
+namespace PathUtil {
+
+bool IsSeparator(WCHAR ch)
+{
+	return ch == L'\\' || ch == L'/';
+}
+
+bool IsDotOrDotDot(const WCHAR* path)
+{
+	return path[0] == L'.' && (path[1] == L'\0' || (path[1] == L'.' && path[2] == L'\0'));
+}
+
+bool IsUNC(const std::wstring& path)
+{
+	return path.length() >= 2 && IsSeparator(path[0]) && IsSeparator(path[1]);
+}
+
+bool IsAbsolute(const std::wstring& path)
+{
+	return (path.find(L":\\") != std::wstring::npos ||
+		path.find(L":/") != std::wstring::npos ||
+		IsUNC(path));
+}
+
+void AppendBacklashIfMissing(std::wstring& path)
+{
+	if (!path.empty() && !IsSeparator(path[path.length() - 1]))
+	{
+		path += L'\\';
+	}
+}
+
+std::wstring GetFolderFromFilePath(const std::wstring& filePath)
+{
+	std::wstring::size_type pos = filePath.find_last_of(L"\\/");
+	if (pos != std::wstring::npos)
+	{
+		return filePath.substr(0, pos + 1);
+	}
+	return L".\\";
+}
+
+/*
+** Extracts volume path from program path.
+** E.g.:
+**   "C:\path\" to "C:"
+**   "\\server\share\" to "\\server\share"
+**   "\\server\C:\path\" to "\\server\C:"
+*/
+std::wstring GetVolume(const std::wstring& path)
+{
+	std::wstring::size_type pos;
+	if ((pos = path.find_first_of(L':')) != std::wstring::npos)
+	{
+		return path.substr(0, pos + 1);
+	}
+	else if (IsUNC(path))
+	{
+		if ((pos = path.find_first_of(L"\\/", 2)) != std::wstring::npos)
+		{
+			std::wstring::size_type pos2;
+			if ((pos2 = path.find_first_of(L"\\/", pos + 1)) != std::wstring::npos ||
+				pos != (path.length() - 1))
+			{
+				pos = pos2;
+			}
+		}
+
+		return path.substr(0, pos);
+	}
+
+	return std::wstring();
+}
+
+void ExpandEnvironmentVariables(std::wstring& path)
+{
+	std::wstring::size_type pos;
+	if ((pos = path.find(L'%')) != std::wstring::npos &&
+		path.find(L'%', pos + 2) != std::wstring::npos)
+	{
+		DWORD bufSize = 4096;
+		WCHAR* buffer = new WCHAR[bufSize];
+
+		// %APPDATA% is a special casem
+		pos = path.find(L"%APPDATA%", pos);
+		if (pos != std::wstring::npos)
+		{
+			HRESULT hr = SHGetFolderPath(nullptr, CSIDL_APPDATA, nullptr, SHGFP_TYPE_CURRENT, buffer);
+			if (SUCCEEDED(hr))
+			{
+				size_t len = wcslen(buffer);
+				do
+				{
+					path.replace(pos, 9, buffer, len);
+				}
+				while ((pos = path.find(L"%APPDATA%", pos + len)) != std::wstring::npos);
+			}
+		}
+
+		if ((pos = path.find(L'%')) != std::wstring::npos &&
+			path.find(L'%', pos + 2) != std::wstring::npos)
+		{
+			// Expand the environment variables.
+			do
+			{
+				DWORD ret = ExpandEnvironmentStrings(path.c_str(), buffer, bufSize);
+				if (ret == 0)  // Error
+				{
+					break;
+				}
+				if (ret <= bufSize)  // Fits in the buffer
+				{
+					path.assign(buffer, ret - 1);
+					break;
+				}
+
+				delete [] buffer;
+				bufSize = ret;
+				buffer = new WCHAR[bufSize];
+			}
+			while (true);
+		}
+
+		delete [] buffer;
+	}
+}
+
+
+}  // namespace PathUtil

Common/PathUtil.h

@@ -0,0 +1,45 @@
+/*
+  Copyright (C) 2013 Rainmeter Team
+
+  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 2
+  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, write to the Free Software
+  Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.
+*/
+
+#ifndef RM_COMMON_PATHUTIL_H_
+#define RM_COMMON_PATHUTIL_H_
+
+#include <Windows.h>
+#include <string>
+
+namespace PathUtil {
+
+bool IsSeparator(WCHAR ch);
+
+bool IsDotOrDotDot(const WCHAR* path);
+
+bool IsUNC(const std::wstring& path);
+
+bool IsAbsolute(const std::wstring& path);
+
+void AppendBacklashIfMissing(std::wstring& path);
+
+std::wstring GetFolderFromFilePath(const std::wstring& filePath);
+
+std::wstring GetVolume(const std::wstring& path);
+
+void ExpandEnvironmentVariables(std::wstring& strPath);
+
+}  // namespace PathUtil
+
+#endif