Windows Kernel

Development Platform:
Visual C++

ftadv.cpp：Code Content
							#include "shellprv.h"
#include "ids.h"
#include "help.h"
#include "apithk.h"
#include "ascstr.h"
#include "filetype.h"
#include "ftdlg.h"
#include "ftadv.h"
#include "ftaction.h"
const static DWORD cs_rgdwHelpIDsArray[] =
{  // Context Help IDs
    IDC_NO_HELP_1,              NO_HELP,
    IDC_FT_EDIT_DOCICON,        IDH_FCAB_FT_EDIT_DOCICON,
    IDC_FT_EDIT_DESC,           IDH_FCAB_FT_EDIT_DESC,
    IDC_FT_EDIT_CHANGEICON,     IDH_FCAB_FT_EDIT_CHANGEICON,
    IDC_FT_EDIT_LV_CMDSTEXT,    IDH_FCAB_FT_EDIT_LV_CMDS,
    IDC_FT_EDIT_LV_CMDS,        IDH_FCAB_FT_EDIT_LV_CMDS,
    IDC_FT_EDIT_NEW,            IDH_FCAB_FT_EDIT_NEW,
    IDC_FT_EDIT_EDIT,           IDH_FCAB_FT_EDIT_EDIT,
    IDC_FT_EDIT_REMOVE,         IDH_FCAB_FT_EDIT_REMOVE,
    IDC_FT_EDIT_DEFAULT,        IDH_FCAB_FT_EDIT_DEFAULT,
    IDC_FT_EDIT_CONFIRM_OPEN,   IDH_CONFIRM_OPEN,
    IDC_FT_EDIT_SHOWEXT,        IDH_FCAB_FT_EDIT_SHOWEXT,
    IDC_FT_EDIT_BROWSEINPLACE,  IDH_SAME_WINDOW,
    0, 0
};
///////////////////////////////////////////////////////////////////////////////
///////////////////////////////////////////////////////////////////////////////
CFTAdvDlg::CFTAdvDlg(LPTSTR pszProgID, BOOL fAutoDelete) :
    CFTDlg((ULONG_PTR)cs_rgdwHelpIDsArray, fAutoDelete),
    _iDefaultAction(-1), _iLVSel(-1)
{
    if (pszProgID)
        StrCpyN(_szProgID, pszProgID, ARRAYSIZE(_szProgID));
    _hdpaActions = DPA_Create(4);
    _hdpaRemovedActions = DPA_Create(1);
}
static int _DeleteLocalAllocCB(void *pItem, void *pData)
{
    LocalFree((HLOCAL)pItem);
    return 1;
}
CFTAdvDlg::~CFTAdvDlg()
{
    if (_hIcon)
        DeleteObject(_hIcon);
    if (_hfontReg)
        DeleteObject(_hfontReg);
    if (_hfontBold)
        DeleteObject(_hfontBold);
    if (_hdpaActions)
        DPA_DestroyCallback(_hdpaActions, _DeleteLocalAllocCB, NULL);
    if (_hdpaRemovedActions)
        DPA_DestroyCallback(_hdpaRemovedActions, _DeleteLocalAllocCB, NULL);
}
///////////////////////////////////////////////////////////////////////////////
// Logic specific to our problem
LRESULT CFTAdvDlg::OnInitDialog(WPARAM wParam, LPARAM lParam)
{
    HRESULT hres = _InitAssocStore();
    DECLAREWAITCURSOR;
    SetWaitCursor();
    if (SUCCEEDED(hres))
    {
        _InitListView();
        _InitDefaultActionFont();
    // FTEdit_AreDefaultViewersInstalled ????
        if (*_szProgID)
        {
            _SetDocIcon();
            _InitDescription();
            _FillListView();
            _InitDefaultAction();
            _SelectListViewItem(0);
            _InitChangeIconButton();
            _UpdateCheckBoxes();
        }   
    }
    else
        EndDialog(_hwnd, -1);
    ResetWaitCursor();
    // Return TRUE so that system set focus
    return TRUE;
}
HRESULT CFTAdvDlg::_SetDocIcon(int iIndex)
{
    HRESULT hres = E_FAIL;
    if (-1 == iIndex)
    {
        IAssocInfo* pAI = NULL;
        hres = _pAssocStore->GetAssocInfo(_szProgID, AIINIT_PROGID, &pAI);
        if (SUCCEEDED(hres))
        {
            hres = pAI->GetDWORD(AIDWORD_DOCLARGEICON, (DWORD*)&iIndex);
            _iOriginalIcon = iIndex;
            pAI->Release();
        }
    }
    if (-1 != iIndex)
    {
        HIMAGELIST hIL = NULL;
        Shell_GetImageLists(&hIL, NULL);
        if (_hIcon)
        {
            DeleteObject(_hIcon);
            _hIcon = NULL;
        }
        if (hIL)
        {
            _hIcon = ImageList_ExtractIcon(g_hinst, hIL, iIndex);
            _hIcon = (HICON)CopyImage(_hIcon, IMAGE_ICON, 32, 32, LR_COPYDELETEORG);
            HICON hiOld = (HICON)SendDlgItemMessage(_hwnd, IDC_FT_EDIT_DOCICON, STM_SETIMAGE, IMAGE_ICON,
                (LPARAM)_hIcon);
            if (hiOld)
                DestroyIcon(hiOld);
        }
    }
    return hres;
}
LRESULT CFTAdvDlg::OnListViewSelItem(int iItem, LPARAM lParam)
{
    _UpdateActionButtons();
    return TRUE;
}
LRESULT CFTAdvDlg::OnMeasureItem(WPARAM wParam, LPARAM lParam)
{
    TEXTMETRIC tm = {0};
    RECT rect;
    LPMEASUREITEMSTRUCT lpmis = (LPMEASUREITEMSTRUCT)lParam;
    HDC hdc = GetDC(NULL);
    HFONT hfontOld = (HFONT)SelectObject(hdc, _hfontBold);
    GetTextMetrics(hdc, &tm);
    GetClientRect(_GetLVHWND(), &rect);
    lpmis->itemWidth = rect.right;
    lpmis->itemHeight = tm.tmHeight;
    SelectObject(hdc, hfontOld);
    ReleaseDC(NULL, hdc);
    return TRUE;
}
LRESULT CFTAdvDlg::OnDrawItem(WPARAM wParam, LPARAM lParam)
{
    LPDRAWITEMSTRUCT lpDIS = (LPDRAWITEMSTRUCT)lParam;
    LRESULT lRet = FALSE;
    
    if (ODT_LISTVIEW == lpDIS->CtlType)
    {
        HWND hwndLV = _GetLVHWND();
        LVITEM lvItem = {0};
        HFONT hfontOld = NULL;
        BOOL fSel = FALSE;
        BOOL fListFocus = FALSE;
        TCHAR szAction[MAX_ACTION];
        COLORREF crBkgd = 0;
        COLORREF crOldText = 0;
        
        lvItem.mask = LVIF_PARAM | LVIF_TEXT | LVIF_STATE;
        lvItem.iItem = lpDIS->itemID;
        lvItem.stateMask = LVIS_SELECTED|LVIS_FOCUSED;
        lvItem.pszText = szAction;
        lvItem.cchTextMax = ARRAYSIZE(szAction);
        ListView_GetItem(hwndLV, &lvItem);
        fSel = (lvItem.state & LVIS_SELECTED);
        fListFocus = (GetFocus() == hwndLV);
        
        crBkgd = (fSel ? (fListFocus ? COLOR_HIGHLIGHT : COLOR_3DFACE) : COLOR_WINDOW);
        SetBkColor(lpDIS->hDC, GetSysColor(crBkgd));
        FillRect(lpDIS->hDC, &lpDIS->rcItem, (HBRUSH)(crBkgd + 1));
        crOldText = SetTextColor(lpDIS->hDC, 
            GetSysColor(fSel ? (fListFocus ? COLOR_HIGHLIGHTTEXT : COLOR_WINDOWTEXT) :
            COLOR_WINDOWTEXT));
        // Use Bold font for default action
        hfontOld = (HFONT)SelectObject(lpDIS->hDC, 
            _IsDefaultAction(szAction) ? _hfontBold : _hfontReg);
        
        int iOldBkMode = SetBkMode(lpDIS->hDC, OPAQUE);
//        DrawText(lpDIS->hDC, szAction, lstrlen(szAction), &lpDIS->rcItem, DT_HIDEPREFIX);
        // BUGBUG: DT_HIDEPREFIX
        DrawText(lpDIS->hDC, szAction, lstrlen(szAction), &lpDIS->rcItem, 0);
        SetBkMode(lpDIS->hDC, iOldBkMode);
        SetTextColor(lpDIS->hDC, crOldText);
        SelectObject(lpDIS->hDC, hfontOld);
        
        if(fListFocus && (lvItem.state & LVIS_FOCUSED))
            DrawFocusRect(lpDIS->hDC, &lpDIS->rcItem);
        lRet = TRUE;
    }
    return lRet;
}
HRESULT CFTAdvDlg::_InitDefaultActionFont()
{
    HFONT hfontDlg = GetWindowFont(_hwnd);
    LOGFONT lf = {0};
    LOGFONT lfDlg = {0};
    GetObject(hfontDlg, sizeof(LOGFONT), &lfDlg);
    SystemParametersInfo(SPI_GETICONTITLELOGFONT, SIZEOF(lf), &lf, FALSE);
    // Normal font
    lf.lfWeight = FW_NORMAL;
    lf.lfHeight = lfDlg.lfHeight;    
    _hfontReg = CreateFontIndirect(&lf);
    // Bold font
    lf.lfWeight = FW_BOLD;
    _hfontBold = CreateFontIndirect(&lf);
    return (_hfontReg && _hfontBold) ? S_OK : E_FAIL;
}
HRESULT CFTAdvDlg::_SelectListViewItem(int i)
{
    LVITEM lvItem = {0};
    lvItem.iItem = i;
    lvItem.mask = LVIF_STATE;
    lvItem.stateMask = LVIS_SELECTED | LVIS_FOCUSED;
    lvItem.state = LVIS_SELECTED | LVIS_FOCUSED;
    ListView_SetItem(_GetLVHWND(), &lvItem);
    return S_OK;
}
HRESULT CFTAdvDlg::_InitDefaultAction()
{
    HRESULT hres = E_FAIL;
    // Get it from the classstore
    IAssocInfo* pAI = NULL;
    hres = _pAssocStore->GetAssocInfo(_szProgID, AIINIT_PROGID, &pAI);
    if (SUCCEEDED(hres))
    {
        TCHAR szAction[MAX_ACTION];
        DWORD cchAction = ARRAYSIZE(szAction);
        HWND hwndLV = _GetLVHWND();
        LVFINDINFO lvFindInfo = {0};
        int iIndex = -1;
        hres = pAI->GetString(AISTR_PROGIDDEFAULTACTION, szAction, &cchAction);
        if (SUCCEEDED(hres))
        {
            lvFindInfo.flags = LVFI_STRING;
            lvFindInfo.psz = szAction;
            iIndex = ListView_FindItem(hwndLV, -1, &lvFindInfo);
            if (-1 != iIndex)
                hres = _SetDefaultAction(iIndex);
            else
                hres = S_OK;
        }
        pAI->Release();
    }
    return hres;
}
BOOL CFTAdvDlg::_IsDefaultAction(LPTSTR pszAction)
{
    BOOL fRet = FALSE;
    HWND hwndLV = _GetLVHWND();
    LVFINDINFO lvFindInfo = {0};
    int iIndex = -1;
    lvFindInfo.flags = LVFI_PARAM;
    lvFindInfo.lParam = 1;
    iIndex = ListView_FindItem(hwndLV, -1, &lvFindInfo);
    if (-1 != iIndex)
    {
        LVITEM lvItem = {0};
        TCHAR szAction[MAX_ACTION];
        lvItem.iItem = iIndex;
        lvItem.mask = LVIF_TEXT;
        lvItem.pszText = szAction;
        lvItem.cchTextMax = ARRAYSIZE(szAction);
        if (ListView_GetItem(hwndLV, &lvItem))
        {
            if (!lstrcmpi(szAction ,pszAction))
                fRet = TRUE;
        }
    }
    return fRet;
}
void CFTAdvDlg::_CheckDefaultAction()
{
    HWND hwndLV = _GetLVHWND();
    // Is there only one elem?
    if (1 == ListView_GetItemCount(hwndLV))
    {
        // Yes, make it the default action
        LVITEM lvItem = {0};
        lvItem.mask = LVIF_PARAM;
        // lvItem.iItem = 0;        // Want the Zero item.
        if (ListView_GetItem(hwndLV, &lvItem))
        {
            lvItem.lParam = 1;
            ListView_SetItem(hwndLV, &lvItem);
        }
    }
}
HRESULT CFTAdvDlg::_SetDefaultAction(int iIndex)
{
    HWND hwndLV = _GetLVHWND();
    LVITEM lvItem = {0};
    lvItem.mask = LVIF_PARAM;
    lvItem.iItem = _iDefaultAction;
    // lvItem.lParam = 0;  //lParam == 0 means "Not Default"
    // Remove previous default if any
    ListView_SetItem(hwndLV, &lvItem);
    ListView_RedrawItems(hwndLV, lvItem.iItem, lvItem.iItem);
    // Set new
    _iDefaultAction = iIndex;
    // iIndex == -1 means no default
    if (iIndex >= 0)    
    {
        lvItem.iItem = _iDefaultAction;
        lvItem.lParam = 1;
        ListView_SetItem(hwndLV, &lvItem);
        ListView_RedrawItems(hwndLV, lvItem.iItem, lvItem.iItem);
    }
    
    return S_OK;
}
HRESULT CFTAdvDlg::_InitListView()
{
    LVCOLUMN lvColumn = {0};
    HWND hwndLV = _GetLVHWND();
    RECT rc = {0};
    {
        // What's this?
        // We need to handle the WM_MEASUREITEM message from the listview.  This msg
        // is sent before we receive the WM_INITDIALOG and thus before we connect the
        // this C++ obj to the HWND.  By changing the style here we receive the msg
        // after the C++ obj and the HWND are connected.
        LONG lStyle = GetWindowLong(hwndLV, GWL_STYLE);
        lStyle &= ~LVS_LIST;
        SetWindowLong(hwndLV, GWL_STYLE, lStyle | LVS_REPORT);
    }
    //
    // Set the columns
    //
    GetClientRect(hwndLV, &rc);
    lvColumn.mask = LVCF_SUBITEM|LVCF_WIDTH;
    lvColumn.cx = rc.right - GetSystemMetrics(SM_CXBORDER);
    lvColumn.iSubItem = 0;
    ListView_InsertColumn(hwndLV, 0, &lvColumn);
    return S_OK;
}
HRESULT CFTAdvDlg::_UpdateActionButtons()
{
    HRESULT hres = E_FAIL;
    TCHAR szAction[MAX_ACTION];
    BOOL bRet = FALSE;
    LVITEM lvItem = {0};
    lvItem.pszText = szAction;
    lvItem.cchTextMax = ARRAYSIZE(szAction);
    bRet = _GetListViewSelectedItem(LVIF_TEXT | LVIF_PARAM, 0, &lvItem);
    // If we don't have a selected item Or we don't have any text for that item.
    if (!bRet || !(*(lvItem.pszText)))
    {
        EnableWindow(GetDlgItem(_hwnd, IDC_FT_EDIT_EDIT), FALSE);
        EnableWindow(GetDlgItem(_hwnd, IDC_FT_EDIT_REMOVE), FALSE);
        EnableWindow(GetDlgItem(_hwnd, IDC_FT_EDIT_DEFAULT), TRUE);        
        hres = S_OK;
    }
    else
    {
        if (_IsNewPROGIDACTION(lvItem.pszText))
        {
            EnableWindow(GetDlgItem(_hwnd, IDC_FT_EDIT_EDIT), TRUE);
            EnableWindow(GetDlgItem(_hwnd, IDC_FT_EDIT_REMOVE), TRUE);
            hres = S_OK;
        }
        else
        {
            IAssocInfo* pAI = NULL;
            hres = _pAssocStore->GetAssocInfo(_szProgID, AIINIT_PROGID, &pAI);
            if (SUCCEEDED(hres))
            {
                DWORD dwAttributes;
                HWND hwndLV = _GetLVHWND();
                // BUGBUG (begin): This code should be in ftassoc.cpp, and we should have
                // more AIBOOL_ flags for this
                hres = pAI->GetDWORD(AIDWORD_PROGIDEDITFLAGS, &dwAttributes);
                if (FAILED(hres))
                {
                    // It failed, probably there is no EditFlags value for this progID, let's
                    // set some default value for dwAttributes
                    dwAttributes = 0;
                }
                // BUGBUG (end)
                EnableWindow(GetDlgItem(_hwnd, IDC_FT_EDIT_EDIT),
                    !((dwAttributes & FTA_NoEditVerb) &&
                    !(dwAttributes & FTAV_UserDefVerb)));
                EnableWindow(GetDlgItem(_hwnd, IDC_FT_EDIT_REMOVE),
                    !((dwAttributes & FTA_NoRemoveVerb) &&
                    !(dwAttributes & FTAV_UserDefVerb)));
                EnableWindow(GetDlgItem(_hwnd, IDC_FT_EDIT_DEFAULT),
                    !(dwAttributes & FTA_NoEditDflt));  
                // Enable the default button only if the action is not already
                // the default action
                EnableWindow(GetDlgItem(_hwnd, IDC_FT_EDIT_DEFAULT),
                    lvItem.lParam ? FALSE : TRUE);
                pAI->Release();
            }
        }
    }
    return hres;
}
HRESULT CFTAdvDlg::_UpdateCheckBoxes()
{
    HRESULT hres = E_FAIL;
    BOOL fBool;
    IAssocInfo* pAI = NULL;
    hres = _pAssocStore->GetAssocInfo(_szProgID, AIINIT_PROGID, &pAI);
    if (SUCCEEDED(hres))
    {
        hres = pAI->GetBOOL(AIBOOL_CONFIRMOPEN, &fBool);
        if (SUCCEEDED(hres))
            CheckDlgButton(_hwnd, IDC_FT_EDIT_CONFIRM_OPEN, !fBool);
        hres = pAI->GetBOOL(AIBOOL_ALWAYSSHOWEXT, &fBool);
        if (SUCCEEDED(hres))
            CheckDlgButton(_hwnd, IDC_FT_EDIT_SHOWEXT, fBool);
        hres = pAI->GetBOOL(AIBOOL_BROWSEINPLACEENABLED, &fBool);
        if (SUCCEEDED(hres))
        {
            EnableWindow(GetDlgItem(_hwnd, IDC_FT_EDIT_BROWSEINPLACE), fBool);
            if (fBool)
            {
                hres = pAI->GetBOOL(AIBOOL_BROWSEINPLACE, &fBool);
                if (SUCCEEDED(hres))
                    CheckDlgButton(_hwnd, IDC_FT_EDIT_BROWSEINPLACE, fBool);
            }
            else
                CheckDlgButton(_hwnd, IDC_FT_EDIT_BROWSEINPLACE, FALSE);
        }
        pAI->Release();
    }
    return hres;
}
HRESULT CFTAdvDlg::_InitChangeIconButton()
{
    HRESULT hres = E_FAIL;
    BOOL fChangeIcon = TRUE;
    IAssocInfo* pAI = NULL;
    hres = _pAssocStore->GetAssocInfo(_szProgID, AIINIT_PROGID, &pAI);
    if (SUCCEEDED(hres))
    {
        hres = pAI->GetBOOL(AIBOOL_EDITDOCICON, &fChangeIcon);
    
        if (SUCCEEDED(hres))
            EnableWindow(GetDlgItem(_hwnd, IDC_FT_EDIT_CHANGEICON), fChangeIcon);
        pAI->Release();
    }
    return hres;
}
HRESULT CFTAdvDlg::_InitDescription()
{
    HRESULT hres = E_FAIL;
    BOOL fEditDescr = TRUE;
    IAssocInfo* pAI = NULL;
    hres = _pAssocStore->GetAssocInfo(_szProgID, AIINIT_PROGID, &pAI);
    if (SUCCEEDED(hres))
    {
        TCHAR szProgIDDescr[MAX_PROGIDDESCR];
        DWORD cchProgIDDescr = ARRAYSIZE(szProgIDDescr);
        hres = pAI->GetString(AISTR_PROGIDDESCR, szProgIDDescr, &cchProgIDDescr);
    
        if (SUCCEEDED(hres))
            SetDlgItemText(_hwnd, IDC_FT_EDIT_DESC, szProgIDDescr);
        hres = pAI->GetBOOL(AIBOOL_EDITDESCR, &fEditDescr);
        EnableWindow(GetDlgItem(_hwnd, IDC_FT_EDIT_DESC), fEditDescr);
        pAI->Release();
        Edit_LimitText(GetDlgItem(_hwnd, IDC_FT_EDIT_DESC), MAX_PROGIDDESCR - 1);
    }
    return hres;
}
HRESULT CFTAdvDlg::_FillListView()
{
    HRESULT hres = E_FAIL;
    IEnumAssocInfo* pEnum = NULL;
    hres = _pAssocStore->EnumAssocInfo(ASENUM_ACTION, _szProgID, AIINIT_PROGID, &pEnum);
    if (SUCCEEDED(hres))
    {
        int iItem = 0;
        IAssocInfo* pAI = NULL;
        while (S_OK == pEnum->Next(&pAI))
        {
            TCHAR szAction[MAX_ACTION];
            DWORD cchAction = ARRAYSIZE(szAction);
            hres = pAI->GetString(AISTR_ACTION, szAction, &cchAction);
            if (SUCCEEDED(hres))
            {
                _InsertListViewItem(iItem, szAction);
                ++iItem;
            }
            pAI->Release();
        }
        pEnum->Release();
    }
    return hres;
}
LRESULT CFTAdvDlg::OnChangeIcon(WORD wNotif)
{
    HRESULT hres = E_FAIL;
    int iIcon = -1;
    IAssocInfo* pAI = NULL;
    hres = _pAssocStore->GetAssocInfo(_szProgID, AIINIT_PROGID, &pAI);
    if (SUCCEEDED(hres))
    {
        DWORD cchIconLoc = ARRAYSIZE(_szIconLoc);
        hres = pAI->GetString(AISTR_ICONLOCATION, _szIconLoc, &cchIconLoc);
        iIcon = PathParseIconLocation(_szIconLoc);
        pAI->Release();
    }
    if (FAILED(hres))
    {
        StrCpyN(_szIconLoc, TEXT("shell32.dll"), ARRAYSIZE(_szIconLoc));
        iIcon = -(IDI_SYSFILE);
        hres = S_OK;
    }
    if (SUCCEEDED(hres))
    {
        if (PickIconDlg(_hwnd, _szIconLoc, ARRAYSIZE(_szIconLoc), &iIcon))
        {
            HICON hIcon = NULL;
            HIMAGELIST hIL = NULL;
            int iLen = 0;
            _iNewIcon = Shell_GetCachedImageIndex(_szIconLoc, iIcon, 0);
            _SetDocIcon(_iNewIcon);
            // Format the _szIconLoc
            iLen = lstrlen(_szIconLoc);
            wnsprintf(_szIconLoc + iLen, ARRAYSIZE(_szIconLoc) - iLen, TEXT(",%d"), iIcon);
        }
        else
            *_szIconLoc = 0;
    }
    return FALSE;
}
// Return value: 
//  TRUE:  Check succeeded, everything is OK
//  FALSE: Check failed
BOOL CFTAdvDlg::_CheckForDuplicateNewAction(LPTSTR pszAction)
{
    // we just go through the listview content
    HWND hwndLV = _GetLVHWND();
    int cItem = ListView_GetItemCount(hwndLV);
    BOOL fRet = TRUE;
    for (int i = 0; (i < cItem) && fRet; ++i)
    {
        TCHAR szAction[MAX_ACTION];
        ListView_GetItemText(hwndLV, i, 0, szAction, ARRAYSIZE(szAction));
        if (!lstrcmpi(szAction, pszAction))
        {
            fRet = FALSE;
        }
    }
    return fRet;
}
// Return value: 
//  TRUE:  Check succeeded, everything is OK
//  FALSE: Check failed
BOOL CFTAdvDlg::_CheckForDuplicateEditAction(LPTSTR pszActionOriginal, LPTSTR pszAction)
{
    // we just go through the listview content
    HWND hwndLV = _GetLVHWND();
    int cItem = ListView_GetItemCount(hwndLV);
    BOOL fRet = TRUE;
    for (int i = 0; (i < cItem) && fRet; ++i)
    {
        TCHAR szAction[MAX_ACTION];
        ListView_GetItemText(hwndLV, i, 0, szAction, ARRAYSIZE(szAction));
        if (!lstrcmpi(szAction, pszAction))
        {
            // they are the same, this can happen if this is the Action we were editing
            // and we did not change the action name
            // Is this the original one we were editing?
            if (lstrcmpi(szAction, pszActionOriginal))
            {
                // No, it's not the original, we have a dup
                fRet = FALSE;
            }
        }
    }
    return fRet;
}
LRESULT CFTAdvDlg::OnNewButton(WORD wNotif)
{
    TCHAR szProgIDDescr[MAX_PROGIDDESCR];
    PROGIDACTION pida = {0};
    CFTActionDlg* pActionDlg = NULL;
    pida.fNew = TRUE;
    GetDlgItemText(_hwnd, IDC_FT_EDIT_DESC, szProgIDDescr, ARRAYSIZE(szProgIDDescr));
    // FALSE: New (not-Edit)
    pActionDlg = new CFTActionDlg(&pida, szProgIDDescr, FALSE);
    if (pActionDlg)
    {
        BOOL fShowAgain;
        do
        {
            fShowAgain = FALSE;
            if (IDOK == pActionDlg->DoModal(g_hinst, MAKEINTRESOURCE(DLG_FILETYPEOPTIONSCMD), _hwnd))
            {
                // Do we have duplicate actions?
                if (_CheckForDuplicateNewAction(pida.szAction))
                {
                    // No
                    HRESULT hres = _AppendPROGIDACTION(&pida);
                    if (SUCCEEDED(hres))
                    {
                        int iItem = _InsertListViewItem(0, pida.szAction);
                        hres = S_OK;
                        if (-1 != iItem)
                            _SelectListViewItem(iItem);
                    }
                }
                else
                {
                    // Yes
                    fShowAgain = TRUE;
                    pActionDlg->SetShowAgain();
                }
            }
            if (fShowAgain)
            {
                ShellMessageBox(g_hinst, _hwnd, MAKEINTRESOURCE(IDS_FT_MB_EXISTINGACTION), 
                    MAKEINTRESOURCE(IDS_FT), MB_OK | MB_ICONSTOP, pida.szAction);
            }
        } while (fShowAgain);
        delete pActionDlg;
    }
    _CheckDefaultAction();
    return FALSE;
}
LRESULT CFTAdvDlg::OnEditButton(WORD wNotif)
{
    TCHAR szAction[MAX_ACTION];
    HRESULT hres = E_FAIL;
    LONG lRes = 0;
    LVITEM lvItem = {0};
    // lvItem.iSubItem = 0;
    lvItem.pszText = szAction;
    lvItem.cchTextMax = ARRAYSIZE(szAction);
    if (_GetListViewSelectedItem(LVIF_TEXT, 0, &lvItem))
    {
        TCHAR szProgIDDescr[MAX_PROGIDDESCR];
        PROGIDACTION* pPIDA = NULL;
        PROGIDACTION pida = {0};
        GetDlgItemText(_hwnd, IDC_FT_EDIT_DESC, szProgIDDescr, ARRAYSIZE(szProgIDDescr));
        BOOL fNewOrEdit = SUCCEEDED(_GetPROGIDACTION(lvItem.pszText, &pPIDA));
        if (!fNewOrEdit)
        {
            hres = _FillPROGIDACTION(&pida, szAction);
            pPIDA = &pida;
        }
        else
        {
            hres = S_OK;
        }
        if (SUCCEEDED(hres))
        {
            // TRUE: Edit
            CFTActionDlg* pActionDlg = new CFTActionDlg(pPIDA, szProgIDDescr, TRUE);
            if (pActionDlg)
            {
                BOOL fShowAgain;
                do
                {
                    fShowAgain = FALSE;
                    if (IDOK == pActionDlg->DoModal(g_hinst, MAKEINTRESOURCE(DLG_FILETYPEOPTIONSCMD), _hwnd))
                    {
                        // Do we have duplicate actions?
                        if (_CheckForDuplicateEditAction(lvItem.pszText, pPIDA->szAction))
                        {
                            // No
                            if (!fNewOrEdit)
                            {
                                hres = _AppendPROGIDACTION(pPIDA);
                            }
                            else
                            {
                                hres = S_OK;
                            }
                            if (SUCCEEDED(hres))
                            {
                                // Replace the current item text
                                StrCpyN(lvItem.pszText, pPIDA->szAction, ARRAYSIZE(pPIDA->szAction));
                                ListView_SetItem(_GetLVHWND(), &lvItem);
                            }
                        }
                        else
                        {
                            // Yes
                            fShowAgain = TRUE;
                            pActionDlg->SetShowAgain();
                        }
                    }
                    if (fShowAgain)
                    {
                        ShellMessageBox(g_hinst, _hwnd, MAKEINTRESOURCE(IDS_FT_MB_EXISTINGACTION), 
                            MAKEINTRESOURCE(IDS_FT), MB_OK | MB_ICONSTOP, pPIDA->szAction);
                    }
                } while (fShowAgain);
                delete pActionDlg;
            }
        }
    }
    return FALSE;
}
LRESULT CFTAdvDlg::OnSetDefault(WORD wNotif)
{
    BOOL bRet;
    LVITEM lvItem = {0};
    // lvItem.iSubItem = 0;
    bRet = _GetListViewSelectedItem(0, 0, &lvItem);
    if (bRet)
        _SetDefaultAction(lvItem.iItem);
    else
        _SetDefaultAction(-1);
        
    return FALSE;
}
LRESULT CFTAdvDlg::OnRemoveButton(WORD wNotif)
{
    TCHAR szExt[MAX_EXT];
    HRESULT hres = E_FAIL;
    LONG lRes = 0;
    LVITEM lvItem = {0};
    // lvItem.iSubItem = 0;
    lvItem.pszText = szExt;
    lvItem.cchTextMax = ARRAYSIZE(szExt);
    if (_GetListViewSelectedItem(LVIF_TEXT, 0, &lvItem))
    {
        if (IDYES == ShellMessageBox(g_hinst, _hwnd, MAKEINTRESOURCE(IDS_FT_MB_REMOVEACTION),
            MAKEINTRESOURCE(IDS_FT), MB_YESNO | MB_ICONQUESTION))
        {
            //
            // First take care of data side
            //
            // Yes.  Is this a new Action?
            PROGIDACTION* pPIDA = NULL;
            if (SUCCEEDED(_GetPROGIDACTION(lvItem.pszText, &pPIDA)) && pPIDA->fNew)
            {
                // Yes, we'll just remove it from the DPA
                hres = _RemovePROGIDACTION(pPIDA);
            }
            else
            {
                // No, add its name to the list to delete if user press OK
                LPTSTR pszActionToRemove = (LPTSTR)LocalAlloc(LPTR, 
                    MAX_ACTION * sizeof(TCHAR));
                hres = E_OUTOFMEMORY;
                if (pszActionToRemove)
                {
                    StrCpyN(pszActionToRemove, lvItem.pszText, MAX_ACTION);
                    if (-1 != DPA_AppendPtr(_hdpaRemovedActions, pszActionToRemove))
                        hres = S_OK;
                    else
                        LocalFree((HLOCAL)pszActionToRemove);
                }
                if (E_OUTOFMEMORY == hres)
                {
                    //Out of memory
                    ShellMessageBox(g_hinst, _hwnd, MAKEINTRESOURCE(IDS_ERROR + 
                        ERROR_NOT_ENOUGH_MEMORY), MAKEINTRESOURCE(IDS_FT), 
                        MB_OK | MB_ICONSTOP);
                }
            }
            //
            // Then update UI, I/A
            //
            if (SUCCEEDED(hres))
            {
                HWND hwndLV = _GetLVHWND();
                int iCount = ListView_GetItemCount(hwndLV);
                int iNextSel = -1;        
        
                ListView_DeleteItem(hwndLV, lvItem.iItem);
        
                if (iCount > lvItem.iItem)
                    iNextSel = lvItem.iItem;
                else
                    if (lvItem.iItem > 0)
                        iNextSel = lvItem.iItem - 1;
                if (-1 != iNextSel)
                    _SelectListViewItem(iNextSel);
            }
        }
        else
            hres = S_FALSE;
    }
    
    _CheckDefaultAction();
    return FALSE;
}
LRESULT CFTAdvDlg::OnOK(WORD wNotif)
{
    BOOL fChecksPassed = FALSE;
    // Yes, we need to:
    //  - remove "removed" items, modify "edited" ones,
    //    and add "New" ones
    //  - update checkboxes related stuff
    //  - set the default action
    //  - set the icon
    //  - set the description
    {
        int n = DPA_GetPtrCount(_hdpaRemovedActions);
        if (n)
        {
            IAssocInfo* pAI;
            HRESULT hres = E_FAIL;
            for (int i = 0; i < n; ++i)
            {
                LPTSTR pszActionToRemove = (LPTSTR)DPA_GetPtr(_hdpaRemovedActions, i);
                if (pszActionToRemove && *pszActionToRemove)
                {
                    hres = _pAssocStore->GetComplexAssocInfo(_szProgID, AIINIT_PROGID, 
                        pszActionToRemove, AIINIT_ACTION, &pAI);
                    if (SUCCEEDED(hres))
                    {
                        pAI->Delete(AIALL_NONE);
                        pAI->Release();
                    }
                    LocalFree((HLOCAL)pszActionToRemove);
                    DPA_DeletePtr(_hdpaRemovedActions, i);
                }
            }
        }
    }
    {
        int n = DPA_GetPtrCount(_hdpaActions);
        if (n)
        {
            IAssocInfo* pAI = NULL;
            HRESULT hres = E_FAIL;
            for (int i = n - 1; i >= 0; --i)
            {
                PROGIDACTION* pPIDAFromList = (PROGIDACTION*)DPA_GetPtr(_hdpaActions, i);
                if (pPIDAFromList)
                {   
                    // Is it an Edited one?
                    if (!pPIDAFromList->fNew)
                    {
                        // Yes, remove the old one first
                        hres = _pAssocStore->GetComplexAssocInfo(_szProgID, AIINIT_PROGID, 
                            pPIDAFromList->szOldAction, AIINIT_ACTION, &pAI);
                        if (SUCCEEDED(hres))
                        {
                            pAI->Delete(AIALL_NONE);                        
                            pAI->Release();
                        }
                    }
                    // Add new data
                    hres = _pAssocStore->GetComplexAssocInfo(_szProgID, AIINIT_PROGID, 
                        pPIDAFromList->szAction, AIINIT_ACTION, &pAI);
                    if (SUCCEEDED(hres))
                    {
                        hres = pAI->SetData(AIDATA_PROGIDACTION, (PBYTE)pPIDAFromList,
                            sizeof(*pPIDAFromList));
                        pAI->Release();
                    }
                    // Clean up DPA
                    _DeletePROGIDACTION(pPIDAFromList);
                    DPA_DeletePtr(_hdpaActions, i);
                }
            }
        }
    }
    {
        IAssocInfo* pAI = NULL;
        HWND hwndLV = _GetLVHWND();
        LVFINDINFO lvFindInfo = {0};
        int iIndex = -1;
        HRESULT hres = _pAssocStore->GetAssocInfo(_szProgID, AIINIT_PROGID, &pAI);
        if (SUCCEEDED(hres))
        {
            hres = pAI->SetBOOL(AIBOOL_CONFIRMOPEN, 
                !IsDlgButtonChecked(_hwnd, IDC_FT_EDIT_CONFIRM_OPEN));
            hres = pAI->SetBOOL(AIBOOL_ALWAYSSHOWEXT, 
                IsDlgButtonChecked(_hwnd, IDC_FT_EDIT_SHOWEXT));
            hres = pAI->SetBOOL(AIBOOL_BROWSEINPLACE, 
                IsDlgButtonChecked(_hwnd, IDC_FT_EDIT_BROWSEINPLACE));
            // Set the default action, if any
            lvFindInfo.flags = LVFI_PARAM;
            lvFindInfo.lParam = 1;
            iIndex = ListView_FindItem(hwndLV, -1, &lvFindInfo);
            if (-1 != iIndex)
            {
                LVITEM lvItem = {0};
                TCHAR szAction[MAX_ACTION];
                lvItem.iItem = iIndex;
                lvItem.mask = LVIF_TEXT;
                lvItem.pszText = szAction;
                lvItem.cchTextMax = ARRAYSIZE(szAction);
                if (ListView_GetItem(hwndLV, &lvItem))
                    hres = pAI->SetString(AISTR_PROGIDDEFAULTACTION, lvItem.pszText);
            }
            else
            {
                hres = pAI->SetString(AISTR_PROGIDDEFAULTACTION, TEXT(""));
            }
            // Set the icon, if changed
            if (*_szIconLoc)
            {
                TCHAR szIconLoc[MAX_ICONLOCATION];
                int iIcon = 0;
                LPTSTR pszFileName = NULL;
                // Set it in the registry
                hres = pAI->SetString(AISTR_ICONLOCATION, _szIconLoc);
                // Set it in the cache
                StrCpyN(szIconLoc, _szIconLoc, ARRAYSIZE(szIconLoc));
                iIcon = PathParseIconLocation(szIconLoc);
                pszFileName = PathFindFileName(szIconLoc);
                SHUpdateImage(pszFileName, iIcon, 0, _iNewIcon);
#if 0
                SHChangeNotify(SHCNE_ASSOCCHANGED, SHCNF_IDLIST, NULL, NULL);
#endif
            }
            // Set the description
            {
                TCHAR szProgIDDescr[MAX_PROGIDDESCR];
                GetDlgItemText(_hwnd, IDC_FT_EDIT_DESC, szProgIDDescr,
                    ARRAYSIZE(szProgIDDescr));
                hres = pAI->SetString(AISTR_PROGIDDESCR, szProgIDDescr);
            }
            pAI->Release();
        }
    }
    EndDialog(_hwnd, IDOK);
    return FALSE;
}
LRESULT CFTAdvDlg::OnNotifyListView(UINT uCode, LPNMHDR pNMHDR)
{
    HWND hwndLV = _GetLVHWND();
    LRESULT lres = FALSE;
    switch(uCode)
    {
        case NM_DBLCLK:
            if (IsWindowEnabled(GetDlgItem(_hwnd, IDC_FT_EDIT_EDIT)))
                PostMessage(_hwnd, WM_COMMAND, (WPARAM)IDC_FT_EDIT_EDIT, 0);
            break;
//review: do I really need to do this?
        case NM_SETFOCUS:
        case NM_KILLFOCUS:
            // update list view
            ListView_RedrawItems(hwndLV, 0, ListView_GetItemCount(hwndLV));
            UpdateWindow(hwndLV);
            break;
        case LVN_ITEMCHANGED:
        {
            NMLISTVIEW* pNMLV = (NMLISTVIEW*)pNMHDR;
            // Is a new item being selected/unselected? 
            if (pNMLV->uChanged & LVIF_STATE)
            {
                // Yes
                OnListViewSelItem(pNMLV->iItem, pNMLV->lParam);
            }
            break;
        }
    }
    return lres;
}
LRESULT CFTAdvDlg::OnCancel(WORD wNotif)
{
    EndDialog(_hwnd, IDCANCEL);
    return FALSE;
}
LRESULT CFTAdvDlg::OnDestroy(WPARAM wParam, LPARAM lParam)
{
    CFTDlg::OnDestroy(wParam, lParam);
    return FALSE;
}
BOOL CFTAdvDlg::_GetListViewSelectedItem(UINT uMask, UINT uStateMask, LVITEM* plvItem)
{
    BOOL fSel = FALSE;
    HWND hwndLV = _GetLVHWND();
    plvItem->mask = uMask | LVIF_STATE;
    plvItem->stateMask = uStateMask | LVIS_SELECTED;
    // Do we have the selection cached?
    if (-1 != _iLVSel)
    {
        // Yes, make sure it's valid
        plvItem->iItem = _iLVSel;
        ListView_GetItem(hwndLV, plvItem);
        if (plvItem->state & LVIS_SELECTED)
            fSel = TRUE;
    }
 
    // Cache was wrong
    if (!fSel)
    {
        int iCount = ListView_GetItemCount(hwndLV);
        for (int i=0; (i < iCount) && !fSel; ++i)
        {
            plvItem->iItem = i;
            ListView_GetItem(hwndLV, plvItem);
            if (plvItem->state & LVIS_SELECTED)
                fSel = TRUE;
        }
        if (fSel)
            _iLVSel = i;
    }
    return fSel;
}
int CFTAdvDlg::_InsertListViewItem(int iItem, LPTSTR pszAction)
{
    HWND hwndLV = _GetLVHWND();
    LVITEM lvItem = {0};
    lvItem.mask = LVIF_TEXT;
    // Extension
    lvItem.iItem = iItem;
    lvItem.pszText = pszAction;
    lvItem.cchTextMax = lstrlen(pszAction);
    
    return ListView_InsertItem(hwndLV, &lvItem);
}
HWND CFTAdvDlg::_GetLVHWND()
{
    return GetDlgItem(_hwnd, IDC_FT_EDIT_LV_CMDS);
}
void CFTAdvDlg::_DeletePROGIDACTION(PROGIDACTION* pPIDA)
{
    if (pPIDA)
        LocalFree((HLOCAL)pPIDA);
}
HRESULT CFTAdvDlg::_RemovePROGIDACTION(PROGIDACTION* pPIDA)
{
    HRESULT hres = E_FAIL;
    int n = DPA_GetPtrCount(_hdpaActions);
    for (int i = 0; (i < n) && FAILED(hres); ++i)
    {
        PROGIDACTION* pPIDAFromList = (PROGIDACTION*)DPA_GetPtr(_hdpaActions, i);
        if (pPIDAFromList == pPIDA)
        {
            _DeletePROGIDACTION(pPIDAFromList);
            DPA_DeletePtr(_hdpaActions, i);
            hres = S_OK;
        }
    }
    return hres;
}
HRESULT CFTAdvDlg::_CreatePROGIDACTION(PROGIDACTION** ppPIDA)
{
    HRESULT hres = E_OUTOFMEMORY;
    *ppPIDA = (PROGIDACTION*)LocalAlloc(LPTR, sizeof(PROGIDACTION));
    
    if (*ppPIDA)
        hres = S_OK;
    return hres;
}
HRESULT CFTAdvDlg::_CopyPROGIDACTION(PROGIDACTION* pPIDADest, PROGIDACTION* pPIDASrc)
{
    memcpy(pPIDADest, pPIDASrc, sizeof(PROGIDACTION));
    return S_OK;
}
HRESULT CFTAdvDlg::_GetPROGIDACTION(LPTSTR pszAction, PROGIDACTION** ppPIDA)
{
    HRESULT hres = E_FAIL;
    *ppPIDA = NULL;
    if (pszAction && *pszAction)
    {
        int n = DPA_GetPtrCount(_hdpaActions);
        for (int i = 0; (i < n) && FAILED(hres); ++i)
        {
            *ppPIDA = (PROGIDACTION*)DPA_GetPtr(_hdpaActions, i);
            if (!StrCmpN((*ppPIDA)->szAction, pszAction, ARRAYSIZE((*ppPIDA)->szAction)))
                hres = S_OK;
        }
    }
    if (FAILED(hres))
        *ppPIDA = NULL;
    return hres;
}
HRESULT CFTAdvDlg::_AppendPROGIDACTION(PROGIDACTION* pPIDA)
{
    PROGIDACTION* pPIDANew = NULL;
    HRESULT hres = _CreatePROGIDACTION(&pPIDANew);
    if (SUCCEEDED(hres))
    {
        _CopyPROGIDACTION(pPIDANew, pPIDA);
        if (-1 != DPA_AppendPtr(_hdpaActions, pPIDANew))
        {
            hres = S_OK;
        }
        else
        {
            _DeletePROGIDACTION(pPIDANew);
            hres = E_OUTOFMEMORY;
        }
    }
    if (E_OUTOFMEMORY == hres)
    {
        //Out of memory
        ShellMessageBox(g_hinst, _hwnd, MAKEINTRESOURCE(IDS_ERROR + 
            ERROR_NOT_ENOUGH_MEMORY), MAKEINTRESOURCE(IDS_FT), 
            MB_OK | MB_ICONSTOP);
    }
    return hres;
}
BOOL CFTAdvDlg::_IsNewPROGIDACTION(LPTSTR pszAction)
{
    BOOL fRet = FALSE;
    PROGIDACTION* pPIDA = NULL;
    HRESULT hres = _GetPROGIDACTION(pszAction, &pPIDA);
    if (SUCCEEDED(hres))
        if (pPIDA->fNew)
            fRet = TRUE;
    return fRet;
}
HRESULT CFTAdvDlg::_FillPROGIDACTION(PROGIDACTION* pPIDA, LPTSTR pszAction)
{
    PROGIDACTION* pPIDAList = NULL;
    HRESULT hres = _GetPROGIDACTION(pszAction, &pPIDAList);
    if (SUCCEEDED(hres))
    {
        _CopyPROGIDACTION(pPIDA, pPIDAList);
    }
    else
    {
        IAssocInfo* pAI = NULL;
        hres = _pAssocStore->GetComplexAssocInfo(_szProgID, AIINIT_PROGID, 
            pszAction, AIINIT_ACTION, &pAI);
        if (SUCCEEDED(hres))
        {
            DWORD cbPIDA = sizeof(*pPIDA);
            hres = pAI->GetData(AIDATA_PROGIDACTION, (PBYTE)pPIDA, &cbPIDA);
            pAI->Release();
        }
    }
    return hres;    
}
///////////////////////////////////////////////////////////////////////////////
// Windows boiler plate code
LRESULT CFTAdvDlg::OnCommand(WPARAM wParam, LPARAM lParam)
{
    LRESULT lRes = FALSE;
    switch(GET_WM_COMMAND_ID(wParam, lParam))
    {
        case IDC_FT_EDIT_NEW:
            lRes = OnNewButton(GET_WM_COMMAND_CMD(wParam, lParam));
            break;
        case IDC_FT_EDIT_REMOVE:
            lRes = OnRemoveButton(GET_WM_COMMAND_CMD(wParam, lParam));
            break;
        case IDC_FT_EDIT_EDIT:
            lRes = OnEditButton(GET_WM_COMMAND_CMD(wParam, lParam));
            break;
        case IDC_FT_EDIT_CHANGEICON:
            lRes = OnChangeIcon(GET_WM_COMMAND_CMD(wParam, lParam));
            break;
        case IDC_FT_EDIT_DEFAULT:
            lRes = OnSetDefault(GET_WM_COMMAND_CMD(wParam, lParam));
            break;
        default:
            lRes = CFTDlg::OnCommand(wParam, lParam);
            break;
    }
    return lRes;    
}
LRESULT CFTAdvDlg::OnNotify(WPARAM wParam, LPARAM lParam)
{
    LRESULT lRes = FALSE;
    LPNMHDR pNMHDR = (LPNMHDR)lParam;
    UINT_PTR idFrom = pNMHDR->idFrom;
    UINT uCode = pNMHDR->code;
    //GET_WM_COMMAND_CMD
    switch(idFrom)
    {
        case IDC_FT_EDIT_LV_CMDS:
            OnNotifyListView(uCode, pNMHDR);
            lRes = CFTDlg::OnNotify(wParam, lParam);
            break;
        default:
            lRes = CFTDlg::OnNotify(wParam, lParam);
            break;
    }
    return lRes;    
}
LRESULT CFTAdvDlg::WndProc(UINT uMsg, WPARAM wParam, LPARAM lParam)
{
    LRESULT lRes = FALSE;
    switch(uMsg)
    {
        case WM_DRAWITEM:
            lRes = OnDrawItem(wParam, lParam);
            break;
        case WM_MEASUREITEM:
            lRes = OnMeasureItem(wParam, lParam);
            break;
        default:
            lRes = CFTDlg::WndProc(uMsg, wParam, lParam);
            break;
    }
    return lRes;
}

						