/*****************************************************************************
 * plugin.cpp: ActiveX control for VLC
 *****************************************************************************
 * Copyright (C) 2005 VideoLAN
 *
 * Authors: Damien Fouilleul <Damien.Fouilleul@laposte.net>
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 2 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111, USA.
 *****************************************************************************/

#include "plugin.h"

#include "oleobject.h"
#include "olecontrol.h"
#include "oleinplaceobject.h"
#include "oleinplaceactiveobject.h"
#include "persistpropbag.h"
#include "persiststreaminit.h"
#include "persiststorage.h"
#include "provideclassinfo.h"
#include "connectioncontainer.h"
#include "objectsafety.h"
#include "vlccontrol.h"

#include "utils.h"

#include <string.h>
#include <winreg.h>

using namespace std;

////////////////////////////////////////////////////////////////////////
//class factory

// {E23FE9C6-778E-49d4-B537-38FCDE4887D8}
//const GUID CLSID_VLCPlugin = 
//    { 0xe23fe9c6, 0x778e, 0x49d4, { 0xb5, 0x37, 0x38, 0xfc, 0xde, 0x48, 0x87, 0xd8 } };

static LRESULT CALLBACK VLCInPlaceClassWndProc(HWND hWnd, UINT uMsg, WPARAM wParam, LPARAM lParam) {
    switch( uMsg )
    {
        case WM_ERASEBKGND:
            return 1L;

        case WM_PAINT:
            PAINTSTRUCT ps;
            if( GetUpdateRect(hWnd, NULL, FALSE) )
            {
                BeginPaint(hWnd, &ps);
                EndPaint(hWnd, &ps);
            }
            return 0L;

        default:
            return DefWindowProc(hWnd, uMsg, wParam, lParam);
    }
};

static LRESULT CALLBACK VLCVideoClassWndProc(HWND hWnd, UINT uMsg, WPARAM wParam, LPARAM lParam) {
    VLCPlugin *p_instance = reinterpret_cast<VLCPlugin *>(GetWindowLongPtr(hWnd, GWLP_USERDATA));

    switch( uMsg )
    {
        case WM_ERASEBKGND:
            return 1L;

        case WM_PAINT:
            PAINTSTRUCT ps;
            RECT pr;
            if( GetUpdateRect(hWnd, &pr, FALSE) )
            {
                BeginPaint(hWnd, &ps);
                p_instance->onPaint(ps, pr);
                EndPaint(hWnd, &ps);
            }
            return 0L;

        default:
            return DefWindowProc(hWnd, uMsg, wParam, lParam);
    }
};

VLCPluginClass::VLCPluginClass(LONG *p_class_ref, HINSTANCE hInstance) :
    _p_class_ref(p_class_ref),
    _hinstance(hInstance)
{
    WNDCLASS wClass;

    if( ! GetClassInfo(hInstance, getInPlaceWndClassName(), &wClass) )
    {
        wClass.style          = CS_NOCLOSE|CS_HREDRAW|CS_VREDRAW|CS_DBLCLKS;
        wClass.lpfnWndProc    = VLCInPlaceClassWndProc;
        wClass.cbClsExtra     = 0;
        wClass.cbWndExtra     = 0;
        wClass.hInstance      = hInstance;
        wClass.hIcon          = NULL;
        wClass.hCursor        = LoadCursor(NULL, IDC_ARROW);
        wClass.hbrBackground  = NULL;
        wClass.lpszMenuName   = NULL;
        wClass.lpszClassName  = getInPlaceWndClassName();
       
        _inplace_wndclass_atom = RegisterClass(&wClass);
    }
    else
    {
        _inplace_wndclass_atom = 0;
    }

    if( ! GetClassInfo(hInstance, getVideoWndClassName(), &wClass) )
    {
        wClass.style          = CS_NOCLOSE|CS_HREDRAW|CS_VREDRAW;
        wClass.lpfnWndProc    = VLCVideoClassWndProc;
        wClass.cbClsExtra     = 0;
        wClass.cbWndExtra     = 0;
        wClass.hInstance      = hInstance;
        wClass.hIcon          = NULL;
        wClass.hCursor        = LoadCursor(NULL, IDC_ARROW);
        wClass.hbrBackground  = NULL;
        wClass.lpszMenuName   = NULL;
        wClass.lpszClassName  = getVideoWndClassName();
       
        _video_wndclass_atom = RegisterClass(&wClass);
    }
    else
    {
        _video_wndclass_atom = 0;
    }

    _inplace_hbitmap = (HBITMAP)LoadImage(getHInstance(), TEXT("INPLACE-PICT"), IMAGE_BITMAP, 0, 0, LR_DEFAULTCOLOR);

    AddRef();
};

