357 lines
9.5 KiB
C++
357 lines
9.5 KiB
C++
/// 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);
|
|
}
|