#include "shellprv.h"
#include <sfview.h>
#include "defviewp.h"

int CGenList::Add(LPVOID pv, int nInsert)
{
    if (!_hList)
    {
        _hList = DSA_Create(_cbItem, 8);
        if (!_hList)
        {
            return -1;
        }
    }
    return DSA_InsertItem(_hList, nInsert, pv);
}


int CViewsList::Add(const SFVVIEWSDATA*pView, int nInsert, BOOL bCopy)
{
    if (bCopy)
    {
        pView = CopyData(pView);
        if (!pView)
        {
            return -1;
        }
    }

    int iIndex = CGenList::Add((LPVOID)(&pView), nInsert);

    if (bCopy && iIndex<0)
    {
        SHFree((LPVOID)pView);
    }

    return iIndex;
}


TCHAR const c_szExtViews[] = TEXT("ExtShellFolderViews");

void CViewsList::AddReg(HKEY hkParent, LPCTSTR pszSubKey)
{
    CSHRegKey ckClass(hkParent, pszSubKey);
    if (!ckClass)
    {
        return;
    }

    CSHRegKey ckShellEx(ckClass, TEXT("shellex"));
    if (!ckShellEx)
    {
        return;
    }

    CSHRegKey ckViews(ckShellEx, c_szExtViews);
    if (!ckViews)
    {
        return;
    }

    TCHAR szKey[40];
    DWORD dwLen = sizeof(szKey);
    SHELLVIEWID vid;

    if (ERROR_SUCCESS==SHRegQueryValue(ckViews, NULL, szKey, (LONG*)&dwLen)
        && SUCCEEDED(SHCLSIDFromString(szKey, &vid)))
    {
        _vidDef = vid;
        _bGotDef = TRUE;
    }


    for (int i=0; ; ++i)
    {
        LONG lRet = RegEnumKey(ckViews, i, szKey, ARRAYSIZE(szKey));
        if (lRet == ERROR_MORE_DATA)
        {
            continue;
        }
        else if (lRet != ERROR_SUCCESS)
        {
            // I assume this is ERROR_NO_MORE_ITEMS
            break;
        }

        SFVVIEWSDATA sView;
        ZeroMemory(&sView, sizeof(sView));

        if (FAILED(SHCLSIDFromString(szKey, &sView.idView)))
        {
            continue;
        }

        CSHRegKey ckView(ckViews, szKey);
        if (ckView)
        {
            TCHAR szFile[ARRAYSIZE(sView.wszMoniker)];
            DWORD dwType;

            // NOTE: This app "Nuts&Bolts" munges the registry and remove the last NULL byte
            // from the PersistMoniker string. When we read that string into un-initialized 
            // local buffer, we do not get a properly null terminated string and we fail to 
            // create the moniker. So, I zero Init the mem here.
            ZeroMemory(szFile, sizeof(szFile));
            
            // Attributes affect all extended views
            dwLen = sizeof(sView.dwFlags);
            if ((ERROR_SUCCESS != SHQueryValueEx(ckView, TEXT("Attributes"),
                    NULL, &dwType, &sView.dwFlags, &dwLen))
                || dwLen != sizeof(sView.dwFlags)
                || !(REG_DWORD==dwType || REG_BINARY==dwType))
            {
                sView.dwFlags = 0;
            }

            // We either have a PersistMoniker (docobj) extended view
            // or we have an IShellView extended view
            //
            dwLen = sizeof(szFile);
            if (ERROR_SUCCESS == SHQueryValueEx(ckView, TEXT("PersistMoniker"),
                    NULL, &dwType, szFile, &dwLen) && REG_SZ == dwType)
            {
                //if the %UserAppData% exists, expand it!
                ExpandOtherVariables(szFile, ARRAYSIZE(szFile));
                SHTCharToUnicode(szFile, sView.wszMoniker, ARRAYSIZE(sView.wszMoniker));
            }
            else
            {
                dwLen = sizeof(szKey);
                if (ERROR_SUCCESS == SHQueryValueEx(ckView, TEXT("ISV"),
                    NULL, &dwType, szKey, &dwLen) && REG_SZ == dwType
                    && SUCCEEDED(SHCLSIDFromString(szKey, &vid)))
                {
                    sView.idExtShellView = vid;

                    // Only IShellView extended vies use LParams
                    dwLen = sizeof(sView.lParam);
                    if ((ERROR_SUCCESS != SHQueryValueEx(ckView, TEXT("LParam"),
                            NULL, &dwType, &sView.lParam, &dwLen))
                        || dwLen != sizeof(sView.lParam)
                        || !(REG_DWORD==dwType || REG_BINARY==dwType))
                    {
                        sView.lParam = 0;
                    }
        
                }
                else
                {
                    if (VID_FolderState != sView.idView)
                    {
                        // No moniker, no IShellView extension, this must be a VID_FolderState
                        // kinda thing. (Otherwise it's a bad desktop.ini.)
                        //
                        RIPMSG(0, "Extended view is registered incorrectly.");
                        continue;
                    }
                }
            }

            // It has been requested (by OEMs) to allow specifying background
            // bitmap and text colors for the regular views. That way they could
            // brand the Control Panel page by putting their logo recessed on
            // the background. We'd do that here by pulling the stuff out of
            // the registry and putting it in the LPCUSTOMVIEWSDATA section...
        }

        // if docobjextended view that is not webview, DO NOT ADD IT, UNSUPPORTED.
        if (!(sView.dwFlags & SFVF_NOWEBVIEWFOLDERCONTENTS)
                && !IsEqualGUID(sView.idView, VID_WebView))
            continue;
        Add(&sView);
    }
}