VLCPluginClass::~VLCPluginClass()
{
    if( 0 != _inplace_wndclass_atom )
        UnregisterClass(MAKEINTATOM(_inplace_wndclass_atom), _hinstance);

    if( 0 != _video_wndclass_atom )
        UnregisterClass(MAKEINTATOM(_video_wndclass_atom), _hinstance);

    if( NULL != _inplace_hbitmap )
        DeleteObject(_inplace_hbitmap);
};

STDMETHODIMP VLCPluginClass::QueryInterface(REFIID riid, void **ppv)
{
    if( NULL == ppv )
        return E_INVALIDARG;

    if( (IID_IUnknown == riid) || (IID_IClassFactory == riid) )
    {
        AddRef();
        *ppv = reinterpret_cast<LPVOID>(this);

        return NOERROR;
    }

    *ppv = NULL;

    return E_NOINTERFACE;
};

STDMETHODIMP_(ULONG) VLCPluginClass::AddRef(void)
{
    return InterlockedIncrement(_p_class_ref);
};

STDMETHODIMP_(ULONG) VLCPluginClass::Release(void)
{
    ULONG refcount = InterlockedDecrement(_p_class_ref);
    if( 0 == refcount )
    {
        delete this;
        return 0;
    }
    return refcount;
};

STDMETHODIMP VLCPluginClass::CreateInstance(IUnknown *pUnkOuter, REFIID riid, void **ppv)
{
    if( NULL == ppv )
        return E_POINTER;

    *ppv = NULL;

    if( NULL != pUnkOuter )
        return CLASS_E_NOAGGREGATION;

    VLCPlugin *plugin = new VLCPlugin(this);
    if( NULL != plugin )
    {
        HRESULT hr = plugin->QueryInterface(riid, ppv);
        plugin->Release();
        return hr;
    }
    return E_OUTOFMEMORY;
};

STDMETHODIMP VLCPluginClass::LockServer(BOOL fLock)
{
    if( fLock )
        AddRef();
    else 
        Release();

    return S_OK;
};

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

VLCPlugin::VLCPlugin(VLCPluginClass *p_class) :
    _inplacewnd(NULL),
    _p_class(p_class),
    _i_ref(1UL),
    _codepage(CP_ACP),
    _psz_src(NULL),
    _b_autostart(TRUE),
    _b_loopmode(FALSE),
    _b_showdisplay(TRUE),
    _b_sendevents(TRUE),
    _i_vlc(0)
{
    p_class->AddRef();

    vlcOleObject = new VLCOleObject(this);
    vlcOleControl = new VLCOleControl(this);
    vlcOleInPlaceObject = new VLCOleInPlaceObject(this);
    vlcOleInPlaceActiveObject = new VLCOleInPlaceActiveObject(this);
    vlcPersistStorage = new VLCPersistStorage(this);
    vlcPersistStreamInit = new VLCPersistStreamInit(this);
    vlcPersistPropertyBag = new VLCPersistPropertyBag(this);
    vlcProvideClassInfo = new VLCProvideClassInfo(this);
    vlcConnectionPointContainer = new VLCConnectionPointContainer(this);
    vlcObjectSafety = new VLCObjectSafety(this);
    vlcControl = new VLCControl(this);
};

