///////////////////////////////////////////////////////////////////////////////////
//
// Microsoft WMIOLE DB Provider
// (C) Copyright 1999 Microsoft Corporation. All Rights Reserved.
//
// @module ROW.H | CRow base object and contained interface
// 
//
///////////////////////////////////////////////////////////////////////////////////

#ifndef _ROW_H_
#define _ROW_H_



typedef enum tagROWCREATEBINDFLAG
{
	ROWOPEN =0,ROWCREATE =1 , ROWOVERWRITE = 1

}ROWCREATEBINDFLAG;

typedef enum tagROWTYPE
{
	ROW_SCOPE =0,ROW_INSTANCE =1

}ROWTYPE;


#include "baserowobj.h"
class CImpIRowChange;
class CImpIColumnsInfo;
class CImpIGetSession;
class CImpIRow;
class CImpIScopedOperations;

//class CRowFetchObj;


typedef CImpIRowChange*			PIMPIROWCHANGE;
typedef CImpIRow*				PIMPIROW;
typedef CImpIGetSession*		PIMPIGETSESSION;
typedef CImpIScopedOperations*	PIMPISCOPEDOPERATIONS;

typedef CRowFetchObj*		PROWFETCHOBJ;


class CRow : public CBaseRowObj
{
	friend class CImpIRowChange;
	friend class CImpIRow;
	friend class CImpIGetSession;
	friend class CImpIScopedOperations;

	friend class CRowFetchObj;
	friend class CDBSession;

	PIMPIROW						m_pIRow;						// Contained IRow
	PIMPIGETSESSION					m_pIGetSession;					// Contained IGetSession
	PIMPIROWCHANGE					m_pIRowChange;					// Contained IRowChange;
	PIMPISCOPEDOPERATIONS			m_pIScopedOperations;			// Contained IScopedOperations


	UDWORD          				m_dwStatus;                     // status word for the entire cursor
	cRowColumnInfoMemMgr *			m_pRowColumns;						// column information for all the columns other than
																	// than the source rowset

	WCHAR *							m_strURL;						// URL of the row
	WCHAR *							m_strKey;						// Key of the row
	CWbemClassInstanceWrapper *		m_pInstance;						// Pointer to the instance corresponding to the row


	BOOL							m_bHelperFunctionCreated;		// flag to indicate whethe helper function are created or not
	BOOL							m_bClassChanged;				// flag which stores if the instance being represented, is
																	// is of different type ( belong to different class) as
																	// compared to the rowset which created it. This will 
																	// have meaning only when row is created from rowset

	CRowset	*						m_pSourcesRowset;				// Pointer to the rowset , if it this row is created by the row
	ROWTYPE							m_rowType;						// type of the object which the row represents							
	
	void	InitVars();
	HRESULT GetColumns(DBORDINAL cColumns,DBCOLUMNACCESS   rgColumns[ ]);
	HRESULT GetColumnsFromRowset(DBORDINAL cColumns,DBCOLUMNACCESS   rgColumns[]);
	HRESULT UpdateRow(DBORDINAL cColumns,DBCOLUMNACCESS rgColumns[ ]);
	HRESULT UpdateColumnsFromRowset(DBORDINAL cColumns,DBCOLUMNACCESS rgColumns[ ]);

	HRESULT GetColumnInfo(DBORDINAL* pcColumns, DBCOLUMNINFO** prgInfo,WCHAR** ppStringsBuffer);
	HRESULT GetRowColumnInfo();
	HRESULT GatherColumnInfo();
	HRESULT GetColumnInfo();
	HRESULT GetColumnData(DBCOLUMNACCESS &col);
	HRESULT SetColumnData(DBCOLUMNACCESS &col , BOOL bNewRow = FALSE);


	HRESULT GetRowData(DBORDINAL cColumns,DBCOLUMNACCESS   rgColumns[]);
	HRESULT SetRowData(DBORDINAL cColumns,DBCOLUMNACCESS   rgColumns[]);

