filehandler.h

小巧的IE ATL组件。简单修改即可帮助你实现很多功能。让你的IE更加智能

// FileHandler.h : Declaration of the CFileHandler

#ifndef __FILEHANDLER_H_
#define __FILEHANDLER_H_

#include "resource.h"       // main symbols
#include "atlctl.h"
#define INTERFACE_UNSAFE 0

/////////////////////////////////////////////////////////////////////////////
// CFileHandler
class ATL_NO_VTABLE CFileHandler : 
	public CComObjectRootEx<CComSingleThreadModel>,
	public CComCoClass<CFileHandler, &CLSID_FileHandler>,
	public IDispatchImpl<IFileHandler, &IID_IFileHandler, &LIBID_OpenSave>,
	public IObjectSafetyImpl<CFileHandler,INTERFACE_UNSAFE>,
	public IObjectWithSiteImpl<CFileHandler>
{
public:
	CFileHandler()
	{
		m_pClientSite = NULL;
	}

DECLARE_REGISTRY_RESOURCEID(IDR_FILEHANDLER)

DECLARE_PROTECT_FINAL_CONSTRUCT()

BEGIN_COM_MAP(CFileHandler)
	COM_INTERFACE_ENTRY(IFileHandler)
	COM_INTERFACE_ENTRY(IDispatch)
	COM_INTERFACE_ENTRY(IObjectSafety)
	COM_INTERFACE_ENTRY(IObjectWithSite)
END_COM_MAP()

// IFileHandler
public:
	STDMETHOD(saveFile)(/*[in]*/ BSTR bstrPath, /*[in]*/ BSTR bstrFileText,/*[out,retval]*/ long *bSuccess);
	STDMETHOD(loadFile)(/*[in]*/ BSTR bstrPath, /*[out, retval]*/ BSTR *bstrFileText);
	STDMETHOD(getFilename)(/*[in]*/BSTR bstrAction,/*[in]*/ BSTR defaultName,/*[in]*/ BSTR defaultExt,/*[in]*/ BSTR defaultDescription, /*[out,retval]*/ BSTR *pName);
//IObjectWithSite
	STDMETHOD(SetSite)(IUnknown* pSite);
	IUnknown* m_pClientSite;
private:
	BOOL isSafeClientSite();
};

#endif //__FILEHANDLER_H_