VLCPlugin::~VLCPlugin()
{
    vlcOleInPlaceObject->UIDeactivate();
    vlcOleInPlaceObject->InPlaceDeactivate();

    delete vlcControl;
    delete vlcObjectSafety;
    delete vlcConnectionPointContainer;
    delete vlcProvideClassInfo;
    delete vlcPersistPropertyBag;
    delete vlcPersistStreamInit;
    delete vlcPersistStorage;
    delete vlcOleInPlaceActiveObject;
    delete vlcOleInPlaceObject;
    delete vlcOleControl;
    delete vlcOleObject;

    if( _psz_src )
        free(_psz_src);

    _p_class->Release();
};

STDMETHODIMP VLCPlugin::QueryInterface(REFIID riid, void **ppv)
{
    if( NULL == ppv )
        return E_INVALIDARG;

    if( IID_IUnknown == riid )
    {
        AddRef();
        *ppv = reinterpret_cast<LPVOID>(this);
        return NOERROR;
    }
    else if( IID_IOleObject == riid )
    {
        AddRef();
        *ppv = reinterpret_cast<LPVOID>(vlcOleObject);
        return NOERROR;
    }
    else if( IID_IOleControl == riid )
    {
        AddRef();
        *ppv = reinterpret_cast<LPVOID>(vlcOleControl);
        return NOERROR;
    }
    else if( IID_IOleWindow == riid )
    {
        AddRef();
        *ppv = reinterpret_cast<LPVOID>(vlcOleInPlaceObject);
        return NOERROR;
    }
    else if( IID_IOleInPlaceObject == riid )
    {
        AddRef();
        *ppv = reinterpret_cast<LPVOID>(vlcOleInPlaceObject);
        return NOERROR;
    }
    else if( IID_IOleInPlaceActiveObject == riid )
    {
        AddRef();
        *ppv = reinterpret_cast<LPVOID>(vlcOleInPlaceActiveObject);
        return NOERROR;
    }
    else if( IID_IPersist == riid )
    {
        AddRef();
        *ppv = reinterpret_cast<LPVOID>(vlcPersistPropertyBag);
        return NOERROR;
    }
    else if( IID_IPersistStreamInit == riid )
    {
        AddRef();
        *ppv = reinterpret_cast<LPVOID>(vlcPersistStreamInit);
        return NOERROR;
    }
    else if( IID_IPersistStorage == riid )
    {
        AddRef();
        *ppv = reinterpret_cast<LPVOID>(vlcPersistStorage);
        return NOERROR;
    }
    else if( IID_IPersistPropertyBag == riid )
    {
        AddRef();
        *ppv = reinterpret_cast<LPVOID>(vlcPersistPropertyBag);
        return NOERROR;
    }
    else if( IID_IProvideClassInfo == riid )
    {
        AddRef();
        *ppv = reinterpret_cast<LPVOID>(vlcProvideClassInfo);
        return NOERROR;
    }
    else if( IID_IProvideClassInfo2 == riid )
    {
        AddRef();
        *ppv = reinterpret_cast<LPVOID>(vlcProvideClassInfo);
        return NOERROR;
    }
    else if( IID_IConnectionPointContainer == riid )
    {
        AddRef();
        *ppv = reinterpret_cast<LPVOID>(vlcConnectionPointContainer);
        return NOERROR;
    }
    else if( IID_IObjectSafety == riid )
    {
        AddRef();
        *ppv = reinterpret_cast<LPVOID>(vlcObjectSafety);
        return NOERROR;
    }
    else if( IID_IDispatch == riid )
    {
        AddRef();
        *ppv = reinterpret_cast<LPVOID>(vlcControl);
        return NOERROR;
    }
    else if( IID_IVLCControl == riid )
    {
        AddRef();
        *ppv = reinterpret_cast<LPVOID>(vlcControl);
        return NOERROR;
    }

    *ppv = NULL;

    return E_NOINTERFACE;
};

