GEEKING

Installer/Bootstrapper/FileSystem.cpp

@@ -0,0 +1,157 @@
+#include "StdAfx.h"
+#include "FileSystem.h"
+#include "format_string.h"
+#include "VistaTools.h"
+
+#include "ATLPath.h"
+
+
+std::wstring FileSystem::getSpecialFolder(FolderType folder, bool create, const char* subDirectory, bool appendRoblox)
+{
+	std::string robloxDir = "";
+	if (appendRoblox)
+	{
+		robloxDir = "\\Watrbx";
+	}
+		
+	switch (folder)
+	{
+	case RobloxUserPrograms:
+	case RobloxProgramFiles:
+		// TODO: Nuke "(Test)" when we go to production
+		//robloxDir += " (Test)";
+		break;
+	}
+	
+	VistaAPIs apis;
+	if (apis.isVistaOrBetter())
+	{
+		CLSID folderID;
+		switch (folder)
+		{
+		case AppData:
+		case RobloxUserApplicationData: // %LOCALAPPDATA% (%USERPROFILE%\AppData\Local)
+			folderID = FOLDERID_LocalAppData;
+			break;
+		case RobloxUserApplicationDataLow: // %USERPROFILE%\AppData\LocalLow
+			folderID = FOLDERID_LocalAppDataLow;
+			break;
+		case RobloxCommonApplicationData: // %ALLUSERSPROFILE% (%ProgramData%, %SystemDrive%\ProgramData)
+			folderID = FOLDERID_ProgramData;
+			break;
+		case RobloxUserPrograms: // %APPDATA%\Microsoft\Windows\Start Menu\Programs
+			folderID = FOLDERID_Programs;
+			break;
+		case RobloxProgramFiles: // %ProgramFiles% (%SystemDrive%\Program Files)
+			folderID = FOLDERID_ProgramFilesX86;
+			break;
+		case RobloxCommonPrograms: //%ALLUSERSPROFILE%\Microsoft\Windows\Start Menu\Programs	
+			folderID = FOLDERID_CommonPrograms;
+			break;
+		case Desktop: // %PUBLIC%\Desktop
+			folderID = FOLDERID_Desktop;
+			break;
+		case RoamingAppData: // %APPDATA% (%USERPROFILE%\AppData\Roaming)
+			folderID = FOLDERID_RoamingAppData;
+			break;
+		default:
+			ATLASSERT(false);
+			break;
+		}
+		
+		LPWSTR lppwstrPath = NULL;
+		if (apis.SHGetKnownFolderPath(folderID, create ? KF_FLAG_CREATE : 0, NULL, &lppwstrPath)!=E_NOTIMPL)
+		{
+			CString s = lppwstrPath;
+
+			s += robloxDir.c_str();
+			s += "\\";
+			if (create)
+				if (!::CreateDirectory(s, NULL) && ::GetLastError()!=ERROR_ALREADY_EXISTS)
+					throw std::runtime_error(format_string("Failed to create directory '%S', error=%d",s,::GetLastError()));
+
+			if (subDirectory)
+			{
+				s += subDirectory;
+				if (create)
+					if (!::CreateDirectory(s, NULL) && ::GetLastError()!=ERROR_ALREADY_EXISTS)
+						throw std::runtime_error(format_string("Failed to create subdirectory '%S', error=%d",s,::GetLastError()));
+			}
+			ATL::CPath path(s);
+			path.AddBackslash();
+			return std::wstring(path);
+		}
+	}
+
+	DWORD flags = create ? CSIDL_FLAG_CREATE : 0;
+	std::wstring s = convert_s2w(robloxDir);
+	if (subDirectory)
+		s += convert_s2w(subDirectory);
+
+	int csidl;
+	switch (folder)
+	{
+	case RobloxUserApplicationData:
+		csidl = CSIDL_LOCAL_APPDATA;
+		break;
+	case RobloxUserApplicationDataLow:
+		csidl = CSIDL_LOCAL_APPDATA;
+		break;
+	case RobloxCommonApplicationData:
+		csidl = CSIDL_COMMON_APPDATA;
+		break;
+	case RobloxUserPrograms:
+		csidl = CSIDL_PROGRAMS;
+		break;
+	case RobloxCommonPrograms:
+		csidl = CSIDL_COMMON_PROGRAMS;
+		break;
+	case RobloxProgramFiles:
+		csidl = CSIDL_PROGRAM_FILES;
+		break;
+	case Desktop:
+		csidl = CSIDL_DESKTOP;
+		break;
+	case AppData:
+		csidl = CSIDL_APPDATA;
+		break;
+	default:
+		ATLASSERT(false);
+		break;
+	}
+
+	wchar_t strPath[_MAX_PATH];
+	HRESULT hr = SHGetFolderPathAndSubDirW(NULL, csidl | flags, NULL, SHGFP_TYPE_CURRENT, s.c_str(), strPath);	
+
+	if (hr==S_OK)
+	{
+		CString s = strPath;
+
+		ATL::CPath path(s);
+		path.AddBackslash();
+		return std::wstring(path);
+	}
+	else{
+		throw std::runtime_error(format_string("Bad result from SHGetFolderPathAndSubDirW=%d",hr));
+	}
+}
+
+bool FileSystem::IsFileExists(const TCHAR* name)
+{
+	HANDLE file = CreateFile(name, GENERIC_READ, FILE_SHARE_READ, NULL, OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, NULL);
+	bool exists = false;
+	if (file != INVALID_HANDLE_VALUE) 
+	{
+		exists = true;
+		CloseHandle(file);
+	}
+
+	return exists;
+}
+
+bool FileSystem::IsFolderExists(const TCHAR* name)
+{
+	DWORD dwAttrib = GetFileAttributes(name);
+	return (dwAttrib != INVALID_FILE_ATTRIBUTES && (dwAttrib & FILE_ATTRIBUTE_DIRECTORY));
+}
+