void CViewsList::AddCLSID(CLSID const* pclsid)
{
    CSHRegKey ckCLSID(HKEY_CLASSES_ROOT, TEXT("CLSID"));
    if (!ckCLSID)
    {
        return;
    }

    TCHAR szCLSID[40];
    SHStringFromGUID(*pclsid, szCLSID, ARRAYSIZE(szCLSID));

    AddReg(ckCLSID, szCLSID);
}

#ifdef DEBUG
//In debug, I want to see if all the realloc code-path works fine!
//So, I deliberately alloc very small amounts.
#define CUSTOM_INITIAL_ALLOC      16
#define CUSTOM_REALLOC_INCREMENT  16
#else
#define CUSTOM_INITIAL_ALLOC      20*64
#define CUSTOM_REALLOC_INCREMENT  512
#endif //DEBUG

//Returns the offset of the string read into the block of memory (in chars)
// NOTE: the index (piCurOffset) and the sizes *piTotalSize, *piSizeRemaining are 
// NOTE: in WCHARs, not BYTES
int GetCustomStrData(LPWSTR *pDataBegin, int *piSizeRemaining, int *piTotalSize, 
                       int *piCurOffset, LPCTSTR szSectionName, LPCTSTR szKeyName, 
                       LPCTSTR szIniFile, LPCTSTR lpszPath)
{
    TCHAR szStrData[INFOTIPSIZE], szTemp[INFOTIPSIZE];
#ifndef UNICODE
    WCHAR wszStrData[MAX_PATH];
#endif
    LPWSTR pszStrData;
    int iLen, iOffsetBegin = *piCurOffset;

    //See if the data is present.
    if (!SHGetIniString(szSectionName, szKeyName, szTemp, ARRAYSIZE(szTemp), szIniFile))
    {
        return -1;  //The given data is not present.
    }

    SHExpandEnvironmentStrings(szTemp, szStrData, ARRAYSIZE(szStrData));   // Expand the env vars if any
    
    //Get the full pathname if required.
    if (lpszPath)
        PathCombine(szStrData, lpszPath, szStrData);

#ifdef UNICODE
    iLen = lstrlen(szStrData);
    pszStrData = szStrData;
#else
    iLen = MultiByteToWideChar(CP_ACP, 0, szStrData, -1, wszStrData, ARRAYSIZE(wszStrData));
    pszStrData = wszStrData;
#endif
    iLen++;   //Include the NULL character.

    while(*piSizeRemaining < iLen)
    {
        LPWSTR lpNew;
        //We need to realloc the block of memory
        if (NULL == (lpNew = (LPWSTR)SHRealloc(*pDataBegin, ( *piTotalSize + CUSTOM_REALLOC_INCREMENT) * sizeof(WCHAR))))
            return -1;  //Unable to realloc; out of mem.
        
        //Note: The begining address of the block could have changed.
        *pDataBegin = lpNew;
        *piTotalSize += CUSTOM_REALLOC_INCREMENT;
        *piSizeRemaining += CUSTOM_REALLOC_INCREMENT;
    }

    //Add the current directory if required.
    StrCpyW((*pDataBegin)+(*piCurOffset), pszStrData);

    *piSizeRemaining -= iLen;
    *piCurOffset += iLen;

    return iOffsetBegin;
}

HRESULT ReadWebViewTemplate(LPCTSTR pszPath, LPTSTR pszWebViewTemplate, int cchWebViewTemplate)
{
    SHFOLDERCUSTOMSETTINGS fcs = {sizeof(fcs), FCSM_WEBVIEWTEMPLATE, 0};
    fcs.pszWebViewTemplate = pszWebViewTemplate;   // template path
    fcs.cchWebViewTemplate = cchWebViewTemplate;
    return SHGetSetFolderCustomSettings(&fcs, pszPath, FCS_READ);
}

