/*
 * Copyright 2015 Mark Jansen
 *
 * This library is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either
 * version 2.1 of the License, or (at your option) any later version.
 *
 * This library 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
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
 * License along with this library; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
 */

#include "precomp.h"
#include <windowsx.h>
#include <sfc.h>

const GUID CLSID_CLayerUIPropPage = { 0x513D916F, 0x2A8E, 0x4F51, { 0xAE, 0xAB, 0x0C, 0xBC, 0x76, 0xFB, 0x1A, 0xF8 } };
#define ACP_WNDPROP L"{513D916F-2A8E-4F51-AEAB-0CBC76FB1AF8}.Prop"

#define GPLK_USER 1
#define GPLK_MACHINE 2
#define MAX_LAYER_LENGTH 256

void ACDBG_FN(PCSTR FunctionName, PCWSTR Format, ...)
{
    WCHAR Buffer[512];
    WCHAR* Current = Buffer;
    size_t Length = _countof(Buffer);

    StringCchPrintfExW(Current, Length, &Current, &Length, STRSAFE_NULL_ON_FAILURE, L"[%-20S] ", FunctionName);
    va_list ArgList;
    va_start(ArgList, Format);
    StringCchVPrintfExW(Current, Length, &Current, &Length, STRSAFE_NULL_ON_FAILURE, Format, ArgList);
    va_end(ArgList);
    OutputDebugStringW(Buffer);
}

#define ACDBG(fmt, ...)  ACDBG_FN(__FUNCTION__, fmt, ##__VA_ARGS__ )



CLayerUIPropPage::CLayerUIPropPage()
:m_Filename(NULL)
, m_IsSfcProtected(FALSE)
, m_AllowPermLayer(FALSE)
, m_LayerQueryFlags(GPLK_USER)
, m_RegistryOSMode(0)
, m_OSMode(0)
, m_RegistryEnabledLayers(0)
, m_EnabledLayers(0)
{
}

CLayerUIPropPage::~CLayerUIPropPage()
{
}


#if 0
HKCU\Software\Microsoft\Windows NT\CurrentVersion\AppCompatFlags\Layers
WINXPSP3 256COLOR 640X480 DISABLETHEMES DISABLEDWM HIGHDPIAWARE RUNASADMIN
#endif

static struct {
    const PCWSTR Display;
    const PCWSTR Name;
} g_CompatModes[] = {
    { L"Windows 95", L"WIN95" },
    { L"Windows 98", L"WIN98" },
    { L"Windows NT 4.0 (SP5)", L"NT4SP5" },
    { L"Windows 2000", L"WIN2000" },
    { L"Windows XP (SP2)", L"WINXPSP2" },
    { L"Windows XP (SP3)", L"WINXPSP3" },
    { L"Windows Server 2003 (SP1)", L"WINSRV03SP1" },
#if 0
    { L"Windows Server 2008 (SP1)", L"WINSRV08SP1" },
    { L"Windows Vista", L"VISTARTM" },
    { L"Windows Vista (SP1)", L"VISTASP1" },
    { L"Windows Vista (SP2)", L"VISTASP2" },
    { L"Windows 7", L"WIN7RTM" },
#endif
    { NULL, NULL }
};

static struct {
    const PCWSTR Name;
    DWORD Id;
    BOOL Enabled;
} g_Layers[] = {
    { L"256COLOR", IDC_CHKRUNIN256COLORS, TRUE },
    { L"640X480", IDC_CHKRUNIN640480RES, TRUE },
    { L"DISABLETHEMES", IDC_CHKDISABLEVISUALTHEMES, TRUE },
    { NULL, 0, FALSE }
};

static const WCHAR* g_AllowedExtensions[] = {
    L".exe",
    L".msi",
    L".pif",
    L".bat",
    L".cmd",
    0
};