STDMETHODIMP_(ULONG) VLCPlugin::AddRef(void)
{
    return InterlockedIncrement((LONG *)&_i_ref);
};

STDMETHODIMP_(ULONG) VLCPlugin::Release(void)
{
    if( ! InterlockedDecrement((LONG *)&_i_ref) )
    {
        delete this;
        return 0;
    }
    return _i_ref;
};

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

/*
** we use an in-place child window to represent plugin viewport,
** whose size is limited by the clipping rectangle
** all drawing within this window must follow 
** cartesian coordinate system represented by _bounds.
*/

void VLCPlugin::calcPositionChange(LPRECT lprPosRect, LPCRECT lprcClipRect)
{
    _bounds.right  = lprPosRect->right-lprPosRect->left;

    if( lprcClipRect->left <= lprPosRect->left )
    {
        // left side is not clipped out
        _bounds.left = 0;

        if( lprcClipRect->right >= lprPosRect->right )
        {
            // right side is not clipped out, no change
        }
        else if( lprcClipRect->right >= lprPosRect->left )
        {
            // right side is clipped out
            lprPosRect->right = lprcClipRect->right;
        }
        else
        {
            // outside of clipping rectange, not visible
            lprPosRect->right = lprPosRect->left;
        }
    }
    else
    {
        // left side is clipped out
        _bounds.left = lprPosRect->left-lprcClipRect->left;
        _bounds.right += _bounds.left;

        lprPosRect->left = lprcClipRect->left;
        if( lprcClipRect->right >= lprPosRect->right )
        {
            // right side is not clipped out
        }
        else
        {
            // right side is clipped out
            lprPosRect->right = lprcClipRect->right;
        }
    }

    _bounds.bottom = lprPosRect->bottom-lprPosRect->top;

    if( lprcClipRect->top <= lprPosRect->top )
    {
        // top side is not clipped out
        _bounds.top = 0;

        if( lprcClipRect->bottom >= lprPosRect->bottom )
        {
            // bottom side is not clipped out, no change
        }
        else if( lprcClipRect->bottom >= lprPosRect->top )
        {
            // bottom side is clipped out
            lprPosRect->bottom = lprcClipRect->bottom;
        }
        else
        {
            // outside of clipping rectange, not visible
            lprPosRect->right = lprPosRect->left;
        }
    }
    else
    {
        _bounds.top = lprPosRect->top-lprcClipRect->top;
        _bounds.bottom += _bounds.top;

        lprPosRect->top = lprcClipRect->top;
        if( lprcClipRect->bottom >= lprPosRect->bottom )
        {
            // bottom side is not clipped out
        }
        else
        {
            // bottom side is clipped out
            lprPosRect->bottom = lprcClipRect->bottom;
        }
    }
};

HRESULT VLCPlugin::onInitNew(void)
{
    if( 0 == _i_vlc )
    {
        char *ppsz_argv[] = { "vlc", "-vv" };
        HKEY h_key;
        DWORD i_type, i_data = MAX_PATH + 1;
        char p_data[MAX_PATH + 1];
        if( RegOpenKeyEx( HKEY_LOCAL_MACHINE, "Software\\VideoLAN\\VLC",
                          0, KEY_READ, &h_key ) == ERROR_SUCCESS )
        {
             if( RegQueryValueEx( h_key, "InstallDir", 0, &i_type,
                                  (LPBYTE)p_data, &i_data ) == ERROR_SUCCESS )
             {
                 if( i_type == REG_SZ )
                 {
                     strcat( p_data, "\\vlc" );
                     ppsz_argv[0] = p_data;
                 }
             }
             RegCloseKey( h_key );
        }

#if 0
        ppsz_argv[0] = "C:\\cygwin\\home\\Damien_Fouilleul\\dev\\videolan\\vlc-trunk\\vlc";
#endif

        _i_vlc = VLC_Create();
        
        if( VLC_Init(_i_vlc, sizeof(ppsz_argv)/sizeof(char*), ppsz_argv) )
        {
            VLC_Destroy(_i_vlc);
            _i_vlc = 0;
            return E_FAIL;
        }
        return S_OK;
    }
    return E_UNEXPECTED;
};
    
