/*
 * inetps.cpp - Property sheet implementation for Internet class.
 */


/* Headers
 **********/

#include "project.hpp"
#pragma hdrstop

#include "clsfact.h"
#include "inetcpl.h"
#include "inetps.hpp"


/***************************** Private Functions *****************************/


#pragma warning(disable:4100) /* "unreferenced formal parameter" warning */

/*
** InternetPropSheetCallback()
**
** Callback function called on Internet property sheets by PropertySheet().
**
** Arguments:
**
** Returns:
**
** Side Effects:  none
**
** N.b., a property sheet only receives a PSPCB_CREATE when it is activated,
** but a property sheet always receives a PSPCB_RELEASE.  A property sheet may
** recieve a PSPCB_RELEASE even though it has never recieved a PSPCB_CREATE.
*/
PRIVATE_CODE UINT CALLBACK InternetPropSheetCallback(HWND hwnd, UINT uMsg,
                                                     LPPROPSHEETPAGE ppsp)
{
   UINT uResult = TRUE;

   // uMsg may be any value.

   ASSERT(! hwnd ||
          IS_VALID_HANDLE(hwnd, WND));
   ASSERT(IS_VALID_STRUCT_PTR(ppsp, CPROPSHEETPAGE));

   switch (uMsg)
   {
      case PSPCB_CREATE:
         TRACE_OUT(("InternetPropSheetCallback(): Received PSPCB_CREATE.  Ignoring."));
         break;

      case PSPCB_RELEASE:
         TRACE_OUT(("InternetPropSheetCallback(): Received PSPCB_RELEASE."));
         /*
          * Assume that a property sheet's parent reference count is
          * decremented before its callback is called with PSPCB_RELEASE.
          */
         if (InternetCPLCanUnloadNow() == S_OK)
            UnloadInternetCPL();
         break;

      default:
         TRACE_OUT(("InternetPropSheetCallback(): Unhandled message %u.",
                    uMsg));
         break;
   }

   return(uResult);
}

#pragma warning(default:4100) /* "unreferenced formal parameter" warning */


PRIVATE_CODE HRESULT AddInternetPS(LPFNADDPROPSHEETPAGE pfnAddPage,
                                   LPARAM lparam)
{
   // lparam may be any value.

   ASSERT(IS_VALID_CODE_PTR(pfnAddPage, LPFNADDPROPSHEETPAGE));

   return(AddInternetPropertySheets(pfnAddPage, lparam,
                                    (PUINT)GetInternetCPLRefCountPtr(),
                                    &InternetPropSheetCallback));
}


/****************************** Public Functions *****************************/


#ifdef DEBUG

PUBLIC_CODE BOOL IsValidPCInternet(PCInternet pcinet)
{
   return(IS_VALID_READ_PTR(pcinet, CInternet) &&
          IS_VALID_STRUCT_PTR((PCRefCount)pcinet, CRefCount) &&
          IS_VALID_INTERFACE_PTR((PCIShellExtInit)pcinet, IShellExtInit) &&
          IS_VALID_INTERFACE_PTR((PCIShellPropSheetExt)pcinet, IShellPropSheetExt));
}

#endif


/********************************** Methods **********************************/


#pragma warning(disable:4705) /* "statement has no effect" warning - cl bug, see KB Q98989 */

Internet::Internet(OBJECTDESTROYEDPROC ObjectDestroyed) :
   RefCount(ObjectDestroyed)
{
   DebugEntry(Internet::Internet);

   // Don't validate this until after construction.

   ASSERT(! ObjectDestroyed ||
          IS_VALID_CODE_PTR(ObjectDestroyed, OBJECTDESTROYEDPROC));

   ASSERT(IS_VALID_STRUCT_PTR(this, CInternet));

   DebugExitVOID(Internet::Internet);

   return;
}

#pragma warning(default:4705) /* "statement has no effect" warning - cl bug, see KB Q98989 */


Internet::~Internet(void)
{
   DebugEntry(Internet::~Internet);

   ASSERT(IS_VALID_STRUCT_PTR(this, CInternet));

   DebugExitVOID(Internet::~Internet);

   return;
}


ULONG STDMETHODCALLTYPE Internet::AddRef(void)
{
   ULONG ulcRef;

   DebugEntry(Internet::AddRef);

   ASSERT(IS_VALID_STRUCT_PTR(this, CInternet));

   ulcRef = RefCount::AddRef();

   ASSERT(IS_VALID_STRUCT_PTR(this, CInternet));

   DebugExitULONG(Internet::AddRef, ulcRef);

   return(ulcRef);
}


