//+---------------------------------------------------------------------------
//
//  Microsoft Windows
//  Copyright (C) Microsoft Corporation, 1992 - 1997.
//
//  File:       cdlbsc.cxx
//
//  Contents:
//
//  Classes:
//
//  Functions:
//
//  History:    01-27-97   t-alans (Alan Shi)   Created
//
//----------------------------------------------------------------------------

#include <eapp.h>
#include "cdlbsc.hxx"

//+---------------------------------------------------------------------------
//
//  Method:     CCodeDLBSC::CCodeDLBSC
//
//  Synopsis:
//
//  Arguments:  
//              
//              
//
//  Returns:
//
//  History:    01-27-1997   t-alans (Alan Shi)   Created
//
//  Notes:      
//
//----------------------------------------------------------------------------

CCodeDLBSC::CCodeDLBSC(IOInetProtocolSink *pIOInetProtocolSink,
                       IOInetBindInfo *pIOInetBindInfo,
                       CCdlProtocol *pCDLProtocol,
                       BOOL fGetClassObject)
{
    _cRef = 1;
    _pIBinding = NULL;
    _pOInetProtocolSink = pIOInetProtocolSink;
    _fGetClassObject = fGetClassObject;

    if (_pOInetProtocolSink != NULL)
    {
        _pOInetProtocolSink->AddRef();
    }
    _pIOInetBindInfo = pIOInetBindInfo;
    if (_pIOInetBindInfo != NULL)
    {
        _pIOInetBindInfo->AddRef();
    }
    _pCDLProtocol = pCDLProtocol;
    if (_pCDLProtocol != NULL)
    {
        _pCDLProtocol->AddRef();
    }
}

//+---------------------------------------------------------------------------
//
//  Method:     CCodeDLBSC::~CCodeDLBSC
//
//  Synopsis:
//
//  Arguments:  
//              
//              
//
//  Returns:
//
//  History:    01-27-1997   t-alans (Alan Shi)   Created
//
//  Notes:      
//
//----------------------------------------------------------------------------

CCodeDLBSC::~CCodeDLBSC()
{
    if (_pOInetProtocolSink != NULL)
    {
        _pOInetProtocolSink->Release();
    }
    if (_pIOInetBindInfo != NULL)
    {
        _pIOInetBindInfo->Release();
    }
    if (_pCDLProtocol != NULL)
    {
        _pCDLProtocol->ClearCodeDLBSC();
        _pCDLProtocol->Release();
    }
}

//+---------------------------------------------------------------------------
//
//  Method:     CCodeDLBSC::Abort
//
//  Synopsis:
//
//  Arguments:  
//              
//              
//
//  Returns:
//
//  History:    01-27-1997   t-alans (Alan Shi)   Created
//
//  Notes:      
//
//----------------------------------------------------------------------------

HRESULT CCodeDLBSC::Abort()
{
    if (_pIBinding)
    {
        return _pIBinding->Abort();
    }
    else
    {
        return S_OK;
    }
}

/*
 *
 * IUnknown Methods
 *
 */

//+---------------------------------------------------------------------------
//
//  Method:     CCodeDLBSC::QueryInterface
//
//  Synopsis:
//
//  Arguments:  
//              
//              
//
//  Returns:
//
//  History:    01-27-1997   t-alans (Alan Shi)   Created
//
//  Notes:      
//
//----------------------------------------------------------------------------

STDMETHODIMP CCodeDLBSC::QueryInterface(REFIID riid, void **ppv)
{
    HRESULT          hr = E_NOINTERFACE;

    *ppv = NULL;
    if (riid == IID_IUnknown || riid == IID_IBindStatusCallback)
    {
        *ppv = (IBindStatusCallback *)this;
    }
    else if (riid == IID_IServiceProvider)
    {
        *ppv = (IServiceProvider *)this;
    }
    if (*ppv != NULL)
    {
        ((IUnknown *)*ppv)->AddRef();
        hr = S_OK;
    }

    return hr;
}