#define ID_EXTVIEWICONAREAIMAGE 3
#define ID_EXTVIEWCOLORSFIRST   4
#define ID_EXTVIEWSTRLAST       5
#define ID_EXTVIEWUICOUNT       6

const LPCTSTR c_szExtViewUIRegKeys[ID_EXTVIEWUICOUNT] =
{
    TEXT("MenuName"),
    TEXT("HelpText"),
    TEXT("TooltipText"),
    TEXT("IconArea_Image"),
    TEXT("IconArea_TextBackground"),
    TEXT("IconArea_Text")
};

void CViewsList::AddIni(LPCTSTR szIniFile, LPCTSTR szPath)
{
    TCHAR szViewIDs[12*45];  // Room for about 12 GUIDs including Default=
    SHELLVIEWID vid;

    //
    //First check if the INI file exists before trying to get data from it.
    //
    if (!PathFileExistsAndAttributes(szIniFile, NULL))
        return;

    if (GetPrivateProfileString(c_szExtViews, TEXT("Default"), c_szNULL,
        szViewIDs, ARRAYSIZE(szViewIDs), szIniFile)
        && SUCCEEDED(SHCLSIDFromString(szViewIDs, &vid)))
    {
        _vidDef = vid;
        _bGotDef = TRUE;
    }

    GetPrivateProfileString(c_szExtViews, NULL, c_szNULL,
        szViewIDs, ARRAYSIZE(szViewIDs), szIniFile);

    for (LPCTSTR pNextID=szViewIDs; *pNextID; pNextID+=lstrlen(pNextID)+1)
    {
        SFVVIEWSDATA sViewData;
        CUSTOMVIEWSDATA sCustomData;
        LPWSTR         pszDataBegin = NULL;
        int            iSizeRemaining = CUSTOM_INITIAL_ALLOC; //Let's begin with 12 strings.
        int            iTotalSize;
        int            iCurOffset;

        ZeroMemory(&sViewData, sizeof(sViewData));

        ZeroMemory(&sCustomData, sizeof(sCustomData));

        // there must be a view id
        if (FAILED(SHCLSIDFromString(pNextID, &sViewData.idView)))
        {
            continue;
        }

        // we blow off IE4b2 customized views. This forces them to run
        // the wizard again which will clean this junk up
        if (IsEqualIID(sViewData.idView, VID_DefaultCustomWebView))
        {
            continue;
        }

        // get the IShellView extended view, if any
        BOOL fExtShellView = FALSE;
        TCHAR szPreProcName[45];
        if (GetPrivateProfileString(c_szExtViews, pNextID, c_szNULL,
            szPreProcName, ARRAYSIZE(szPreProcName), szIniFile))
        {
            fExtShellView = SUCCEEDED(SHCLSIDFromString(szPreProcName, &sViewData.idExtShellView));
        }

        // All extended views use Attributes
        sViewData.dwFlags = GetPrivateProfileInt(pNextID, TEXT("Attributes"), 0, szIniFile) | SFVF_CUSTOMIZEDVIEW;

        // For some reason this code uses a much larger buffer
        // than seems necessary. I don't know why... [mikesh 29jul97]
        TCHAR szViewData[MAX_PATH+MAX_PATH];
        szViewData[0] = TEXT('\0'); // For the non-webview case

        if (IsEqualGUID(sViewData.idView, VID_WebView) && SUCCEEDED(ReadWebViewTemplate(szPath, szViewData, MAX_PATH)))
        {
            LPTSTR pszPath = szViewData;
            // We want to allow relative paths for the file: protocol
            //
            if (0 == StrCmpNI(TEXT("file://"), szViewData, 7)) // ARRAYSIZE(TEXT("file://"))
            {
                pszPath += 7;   // ARRAYSIZE(TEXT("file://"))
            }
            // for webview:// compatibility, keep this working:
            else if (0 == StrCmpNI(TEXT("webview://file://"), szViewData, 17)) // ARRAYSIZE(TEXT("file://"))
            {
                pszPath += 17;  // ARRAYSIZE(TEXT("webview://file://"))
            }
            // handle relative references...
            PathCombine(pszPath, szPath, pszPath);

            // Avoid overwriting buffers
            szViewData[MAX_PATH-1] = NULL;
        }
        else
        {
            // only IShellView extensions use LParams
            sViewData.lParam = GetPrivateProfileInt( pNextID, TEXT("LParam"), 0, szIniFile );

            if (!fExtShellView && VID_FolderState != sViewData.idView)
            {
                // No moniker, no IShellView extension, this must be a VID_FolderState
                // kinda thing. (Otherwise it's a bad desktop.ini.)
                //
                RIPMSG(0, "Extended view is registered incorrectly.");
                continue;
            }
        }
        SHTCharToUnicode(szViewData, sViewData.wszMoniker, ARRAYSIZE(sViewData.wszMoniker));

        // NOTE: the size is in WCHARs not in BYTES
        pszDataBegin = (LPWSTR)SHAlloc(iSizeRemaining * sizeof(WCHAR));
        if (NULL == pszDataBegin)
            continue;

        iTotalSize = iSizeRemaining;
        iCurOffset = 0;

        // Read the custom colors
        for (int i = 0; i < CRID_COLORCOUNT; i++)
            sCustomData.crCustomColors[i] = GetPrivateProfileInt(pNextID, c_szExtViewUIRegKeys[ID_EXTVIEWCOLORSFIRST + i], CLR_MYINVALID, szIniFile);
        
        // Read the extended view strings
        for (i = 0; i <= ID_EXTVIEWSTRLAST; i++)
        {
            sCustomData.acchOffExtViewUIstr[i] = GetCustomStrData(&pszDataBegin,
                       &iSizeRemaining, &iTotalSize, &iCurOffset,
                       pNextID, c_szExtViewUIRegKeys[i], szIniFile,
                       (i == ID_EXTVIEWICONAREAIMAGE ? szPath : NULL));
        }

        sCustomData.cchSizeOfBlock = (iTotalSize - iSizeRemaining);
        sCustomData.lpDataBlock = pszDataBegin;
        sViewData.pCustomData = &sCustomData;

        // if docobjextended view that is not webview, DO NOT ADD IT, UNSUPPORTED.
        if (!(sViewData.dwFlags & SFVF_NOWEBVIEWFOLDERCONTENTS)
                && !IsEqualGUID(sViewData.idView, VID_WebView)
                && !IsEqualGUID(sViewData.idView, VID_FolderState))
            continue;
        Add(&sViewData);

        //We already copied the data. So, we can free it!
        SHFree(pszDataBegin);
    }
}


