// ============================================================================
//  Copyright (C) 2001-2011 by Jozef Starosczyk
//  ixen@copyhandler.com
//
//  This program is free software; you can redistribute it and/or modify
//  it under the terms of the GNU Library General Public License
//  (version 2) as published by the Free Software Foundation;
//
//  This program 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 General Public License for more details.
//
//  You should have received a copy of the GNU Library General Public
//  License along with this program; if not, write to the
//  Free Software Foundation, Inc.,
//  59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
// ============================================================================
/// @file  TContextMenuHandler.h
/// @date  2011/05/28
/// @brief Contains class which handles context menu.
// ============================================================================
#ifndef __TCONTEXTMENUHANDLER_H__
#define __TCONTEXTMENUHANDLER_H__

#include "TShellExtData.h"
#include "../libchengine/TLocalFilesystem.h"
#include "../libchengine/TSizeFormatter.h"

class TShellMenuItem;

typedef std::shared_ptr<TShellMenuItem> TShellMenuItemPtr;

class TContextMenuHandler
{
private:
	const unsigned long CHItemMarker = L'C' << 16 | L'H';

public:
	TContextMenuHandler();
	~TContextMenuHandler();

	void Init(const TShellMenuItemPtr& spRootMenuItem, HMENU hMenu, UINT uiFirstItemID, UINT uiFirstItemPosition, const TShellExtData& rShellExtData,
		const chengine::TSizeFormatterPtr& spFormatter, bool bShowFreeSpace,
		bool bEnableOwnerDrawnPaths, bool bOverrideDefaultItem);
	bool HasCHItems(HMENU hMenu) const;
	void Clear();

	int FindMenuInsertLocation(HMENU hMenu);

	UINT GetLastCommandID() const { return m_uiNextMenuID; }
	TShellMenuItemPtr GetCommandByMenuItemOffset(UINT uiOffset);
	TShellMenuItemPtr GetCommandByItemID(UINT uiOffset);

protected:
	void UpdateMenuRecursive(const TShellMenuItemPtr& spRootMenuItem, HMENU hMenu, UINT uiFirstItemPosition, const TShellExtData& rShellExtData,
		const chengine::TSizeFormatterPtr& spFormatter, bool bShowFreeSpace, bool bOverrideDefaultItem);
	std::wstring GetDisplayText(const TShellMenuItemPtr& spMenuItem, const chengine::TSizeFormatterPtr& spFormatter, bool bShowFreeSpace);

private:
	std::map<UINT, TShellMenuItemPtr> m_mapMenuItems;
	std::vector<HMENU> m_vHandlesToFree;

	UINT m_uiFirstMenuID;		// menu ID from which the numbering started
	UINT m_uiNextMenuID;		// next menu ID to be used

	bool m_bEnableOwnerDrawnPaths;
	chengine::TLocalFilesystem m_fsLocal;
};
#endif