HRESULT VLCPlugin::onClose(DWORD dwSaveOption)
{
    if( _i_vlc )
    {
        if( isInPlaceActive() )
        {
            onInPlaceDeactivate();
        }

        VLC_CleanUp(_i_vlc);
        VLC_Destroy(_i_vlc);
        _i_vlc = 0;
    }
    return S_OK;
};

BOOL VLCPlugin::isInPlaceActive(void)
{
    return (NULL != _inplacewnd);
};

HRESULT VLCPlugin::onActivateInPlace(LPMSG lpMesg, HWND hwndParent, LPCRECT lprcPosRect, LPCRECT lprcClipRect)
{
    RECT posRect = *lprcPosRect;

    calcPositionChange(&posRect, lprcClipRect);

    _inplacewnd = CreateWindow(_p_class->getInPlaceWndClassName(),
            "VLC Plugin In-Place Window",
            WS_CHILD|WS_CLIPCHILDREN|WS_TABSTOP,
            posRect.left,
            posRect.top,
            posRect.right-posRect.left,
            posRect.bottom-posRect.top,
            hwndParent,
            0,
            _p_class->getHInstance(),
            NULL
           );

    if( NULL == _inplacewnd )
        return E_FAIL;

    SetWindowLongPtr(_inplacewnd, GWLP_USERDATA, reinterpret_cast<LONG_PTR>(this));

    _videownd = CreateWindow(_p_class->getVideoWndClassName(),
            "VLC Plugin Video Window",
            WS_CHILD|WS_CLIPCHILDREN|WS_VISIBLE,
            _bounds.left,
            _bounds.top,
            _bounds.right-_bounds.left,
            _bounds.bottom-_bounds.top,
            _inplacewnd,
            0,
            _p_class->getHInstance(),
            NULL
           );

    if( NULL == _videownd )
    {
        DestroyWindow(_inplacewnd);
        return E_FAIL;
    }

    SetWindowLongPtr(_videownd, GWLP_USERDATA, reinterpret_cast<LONG_PTR>(this));

    /* horrible cast there */
    vlc_value_t val;
    val.i_int = reinterpret_cast<int>(_videownd);
    VLC_VariableSet(_i_vlc, "drawable", val);

    setVisible(_b_showdisplay);

    if( NULL != _psz_src )
    {
        // add target to playlist
        char *cOptions[1];
        int  cOptionsCount = 0;

        if( _b_loopmode )
        {
            cOptions[cOptionsCount++] = "loop";
        }
        VLC_AddTarget(_i_vlc, _psz_src, (const char **)&cOptions, cOptionsCount, PLAYLIST_APPEND, PLAYLIST_END);

        if( _b_autostart )
        {
            VLC_Play(_i_vlc);
            fireOnPlayEvent();
        }
    }
    return S_OK;
};

HRESULT VLCPlugin::onInPlaceDeactivate(void)
{
    VLC_Stop(_i_vlc);
    fireOnStopEvent();

    DestroyWindow(_videownd);
    _videownd = NULL;
    DestroyWindow(_inplacewnd);
    _inplacewnd = NULL;
 
    return S_OK;
};

BOOL VLCPlugin::isVisible(void)
{
    return GetWindowLong(_inplacewnd, GWL_STYLE) & WS_VISIBLE;
};

void VLCPlugin::setVisible(BOOL fVisible)
{
    ShowWindow(_inplacewnd, fVisible ? SW_SHOW : SW_HIDE);
};

void VLCPlugin::setFocus(BOOL fFocus)
{
    if( fFocus )
        SetActiveWindow(_inplacewnd);
};

BOOL VLCPlugin::hasFocus(void)
{
    return GetActiveWindow() == _inplacewnd;
};