void CViewsList::Empty()
{
    _bGotDef = FALSE;

    for (int i=GetItemCount()-1; i>=0; --i)
    {
        SFVVIEWSDATA  *sfvData = GetPtr(i);
        
        ASSERT(sfvData);
        if (sfvData->dwFlags & SFVF_CUSTOMIZEDVIEW)
        {
            CUSTOMVIEWSDATA  *pCustomPtr = sfvData->pCustomData;
            if (pCustomPtr)
            {
                if (pCustomPtr->lpDataBlock)
                    SHFree(pCustomPtr->lpDataBlock);
                SHFree(pCustomPtr);
            }
        }
        SHFree(sfvData);
    }

    CGenList::Empty();
}


SFVVIEWSDATA* CViewsList::CopyData(const SFVVIEWSDATA* pData)
{
    SFVVIEWSDATA* pCopy = (SFVVIEWSDATA*)SHAlloc(sizeof(SFVVIEWSDATA));
    if (pCopy)
    {
        memcpy(pCopy, pData, sizeof(SFVVIEWSDATA));
        if ((pData->dwFlags & SFVF_CUSTOMIZEDVIEW) && pData->pCustomData)
        {
            CUSTOMVIEWSDATA *pCustomData = (CUSTOMVIEWSDATA *)SHAlloc(sizeof(CUSTOMVIEWSDATA));
            if (pCustomData)
            {
                memcpy(pCustomData, pData->pCustomData, sizeof(CUSTOMVIEWSDATA));
                pCopy->pCustomData = pCustomData;

                if (pCustomData->lpDataBlock)
                {
                    // NOTE: DataBlock size is in WCHARs
                    LPWSTR lpDataBlock = (LPWSTR)SHAlloc(pCustomData->cchSizeOfBlock * sizeof(WCHAR));
                    if (lpDataBlock)
                    {
                        // NOTE: DataBlock size is in WCHARs
                        memcpy(lpDataBlock, pCustomData->lpDataBlock, pCustomData->cchSizeOfBlock * sizeof(WCHAR));
                        pCustomData->lpDataBlock = lpDataBlock;
                    }
                    else
                    {
                        SHFree(pCustomData);
                        goto Failed;
                    }
                }
            }
            else
            {
Failed:
                SHFree(pCopy);
                pCopy = NULL;
            }
        }
    }

    return pCopy;
}


