mirror of
https://github.com/Xevion/easy7zip.git
synced 2025-12-06 19:14:56 -06:00
Initialer Commit
This commit is contained in:
Tino Reichardt
356
CPP/7zip/UI/FileManager/PanelCopy.cpp
Normal file
356
CPP/7zip/UI/FileManager/PanelCopy.cpp
Normal file
@@ -0,0 +1,356 @@
|
||||
/// PanelCopy.cpp
|
||||
|
||||
#include "StdAfx.h"
|
||||
|
||||
#include "../../../Common/MyException.h"
|
||||
|
||||
#include "../GUI/HashGUI.h"
|
||||
|
||||
#include "ExtractCallback.h"
|
||||
#include "LangUtils.h"
|
||||
#include "Panel.h"
|
||||
#include "resource.h"
|
||||
#include "UpdateCallback100.h"
|
||||
|
||||
using namespace NWindows;
|
||||
|
||||
class CPanelCopyThread: public CProgressThreadVirt
|
||||
{
|
||||
HRESULT ProcessVirt();
|
||||
public:
|
||||
const CCopyToOptions *options;
|
||||
CMyComPtr<IFolderOperations> FolderOperations;
|
||||
CRecordVector<UInt32> Indices;
|
||||
CExtractCallbackImp *ExtractCallbackSpec;
|
||||
CMyComPtr<IFolderOperationsExtractCallback> ExtractCallback;
|
||||
|
||||
CHashBundle Hash;
|
||||
UString FirstFilePath;
|
||||
|
||||
HRESULT Result;
|
||||
|
||||
|
||||
CPanelCopyThread(): Result(E_FAIL) {}
|
||||
};
|
||||
|
||||
HRESULT CPanelCopyThread::ProcessVirt()
|
||||
{
|
||||
/*
|
||||
CMyComPtr<IFolderSetReplaceAltStreamCharsMode> iReplace;
|
||||
FolderOperations.QueryInterface(IID_IFolderSetReplaceAltStreamCharsMode, &iReplace);
|
||||
if (iReplace)
|
||||
{
|
||||
RINOK(iReplace->SetReplaceAltStreamCharsMode(ReplaceAltStreamChars ? 1 : 0));
|
||||
}
|
||||
*/
|
||||
|
||||
if (options->testMode)
|
||||
{
|
||||
CMyComPtr<IArchiveFolder> archiveFolder;
|
||||
FolderOperations.QueryInterface(IID_IArchiveFolder, &archiveFolder);
|
||||
if (!archiveFolder)
|
||||
return E_NOTIMPL;
|
||||
CMyComPtr<IFolderArchiveExtractCallback> extractCallback2;
|
||||
RINOK(ExtractCallback.QueryInterface(IID_IFolderArchiveExtractCallback, &extractCallback2));
|
||||
NExtract::NPathMode::EEnum pathMode =
|
||||
NExtract::NPathMode::kCurPaths;
|
||||
// NExtract::NPathMode::kFullPathnames;
|
||||
Result = archiveFolder->Extract(&Indices.Front(), Indices.Size(),
|
||||
BoolToInt(options->includeAltStreams),
|
||||
BoolToInt(options->replaceAltStreamChars),
|
||||
pathMode, NExtract::NOverwriteMode::kAsk,
|
||||
options->folder, BoolToInt(true), extractCallback2);
|
||||
}
|
||||
else
|
||||
Result = FolderOperations->CopyTo(
|
||||
BoolToInt(options->moveMode),
|
||||
&Indices.Front(), Indices.Size(),
|
||||
BoolToInt(options->includeAltStreams),
|
||||
BoolToInt(options->replaceAltStreamChars),
|
||||
options->folder, ExtractCallback);
|
||||
|
||||
if (Result == S_OK && !ExtractCallbackSpec->ThereAreMessageErrors &&
|
||||
(!options->hashMethods.IsEmpty() || options->testMode))
|
||||
{
|
||||
CProgressMessageBoxPair &pair = GetMessagePair(false); // GetMessagePair(ExtractCallbackSpec->Hash.NumErrors != 0);
|
||||
AddHashBundleRes(pair.Message, Hash, FirstFilePath);
|
||||
}
|
||||
|
||||
return Result;
|
||||
}
|
||||
|
||||
|
||||
/*
|
||||
#ifdef EXTERNAL_CODECS
|
||||
|
||||
static void ThrowException_if_Error(HRESULT res)
|
||||
{
|
||||
if (res != S_OK)
|
||||
throw CSystemException(res);
|
||||
}
|
||||
|
||||
#endif
|
||||
*/
|
||||
|
||||
|
||||
HRESULT CPanel::CopyTo(CCopyToOptions &options, const CRecordVector<UInt32> &indices,
|
||||
UStringVector *messages,
|
||||
bool &usePassword, UString &password)
|
||||
{
|
||||
if (!_folderOperations)
|
||||
{
|
||||
UString errorMessage = LangString(IDS_OPERATION_IS_NOT_SUPPORTED);
|
||||
if (options.showErrorMessages)
|
||||
MessageBox(errorMessage);
|
||||
else if (messages != 0)
|
||||
messages->Add(errorMessage);
|
||||
return E_FAIL;
|
||||
}
|
||||
|
||||
HRESULT res = S_OK;
|
||||
|
||||
{
|
||||
/*
|
||||
#ifdef EXTERNAL_CODECS
|
||||
CExternalCodecs g_ExternalCodecs;
|
||||
#endif
|
||||
*/
|
||||
/* extracter.Hash uses g_ExternalCodecs
|
||||
extracter must be declared after g_ExternalCodecs for correct destructor order !!! */
|
||||
|
||||
CPanelCopyThread extracter;
|
||||
|
||||
extracter.ExtractCallbackSpec = new CExtractCallbackImp;
|
||||
extracter.ExtractCallback = extracter.ExtractCallbackSpec;
|
||||
|
||||
extracter.options = &options;
|
||||
extracter.ExtractCallbackSpec->ProgressDialog = &extracter.ProgressDialog;
|
||||
extracter.ProgressDialog.CompressingMode = false;
|
||||
|
||||
extracter.ExtractCallbackSpec->StreamMode = options.streamMode;
|
||||
|
||||
|
||||
if (indices.Size() == 1)
|
||||
extracter.FirstFilePath = GetItemRelPath(indices[0]);
|
||||
|
||||
if (options.VirtFileSystem)
|
||||
{
|
||||
extracter.ExtractCallbackSpec->VirtFileSystem = options.VirtFileSystem;
|
||||
extracter.ExtractCallbackSpec->VirtFileSystemSpec = options.VirtFileSystemSpec;
|
||||
}
|
||||
extracter.ExtractCallbackSpec->ProcessAltStreams = options.includeAltStreams;
|
||||
|
||||
if (!options.hashMethods.IsEmpty())
|
||||
{
|
||||
/* this code is used when we call CRC calculation for files in side archive
|
||||
But new code uses global codecs so we don't need to call LoadGlobalCodecs again */
|
||||
|
||||
/*
|
||||
#ifdef EXTERNAL_CODECS
|
||||
ThrowException_if_Error(LoadGlobalCodecs());
|
||||
#endif
|
||||
*/
|
||||
|
||||
extracter.Hash.SetMethods(EXTERNAL_CODECS_VARS_G options.hashMethods);
|
||||
extracter.ExtractCallbackSpec->SetHashMethods(&extracter.Hash);
|
||||
}
|
||||
else if (options.testMode)
|
||||
{
|
||||
extracter.ExtractCallbackSpec->SetHashCalc(&extracter.Hash);
|
||||
}
|
||||
|
||||
extracter.Hash.Init();
|
||||
|
||||
UString title;
|
||||
{
|
||||
UInt32 titleID = IDS_COPYING;
|
||||
if (options.moveMode)
|
||||
titleID = IDS_MOVING;
|
||||
else if (!options.hashMethods.IsEmpty() && options.streamMode)
|
||||
{
|
||||
titleID = IDS_CHECKSUM_CALCULATING;
|
||||
if (options.hashMethods.Size() == 1)
|
||||
{
|
||||
const UString &s = options.hashMethods[0];
|
||||
if (s != L"*")
|
||||
title = s;
|
||||
}
|
||||
}
|
||||
else if (options.testMode)
|
||||
titleID = IDS_PROGRESS_TESTING;
|
||||
|
||||
if (title.IsEmpty())
|
||||
title = LangString(titleID);
|
||||
}
|
||||
|
||||
UString progressWindowTitle = L"7-Zip"; // LangString(IDS_APP_TITLE);
|
||||
|
||||
extracter.ProgressDialog.MainWindow = GetParent();
|
||||
extracter.ProgressDialog.MainTitle = progressWindowTitle;
|
||||
extracter.ProgressDialog.MainAddTitle = title + L' ';
|
||||
|
||||
extracter.ExtractCallbackSpec->OverwriteMode = NExtract::NOverwriteMode::kAsk;
|
||||
extracter.ExtractCallbackSpec->Init();
|
||||
extracter.Indices = indices;
|
||||
extracter.FolderOperations = _folderOperations;
|
||||
|
||||
extracter.ExtractCallbackSpec->PasswordIsDefined = usePassword;
|
||||
extracter.ExtractCallbackSpec->Password = password;
|
||||
|
||||
RINOK(extracter.Create(title, GetParent()));
|
||||
|
||||
if (messages != 0)
|
||||
*messages = extracter.ProgressDialog.Sync.Messages;
|
||||
res = extracter.Result;
|
||||
|
||||
if (res == S_OK && extracter.ExtractCallbackSpec->IsOK())
|
||||
{
|
||||
usePassword = extracter.ExtractCallbackSpec->PasswordIsDefined;
|
||||
password = extracter.ExtractCallbackSpec->Password;
|
||||
}
|
||||
}
|
||||
|
||||
RefreshTitleAlways();
|
||||
return res;
|
||||
}
|
||||
|
||||
|
||||
struct CThreadUpdate
|
||||
{
|
||||
CMyComPtr<IFolderOperations> FolderOperations;
|
||||
UString FolderPrefix;
|
||||
UStringVector FileNames;
|
||||
CRecordVector<const wchar_t *> FileNamePointers;
|
||||
CProgressDialog ProgressDialog;
|
||||
CMyComPtr<IFolderArchiveUpdateCallback> UpdateCallback;
|
||||
CUpdateCallback100Imp *UpdateCallbackSpec;
|
||||
HRESULT Result;
|
||||
bool MoveMode;
|
||||
|
||||
void Process()
|
||||
{
|
||||
try
|
||||
{
|
||||
CProgressCloser closer(ProgressDialog);
|
||||
Result = FolderOperations->CopyFrom(
|
||||
MoveMode,
|
||||
FolderPrefix,
|
||||
&FileNamePointers.Front(),
|
||||
FileNamePointers.Size(),
|
||||
UpdateCallback);
|
||||
}
|
||||
catch(...) { Result = E_FAIL; }
|
||||
}
|
||||
static THREAD_FUNC_DECL MyThreadFunction(void *param)
|
||||
{
|
||||
((CThreadUpdate *)param)->Process();
|
||||
return 0;
|
||||
}
|
||||
};
|
||||
|
||||
HRESULT CPanel::CopyFrom(bool moveMode, const UString &folderPrefix, const UStringVector &filePaths,
|
||||
bool showErrorMessages, UStringVector *messages)
|
||||
{
|
||||
HRESULT res;
|
||||
if (!_folderOperations)
|
||||
res = E_NOINTERFACE;
|
||||
else
|
||||
{
|
||||
CThreadUpdate updater;
|
||||
updater.MoveMode = moveMode;
|
||||
updater.UpdateCallbackSpec = new CUpdateCallback100Imp;
|
||||
updater.UpdateCallback = updater.UpdateCallbackSpec;
|
||||
updater.UpdateCallbackSpec->Init();
|
||||
|
||||
updater.UpdateCallbackSpec->ProgressDialog = &updater.ProgressDialog;
|
||||
|
||||
UString title = LangString(IDS_COPYING);
|
||||
UString progressWindowTitle = L"7-Zip"; // LangString(IDS_APP_TITLE);
|
||||
|
||||
updater.ProgressDialog.MainWindow = GetParent();
|
||||
updater.ProgressDialog.MainTitle = progressWindowTitle;
|
||||
updater.ProgressDialog.MainAddTitle = title + L' ';
|
||||
|
||||
{
|
||||
if (!_parentFolders.IsEmpty())
|
||||
{
|
||||
const CFolderLink &fl = _parentFolders.Back();
|
||||
updater.UpdateCallbackSpec->PasswordIsDefined = fl.UsePassword;
|
||||
updater.UpdateCallbackSpec->Password = fl.Password;
|
||||
}
|
||||
}
|
||||
|
||||
updater.FolderOperations = _folderOperations;
|
||||
updater.FolderPrefix = folderPrefix;
|
||||
updater.FileNames.ClearAndReserve(filePaths.Size());
|
||||
unsigned i;
|
||||
for (i = 0; i < filePaths.Size(); i++)
|
||||
updater.FileNames.AddInReserved(filePaths[i]);
|
||||
updater.FileNamePointers.ClearAndReserve(updater.FileNames.Size());
|
||||
for (i = 0; i < updater.FileNames.Size(); i++)
|
||||
updater.FileNamePointers.AddInReserved(updater.FileNames[i]);
|
||||
|
||||
NWindows::CThread thread;
|
||||
RINOK(thread.Create(CThreadUpdate::MyThreadFunction, &updater));
|
||||
updater.ProgressDialog.Create(title, thread, GetParent());
|
||||
|
||||
if (messages != 0)
|
||||
*messages = updater.ProgressDialog.Sync.Messages;
|
||||
|
||||
res = updater.Result;
|
||||
}
|
||||
|
||||
if (res == E_NOINTERFACE)
|
||||
{
|
||||
UString errorMessage = LangString(IDS_OPERATION_IS_NOT_SUPPORTED);
|
||||
if (showErrorMessages)
|
||||
MessageBox(errorMessage);
|
||||
else if (messages != 0)
|
||||
messages->Add(errorMessage);
|
||||
return E_ABORT;
|
||||
}
|
||||
|
||||
RefreshTitleAlways();
|
||||
return res;
|
||||
}
|
||||
|
||||
void CPanel::CopyFromNoAsk(const UStringVector &filePaths)
|
||||
{
|
||||
CDisableTimerProcessing disableTimerProcessing(*this);
|
||||
|
||||
CSelectedState srcSelState;
|
||||
SaveSelectedState(srcSelState);
|
||||
|
||||
HRESULT result = CopyFrom(false, L"", filePaths, true, 0);
|
||||
|
||||
CDisableNotify disableNotify(*this);
|
||||
|
||||
if (result != S_OK)
|
||||
{
|
||||
disableNotify.Restore();
|
||||
// For Password:
|
||||
SetFocusToList();
|
||||
if (result != E_ABORT)
|
||||
MessageBoxError(result);
|
||||
return;
|
||||
}
|
||||
|
||||
RefreshListCtrl(srcSelState);
|
||||
|
||||
disableNotify.Restore();
|
||||
SetFocusToList();
|
||||
}
|
||||
|
||||
void CPanel::CopyFromAsk(const UStringVector &filePaths)
|
||||
{
|
||||
UString title = LangString(IDS_CONFIRM_FILE_COPY);
|
||||
UString message = LangString(IDS_WANT_TO_COPY_FILES);
|
||||
message += L"\n\'";
|
||||
message += _currentFolderPrefix;
|
||||
message += L"\' ?";
|
||||
int res = ::MessageBoxW(*(this), message, title, MB_YESNOCANCEL | MB_ICONQUESTION);
|
||||
if (res != IDYES)
|
||||
return;
|
||||
|
||||
CopyFromNoAsk(filePaths);
|
||||
}
|
||||
Reference in New Issue
Block a user