/*

 * Copyright (C) 2007 Apple Inc.  All rights reserved.

 *

 * Redistribution and use in source and binary forms, with or without

 * modification, are permitted provided that the following conditions

 * are met:

 * 1. Redistributions of source code must retain the above copyright

 *    notice, this list of conditions and the following disclaimer.

 * 2. Redistributions in binary form must reproduce the above copyright

 *    notice, this list of conditions and the following disclaimer in the

 *    documentation and/or other materials provided with the distribution.

 *

 * THIS SOFTWARE IS PROVIDED BY APPLE COMPUTER, INC. ``AS IS'' AND ANY

 * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE

 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR

 * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL APPLE COMPUTER, INC. OR

 * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,

 * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,

 * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR

 * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY

 * OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT

 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE

 * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 

 */

#include "WCDataObject.h"

#include "PlatformString.h"

namespace WebCore {

class WCEnumFormatEtc : public IEnumFORMATETC

{

public:

    WCEnumFormatEtc(const Vector<FORMATETC>& formats);

    WCEnumFormatEtc(const Vector<FORMATETC*>& formats);

    //IUnknown members

    STDMETHOD(QueryInterface)(REFIID, void**);

    STDMETHOD_(ULONG, AddRef)(void);

    STDMETHOD_(ULONG, Release)(void);

    //IEnumFORMATETC members

    STDMETHOD(Next)(ULONG, LPFORMATETC, ULONG*);

    STDMETHOD(Skip)(ULONG);

    STDMETHOD(Reset)(void);

    STDMETHOD(Clone)(IEnumFORMATETC**);

private:

    long m_ref;

    Vector<FORMATETC>  m_formats;

