mame/3rdparty/lzma/CPP/7zip/UI/Common/WorkDir.cpp
2023-12-06 07:58:49 +11:00

87 lines
2.2 KiB
C++

// WorkDir.cpp
#include "StdAfx.h"
#include "../../../Windows/FileName.h"
#include "../../../Windows/FileSystem.h"
#include "WorkDir.h"
using namespace NWindows;
using namespace NFile;
using namespace NDir;
FString GetWorkDir(const NWorkDir::CInfo &workDirInfo, const FString &path, FString &fileName)
{
NWorkDir::NMode::EEnum mode = workDirInfo.Mode;
#if defined(_WIN32) && !defined(UNDER_CE)
if (workDirInfo.ForRemovableOnly)
{
mode = NWorkDir::NMode::kCurrent;
const FString prefix = path.Left(3);
if (NName::IsDrivePath(prefix))
{
const UINT driveType = NSystem::MyGetDriveType(prefix);
if (driveType == DRIVE_CDROM || driveType == DRIVE_REMOVABLE)
mode = workDirInfo.Mode;
}
/*
CParsedPath parsedPath;
parsedPath.ParsePath(archiveName);
UINT driveType = GetDriveType(parsedPath.Prefix);
if ((driveType != DRIVE_CDROM) && (driveType != DRIVE_REMOVABLE))
mode = NZipSettings::NWorkDir::NMode::kCurrent;
*/
}
#endif
const int pos = path.ReverseFind_PathSepar() + 1;
fileName = path.Ptr((unsigned)pos);
FString tempDir;
switch ((int)mode)
{
case NWorkDir::NMode::kCurrent:
tempDir = path.Left((unsigned)pos);
break;
case NWorkDir::NMode::kSpecified:
tempDir = workDirInfo.Path;
break;
// case NWorkDir::NMode::kSystem:
default:
if (!MyGetTempPath(tempDir))
throw 141717;
break;
}
NName::NormalizeDirPathPrefix(tempDir);
return tempDir;
}
HRESULT CWorkDirTempFile::CreateTempFile(const FString &originalPath)
{
NWorkDir::CInfo workDirInfo;
workDirInfo.Load();
FString namePart;
const FString workDir = GetWorkDir(workDirInfo, originalPath, namePart);
CreateComplexDir(workDir);
_outStreamSpec = new COutFileStream;
OutStream = _outStreamSpec;
if (!_tempFile.Create(workDir + namePart, &_outStreamSpec->File))
{
return GetLastError_noZero_HRESULT();
}
_originalPath = originalPath;
return S_OK;
}
HRESULT CWorkDirTempFile::MoveToOriginal(bool deleteOriginal)
{
OutStream.Release();
if (!_tempFile.MoveTo(_originalPath, deleteOriginal))
{
return GetLastError_noZero_HRESULT();
}
return S_OK;
}