//+---------------------------------------------------------------------------
//
//  Method:     CCodeDLBSC::AddRef
//
//  Synopsis:
//
//  Arguments:  
//              
//              
//
//  Returns:
//
//  History:    01-27-1997   t-alans (Alan Shi)   Created
//
//  Notes:      
//
//----------------------------------------------------------------------------

STDMETHODIMP_(ULONG) CCodeDLBSC::AddRef()
{
    return ++_cRef;
}

//+---------------------------------------------------------------------------
//
//  Method:     CCodeDLBSC::Release
//
//  Synopsis:
//
//  Arguments:  
//              
//              
//
//  Returns:
//
//  History:    01-27-1997   t-alans (Alan Shi)   Created
//
//  Notes:      
//
//----------------------------------------------------------------------------

STDMETHODIMP_(ULONG) CCodeDLBSC::Release()
{
    if (--_cRef)
    {
        return _cRef;
    }
    delete this;

    return 0;
}

/*
 *
 * IBindStatusCallback Methods
 *
 */

//+---------------------------------------------------------------------------
//
//  Method:     CCodeDLBSC::OnStartBinding
//
//  Synopsis:
//
//  Arguments:  
//              
//              
//
//  Returns:
//
//  History:    01-27-1997   t-alans (Alan Shi)   Created
//
//  Notes:      
//
//----------------------------------------------------------------------------

STDMETHODIMP CCodeDLBSC::OnStartBinding(DWORD grfBSCOption, IBinding *pib)
{
    if (_pIBinding != NULL)
    {
        _pIBinding->Release();
    }
    _pIBinding = pib;

    if (_pIBinding != NULL)
    {
        _pIBinding->AddRef();
    }

    return S_OK;
}

//+---------------------------------------------------------------------------
//
//  Method:     CCodeDLBSC::OnStopBinding
//
//  Synopsis:
//
//  Arguments:  
//              
//              
//
//  Returns:
//
//  History:    01-27-1997   t-alans (Alan Shi)   Created
//
//  Notes:      
//
//----------------------------------------------------------------------------

STDMETHODIMP CCodeDLBSC::OnStopBinding(HRESULT hresult, LPCWSTR szError)
{
    HRESULT                   hr = S_OK;
    DWORD                     dwError = 0;
    DWORD                     grfBINDF = 0;
    BINDINFO                  bindinfo;

    _pCDLProtocol->SetDataPending(FALSE);

    if (SUCCEEDED(hresult))
    {
        if (_fGetClassObject)
        {
            EProtAssert(_pUnk);

            // put _pUnk into the bind context for CBinding to retrieve

            hresult = _pCDLProtocol->RegisterIUnknown(_pUnk);

            // no need for _pUnk anymore, release it

            _pUnk->Release();
            _pUnk = NULL;

            if (SUCCEEDED(hresult))
            {
                _pOInetProtocolSink->ReportProgress(BINDSTATUS_IUNKNOWNAVAILABLE,
                                                    NULL);
            }
        }
        else
        {
            if (!IsEqualGUID(_pCDLProtocol->GetClsid() , CLSID_NULL))
            {
                LPOLESTR pwzStrClsId;
                StringFromCLSID(_pCDLProtocol->GetClsid(), &pwzStrClsId);
                _pOInetProtocolSink->ReportProgress(BINDSTATUS_CLSIDCANINSTANTIATE, pwzStrClsId);
        
                delete [] pwzStrClsId;
            }
        }
    }

    hr = _pOInetProtocolSink->ReportResult(hresult, dwError, szError);

    return hr;
}

//+---------------------------------------------------------------------------
//
//  Method:     CCodeDLBSC::OnObjectAvailable
//
//  Synopsis:
//
//  Arguments:  
//              
//              
//
//  Returns:
//
//  History:    01-27-1997   t-alans (Alan Shi)   Created
//
//  Notes:      
//
//----------------------------------------------------------------------------

STDMETHODIMP CCodeDLBSC::OnObjectAvailable(REFIID riid, IUnknown *punk)
{
    EProtAssert(!_pUnk && punk);

    _pUnk = punk;
    _pUnk->AddRef();

    return S_OK;
}

