Category

Development Platform
Home > SourceCode/Document > Windows Develop > Windows Kernel > View Code
dsubscri.cpp
Package: shell.rar　[view]
Upload User: xhy777
Upload Date: 2007-02-14
Package Size: 24088k
Code Size: 5k
Category:

Windows Kernel

Development Platform:

Visual C++

dsubscri.cpp：Code Content
  1. #include "stdafx.h"
  2. #pragma hdrstop
  4. #ifdef POSTSPLIT
  6. // This function checks if a given URL already has a subscription.
  7. // Returns TRUE: if it aleady has a subscription 
  8. //         FALSE: Otherwise.
  9. //
  10. BOOL CheckForExistingSubscription(LPCTSTR lpcszURL)
  11. {
  12.     HRESULT hr;
  13.     ISubscriptionMgr *psm;
  14.     BOOL    fRet = FALSE;  //Assume failure.
  16.     //Create the subscription Manager.
  17.     hr = CoCreateInstance(CLSID_SubscriptionMgr, NULL,
  18.                           CLSCTX_INPROC_SERVER,
  19.                           IID_ISubscriptionMgr,
  20.                           (void**)&psm);
  22.     if (SUCCEEDED(hr))
  23.     {
  24.         BSTR bstrURL = SysAllocStringT(lpcszURL);
  25.         if (bstrURL)
  26.         {
  27.             psm->IsSubscribed(bstrURL, &fRet);
  28.             SysFreeString(bstrURL);
  29.         }
  31.         psm->Release();
  32.     }
  34.     return(fRet);
  35. }
  37. BOOL DeleteFromSubscriptionList(LPCTSTR pszURL)
  38. {
  39.     BOOL fRet = FALSE;
  40.     HRESULT hr;
  41.     ISubscriptionMgr *psm;
  43.     hr = CoCreateInstance(CLSID_SubscriptionMgr, NULL,
  44.                           CLSCTX_INPROC_SERVER,
  45.                           IID_ISubscriptionMgr,
  46.                           (void**)&psm);
  48.     if (SUCCEEDED(hr))
  49.     {
  50.         BSTR bstrURL = SysAllocStringT(pszURL);     // Call TSTR version
  51.         if (bstrURL)
  52.         {
  53.             //  Looks like all code paths going through this has already
  54.             //  put up some UI.
  55.             if (SUCCEEDED(psm->DeleteSubscription(bstrURL, NULL)))
  56.             {
  57.                 fRet = TRUE;
  58.             }
  60.             SysFreeString(bstrURL);
  61.         }
  63.         psm->Release();
  64.     }
  66.     return(fRet);
  67. }
  69. BOOL UpdateSubscription(LPCTSTR pszURL)
  70. {
  71.     BOOL fRet = FALSE;
  72.     HRESULT hr;
  73.     ISubscriptionMgr *psm;
  75.     hr = CoCreateInstance(CLSID_SubscriptionMgr, NULL,
  76.                           CLSCTX_INPROC_SERVER,
  77.                           IID_ISubscriptionMgr,
  78.                           (void**)&psm);
  80.     if (SUCCEEDED(hr))
  81.     {
  82.         BSTR bstrURL = SysAllocStringT(pszURL);     // Call TSTR version
  83.         if (bstrURL)
  84.         {
  85.             if (SUCCEEDED(psm->UpdateSubscription(bstrURL)))
  86.             {
  87.                 fRet = TRUE;
  88.             }
  90.             SysFreeString(bstrURL);
  91.         }
  93.         psm->Release();
  94.     }
  96.     return(fRet);
  97. }
  99. //
  100. //
  101. // This function enumerates the URLs of all the desktop components and then
  102. // calls webcheck to see if they are subcribed to and if so asks webcheck to
  103. // deliver those subscriptions right now.
  104. //
  105. //
  107. BOOL UpdateAllDesktopSubscriptions()
  108. {
  109.     IActiveDesktop  *pActiveDesktop;
  110.     ISubscriptionMgr *psm;
  111.     int     iCount; //Count of components.
  112.     HRESULT     hres;
  113.     BOOL        fRet = TRUE;  //Assume success!
  115.     if(FAILED(hres = CActiveDesktop_InternalCreateInstance((LPUNKNOWN *)&pActiveDesktop, IID_IActiveDesktop)))
  116.     {
  117.         TraceMsg(TF_WARNING, "Could not instantiate CActiveDesktop COM object");
  118.         return FALSE;
  119.     }
  121.     pActiveDesktop->GetDesktopItemCount(&iCount, 0);
  123.     if(iCount <= 0)
  124.     {
  125.         TraceMsg(DM_TRACE, "No desktop components to update!");
  126.         return TRUE; //No components to enumerate!
  127.     }
  129.     //Create the subscription Manager.
  130.     hres = CoCreateInstance(CLSID_SubscriptionMgr, NULL,
  131.                             CLSCTX_INPROC_SERVER,
  132.                             IID_ISubscriptionMgr,
  133.                             (void**)&psm);
  135.     if(SUCCEEDED(hres))
  136.     {
  137.         int iIndex;
  138.         BSTR bstrURL;
  140.         //Enumerate the desktop components one by one.
  141.         for(iIndex = 0; iIndex < iCount; iIndex++)
  142.         {
  143.             COMPONENT   Comp;   //We are using the public structure here.
  145.             Comp.dwSize = sizeof(COMPONENT);
  146.             if(SUCCEEDED(pActiveDesktop->GetDesktopItem(iIndex, &Comp, 0)) && 
  147.                         Comp.fChecked)  //Is this component enabled?
  148.             {
  149.                 BOOL    fSubscribed;
  151.                 fSubscribed = FALSE;  //Assume that it is NOT subscribed!
  153.                 bstrURL = SysAllocString(Comp.wszSubscribedURL);
  154.                 if(!bstrURL)
  155.                 {
  156.                     fRet = FALSE;
  157.                     break;  //Out of memory!
  158.                 }
  160.                 psm->IsSubscribed(bstrURL, &fSubscribed);
  162.                 if(fSubscribed)
  163.                     psm->UpdateSubscription(bstrURL);
  165.                 SysFreeString(bstrURL);
  166.             }
  167.             else
  168.                 TraceMsg(TF_WARNING, "Component# %d either failed or not enabled!", iIndex);
  169.         }
  170.         psm->Release();
  171.     }
  172.     else
  173.     {
  174.         TraceMsg(TF_WARNING, "Could not create CLSID_SubscriptionMgr");
  175.     }
  176.         
  177.     pActiveDesktop->Release();
  179.     return (fRet);
  180. }
  181. #endif