    size_t m_current;

};

WCEnumFormatEtc::WCEnumFormatEtc(const Vector<FORMATETC>& formats)

: m_ref(1)

, m_current(0)

{

    for(size_t i = 0; i < formats.size(); ++i)

        m_formats.append(formats[i]);

}

WCEnumFormatEtc::WCEnumFormatEtc(const Vector<FORMATETC*>& formats)

: m_ref(1)

, m_current(0)

{

    for(size_t i = 0; i < formats.size(); ++i)

        m_formats.append(*formats[i]);

}

STDMETHODIMP  WCEnumFormatEtc::QueryInterface(REFIID riid, void** ppvObject)

{

    *ppvObject = 0;

    if (IsEqualIID(riid, IID_IUnknown) || 

        IsEqualIID(riid, IID_IEnumFORMATETC)) {

        *ppvObject = this;

        AddRef();

        return S_OK;

    }

    return E_NOINTERFACE;

}

STDMETHODIMP_(ULONG) WCEnumFormatEtc::AddRef(void)

{

    return InterlockedIncrement(&m_ref);

}

STDMETHODIMP_(ULONG) WCEnumFormatEtc::Release(void)

{

    long c = InterlockedDecrement(&m_ref);

    if (c == 0)

        delete this;

    return c;

}

STDMETHODIMP WCEnumFormatEtc::Next(ULONG celt, LPFORMATETC lpFormatEtc, ULONG* pceltFetched)

{

    if(pceltFetched != 0)

        *pceltFetched=0;

    ULONG cReturn = celt;

    if(celt <= 0 || lpFormatEtc == 0 || m_current >= m_formats.size())

        return S_FALSE;

    if(pceltFetched == 0 && celt != 1) // pceltFetched can be 0 only for 1 item request

        return S_FALSE;

    while (m_current < m_formats.size() && cReturn > 0) {

        *lpFormatEtc++ = m_formats[m_current++];

        --cReturn;

    }

    if (pceltFetched != 0)

        *pceltFetched = celt - cReturn;

    return (cReturn == 0) ? S_OK : S_FALSE;

}

STDMETHODIMP WCEnumFormatEtc::Skip(ULONG celt)

{

    if((m_current + int(celt)) >= m_formats.size())

        return S_FALSE;

    m_current += celt;

    return S_OK;

}

STDMETHODIMP WCEnumFormatEtc::Reset(void)

{

    m_current = 0;

    return S_OK;

}

STDMETHODIMP WCEnumFormatEtc::Clone(IEnumFORMATETC** ppCloneEnumFormatEtc)

{

    if(!ppCloneEnumFormatEtc)

        return E_POINTER;

    WCEnumFormatEtc *newEnum = new WCEnumFormatEtc(m_formats);

    if(!newEnum)

        return E_OUTOFMEMORY;

    newEnum->AddRef();

    newEnum->m_current = m_current;

    *ppCloneEnumFormatEtc = newEnum;

    return S_OK;

}

//////////////////////////////////////////////////////////////////////////

HRESULT WCDataObject::createInstance(WCDataObject** result)

{

    if (!result)

        return E_POINTER;

    *result = new WCDataObject();

    return S_OK;

}

WCDataObject::WCDataObject()

: m_ref(1)

{

}

WCDataObject::~WCDataObject()

{

    for(size_t i = 0; i < m_medium.size(); ++i) {

        ReleaseStgMedium(m_medium[i]);

        delete m_medium[i];

    }

    WTF::deleteAllValues(m_formats);

}

STDMETHODIMP WCDataObject::QueryInterface(REFIID riid,void** ppvObject)

{

    *ppvObject = 0;

    if (IID_IUnknown==riid || IID_IDataObject==riid)

        *ppvObject=this;

    if (*ppvObject) {

        AddRef();

        return S_OK;

    }

    return E_NOINTERFACE;

}

STDMETHODIMP_(ULONG) WCDataObject::AddRef( void)

{

    return InterlockedIncrement(&m_ref);

}

STDMETHODIMP_(ULONG) WCDataObject::Release( void)

{

    long c = InterlockedDecrement(&m_ref);

    if (c == 0)

        delete this;

    return c;

}

STDMETHODIMP WCDataObject::GetData(FORMATETC* pformatetcIn, STGMEDIUM* pmedium)

{ 

    if(!pformatetcIn || !pmedium)

        return E_POINTER;

    pmedium->hGlobal = 0;

    for(size_t i=0; i < m_formats.size(); ++i) {

        if(/*pformatetcIn->tymed & m_formats[i]->tymed &&*/     // tymed can be 0 (TYMED_NULL) - but it can have a medium that contains an pUnkForRelease

            pformatetcIn->lindex == m_formats[i]->lindex &&

            pformatetcIn->dwAspect == m_formats[i]->dwAspect &&

            pformatetcIn->cfFormat == m_formats[i]->cfFormat) {

            CopyMedium(pmedium, m_medium[i], m_formats[i]);

            return S_OK;

        }

    }

    return DV_E_FORMATETC;

}

STDMETHODIMP WCDataObject::GetDataHere(FORMATETC*, STGMEDIUM*)

{ 

    return E_NOTIMPL;

}

STDMETHODIMP WCDataObject::QueryGetData(FORMATETC* pformatetc)

{ 

    if(!pformatetc)

        return E_POINTER;

    if (!(DVASPECT_CONTENT & pformatetc->dwAspect))

        return (DV_E_DVASPECT);

    HRESULT  hr = DV_E_TYMED;

    for(size_t i = 0; i < m_formats.size(); ++i) {

        if(pformatetc->tymed & m_formats[i]->tymed) {

            if(pformatetc->cfFormat == m_formats[i]->cfFormat)

                return S_OK;

            else

                hr = DV_E_CLIPFORMAT;

        }

        else

            hr = DV_E_TYMED;

    }

    return hr;

}

STDMETHODIMP WCDataObject::GetCanonicalFormatEtc(FORMATETC*, FORMATETC*)

{ 

    return DATA_S_SAMEFORMATETC;

}

STDMETHODIMP WCDataObject::SetData(FORMATETC* pformatetc, STGMEDIUM* pmedium, BOOL fRelease)

{ 

    if(!pformatetc || !pmedium)

        return E_POINTER;

    FORMATETC* fetc=new FORMATETC;

    if (!fetc)

        return E_OUTOFMEMORY;

    STGMEDIUM* pStgMed = new STGMEDIUM;

    if(!pStgMed) {

        delete fetc;

        return E_OUTOFMEMORY;

    }

    ZeroMemory(fetc,sizeof(FORMATETC));

    ZeroMemory(pStgMed,sizeof(STGMEDIUM));

    *fetc = *pformatetc;

    m_formats.append(fetc);

    if(fRelease)

        *pStgMed = *pmedium;

    else

        CopyMedium(pStgMed, pmedium, pformatetc);

    m_medium.append(pStgMed);

    return S_OK;

}

void WCDataObject::CopyMedium(STGMEDIUM* pMedDest, STGMEDIUM* pMedSrc, FORMATETC* pFmtSrc)

{

    switch(pMedSrc->tymed)

    {

    case TYMED_HGLOBAL:

        pMedDest->hGlobal = (HGLOBAL)OleDuplicateData(pMedSrc->hGlobal,pFmtSrc->cfFormat, 0);

        break;

    case TYMED_GDI:

        pMedDest->hBitmap = (HBITMAP)OleDuplicateData(pMedSrc->hBitmap,pFmtSrc->cfFormat, 0);

        break;

    case TYMED_MFPICT:

        pMedDest->hMetaFilePict = (HMETAFILEPICT)OleDuplicateData(pMedSrc->hMetaFilePict,pFmtSrc->cfFormat, 0);

        break;

    case TYMED_ENHMF:

        pMedDest->hEnhMetaFile = (HENHMETAFILE)OleDuplicateData(pMedSrc->hEnhMetaFile,pFmtSrc->cfFormat, 0);

        break;

    case TYMED_FILE:

        pMedSrc->lpszFileName = (LPOLESTR)OleDuplicateData(pMedSrc->lpszFileName,pFmtSrc->cfFormat, 0);

        break;

    case TYMED_ISTREAM:

        pMedDest->pstm = pMedSrc->pstm;

        pMedSrc->pstm->AddRef();

        break;

    case TYMED_ISTORAGE:

        pMedDest->pstg = pMedSrc->pstg;

        pMedSrc->pstg->AddRef();

        break;

    default:

        break;

    }

    pMedDest->tymed = pMedSrc->tymed;

    pMedDest->pUnkForRelease = 0;

    if(pMedSrc->pUnkForRelease != 0) {

        pMedDest->pUnkForRelease = pMedSrc->pUnkForRelease;

        pMedSrc->pUnkForRelease->AddRef();

    }

}

STDMETHODIMP WCDataObject::EnumFormatEtc(DWORD dwDirection, IEnumFORMATETC** ppenumFormatEtc)

{ 

    if(!ppenumFormatEtc)

        return E_POINTER;

    *ppenumFormatEtc=0;

    switch (dwDirection)

    {

    case DATADIR_GET:

        *ppenumFormatEtc= new WCEnumFormatEtc(m_formats);

        if(!(*ppenumFormatEtc))

            return E_OUTOFMEMORY;

        break;

    case DATADIR_SET:

    default:

        return E_NOTIMPL;

        break;

    }

    return S_OK;

}

STDMETHODIMP WCDataObject::DAdvise(FORMATETC*, DWORD, IAdviseSink*,DWORD*)

{ 

    return OLE_E_ADVISENOTSUPPORTED;

}

STDMETHODIMP WCDataObject::DUnadvise(DWORD)

{

    return E_NOTIMPL;

}

HRESULT STDMETHODCALLTYPE WCDataObject::EnumDAdvise(IEnumSTATDATA**)

{

    return OLE_E_ADVISENOTSUPPORTED;

}

void WCDataObject::clearData(CLIPFORMAT format)

{

    size_t ptr = 0;

    while (ptr < m_formats.size()) {

        if (m_formats[ptr]->cfFormat == format) {

            FORMATETC* current = m_formats[ptr];

            m_formats[ptr] = m_formats[m_formats.size() - 1];

            m_formats[m_formats.size() - 1] = 0;

            m_formats.removeLast();

            delete current;

            STGMEDIUM* medium = m_medium[ptr];

            m_medium[ptr] = m_medium[m_medium.size() - 1];

            m_medium[m_medium.size() - 1] = 0;

            m_medium.removeLast();

            ReleaseStgMedium(medium);

            delete medium;

            continue;

        }

        ptr++;

    }

}

}