HRESULT CLayerUIPropPage::InitFile(PCWSTR Filename)
{
    PCWSTR pwszExt = PathFindExtensionW(Filename);
    if (!pwszExt)
    {
        ACDBG(L"Failed to find an extension: '%s'\r\n", Filename);
        return E_FAIL;
    }
    if (!wcsicmp(pwszExt, L".lnk"))
    {
        WCHAR Buffer[MAX_PATH];
        if (!GetExeFromLnk(Filename, Buffer, _countof(Buffer)))
        {
            ACDBG(L"Failed to read link target from: '%s'\r\n", Filename);
            return E_FAIL;
        }
        if (!wcsicmp(Buffer, Filename))
        {
            ACDBG(L"Link redirects to itself: '%s'\r\n", Filename);
            return E_FAIL;
        }
        return InitFile(Buffer);
    }
    for (size_t n = 0; g_AllowedExtensions[n]; ++n)
    {
        if (!wcsicmp(g_AllowedExtensions[n], pwszExt))
        {
            m_Filename = Filename;
            ACDBG(L"Got: %s\r\n", Filename);
            m_IsSfcProtected = SfcIsFileProtected(NULL, m_Filename);
            m_AllowPermLayer = AllowPermLayer(Filename);
            return S_OK;
        }
    }
    ACDBG(L"Extension not included: '%s'\r\n", pwszExt);
    return E_FAIL;
}

BOOL GetLayerInfo(BSTR Filename, DWORD QueryFlags, PDWORD OSMode, PDWORD Enabledlayers)
{
    *OSMode = *Enabledlayers = 0;
    WCHAR wszLayers[MAX_LAYER_LENGTH] = { 0 };
    DWORD dwBytes = sizeof(wszLayers);
    if (!SdbGetPermLayerKeys(Filename, wszLayers, &dwBytes, QueryFlags))
        return FALSE;

    for (PWCHAR Layer = wcstok(wszLayers, L" "); Layer; Layer = wcstok(NULL, L" "))
    {
        size_t n;
        for (n = 0; g_Layers[n].Name; ++n)
        {
            if (g_Layers[n].Enabled && !wcsicmp(g_Layers[n].Name, Layer))
            {
                *Enabledlayers |= (1<<n);
                break;
            }
        }
        if (!g_Layers[n].Name)
        {
            for (n = 0; g_CompatModes[n].Name; ++n)
            {
                if (!wcsicmp(g_CompatModes[n].Name, Layer))
                {
                    *OSMode = n+1;
                    break;
                }
            }
        }
    }
    return TRUE;
}

void CLayerUIPropPage::OnRefresh(HWND hWnd)
{
    if (!GetLayerInfo(m_Filename, m_LayerQueryFlags, &m_RegistryOSMode, &m_RegistryEnabledLayers))
        m_RegistryOSMode = m_RegistryEnabledLayers = 0;

    for (size_t n = 0; g_Layers[n].Name; ++n)
        CheckDlgButton(hWnd, g_Layers[n].Id, (m_RegistryEnabledLayers & (1<<n)) ? BST_CHECKED : BST_UNCHECKED);
    CheckDlgButton(hWnd, IDC_CHKRUNCOMPATIBILITY, m_RegistryOSMode ? BST_CHECKED : BST_UNCHECKED);
    if (m_RegistryOSMode)
        ComboBox_SetCurSel(GetDlgItem(hWnd, IDC_COMPATIBILITYMODE), m_RegistryOSMode-1);
    UpdateControls(hWnd);
}

void CLayerUIPropPage::OnApply(HWND hWnd)
{
    if (m_RegistryEnabledLayers != m_EnabledLayers || m_RegistryOSMode != m_OSMode)
    {
        BOOL bMachine = m_LayerQueryFlags == GPLK_MACHINE;
        for (size_t n = 0; g_CompatModes[n].Name; ++n)
            SetPermLayerState(m_Filename, g_CompatModes[n].Name, 0, bMachine, (n+1) == m_OSMode);
        for (size_t n = 0; g_Layers[n].Name; ++n)
        {
            if (g_Layers[n].Enabled)
                SetPermLayerState(m_Filename, g_Layers[n].Name, 0, bMachine, ((1<<n) & m_EnabledLayers) != 0);
        }
        SHChangeNotify(SHCNE_UPDATEITEM, SHCNF_PATHW, (BSTR)m_Filename, NULL);
    }
}