ULONG STDMETHODCALLTYPE Internet::Release(void)
{
   ULONG ulcRef;

   DebugEntry(Internet::Release);

   ASSERT(IS_VALID_STRUCT_PTR(this, CInternet));

   ulcRef = RefCount::Release();

   DebugExitULONG(Internet::Release, ulcRef);

   return(ulcRef);
}


HRESULT STDMETHODCALLTYPE Internet::QueryInterface(REFIID riid,
                                                   PVOID *ppvObject)
{
   HRESULT hr = S_OK;

   DebugEntry(Internet::QueryInterface);

   ASSERT(IS_VALID_STRUCT_PTR(this, CInternet));
   ASSERT(IsValidREFIID(riid));
   ASSERT(IS_VALID_WRITE_PTR(ppvObject, PVOID));

   if (riid == IID_IShellExtInit)
   {
      *ppvObject = (PIShellExtInit)this;
      TRACE_OUT(("Internet::QueryInterface(): Returning IShellExtInit."));
   }
   else if (riid == IID_IShellPropSheetExt)
   {
      *ppvObject = (PIShellPropSheetExt)this;
      TRACE_OUT(("Internet::QueryInterface(): Returning IShellPropSheetExt."));
   }
   else if (riid == IID_IUnknown)
   {
      *ppvObject = (PIUnknown)(PIShellPropSheetExt)this;
      TRACE_OUT(("Internet::QueryInterface(): Returning IUnknown."));
   }
   else
   {
      TRACE_OUT(("Internet::QueryInterface(): Called on unknown interface."));
      *ppvObject = NULL;
      hr = E_NOINTERFACE;
   }

   if (hr == S_OK)
      AddRef();

   ASSERT(IS_VALID_STRUCT_PTR(this, CInternet));

   DebugExitHRESULT(Internet::QueryInterface, hr);

   return(hr);
}


#pragma warning(disable:4100) /* "unreferenced formal parameter" warning */

HRESULT STDMETHODCALLTYPE Internet::Initialize(PCITEMIDLIST pcidlFolder,
                                               PIDataObject pido,
                                               HKEY hkeyProgID)
{
   HRESULT hr;

   DebugEntry(Internet::Initialize);

   ASSERT(IS_VALID_STRUCT_PTR(this, CInternet));
   ASSERT(! pcidlFolder ||
          IS_VALID_STRUCT_PTR(pcidlFolder, CITEMIDLIST));
   ASSERT(IS_VALID_INTERFACE_PTR(pido, IDataObject));
   ASSERT(IS_VALID_HANDLE(hkeyProgID, KEY));

   // An Internet doesn't care where it lives in the name space.

   hr = S_OK;

   ASSERT(IS_VALID_STRUCT_PTR(this, CInternet));

   DebugExitHRESULT(Internet::Initialize, hr);

   return(hr);
}

#pragma warning(default:4100) /* "unreferenced formal parameter" warning */


HRESULT STDMETHODCALLTYPE Internet::AddPages(LPFNADDPROPSHEETPAGE pfnAddPage,
                                             LPARAM lparam)
{
   HRESULT hr;

   DebugEntry(Internet::AddPages);

   // lparam may be any value.

   ASSERT(IS_VALID_STRUCT_PTR(this, CInternet));
   ASSERT(IS_VALID_CODE_PTR(pfnAddPage, LPFNADDPROPSHEETPAGE));

   hr = AddInternetPS(pfnAddPage, lparam);

   ASSERT(IS_VALID_STRUCT_PTR(this, CInternet));

   DebugExitHRESULT(Internet::AddPages, hr);

   return(hr);
}


#pragma warning(disable:4100) /* "unreferenced formal parameter" warning */

HRESULT STDMETHODCALLTYPE Internet::ReplacePage(
                                          UINT uPageID,
                                          LPFNADDPROPSHEETPAGE pfnReplaceWith,
                                          LPARAM lparam)
{
   HRESULT hr;

   DebugEntry(Internet::ReplacePage);

   // lparam may be any value.
   // uPageID may be any value.

   ASSERT(IS_VALID_STRUCT_PTR(this, CInternet));
   ASSERT(IS_VALID_CODE_PTR(pfnReplaceWith, LPFNADDPROPSHEETPAGE));

   // No pages to replace.

   hr = E_NOTIMPL;

   ASSERT(IS_VALID_STRUCT_PTR(this, CInternet));

   DebugExitHRESULT(Internet::ReplacePage, hr);

   return(hr);
}

#pragma warning(default:4100) /* "unreferenced formal parameter" warning */