// Row.h : Declaration of the CRow

#if (!defined(BUILD_FOR_NT40))
#ifndef __ROW_H_
#define __ROW_H_

#define DBGUID_ROWURL   {0x0C733AB6L,0x2A1C,0x11CE,{0xAD,0xE5,0x00,0xAA,0x00,0x44,0x77,0x3D}}
#define NT_SEC_DESC_ATTR L"ntsecuritydescriptor"

extern const OLEDBDECLSPEC DBID DBROWCOL_ROWURL  = {DBGUID_ROWURL, DBKIND_GUID_PROPID, (LPOLESTR)0};

/////////////////////////////////////////////////////////////////////////////
// CRow
//
class ATL_NO_VTABLE CRow :
        INHERIT_TRACKING,
        public CComObjectRootEx<CComMultiThreadModel>,
        public CComCoClass<CRow, &CLSID_Row>,
        public ISupportErrorInfo,
        public IRow,
        public IColumnsInfo2,
        public IConvertType,
        public IGetSession
{
public:
        CRow()
        {
                m_pUnkMarshaler = NULL;

                m_pSession = NULL;
                m_pSourceRowset = NULL;
                m_hRow = DB_NULL_HROW;
                m_cMaxColumns = -1;
                m_pAttrInfo = NULL;
        }

DECLARE_GET_CONTROLLING_UNKNOWN()

BEGIN_COM_MAP(CRow)
        COM_INTERFACE_ENTRY(IRow)
        COM_INTERFACE_ENTRY(IColumnsInfo2)
        COM_INTERFACE_ENTRY(IColumnsInfo)
        COM_INTERFACE_ENTRY(IGetSession)
        COM_INTERFACE_ENTRY(IConvertType)
        COM_INTERFACE_ENTRY_AGGREGATE(IID_IMarshal, m_pUnkMarshaler.p)
END_COM_MAP()

        HRESULT FinalConstruct()
        {
                m_cSourceRowsetColumns = 0;
                m_pSourceRowsetColumnInfo = NULL;
                m_pSourceRowsetStringsBuffer = NULL;
                m_pRowProvider = NULL;

                RRETURN( CoCreateFreeThreadedMarshaler(
                        GetControllingUnknown(), &m_pUnkMarshaler.p));
        }

        void FinalRelease()
        {
                //Release Free Threaded Marshaler.
                m_pUnkMarshaler.Release();

                //If this is a row in a rowset, free rowset
                //column info and release the row handle.
                if (m_pSourceRowset)
                {
                        if (m_cSourceRowsetColumns)
                        {
                                CoTaskMemFree(m_pSourceRowsetColumnInfo);
                                CoTaskMemFree(m_pSourceRowsetStringsBuffer);
                        }
                        auto_rel<IRowset> pRowset;
                        HRESULT hr = m_pSourceRowset->QueryInterface(__uuidof(IRowset),
                                (void **)&pRowset);
                        if (FAILED(hr) || !pRowset)
                                return;
                        pRowset->ReleaseRows(1, &m_hRow, NULL, NULL, NULL);
                        m_pRowProvider = NULL;
                }
                if(m_pAttrInfo)
                        FreeADsMem(m_pAttrInfo);
        }

        CComPtr<IUnknown> m_pUnkMarshaler;

// ISupportsErrorInfo
        STDMETHOD(InterfaceSupportsErrorInfo)(REFIID riid);

// IRow
    STDMETHOD(GetColumns)(
            /* [in] */ DBORDINAL cColumns,
            /* [size_is][out][in] */ DBCOLUMNACCESS rgColumns[  ]);


    STDMETHOD(GetSourceRowset)(
            /* [in] */ REFIID riid,
            /* [iid_is][out] */ IUnknown __RPC_FAR *__RPC_FAR *ppRowset,
            /* [out] */ HROW *phRow);

    STDMETHOD(Open)(
            /* [unique][in] */ IUnknown *pUnkOuter,
            /* [in] */ DBID *pColumnID,
            /* [in] */ REFGUID rguidColumnType,
            /* [in] */ DWORD dwBindFlags,
            /* [in] */ REFIID riid,
            /* [iid_is][out] */ IUnknown **ppUnk);

//IColumnsInfo2 : IColumnsInfo
   STDMETHOD(GetColumnInfo)(
            /* [out][in] */ DBORDINAL *pcColumns,
            /* [size_is][size_is][out] */ DBCOLUMNINFO **prgInfo,
            /* [out] */ OLECHAR **ppStringsBuffer);

   STDMETHOD(MapColumnIDs)(
            /* [in] */ DBORDINAL cColumnIDs,
            /* [size_is][in] */ const DBID rgColumnIDs[  ],
            /* [size_is][out] */ DBORDINAL rgColumns[  ]);

    STDMETHOD(GetRestrictedColumnInfo)(
            /* [in] */ DBORDINAL cColumnIDMasks,
            /* [in] */ const DBID rgColumnIDMasks[  ],
            /* [in] */ DWORD dwFlags,
            /* [out][in] */ DBORDINAL *pcColumns,
            /* [size_is][size_is][out] */ DBID  **prgColumnIDs,
            /* [size_is][size_is][out] */ DBCOLUMNINFO **prgColumnInfo,
            /* [out] */ OLECHAR **ppStringsBuffer);

//IConvertType
    STDMETHOD(CanConvert)(
            /* [in] */ DBTYPE wFromType,
            /* [in] */ DBTYPE wToType,
            /* [in] */ DBCONVERTFLAGS dwConvertFlags);

//IGetSession
    STDMETHOD(GetSession)(
            REFIID riid,
            IUnknown **ppunkSession);

//Internal methods

//public
public:
        //Initializes a standalone CRow object.
        STDMETHOD(Initialize)(  PWSTR pwszURL,
                                                        IUnknown *pUnkSession,
                                                        IAuthenticate* pAuthenticate,
                                                        DWORD dwBindFlags,
                                                        BOOL fIsTearOff,
                                                        BOOL fGetColInfoFromRowset,
                                                        CCredentials *pSessCreds,
                                                        bool fBind = TRUE
                                                  );

        //Initializes a CRow object that represents a Row in a Rowset.
        STDMETHOD(Initialize) ( PWSTR pwszURL,
                                                        IUnknown* pUnkSession,
                                                        IUnknown* pUnkSourceRowset,
                                                        HROW      hRow,
                                                        PWSTR   pwszUserName,
                                                        PWSTR   pwszPassword,
                                                        DWORD   dwBindFlags,
                                                        BOOL fIsTearOff,
                                                        BOOL fGetColInfoFromRowset,
                                                        CRowProvider *pRowProvider
                                                  );

public:

private:
        //Internal methods

        //Function to check if a given column name
        //matches one of the source rowset's columns.
        //If a match is found, it returns the index
        //into the source rowset column array.
        int IsSourceRowsetColumn(PWCHAR pwszColumnName);

        //Functions to check if a given column name matches
        //Column ID Mask criteria.
        bool fMatchesMaskCriteria(PWCHAR pwszColumnName,
                                  ULONG cColumnIDMasks,
                                  const DBID rgColumnIDMasks[  ]);
                                  
        bool fMatchesMaskCriteria(DBID columnid,
                                  ULONG cColumnIDMasks,
                                  const DBID rgColumnIDMasks[  ]);

        //Function to check if a property is multi-valued.
        HRESULT GetSchemaAttributes(
                PWCHAR                  pwszColumnName,
                VARIANT_BOOL    *pfMultiValued,
                long                    *plMaxRange
                );

        //Function to get size and DBTYPE of a property
        HRESULT GetTypeAndSize(
                ADSTYPE                         dwADsType,
                CComBSTR&                       bstrPropName,
                DBTYPE*                         pdbType,
                ULONG*                          pulSise
                );

        //Function to get the root of schema tree eg. "LDAP://Schema"
        //This function gets this from m_pADsobj using ADSI functions
        //and stores the string in m_bstrSchemaRoot.
        HRESULT GetSchemaRoot();

        //Function to get columns from source rowset if this is a tear-off row.
        HRESULT GetSourceRowsetColumns(
            ULONG cColumns,
            DBCOLUMNACCESS rgColumns[  ],
            ULONG *pcErrors
            );

        //Function to get the status from HRESULT
        DBSTATUS StatusFromHRESULT(HRESULT hr);

        HRESULT GetSecurityDescriptor(DBCOLUMNACCESS *pColumn, 
                                          BOOL fMultiValued);
  
        int IgnorecbMaxLen(DBTYPE wType);

        auto_rel<IADs>                          m_pADsObj;
        CCredentials                            m_objCredentials;
        CComBSTR                                        m_objURL;
        CComBSTR                                        m_bstrSchemaRoot;
        auto_cs                                         m_autocs;
        auto_rel<IUnknown>                      m_pSession;
        auto_rel<IUnknown>                      m_pSourceRowset;
        HROW                                            m_hRow;
        DWORD                                           m_dwBindFlags;
        DBORDINAL                                           m_cSourceRowsetColumns;
        DBCOLUMNINFO                            *m_pSourceRowsetColumnInfo;
        OLECHAR                                         *m_pSourceRowsetStringsBuffer;
        CRowProvider                            *m_pRowProvider;
        BOOL                                    m_fIsTearOff;
        BOOL                                    m_fGetColInfoFromRowset;
        PADS_ATTR_INFO                          m_pAttrInfo;
        LONG                                    m_cMaxColumns;
};

#endif //__ROW_H_
#endif