void VLCPlugin::onPaint(PAINTSTRUCT &ps, RECT &pr)
{
    /*
    ** if VLC is playing, it may not display any VIDEO content 
    ** hence, draw control logo
    */ 
    int width = _bounds.right-_bounds.left;
    int height = _bounds.bottom-_bounds.top;

    HBITMAP pict = _p_class->getInPlacePict();
    if( NULL != pict )
    {
        HDC hdcPict = CreateCompatibleDC(ps.hdc);
        if( NULL != hdcPict )
        {
            BITMAP bm;
            if( GetObject(pict, sizeof(BITMAPINFO), &bm) )
            {
                int dstWidth = bm.bmWidth;
                if( dstWidth > width-4 )
                    dstWidth = width-4;

                int dstHeight = bm.bmHeight;
                if( dstHeight > height-4 )
                    dstHeight = height-4;

                int dstX = (width-dstWidth)/2;
                int dstY = (height-dstHeight)/2;

                SelectObject(hdcPict, pict);
                StretchBlt(ps.hdc, dstX, dstY, dstWidth, dstHeight,
                        hdcPict, 0, 0, bm.bmWidth, bm.bmHeight, SRCCOPY);
                DeleteDC(hdcPict);
                ExcludeClipRect(ps.hdc, dstX, dstY, dstWidth+dstX, dstHeight+dstY);
            }
        }
    }

    FillRect(ps.hdc, &pr, (HBRUSH)GetStockObject(WHITE_BRUSH));
    SelectObject(ps.hdc, GetStockObject(BLACK_BRUSH));

    MoveToEx(ps.hdc, 0, 0, NULL);
    LineTo(ps.hdc, width-1, 0);
    LineTo(ps.hdc, width-1, height-1);
    LineTo(ps.hdc, 0, height-1);
    LineTo(ps.hdc, 0, 0);
};

void VLCPlugin::onPositionChange(LPCRECT lprcPosRect, LPCRECT lprcClipRect)
{
    RECT posRect = *lprcPosRect;

    calcPositionChange(&posRect, lprcClipRect);

    /*
    ** change in-place window geometry to match clipping region
    */
    MoveWindow(_inplacewnd,
            posRect.left,
            posRect.top,
            posRect.right-posRect.left,
            posRect.bottom-posRect.top,
            FALSE);

    /*
    ** change video window geometry to match object bounds within clipping region
    */
    MoveWindow(_videownd,
            _bounds.left,
            _bounds.top,
            _bounds.right-_bounds.left,
            _bounds.bottom-_bounds.top,
            FALSE);

    RECT updateRect;

    updateRect.left = -_bounds.left;
    updateRect.top = -_bounds.top;
    updateRect.right = _bounds.right-_bounds.left;
    updateRect.bottom = _bounds.bottom-_bounds.top;

    ValidateRect(_videownd, NULL);
    InvalidateRect(_videownd, &updateRect, FALSE);
    UpdateWindow(_videownd);
};

void VLCPlugin::fireOnPlayEvent(void)
{
    if( _b_sendevents )
    {
        DISPPARAMS dispparamsNoArgs = {NULL, NULL, 0, 0};
        vlcConnectionPointContainer->fireEvent(1, LOCALE_USER_DEFAULT, &dispparamsNoArgs); 
    }
};

void VLCPlugin::fireOnPauseEvent(void)
{
    if( _b_sendevents )
    {
        DISPPARAMS dispparamsNoArgs = {NULL, NULL, 0, 0};
        vlcConnectionPointContainer->fireEvent(2, LOCALE_USER_DEFAULT, &dispparamsNoArgs); 
    }
};

void VLCPlugin::fireOnStopEvent(void)
{
    if( _b_sendevents )
    {
        DISPPARAMS dispparamsNoArgs = {NULL, NULL, 0, 0};
        vlcConnectionPointContainer->fireEvent(3, LOCALE_USER_DEFAULT, &dispparamsNoArgs); 
    }
};