INT_PTR CLayerUIPropPage::InitDialog(HWND hWnd)
{
    HWND cboMode = GetDlgItem(hWnd, IDC_COMPATIBILITYMODE);
    for (size_t n = 0; g_CompatModes[n].Display; ++n)
        ComboBox_AddString(cboMode, g_CompatModes[n].Display);
    ComboBox_SetCurSel(cboMode, 5);
    EnableWindow(GetDlgItem(hWnd, IDC_EDITCOMPATIBILITYMODES), 0);

    CComBSTR explanation;
    if (!m_AllowPermLayer)
    {
        explanation.LoadString(g_hModule, IDS_FAILED_NETWORK);
        DisableControls(hWnd);
        ACDBG(L"AllowPermLayer returned FALSE\r\n");
    }
    else if (m_IsSfcProtected)
    {
        explanation.LoadString(g_hModule, IDS_FAILED_PROTECTED);
        DisableControls(hWnd);
        ACDBG(L"Protected OS file\r\n");
    }
    else
    {
        return TRUE;
    }
    SetDlgItemTextW(hWnd, IDC_EXPLANATION, explanation);
    return TRUE;
}

INT_PTR CLayerUIPropPage::DisableControls(HWND hWnd)
{
    EnableWindow(GetDlgItem(hWnd, IDC_COMPATIBILITYMODE), 0);
    EnableWindow(GetDlgItem(hWnd, IDC_CHKRUNCOMPATIBILITY), 0);
    for (size_t n = 0; g_Layers[n].Name; ++n)
        EnableWindow(GetDlgItem(hWnd, g_Layers[n].Id), 0);
    EnableWindow(GetDlgItem(hWnd, IDC_EDITCOMPATIBILITYMODES), 0);
    return TRUE;
}

void CLayerUIPropPage::UpdateControls(HWND hWnd)
{
    m_OSMode = 0, m_EnabledLayers = 0;
    BOOL ModeEnabled = IsDlgButtonChecked(hWnd, IDC_CHKRUNCOMPATIBILITY);
    if (ModeEnabled)
        m_OSMode = ComboBox_GetCurSel(GetDlgItem(hWnd, IDC_COMPATIBILITYMODE))+1;
    EnableWindow(GetDlgItem(hWnd, IDC_COMPATIBILITYMODE), ModeEnabled);
    for (size_t n = 0; g_Layers[n].Name; ++n)
    {
        if (g_Layers[n].Enabled)
        {
            m_EnabledLayers |= IsDlgButtonChecked(hWnd, g_Layers[n].Id) ? (1<<n) : 0;
            ShowWindow(GetDlgItem(hWnd, g_Layers[n].Id), SW_SHOW);
        }
        else
        {
            ShowWindow(GetDlgItem(hWnd, g_Layers[n].Id), SW_HIDE);
        }
    }
    if (m_RegistryOSMode != m_OSMode || m_RegistryEnabledLayers != m_EnabledLayers)
    {
        PropSheet_Changed(GetParent(hWnd), hWnd);
    }
    else
    {
        PropSheet_UnChanged(GetParent(hWnd), hWnd);
    }
}

INT_PTR CLayerUIPropPage::OnCommand(HWND hWnd, WORD id)
{
    switch (id)
    {
    case IDC_CHKRUNCOMPATIBILITY:
        UpdateControls(hWnd);
        break;
    case IDC_COMPATIBILITYMODE:
        UpdateControls(hWnd);
        break;
    case IDC_CHKRUNIN256COLORS:
    case IDC_CHKRUNIN640480RES:
    case IDC_CHKDISABLEVISUALTHEMES:
        UpdateControls(hWnd);
        break;
    case IDC_EDITCOMPATIBILITYMODES:
        break;
    }
    return FALSE;
}