	HRESULT OpenChild(IUnknown *pUnkOuter,WCHAR *strColName ,REFIID riid, IUnknown **ppUnk,BOOL bIsChildRowset);
	WCHAR * GetPropertyNameFromColName(WCHAR *pColName);
	HRESULT GetEmbededInstancePtrAndSetMapClass(CURLParser *urlParser);

	HRESULT GetChildRowet(	IUnknown *pUnkOuter,
						WCHAR *strColName,
						DBPROPSET *rgPropertySets,
						ULONG cPropSets,
						REFIID riid, 
						IUnknown **ppUnk);
	
	HRESULT GetChildRow(IUnknown *pUnkOuter,
						WCHAR *strColName,
						DBPROPSET *rgPropertySets,
						ULONG cPropSets,
						REFIID riid, 
						IUnknown **ppUnk);

	HRESULT CreateNewRow(CWbemClassInstanceWrapper **pNewInst);
	HRESULT AllocateInterfacePointers();
	HRESULT InitializeWmiOledbMap(LPWSTR strPath, 
								LPWSTR strTableID , 
								DWORD dwFlags);
	HRESULT	AddInterfacesForISupportErrorInfo();
	
	HRESULT OpenRow(LPCOLESTR pwszURL,IUnknown * pUnkOuter,REFIID riid,IUnknown** ppOut);
	HRESULT OpenRowset(LPCOLESTR pwszURL,
						IUnknown * pUnkOuter,
						REFIID riid,
						BOOL bContainer , 
						IUnknown** ppOut,
						ULONG cPropertySets = 0,
						DBPROPSET __RPC_FAR rgPropertySets[] = NULL);

	
	HRESULT SetRowProperties(const ULONG cPropertySets, const DBPROPSET	rgPropertySets[] );
	HRESULT Delete(LPCOLESTR lpszURL,DWORD dwDeleteFlags,DBSTATUS &dbStatus);
	HRESULT MoveObjects(WCHAR * pStrDstURL,WCHAR * pstrSrcURL, WCHAR *& pstrUrl,DBSTATUS &dbStatus);
	HRESULT CopyObjects(WCHAR * pStrDstURL,WCHAR * pstrSrcURL, WCHAR *& pNewURL,DBSTATUS &dbStatus);
	BOOL	GetPathFromURL(BSTR &str);
	BOOL	IsContainer();
	HRESULT GetIUnknownColumnValue(DBORDINAL iCol,REFIID riid,IUnknown ** ppUnk,LPCWSTR pStrColName);


	public:

		CRow(LPUNKNOWN,CRowset *pRowset,PCDBSESSION pObj,CWbemConnectionWrapper *pCon = NULL);
		CRow(LPUNKNOWN pUnkOuter, PCDBSESSION pObj);
		~CRow();

		HRESULT InitRow(HROW hRow = 0, 
					  CWbemClassWrapper *pInst = NULL,
					  ROWCREATEBINDFLAG rowCreateFlag = ROWOPEN);

		HRESULT InitRow(LPWSTR strPath , 
					  LPWSTR strTableID , 
					  DWORD dwFlags = -1,
					  ROWCREATEBINDFLAG rowCreateFlag = ROWOPEN);

		HRESULT UpdateKeysForNewInstance();

		STDMETHODIMP			QueryInterface(REFIID, LPVOID *);
		STDMETHODIMP_(ULONG)	AddRef(void);
		STDMETHODIMP_(ULONG)	Release(void);

};



class CImpIGetSession:public IGetSession
{
private:
		CRow		*m_pObj;											
		DEBUGCODE(ULONG m_cRef);

	public: 
		CImpIGetSession( CRow *pObj )				
		{																
			DEBUGCODE(m_cRef = 0L);										
			m_pObj		= pObj;											
		}																
		~CImpIGetSession()													
		{																
		}

		STDMETHODIMP_(ULONG) AddRef(void)									
		{																
			DEBUGCODE(InterlockedIncrement((long*)&m_cRef));						
			return m_pObj->GetOuterUnknown()->AddRef();								
		}																

