// TortoiseSVN - a Windows shell extension for easy version control

// Copyright (C) 2007-2012, 2014-2015, 2017, 2021 - TortoiseSVN

// 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,
// 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
//
#include "stdafx.h"
#include "DropCopyAddCommand.h"
#include "FormatMessageWrapper.h"
#include "SVNProgressDlg.h"
#include "StringUtils.h"
#include "TortoiseProc.h"

bool DropCopyAddCommand::Execute()
{
    bool    bRet     = false;
    CString dropPath = parser.GetVal(L"droptarget");
    if (CTSVNPath(dropPath).IsAdminDir())
        return FALSE;

    pathList.RemoveAdminPaths();
    CTSVNPathList copiedFiles;
    UINT          defaultRet = 0;
    for (int nPath = 0; nPath < pathList.GetCount(); nPath++)
    {
        if (pathList[nPath].IsEquivalentTo(CTSVNPath(dropPath)))
            continue;

        //copy the file to the new location
        CString name = pathList[nPath].GetFileOrDirectoryName();
        if (::PathFileExists(dropPath + L"\\" + name))
        {
            if (::PathIsDirectory(dropPath + L"\\" + name))
            {
                if (pathList[nPath].IsDirectory())
                    continue;
            }

            UINT    ret = defaultRet;
            CString strMessage;
            strMessage.Format(IDS_PROC_OVERWRITE_CONFIRM, static_cast<LPCWSTR>(dropPath + L"\\" + name));
            if (defaultRet == 0)
            {
                CTaskDialog taskDlg(strMessage,
                                    CString(MAKEINTRESOURCE(IDS_PROC_OVERWRITE_CONFIRM_TASK2)),
                                    L"TortoiseSVN",
                                    0,
                                    TDF_ENABLE_HYPERLINKS | TDF_USE_COMMAND_LINKS | TDF_ALLOW_DIALOG_CANCELLATION | TDF_POSITION_RELATIVE_TO_WINDOW | TDF_SIZE_TO_CONTENT);
                taskDlg.AddCommandControl(IDYES, CString(MAKEINTRESOURCE(IDS_PROC_OVERWRITE_CONFIRM_TASK3)));
                taskDlg.AddCommandControl(IDCANCEL, CString(MAKEINTRESOURCE(IDS_PROC_OVERWRITE_CONFIRM_TASK4)));
                taskDlg.SetDefaultCommandControl(IDCANCEL);
                taskDlg.SetCommonButtons(TDCBF_CANCEL_BUTTON);
                taskDlg.SetVerificationCheckboxText(CString(MAKEINTRESOURCE(IDS_PROC_OVERWRITE_CONFIRM_TASK5)));
                taskDlg.SetMainIcon(TD_WARNING_ICON);
                ret = static_cast<UINT>(taskDlg.DoModal(GetExplorerHWND()));
                if (taskDlg.GetVerificationCheckboxState())
                    defaultRet = ret;
            }

            if (ret == IDCANCEL)
            {
                return FALSE; //cancel the whole operation
            }
            if (ret == IDYES)
            {
                if (!::CopyFile(pathList[nPath].GetWinPath(), dropPath + L"\\" + name, FALSE))
                {
                    //the copy operation failed! Get out of here!
                    ShowErrorMessage();
                    return FALSE;
                }
            }
        }
        else
        {
            if (pathList[nPath].IsDirectory())
            {
                CString fromPath = pathList[nPath].GetWinPathString() + L"||";
                CString toPath   = dropPath + L"\\" + name + L"||";
                auto    fromBuf  = std::make_unique<wchar_t[]>(fromPath.GetLength() + 2LL);
                auto    toBuf    = std::make_unique<wchar_t[]>(toPath.GetLength() + 2LL);
                wcscpy_s(fromBuf.get(), fromPath.GetLength() + 2LL, fromPath);
                wcscpy_s(toBuf.get(), toPath.GetLength() + 2LL, toPath);
                CStringUtils::PipesToNulls(fromBuf.get(), fromPath.GetLength() + 2LL);
                CStringUtils::PipesToNulls(toBuf.get(), toPath.GetLength() + 2LL);

                SHFILEOPSTRUCT fileOp = {nullptr};
                fileOp.wFunc          = FO_COPY;
                fileOp.pFrom          = fromBuf.get();
                fileOp.pTo            = toBuf.get();
                fileOp.fFlags         = FOF_NO_CONNECTED_ELEMENTS | FOF_NOCONFIRMATION | FOF_NOERRORUI | FOF_NOCONFIRMMKDIR | FOF_NOCOPYSECURITYATTRIBS | FOF_SILENT;
                SHFileOperation(&fileOp);
            }
            else if (!CopyFile(pathList[nPath].GetWinPath(), dropPath + L"\\" + name, FALSE))
            {
                //the copy operation failed! Get out of here!
                ShowErrorMessage();
                return FALSE;
            }
        }
        copiedFiles.AddPath(CTSVNPath(dropPath + L"\\" + name)); //add the new filepath
    }
    //now add all the newly copied files to the working copy
    CSVNProgressDlg progDlg;
    theApp.m_pMainWnd = &progDlg;
    progDlg.SetCommand(CSVNProgressDlg::SVNProgress_Add);
    progDlg.SetAutoClose(parser);
    progDlg.SetPathList(copiedFiles);
    ProjectProperties props;
    props.ReadProps(copiedFiles.GetCommonRoot());
    progDlg.SetProjectProperties(props);
    progDlg.DoModal();
    bRet = !progDlg.DidErrorsOccur();
    return bRet;
}

void DropCopyAddCommand::ShowErrorMessage() const
{
    CFormatMessageWrapper errorDetails;
    CString               strMessage;
    strMessage.Format(IDS_ERR_COPYFILES, static_cast<LPCWSTR>(errorDetails));
    MessageBox(GetExplorerHWND(), strMessage, L"TortoiseSVN", MB_OK | MB_ICONINFORMATION);
}