/*
	vfdshext.cpp

	Virtual Floppy Drive for Windows
	Driver control library
	shell extension COM shell extension class

	Copyright (c) 2003-2005 Ken Kato
*/

#define WIN32_LEAN_AND_MEAN
#include <windows.h>
#include <shellapi.h>
#include <shlobj.h>

#include "vfdtypes.h"
#include "vfdapi.h"
#include "vfdlib.h"

//	class header
#include "vfdshext.h"

//
//	Constructor
//
CVfdShExt::CVfdShExt()
{
	VFDTRACE(0, ("CVfdShExt::CVfdShExt()\n"));

	m_cRefCnt		= 0L;
	m_pDataObj		= NULL;
	m_nDevice		= (ULONG)-1;
	m_sTarget[0]	= '\0';
	m_bDragDrop		= FALSE;

	g_cDllRefCnt++;
}

//
//	Destructor
//
CVfdShExt::~CVfdShExt()
{
	VFDTRACE(0, ("CVfdShExt::~CVfdShExt()\n"));

	if (m_pDataObj) {
		m_pDataObj->Release();
	}

	g_cDllRefCnt--;
}

//	IUnknown members

STDMETHODIMP CVfdShExt::QueryInterface(
	REFIID			riid,
	LPVOID			*ppv)
{
	*ppv = NULL;

	if (IsEqualIID(riid, IID_IShellExtInit) ||
		IsEqualIID(riid, IID_IUnknown)) {
		VFDTRACE(0,
			("CVfdShExt::QueryInterface()==>IID_IShellExtInit\n"));

		*ppv = (LPSHELLEXTINIT)this;
	}
	else if (IsEqualIID(riid, IID_IContextMenu)) {
		VFDTRACE(0,
			("CVfdShExt::QueryInterface()==>IID_IContextMenu\n"));

		*ppv = (LPCONTEXTMENU)this;
	}
	else if (IsEqualIID(riid, IID_IShellPropSheetExt)) {
		VFDTRACE(0,
			("CVfdShExt::QueryInterface()==>IID_IShellPropSheetExt\n"));

		*ppv = (LPSHELLPROPSHEETEXT)this;
	}

	if (*ppv) {
		AddRef();

		return NOERROR;
	}

	VFDTRACE(0,
		("CVfdShExt::QueryInterface()==>Unknown Interface!\n"));

	return E_NOINTERFACE;
}

STDMETHODIMP_(ULONG) CVfdShExt::AddRef()
{
	VFDTRACE(0, ("CVfdShExt::AddRef()\n"));

	return ++m_cRefCnt;
}

STDMETHODIMP_(ULONG) CVfdShExt::Release()
{
	VFDTRACE(0, ("CVfdShExt::Release()\n"));

	if (--m_cRefCnt) {
		return m_cRefCnt;
	}

#ifndef __REACTOS__
	delete this;
#endif

	return 0L;
}

//	IShellExtInit members

//
//	Initialize
//	Called by the shell to initialize the shell extension object
//
STDMETHODIMP CVfdShExt::Initialize(
	LPCITEMIDLIST	pIDFolder,
	LPDATAOBJECT	pDataObj,
	HKEY			hRegKey)
{
	CHAR drive = '\0';

	VFDTRACE(0, ("CVfdShExt::Initialize()\n"));

	UNREFERENCED_PARAMETER(hRegKey);

	// Initialize can be called more than once

	if (m_pDataObj) {
		m_pDataObj->Release();
		m_pDataObj = NULL;
	}

	m_nDevice = (ULONG)-1;
	m_sTarget[0] = '\0';

	//	Get the folder name
	if (SHGetPathFromIDList(pIDFolder, m_sTarget)) {

		//	act as a Drag-and-Drop Handler

		VFDTRACE(0, ("Drag-Drop: %s\n", m_sTarget));

		if (GetDriveType(m_sTarget) != DRIVE_REMOVABLE) {
			VFDTRACE(0, ("Not a VFD drive\n"));
			return NOERROR;
		}

		drive = m_sTarget[0];
		m_bDragDrop = TRUE;
	}
	else {

		//	act as a context menu handler

		VFDTRACE(0, ("Context menu:\n"));
		m_bDragDrop = FALSE;
	}

	// Extract the target object name

	if (pDataObj) {

		STGMEDIUM medium;
		FORMATETC fmt = {
			CF_HDROP, NULL, DVASPECT_CONTENT, -1, TYMED_HGLOBAL
		};

		if (SUCCEEDED(pDataObj->GetData(&fmt, &medium))) {
			if (DragQueryFile((HDROP)medium.hGlobal, (UINT)-1, NULL, 0)) {

				DragQueryFile((HDROP)medium.hGlobal,
					0, m_sTarget, sizeof(m_sTarget));
			}

			ReleaseStgMedium(&medium);
		}
	}

	VFDTRACE(0, ("Target %s\n", m_sTarget));

	if (!drive) {
		//	Contect menu handler
		//	-- Data object is the target drive
		drive = m_sTarget[0];
	}

	HANDLE hDevice = VfdOpenDevice(drive);

	if (hDevice == INVALID_HANDLE_VALUE) {
		VFDTRACE(0, ("Not a VFD drive\n"));
		return NOERROR;
	}

	ULONG ret = VfdGetDeviceNumber(hDevice, &m_nDevice);

	CloseHandle(hDevice);

	if (ret != ERROR_SUCCESS) {
		m_nDevice = (ULONG)-1;
		return NOERROR;
	}

	VFDTRACE(0, ("VFD device %d\n", m_nDevice));
	//	Store the data object

	m_pDataObj = pDataObj;
	m_pDataObj->AddRef();

	return NOERROR;
}

/*
STDMETHODIMP CVfdShExt::GetInfoFlags(
	DWORD *pdwFlags)
{
	VFDTRACE(0, ("CVfdShExt::GetInfoFlags\n"));
	*pdwFlags = 0;
	return NOERROR;
}

STDMETHODIMP CVfdShExt::GetInfoTip(
	DWORD dwFlags,
	LPWSTR *ppwszTip)
{
	VFDTRACE(0, ("CVfdShExt::GetInfoTip\n"));
	*ppwszTip = NULL;
	return NOERROR;
}
*/