//+---------------------------------------------------------------------------
//
//  Method:     CCodeDLBSC::GetPriority
//
//  Synopsis:
//
//  Arguments:  
//              
//              
//
//  Returns:
//
//  History:    01-27-1997   t-alans (Alan Shi)   Created
//
//  Notes:      
//
//----------------------------------------------------------------------------

STDMETHODIMP CCodeDLBSC::GetPriority(LONG *pnPriority)
{
    return S_OK;
}

//+---------------------------------------------------------------------------
//
//  Method:     CCodeDLBSC::OnLowResource
//
//  Synopsis:
//
//  Arguments:  
//              
//              
//
//  Returns:
//
//  History:    01-27-1997   t-alans (Alan Shi)   Created
//
//  Notes:      
//
//----------------------------------------------------------------------------

STDMETHODIMP CCodeDLBSC::OnLowResource(DWORD dwReserved)
{
    return S_OK;
}  

//+---------------------------------------------------------------------------
//
//  Method:     CCodeDLBSC::OnProgress
//
//  Synopsis:
//
//  Arguments:  
//              
//              
//
//  Returns:
//
//  History:    01-27-1997   t-alans (Alan Shi)   Created
//
//  Notes:      
//
//----------------------------------------------------------------------------

STDMETHODIMP CCodeDLBSC::OnProgress(ULONG ulProgress, ULONG ulProgressMax,
                                    ULONG ulStatusCode,
                                    LPCWSTR szStatusText)
{
    EProtAssert(_pOInetProtocolSink != NULL);
    return _pOInetProtocolSink->ReportProgress(ulStatusCode, szStatusText);
}

//+---------------------------------------------------------------------------
//
//  Method:     CCodeDLBSC::GetBindInfo
//
//  Synopsis:
//
//  Arguments:  
//              
//              
//
//  Returns:
//
//  History:    01-27-1997   t-alans (Alan Shi)   Created
//
//  Notes:      
//
//----------------------------------------------------------------------------

STDMETHODIMP CCodeDLBSC::GetBindInfo(DWORD *pgrfBINDF, BINDINFO *pbindInfo)
{
    EProtAssert(_pIOInetBindInfo != NULL);
    return _pIOInetBindInfo->GetBindInfo(pgrfBINDF, pbindInfo);
}

//+---------------------------------------------------------------------------
//
//  Method:     CCodeDLBSC::OnDataAvailable
//
//  Synopsis:
//
//  Arguments:  
//              
//              
//
//  Returns:
//
//  History:    01-27-1997   t-alans (Alan Shi)   Created
//
//  Notes:      
//
//----------------------------------------------------------------------------

STDMETHODIMP CCodeDLBSC::OnDataAvailable(DWORD grfBSCF, DWORD dwSize,
                                         FORMATETC *pformatetc,
                                         STGMEDIUM *pstgmed)
{
    return S_OK;
}

/*
 *
 * IWindowForBindingUI Methods
 *
 */

//+---------------------------------------------------------------------------
//
//  Method:     CCodeDLBSC::QueryService
//
//  Synopsis:
//
//  Arguments:  
//              
//              
//
//  Returns:
//
//  History:    01-27-1997   t-alans (Alan Shi)   Created
//
//  Notes:      
//
//----------------------------------------------------------------------------

STDMETHODIMP CCodeDLBSC::QueryService(REFGUID rsid, REFIID riid, void ** ppvObj)
{
    HRESULT     hr = NOERROR;
    IServiceProvider        *pIServiceProvider = NULL;

    EProtAssert(ppvObj);
    if (!ppvObj)
        return E_INVALIDARG;

    *ppvObj = 0;

    hr = _pOInetProtocolSink->QueryInterface(IID_IServiceProvider,
                                             (LPVOID *)&pIServiceProvider);

    if (SUCCEEDED(hr))
    {
        hr = pIServiceProvider->QueryService(rsid, riid, (LPVOID *)ppvObj);
        pIServiceProvider->Release();
    }

    return hr;
}