#include "stdafx.h"
#include "MulSelFileDialog.h"

#ifndef _INC_CDERR
#include <cderr.h>     
#endif // _INC_CDERR

#ifdef _DEBUG
#define new DEBUG_NEW
#undef THIS_FILE
static char THIS_FILE[] = __FILE__;
#endif

/////////////////////////////////////////////////////////////////////////////
// CMulSelFileDialog

IMPLEMENT_DYNAMIC(CMulSelFileDialog, CFileDialog)


CMulSelFileDialog::CMulSelFileDialog(BOOL bOpenFileDialog, 
									 LPCTSTR lpszDefExt, 
									 LPCTSTR lpszFileName,
                                     DWORD dwFlags, 
									 LPCTSTR lpszFilter, 
									 CWnd* pParentWnd) :
CFileDialog(bOpenFileDialog, lpszDefExt, lpszFileName, dwFlags, lpszFilter, pParentWnd)
{
    Files = NULL;
    Folder = NULL;
    bParsed = FALSE;
}


CMulSelFileDialog::~CMulSelFileDialog()
{
    if (Files)
    {
        delete[] Files;
        delete[] Folder;
    }
}

BEGIN_MESSAGE_MAP(CMulSelFileDialog, CFileDialog)
//{{AFX_MSG_MAP(CMulSelFileDialog)
//}}AFX_MSG_MAP
END_MESSAGE_MAP()


int CMulSelFileDialog::DoModal()
{
    if (Files)
    {
        delete[] Files;
        Files = NULL;
        delete[] Folder;
        Folder = NULL;
    }

    int ret = CFileDialog::DoModal();

    if (ret == IDCANCEL)
    {
        DWORD err = CommDlgExtendedError();
        if (err == FNERR_BUFFERTOOSMALL/*0x3003*/ && Files)
		{
            ret = IDOK;
		}
    }

	return ret;
}

CString CMulSelFileDialog::GetNextPathName(POSITION &pos) const
{
    if (!Files)
	{
        return CFileDialog::GetNextPathName(pos);
	}

    ASSERT(pos);    
    TCHAR *ptr = (TCHAR *)pos;

    CString ret = Folder;
    ret += _T("\\");
    ret += ptr;

    ptr += _tcslen(ptr) + 1;
    if (*ptr)
	{
        pos = (POSITION)ptr;
	}
    else
	{
        pos = NULL;
	}

    return ret;
}


POSITION CMulSelFileDialog::GetStartPosition()
{
    if (!Files)
	{
        return CFileDialog::GetStartPosition();
	}

    if (!bParsed)
    {
        CString temp = Files;
        temp.Replace(_T("\" \""), _T("\""));
        temp.Delete(0, 1);                      // remove leading quote mark
        temp.Delete(temp.GetLength() - 1, 1);   // remove trailing space
    
        _tcscpy(Files, temp);
    
        TCHAR *ptr = Files;
        while (*ptr)
        {
            if ('\"' == *ptr)
                *ptr = '\0';
            ++ptr;
        }

        bParsed = TRUE;
    }
    
    return (POSITION)Files;
}

void CMulSelFileDialog::OnFileNameChange()
{
    TCHAR dummy_buffer;
    
    // Get the required size for the 'files' buffer
    UINT nfiles = CommDlg_OpenSave_GetSpec(GetParent()->m_hWnd, 
		                                   &dummy_buffer, 1);

    // Get the required size for the 'folder' buffer
    UINT nfolder = CommDlg_OpenSave_GetFolderPath(GetParent()->m_hWnd, 
		                                          &dummy_buffer, 1);

    // Check if lpstrFile and nMaxFile are large enough
    if (nfiles + nfolder > m_ofn.nMaxFile)
    {
        bParsed = FALSE;
        if (Files)
            delete[] Files;
        Files = new TCHAR[nfiles + 1];
        CommDlg_OpenSave_GetSpec(GetParent()->m_hWnd, Files, nfiles);

        if (Folder)
		{
            delete[] Folder;
		}
        Folder = new TCHAR[nfolder + 1];
        CommDlg_OpenSave_GetFolderPath(GetParent()->m_hWnd, 
			                           Folder, 
									   nfolder);
    }
    else if (Files)
    {
        delete[] Files;
        Files = NULL;
        delete[] Folder;
        Folder = NULL;
    }

    CFileDialog::OnFileNameChange();
}