//+---------------------------------------------------------------------------- // // DS Administration MMC snapin. // // Microsoft Windows // Copyright (C) Microsoft Corporation, 1992 - 1999 // // File: query.h // //-------------------------------------------------------------------------- #ifndef __QUERY_ #define __QUERY_ #include "dscookie.h" #include "dsfilter.h" // CDSAdminPersistQueryFilterImpl //////////////////////////////////////////////////////////////////////////// class CFavoritesNode : public CGenericUINode { public: CFavoritesNode(); void RemoveQueryResults(); void FindCookiesInQueries(LPCWSTR lpszCookieDN, CUINodeList* pNodeList); // // These set the state of the standard context menu items // virtual BOOL IsDeleteAllowed(CDSComponentData* pComponentData, BOOL* pbHide); virtual BOOL IsRenameAllowed(CDSComponentData* pComponentData, BOOL* pbHide); virtual BOOL IsRefreshAllowed(CDSComponentData* pComponentData, BOOL* pbHide); virtual BOOL IsCutAllowed(CDSComponentData* pComponentData, BOOL* pbHide); virtual BOOL IsCopyAllowed(CDSComponentData* pComponentData, BOOL* pbHide); virtual BOOL IsPasteAllowed(CDSComponentData* pComponentData, BOOL* pbHide); virtual BOOL ArePropertiesAllowed(CDSComponentData* pComponentData, BOOL* pbHide); virtual BOOL HasPropertyPages(LPDATAOBJECT) { return TRUE; } virtual CContextMenuVerbs* GetContextMenuVerbsObject(CDSComponentData* pComponentData); virtual HRESULT OnCommand(long lCommandID, CDSComponentData* pComponentData); virtual BOOL IsUniqueName(PCWSTR pszName, CUINode** ppDuplicateNode); virtual HRESULT Rename(LPCWSTR lpszNewName, CDSComponentData* pComponentData); virtual HRESULT Delete(CDSComponentData* pComponentData); virtual HRESULT DeleteMultiselect(CDSComponentData* pComponentData, CInternalFormatCracker* pObjCracker); virtual void Paste(IDataObject* pPasteData, CDSComponentData* pComponentData, LPDATAOBJECT* ppCutDataObj); virtual HRESULT QueryPaste(IDataObject* pPasteData, CDSComponentData* pComponentData); CFavoritesNode* GetFavoritesNode() { return dynamic_cast(GetParent()); } virtual BOOL IsFavoritesRoot() { return m_bFavoritesRoot; } virtual void SetRoot(BOOL bRoot) { m_bFavoritesRoot = bRoot; } virtual LPCWSTR GetDisplayString(int nCol, CDSColumnSet*) { if (IsFavoritesRoot()) { if (nCol == 0) return GetName(); else if (nCol == 2) return GetDesc(); } else { if (nCol == 0) return GetName(); else if (nCol == 1) return GetDesc(); } return L""; } // // Context menu command handlers // void OnImportQuery(CDSComponentData* pComponentData); void OnNewFavoritesFolder(CDSComponentData* pComponentData); void OnNewSavedQuery(CDSComponentData* pComponentData); BOOL DeepCopyChildren(CUINode* pUINodeToCopy); virtual HRESULT XMLSave(IXMLDOMDocument* pXMLDoc, IXMLDOMNode** ppXMLDOMNode); // // Used to load the embedded favorites root node // All other favorites folders are loaded through the static method XMLLoad // HRESULT CFavoritesNode::Load(IXMLDOMNode* pXDN, CDSComponentData* pComponentData); // // static members for XML serialization // static HRESULT XMLLoad(CDSComponentData* pComponentData, IXMLDOMNode* pXDN, CFavoritesNode** ppNode); static LPCWSTR g_szObjXMLTag; virtual HRESULT CreatePropertyPages(LPPROPERTYSHEETCALLBACK pCall, LONG_PTR lNotifyHandle, LPDATAOBJECT pDataObject, CDSComponentData* pComponentData); virtual void InvalidateSavedQueriesContainingObjects(CDSComponentData* pComponentData, const CStringList& refDNList); private: bool IsSameSnapin(IDataObject* pPasteData, CDSComponentData* pComponentData); BOOL m_bFavoritesRoot; }; //////////////////////////////////////////////////////////////////////////// HRESULT FilterDisabledAccounts(CUINodeList* pUINodeList); HRESULT FilterNonExpiringPwds(CUINodeList* pUINodeList); class CSavedQueryNode : public CGenericUINode { public: CSavedQueryNode(MyBasePathsInfo* pBasePathsInfo, SnapinType snapinType = SNAPINTYPE_DS); CSavedQueryNode(const CSavedQueryNode& copyNode); void SetRootPath(LPCWSTR lpszRootPath); void SetQueryString(LPCWSTR lpszQueryString) { m_szQueryString = lpszQueryString;} LPCWSTR GetRootPath(); LPCWSTR GetQueryString(); void AppendLastLogonQuery(CString& szQuery, DWORD dwDays); CFavoritesNode* GetFavoritesNode() { return dynamic_cast(GetParent()); } void SetOneLevel(BOOL bOneLevel) { m_bOneLevel = bOneLevel;} BOOL IsOneLevel() { return m_bOneLevel;} void SetValid(BOOL bValid) { m_bQueryValid = bValid; } BOOL IsValid() { return m_bQueryValid; } void SetQueryPersist(CComObject* pPersistQueryImpl); CComObject* GetQueryPersist() { return m_pPersistQueryImpl; } void FindCookieByDN(LPCWSTR lpszCookieDN, CUINodeList* pNodeList); void SetColumnID(CDSComponentData* pComponentData, PCWSTR pszColumnID); // // These set the state of the standard context menu items // virtual BOOL IsDeleteAllowed(CDSComponentData* pComponentData, BOOL* pbHide); virtual BOOL IsRenameAllowed(CDSComponentData* pComponentData, BOOL* pbHide); virtual BOOL IsRefreshAllowed(CDSComponentData* pComponentData, BOOL* pbHide); virtual BOOL IsPasteAllowed(CDSComponentData* pComponentData, BOOL* pbHide); virtual BOOL IsCutAllowed(CDSComponentData* pComponentData, BOOL* pbHide); virtual BOOL IsCopyAllowed(CDSComponentData* pComponentData, BOOL* pbHide); virtual CContextMenuVerbs* GetContextMenuVerbsObject(CDSComponentData* pComponentData); virtual HRESULT OnCommand(long lCommandID, CDSComponentData* pComponentData); virtual HRESULT Delete(CDSComponentData* pComponentData); virtual HRESULT Rename(LPCWSTR lpszNewName, CDSComponentData* pComponentData); void OnEditQuery(CDSComponentData* pComponentData); void OnExportQuery(CDSComponentData* pComponentData); virtual HRESULT XMLSave(IXMLDOMDocument* pXMLDoc, IXMLDOMNode** ppXMLDOMNode); // // static members for XML serialization // static HRESULT XMLLoad(CDSComponentData* pComponentData, IXMLDOMNode* pXDN, CSavedQueryNode** ppQuery); static LPCWSTR g_szObjXMLTag; static LPCWSTR g_szOneLevelXMLTag; static LPCWSTR g_szQueryStringXMLTag; static LPCWSTR g_szDSQueryPersistTag; static LPCWSTR g_szDisabledAccountsFilterTag; static LPCWSTR g_szNonExpiringPwdFilterTag; static LPCWSTR g_szLastLogonFilterTag; static LPCWSTR g_szColumnIDTag; BOOL IsFilterDisabledAccounts(); BOOL IsFilterNonExpiringPwds(); BOOL IsFilterLastLogon(); void SetLastLogonQuery(DWORD dwLastLogonDays) { if (dwLastLogonDays != (DWORD)-1) { m_bLastLogonFilter = TRUE; } m_dwLastLogonDays = dwLastLogonDays; } DWORD GetLastLogonDays() { return m_dwLastLogonDays; } virtual void InvalidateSavedQueriesContainingObjects(CDSComponentData* pComponentData, const CStringList& refDNList); private: CString m_szRelativeRootPath; CString m_szCurrentFullPath; CString m_szColumnID; CString m_szQueryString; BOOL m_bOneLevel; BOOL m_bQueryValid; BOOL m_bLastLogonFilter; DWORD m_dwLastLogonDays; MyBasePathsInfo* m_pBasePathsInfo; // // for presisting DSQuery dialog info // CComObject* m_pPersistQueryImpl; }; //////////////////////////////////////////////////////////////////////////// class CFavoritesNodesHolder { public: CFavoritesNodesHolder() { AFX_MANAGE_STATE(AfxGetStaticModuleState()); // fill in the root m_favoritesRoot.SetRoot(TRUE); CString szName; szName.LoadString(IDS_SAVED_QUERIES); m_favoritesRoot.SetName(szName); CString szDescription; szDescription.LoadString(IDS_SAVED_QUERIES_DESC); m_favoritesRoot.SetDesc(szDescription); } CFavoritesNode* GetFavoritesRoot() { return &m_favoritesRoot;} /* For test purposes ONLY void BuildTestTree(LPCWSTR lpszXMLFileName, SnapinType snapinType); BOOL BuildTestTreefromXML(LPCWSTR lpszXMLFileName, SnapinType snapinType); void BuildTestTreeHardWired(SnapinType snapinType); */ HRESULT Save(IStream* pStm); HRESULT Load(IStream* pStm, CDSComponentData* pComponentData); virtual void InvalidateSavedQueriesContainingObjects(CDSComponentData* pComponentData, const CStringList& refDNList); private: CFavoritesNode m_favoritesRoot; }; ////////////////////////////////////////////////////////////////////////////// class CDummyStream : public IStream, public CComObjectRoot { BEGIN_COM_MAP(CDummyStream) COM_INTERFACE_ENTRY(IStream) END_COM_MAP() public: CDummyStream() { m_pByteArray = NULL; m_nReadIndex = 0; m_nByteCount = 0; } ~CDummyStream() { if (m_pByteArray != NULL) { delete[] m_pByteArray; m_pByteArray = NULL; } } // // ISequentialStream Interfaces // HRESULT STDMETHODCALLTYPE Read(void *pv, ULONG cb, ULONG *pcbRead); HRESULT STDMETHODCALLTYPE Write(void const *pv, ULONG cb, ULONG *pcbWritten); // // IStream Interfaces (none of these are implemented) // HRESULT STDMETHODCALLTYPE Seek(LARGE_INTEGER, DWORD, ULARGE_INTEGER*) { return E_NOTIMPL; } HRESULT STDMETHODCALLTYPE SetSize(ULARGE_INTEGER) { return E_NOTIMPL; } HRESULT STDMETHODCALLTYPE CopyTo(IStream*, ULARGE_INTEGER, ULARGE_INTEGER*, ULARGE_INTEGER*) { return E_NOTIMPL; } HRESULT STDMETHODCALLTYPE Commit(DWORD) { return E_NOTIMPL; } HRESULT STDMETHODCALLTYPE Revert(void) { return E_NOTIMPL; } HRESULT STDMETHODCALLTYPE LockRegion(ULARGE_INTEGER, ULARGE_INTEGER, DWORD) { return E_NOTIMPL; } HRESULT STDMETHODCALLTYPE UnlockRegion(ULARGE_INTEGER, ULARGE_INTEGER, DWORD) { return E_NOTIMPL; } HRESULT STDMETHODCALLTYPE Stat(STATSTG*, DWORD) { return E_NOTIMPL; } HRESULT STDMETHODCALLTYPE Clone(IStream**) { return E_NOTIMPL; } ULONG GetStreamData(BYTE** ppByteArray); HRESULT SetStreamData(BYTE* pByteArray, ULONG nByteCount); private: BYTE* m_pByteArray; ULONG m_nByteCount; ULONG m_nReadIndex; }; #endif // __QUERY_