		STDMETHODIMP_(ULONG) Release(void)									
		{																
		   DEBUGCODE(long lRef = InterlockedDecrement((long*)&m_cRef);
		   if( lRef < 0 ){
			   ASSERT("Reference count on Object went below 0!")
		   })
		
			return m_pObj->GetOuterUnknown()->Release();								
		}																

		STDMETHODIMP QueryInterface(REFIID riid, LPVOID *ppv)		
		{																	
			return m_pObj->GetOuterUnknown()->QueryInterface(riid, ppv);					
		}

		STDMETHODIMP GetSession(REFIID riid,IUnknown ** ppSession);


};


class CImpIRow:public IRow
{
private:
		CRow		*m_pObj;											
		DEBUGCODE(ULONG m_cRef);


	public: 
		CImpIRow( CRow *pObj )				
		{																
			DEBUGCODE(m_cRef = 0L);										
			m_pObj		= pObj;											
		}																
		~CImpIRow()													
		{																
		}

		STDMETHODIMP_(ULONG) AddRef(void)									
		{																
			DEBUGCODE(InterlockedIncrement((long*)&m_cRef));						
			return m_pObj->GetOuterUnknown()->AddRef();								
		}																

		STDMETHODIMP_(ULONG) Release(void)									
		{																
		   DEBUGCODE(long lRef = InterlockedDecrement((long*)&m_cRef);
		   if( lRef < 0 ){
			   ASSERT("Reference count on Object went below 0!")
		   })
		
			return m_pObj->GetOuterUnknown()->Release();								
		}																

		STDMETHODIMP QueryInterface(REFIID riid, LPVOID *ppv)		
		{																	
			return m_pObj->GetOuterUnknown()->QueryInterface(riid, ppv);					
		}

		STDMETHODIMP GetColumns(DBORDINAL cColumns,DBCOLUMNACCESS   rgColumns[ ]);

		STDMETHODIMP GetSourceRowset(REFIID riid,IUnknown ** ppRowset,HROW *phRow);
		STDMETHODIMP Open(  IUnknown *    pUnkOuter,
					   DBID *        pColumnID,
					   REFGUID       rguidColumnType,
					   DWORD         dwFlags,
					   REFIID        riid,
					   IUnknown **   ppUnk);


};



class CImpIRowChange:public IRowChange
{
private:
		CRow		*m_pObj;											
		DEBUGCODE(ULONG m_cRef);

	public: 
		CImpIRowChange( CRow *pObj )				
		{																
			DEBUGCODE(m_cRef = 0L);										
			m_pObj		= pObj;											
		}																
		~CImpIRowChange()													
		{																
		}

		STDMETHODIMP_(ULONG) AddRef(void)									
		{																
			DEBUGCODE(InterlockedIncrement((long*)&m_cRef));						
			return m_pObj->GetOuterUnknown()->AddRef();								
		}																

		STDMETHODIMP_(ULONG) Release(void)									
		{																
		   DEBUGCODE(long lRef = InterlockedDecrement((long*)&m_cRef);
		   if( lRef < 0 ){
			   ASSERT("Reference count on Object went below 0!")
		   })
		
			return m_pObj->GetOuterUnknown()->Release();								
		}																

		STDMETHODIMP QueryInterface(REFIID riid, LPVOID *ppv)		
		{																	
			return m_pObj->GetOuterUnknown()->QueryInterface(riid, ppv);					
		}

		STDMETHODIMP SetColumns(DBORDINAL cColumns,DBCOLUMNACCESS rgColumns[ ]);




};

class CImpIScopedOperations:public IScopedOperations
{
private:
		CRow		*m_pObj;											
		DEBUGCODE(ULONG m_cRef);

		BOOL	CheckBindURLFlags(DBBINDURLFLAG dwBindURLFlags , REFGUID rguid);	// Function to check if proper flags are
																				// are set for the require object
		// To check if URL Matches the requested type of object
		HRESULT	CheckIfProperURL(LPOLESTR & lpszURL,REFGUID rguid);

