Shell Extensions for .NET Assemblies

// AssemblyIconOverlay.h : Declaration of the CAssemblyIconOverlay

#pragma once
#include "resource.h"       // main symbols
#include "AsmShell.h"
#include "rgsmap.h"

EXTERN_C const CLSID CLSID_AssemblyIconOverlay;

// CAssemblyIconOverlay

class ATL_NO_VTABLE CAssemblyIconOverlay : 
	public CComObjectRootEx<CComSingleThreadModel>,
	public CComCoClass<CAssemblyIconOverlay, &CLSID_AssemblyIconOverlay>,
	public IShellIconOverlayIdentifier
{
public:
	CAssemblyIconOverlay()
	{
	}

DECLARE_REGISTRY_RESOURCEID_EX(IDR_ASSEMBLYICONOVERLAY)

BEGIN_REGISTRY_MAP(CAssemblyIconOverlay)
	REGISTRY_GUID("CLSID", CLSID_AssemblyIconOverlay)
END_REGISTRY_MAP()

BEGIN_COM_MAP(CAssemblyIconOverlay)
	COM_INTERFACE_ENTRY_IID(IID_IShellIconOverlayIdentifier, IShellIconOverlayIdentifier)
END_COM_MAP()

	DECLARE_PROTECT_FINAL_CONSTRUCT()

	HRESULT FinalConstruct()
	{
		HRESULT hr;

		hr = m_spAsmInfo.CoCreateInstance(CLSID_AssemblyInfo);
		return hr;
	}
	
	void FinalRelease()
	{
	}

public:

	// IShellIconOverlayIdentifier methods
	STDMETHOD(GetOverlayInfo)(LPWSTR pwszIconFile, int cchMax, int *pIndex, DWORD *pdwFlags);
	STDMETHOD(GetPriority)(int *pPriority);
	STDMETHOD(IsMemberOf)(LPCWSTR pwszPath, DWORD dwAttrib);

private:
	CComPtr<IAssemblyInfo> m_spAsmInfo;

};

OBJECT_ENTRY_AUTO(CLSID_AssemblyIconOverlay, CAssemblyIconOverlay)