/*
 * PROJECT:     ReactOS Zip Shell Extension
 * LICENSE:     GPL-2.0+ (https://spdx.org/licenses/GPL-2.0+)
 * PURPOSE:     Ask the user for a password
 * COPYRIGHT:   Copyright 2019 Mark Jansen (mark.jansen@reactos.org)
 */

#include "precomp.h"

class CZipPassword : public CDialogImpl<CZipPassword>
{
private:
    CStringA m_Filename;
    CStringA* m_pPassword;
public:
    CZipPassword(const char* filename, CStringA* Password)
        :m_pPassword(Password)
    {
        if (filename != NULL)
            m_Filename = filename;
    }

    LRESULT OnInitDialog(UINT nMsg, WPARAM wParam, LPARAM lParam, BOOL& bHandled)
    {
        CenterWindow(GetParent());

        /* No filename, so this is the question before starting to extract */
        if (m_Filename.IsEmpty())
        {
            CStringA message(MAKEINTRESOURCE(IDS_PASSWORD_ZIP_TEXT));
            ::SetDlgItemTextA(m_hWnd, IDC_MESSAGE, message);
            ::ShowWindow(GetDlgItem(IDSKIP), SW_HIDE);
        }
        else
        {
            /* Our CString does not support FormatMessage yet */
            CStringA message(MAKEINTRESOURCE(IDS_PASSWORD_FILE_TEXT));
            CHeapPtr<CHAR, CLocalAllocator> formatted;

            DWORD_PTR args[2] =
            {
                (DWORD_PTR)m_Filename.GetString(),
                NULL
            };

            ::FormatMessageA(FORMAT_MESSAGE_FROM_STRING | FORMAT_MESSAGE_ALLOCATE_BUFFER | FORMAT_MESSAGE_ARGUMENT_ARRAY,
                             message, 0, 0, (LPSTR)&formatted, 0, (va_list*)args);

            ::SetDlgItemTextA(m_hWnd, IDC_MESSAGE, formatted);
        }
        return TRUE;
    }

    LRESULT OnButton(WORD wNotifyCode, WORD wID, HWND hWndCtl, BOOL& bHandled)
    {
        if (wID == IDOK)
        {
            HWND item = GetDlgItem(IDC_PASSWORD);
            int len = ::GetWindowTextLengthA(item);
            len = ::GetDlgItemTextA(m_hWnd, IDC_PASSWORD, m_pPassword->GetBuffer(len+1), len+1);
            m_pPassword->ReleaseBuffer(len);
        }
        EndDialog(wID);
        return 0;
    }

public:
    enum { IDD = IDD_PASSWORD };

    BEGIN_MSG_MAP(CZipPassword)
        MESSAGE_HANDLER(WM_INITDIALOG, OnInitDialog)
        COMMAND_ID_HANDLER(IDOK, OnButton)
        COMMAND_ID_HANDLER(IDSKIP, OnButton)
        COMMAND_ID_HANDLER(IDCANCEL, OnButton)
    END_MSG_MAP()
};

eZipPasswordResponse _CZipAskPassword(HWND hDlg, const char* filename, CStringA& Password)
{
    if (filename)
        filename = PathFindFileNameA(filename);
    CZipPassword password(filename, &Password);
    INT_PTR Result = password.DoModal(hDlg);
    switch (Result)
    {
    case IDOK: return eAccept;
    case IDSKIP: return eSkip;
    default:
    case IDCANCEL: return eAbort;
    }
}