		HRESULT BindURL(IUnknown *            pUnkOuter,
					 LPCOLESTR             pwszURL,
					 DBBINDURLFLAG         dwBindURLFlags,
					 REFGUID               rguid,
					 REFIID                riid,
					 DBIMPLICITSESSION *   pImplSession,
					 DBBINDURLSTATUS *     pdwBindStatus,
					 IUnknown **           ppUnk);

		// Copy or move objects from one scope/container to another
		HRESULT ManipulateObjects(DBCOUNTITEM cRows,
									LPCOLESTR __RPC_FAR rgpwszSourceURLs[  ],
									LPCOLESTR __RPC_FAR rgpwszDestURLs[  ],
									DBSTATUS __RPC_FAR rgdwStatus[  ],
									LPOLESTR __RPC_FAR rgpwszNewURLs[  ],
									OLECHAR __RPC_FAR *__RPC_FAR *ppStringsBuffer,
									BOOL bMoveObjects);


	public: 
		CImpIScopedOperations( CRow *pObj )				
		{																
			DEBUGCODE(m_cRef = 0L);										
			m_pObj		= pObj;											
		}																
		~CImpIScopedOperations()													
		{																
		}

		STDMETHODIMP_(ULONG) AddRef(void)									
		{																
			DEBUGCODE(InterlockedIncrement((long*)&m_cRef));						
			return m_pObj->GetOuterUnknown()->AddRef();								
		}																

		STDMETHODIMP_(ULONG) Release(void)									
		{																
		   DEBUGCODE(long lRef = InterlockedDecrement((long*)&m_cRef);
		   if( lRef < 0 ){
			   ASSERT("Reference count on Object went below 0!")
		   })
		
			return m_pObj->GetOuterUnknown()->Release();								
		}																

		STDMETHODIMP QueryInterface(REFIID riid, LPVOID *ppv)		
		{																	
			return m_pObj->GetOuterUnknown()->QueryInterface(riid, ppv);					
		}


		STDMETHODIMP Bind(IUnknown *			 pUnkOuter,
						   LPCOLESTR             pwszURL,
						   DBBINDURLFLAG         dwBindURLFlags,
						   REFGUID               rguid,
						   REFIID                riid,
						   IAuthenticate *       pAuthenticate,
						   DBIMPLICITSESSION *   pImplSession,
						   DBBINDURLSTATUS *     pdwBindStatus,
						   IUnknown **           ppUnk);





        STDMETHODIMP Copy(DBCOUNTITEM cRows,
							LPCOLESTR __RPC_FAR rgpwszSourceURLs[  ],
							LPCOLESTR __RPC_FAR rgpwszDestURLs[  ],
							DWORD dwCopyFlags,
							IAuthenticate __RPC_FAR *pAuthenticate,
							DBSTATUS __RPC_FAR rgdwStatus[  ],
							LPOLESTR __RPC_FAR rgpwszNewURLs[  ],
							OLECHAR __RPC_FAR *__RPC_FAR *ppStringsBuffer);
        
        STDMETHODIMP Move(DBCOUNTITEM cRows,
							LPCOLESTR __RPC_FAR rgpwszSourceURLs[  ],
							LPCOLESTR __RPC_FAR rgpwszDestURLs[  ],
							DWORD dwMoveFlags,
							IAuthenticate __RPC_FAR *pAuthenticate,
							DBSTATUS __RPC_FAR rgdwStatus[  ],
							LPOLESTR __RPC_FAR rgpwszNewURLs[  ],
							OLECHAR __RPC_FAR *__RPC_FAR *ppStringsBuffer);
        
        STDMETHODIMP Delete(DBCOUNTITEM cRows,
							LPCOLESTR __RPC_FAR rgpwszURLs[  ],
							DWORD dwDeleteFlags,
							DBSTATUS __RPC_FAR rgdwStatus[  ]);
        
        STDMETHODIMP OpenRowset(IUnknown __RPC_FAR *pUnkOuter,
								DBID __RPC_FAR *pTableID,
								DBID __RPC_FAR *pIndexID,
								REFIID riid,
								ULONG cPropertySets,
								DBPROPSET __RPC_FAR rgPropertySets[  ],
								IUnknown __RPC_FAR *__RPC_FAR *ppRowset);
};



#endif