INT_PTR CALLBACK CLayerUIPropPage::PropDlgProc(HWND hWnd, UINT uMsg, WPARAM wParam, LPARAM lParam)
{
    CLayerUIPropPage* page = NULL;

    switch (uMsg)
    {
    case WM_INITDIALOG:
        page = (CLayerUIPropPage*)((LPPROPSHEETPAGE)lParam)->lParam;
        SetProp(hWnd, ACP_WNDPROP, page);
        return page->InitDialog(hWnd);

    case WM_ENDSESSION:
    case WM_DESTROY:
        page = (CLayerUIPropPage*)GetProp(hWnd, ACP_WNDPROP);
        RemoveProp(hWnd, ACP_WNDPROP);
        page->Release();
        break;

    case WM_COMMAND:
        page = (CLayerUIPropPage*)GetProp(hWnd, ACP_WNDPROP);
        return page->OnCommand(hWnd, LOWORD(wParam));
    case WM_NOTIFY:
        switch (((LPNMHDR)lParam)->code)
        {
        case PSN_SETACTIVE:
            if (((LPNMHDR)lParam)->hwndFrom == GetParent(hWnd))
            {
                page = (CLayerUIPropPage*)GetProp(hWnd, ACP_WNDPROP);
                page->OnRefresh(hWnd);
            }
            break;
        case PSN_APPLY:
            if (((LPNMHDR)lParam)->hwndFrom == GetParent(hWnd))
            {
                page = (CLayerUIPropPage*)GetProp(hWnd, ACP_WNDPROP);
                page->OnApply(hWnd);
            }
            break;
        case NM_CLICK:
        case NM_RETURN:
            if (((LPNMHDR)lParam)->idFrom == IDC_INFOLINK)
            {
                ShellExecute(NULL, L"open", L"https://www.reactos.org/forum/viewforum.php?f=4", NULL, NULL, SW_SHOW);
            }
            break;
        default:
            break;
        }
        break;
    }

    return FALSE;
}

STDMETHODIMP CLayerUIPropPage::Initialize(LPCITEMIDLIST pidlFolder, LPDATAOBJECT pDataObj, HKEY hkeyProgID)
{
    FORMATETC etc = { CF_HDROP, NULL, DVASPECT_CONTENT, -1, TYMED_HGLOBAL };
    STGMEDIUM stg;
    HRESULT hr = pDataObj->GetData(&etc, &stg);
    if (FAILED(hr))
    {
        ACDBG(L"Failed to retrieve Data from pDataObj.\r\n");
        return E_INVALIDARG;
    }
    hr = E_FAIL;
    HDROP hdrop = (HDROP)GlobalLock(stg.hGlobal);
    if (hdrop)
    {
        UINT uNumFiles = DragQueryFileW(hdrop, 0xFFFFFFFF, NULL, 0);
        if (uNumFiles == 1)
        {
            WCHAR szFile[MAX_PATH * 2];
            if (DragQueryFileW(hdrop, 0, szFile, _countof(szFile)))
            {
                this->AddRef();
                hr = InitFile(szFile);
            }
            else
            {
                ACDBG(L"Failed to query the file.\r\n");
            }
        }
        else
        {
            ACDBG(L"Invalid number of files: %d\r\n", uNumFiles);
        }
        GlobalUnlock(stg.hGlobal);
    }
    else
    {
        ACDBG(L"Could not lock stg.hGlobal\r\n");
    }
    ReleaseStgMedium(&stg);
    return hr;
}

STDMETHODIMP CLayerUIPropPage::AddPages(LPFNADDPROPSHEETPAGE pfnAddPage, LPARAM lParam)
{
    PROPSHEETPAGEW psp = { 0 };
    psp.dwSize = sizeof(psp);
    psp.dwFlags = PSP_USEREFPARENT | PSP_USETITLE;
    psp.hInstance = g_hModule;
    psp.pszTemplate = MAKEINTRESOURCE(IDD_ACPPAGESHEET);
    psp.pszTitle = MAKEINTRESOURCE(IDS_TABTITLE);
    psp.pfnDlgProc = PropDlgProc;
    psp.lParam = (LPARAM)this;
    psp.pcRefParent = (PUINT)&g_ModuleRefCnt;
    HPROPSHEETPAGE hPage = CreatePropertySheetPageW(&psp);
    if (hPage && !pfnAddPage(hPage, lParam))
        DestroyPropertySheetPage(hPage);

    return S_OK;
}