int CViewsList::NextUnique(int nLast)
{
    for (int nNext = nLast + 1; ; ++nNext)
    {
        SFVVIEWSDATA* pItem = GetPtr(nNext);
        if (!pItem)
        {
            break;
        }

        for (int nPrev=nNext-1; nPrev>=0; --nPrev)
        {
            SFVVIEWSDATA*pPrev = GetPtr(nPrev);
            if (pItem->idView == pPrev->idView)
            {
                break;
            }
        }

        if (nPrev < 0)
        {
            return nNext;
        }
    }

    return -1;
}


// Note this is 1-based
int CViewsList::NthUnique(int nUnique)
{
    for (int nNext = -1; nUnique > 0; --nUnique)
    {
        nNext = NextUnique(nNext);
        if (nNext < 0)
        {
            return -1;
        }
    }

    return nNext;
}


void CCallback::_GetExtViews(BOOL bForce)
{
    CDefView* pView = IToClass(CDefView, _cCallback, this);

    IEnumSFVViews *pev = NULL;

    if (bForce)
    {
        _bGotViews = FALSE;
    }

    if (_bGotViews)
    {
        return;
    }

    _lViews.Empty();

    SHELLVIEWID vid = VID_LargeIcons;
    if (FAILED(pView->CallCB(SFVM_GETVIEWS, (WPARAM)&vid, (LPARAM)&pev)) ||
        !pev)
    {
        return;
    }

    _lViews.SetDef(&vid);

    SFVVIEWSDATA *pData;
    ULONG uFetched;

    while ((pev->Next(1, &pData, &uFetched) == S_OK) && (uFetched == 1))
    {
        // The list comes to us in general to specific order, but we want
        // to search it in specific->general order. Inverting the list
        // is easiest here, even though it causes a bunch of memcpy calls.
        //
        _lViews.Prepend(pData, FALSE);
    }

    ATOMICRELEASE(pev);

    _bGotViews = TRUE;
}


HRESULT CCallback::TryLegacyGetViews(SFVM_WEBVIEW_TEMPLATE_DATA* pvit)
{
    CDefView* pView = IToClass(CDefView, _cCallback, this);
    HRESULT hr = E_FAIL;

    CLSID clsid;
    HRESULT hr2 = IUnknown_GetClassID(pView->_pshf, &clsid);
    if (FAILED(hr2) || !(SHGetObjectCompatFlags(NULL, &clsid) & OBJCOMPATF_NOLEGACYWEBVIEW))
    {
        _GetExtViews(FALSE);
        if (_bGotViews)
        {
            SFVVIEWSDATA* pItem;
            GetViewIdFromGUID(&VID_WebView, &pItem);
            if (pItem)
            {
                StrCpyNW(pvit->szWebView, pItem->wszMoniker, ARRAYSIZE(pvit->szWebView));
                hr = S_OK;
            }
        }
        else if (SUCCEEDED(hr2))
        {
            // check for PersistMoniker under isf's coclass (Web Folders used this in W2K to get .htt Web View)
            WCHAR szCLSID[GUIDSTR_MAX];
            SHStringFromGUID(clsid, szCLSID, ARRAYSIZE(szCLSID));

            WCHAR szkey[MAX_PATH];
            wnsprintf(szkey, ARRAYSIZE(szkey), L"CLSID\\%s\\shellex\\ExtShellFolderViews\\{5984FFE0-28D4-11CF-AE66-08002B2E1262}", szCLSID);

            DWORD cbSize = sizeof(pvit->szWebView);
            if (ERROR_SUCCESS == SHGetValueW(HKEY_CLASSES_ROOT, szkey, L"PersistMoniker", NULL, pvit->szWebView, &cbSize))
            {
                hr = S_OK;
            }
        }
    }

    return hr;
}

HRESULT CCallback::OnRefreshLegacy(void* pv, BOOL fPrePost)
{
    // If we're using the SFVM_GETVIEWS layer, invalidate it
    if (_bGotViews)
    {
        _lViews.Empty();
        _bGotViews = FALSE;
    }

    return S_OK;
}

int CCallback::GetViewIdFromGUID(SHELLVIEWID const *pvid, SFVVIEWSDATA** ppItem)
{
    int iView = -1;
    for (UINT uView=0; uView<MAX_EXT_VIEWS; ++uView)
    {
        iView = _lViews.NextUnique(iView);

        SFVVIEWSDATA* pItem = _lViews.GetPtr(iView);
        if (!pItem)
        {
            break;
        }

        if (*pvid == pItem->idView)
        {
            if (ppItem)
                *ppItem = pItem;

            return (int)uView;
        }
    }

    if (ppItem)
        *ppItem = NULL;
    return -1;
}