/* -*- Mode: C++; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 4 -*- */
/*
 * This file is part of the Collabora Office project.
 *
 * This Source Code Form is subject to the terms of the Mozilla Public
 * License, v. 2.0. If a copy of the MPL was not distributed with this
 * file, You can obtain one at http://mozilla.org/MPL/2.0/.
 *
 * This file incorporates work covered by the following license notice:
 *
 *   Licensed to the Apache Software Foundation (ASF) under one or more
 *   contributor license agreements. See the NOTICE file distributed
 *   with this work for additional information regarding copyright
 *   ownership. The ASF licenses this file to you under the Apache
 *   License, Version 2.0 (the "License"); you may not use this file
 *   except in compliance with the License. You may obtain a copy of
 *   the License at http://www.apache.org/licenses/LICENSE-2.0 .
 */


#include <cppuhelper/supportsservice.hxx>

#include <com/sun/star/util/CloseVetoException.hpp>
#include <com/sun/star/util/XCloseBroadcaster.hpp>
#include <com/sun/star/util/XCloseable.hpp>
#include <com/sun/star/lang/DisposedException.hpp>
#include <com/sun/star/lang/IllegalArgumentException.hpp>
#include <com/sun/star/frame/XDesktop.hpp>
#include <com/sun/star/frame/TerminationVetoException.hpp>
#include <com/sun/star/frame/DoubleInitializationException.hpp>
#include <com/sun/star/embed/Actions.hpp>
#include <com/sun/star/embed/XActionsApproval.hpp>
#include <utility>

#include "instancelocker.hxx"

namespace com::sun::star::uno { class XComponentContext; }

using namespace ::com::sun::star;


// OInstanceLocker


OInstanceLocker::OInstanceLocker()
: m_bDisposed( false )
, m_bInitialized( false )
{
}


OInstanceLocker::~OInstanceLocker()
{
    if ( !m_bDisposed )
    {
        osl_atomic_increment(&m_refCount); // to call dispose
        try {
            dispose();
        }
        catch ( uno::RuntimeException& )
        {}
    }
}

// XComponent

void OInstanceLocker::dispose()
{
    std::unique_lock aGuard( m_aMutex );

    if ( m_bDisposed )
        throw lang::DisposedException();

    lang::EventObject aSource( static_cast< ::cppu::OWeakObject* >(this) );
    m_aListenersContainer.disposeAndClear( aGuard, aSource );
    if ( m_xLockListener.is() )
    {
        auto tmp = std::move(m_xLockListener);
        aGuard.unlock();
        tmp->Dispose();
        aGuard.lock();
    }

    m_bDisposed = true;
}


void OInstanceLocker::addEventListener( const uno::Reference< lang::XEventListener >& xListener )
{
    std::unique_lock aGuard( m_aMutex );
    if ( m_bDisposed )
        throw lang::DisposedException(); // TODO

    m_aListenersContainer.addInterface( aGuard, xListener );
}


void OInstanceLocker::removeEventListener( const uno::Reference< lang::XEventListener >& xListener )
{
    std::unique_lock aGuard( m_aMutex );
    m_aListenersContainer.removeInterface( aGuard, xListener );
}

// XInitialization

void OInstanceLocker::initialize( const cpo::uno::Sequence< cpo::uno::Any >& aArguments )
{
    std::unique_lock aGuard( m_aMutex );
    if ( m_bInitialized )
        throw frame::DoubleInitializationException();

    if ( m_bDisposed )
        throw lang::DisposedException(); // TODO

    if ( !m_refCount )
        throw uno::RuntimeException(); // the object must be refcounted already!

    uno::Reference< uno::XInterface > xInstance;
    uno::Reference< embed::XActionsApproval > xApproval;

    try
    {
        sal_Int32 nLen = aArguments.getLength();
        if ( nLen < 2 || nLen > 3 )
            throw lang::IllegalArgumentException(
                            u"Wrong count of parameters!"_ustr,
                            uno::Reference< uno::XInterface >(),
                            0 );

        if ( !( aArguments[0] >>= xInstance ) || !xInstance.is() )
            throw lang::IllegalArgumentException(
                    u"Nonempty reference is expected as the first argument!"_ustr,
                    uno::Reference< uno::XInterface >(),
                    0 );

        sal_Int32 nModes = 0;
        if (
            !( aArguments[1] >>= nModes ) ||
            (
              !( nModes & embed::Actions::PREVENT_CLOSE ) &&
              !( nModes & embed::Actions::PREVENT_TERMINATION )
            )
           )
        {
            throw lang::IllegalArgumentException(
                    u"The correct modes set is expected as the second argument!"_ustr,
                    uno::Reference< uno::XInterface >(),
                    0 );
        }

        if ( nLen == 3 && !( aArguments[2] >>= xApproval ) )
            throw lang::IllegalArgumentException(
                    u"If the third argument is provided, it must be XActionsApproval implementation!"_ustr,
                    uno::Reference< uno::XInterface >(),
                    0 );

        m_xLockListener = new OLockListener( uno::Reference< lang::XComponent > ( static_cast< lang::XComponent* >( this ) ),
                                            xInstance,
                                            nModes,
                                            xApproval );
        m_xLockListener->Init();
    }
    catch( uno::Exception& )
    {
        aGuard.unlock();
        dispose();
        throw;
    }

    m_bInitialized = true;
}

// XServiceInfo
OUString OInstanceLocker::getImplementationName(  )
{
    return u"com.sun.star.comp.embed.InstanceLocker"_ustr;
}

bool OInstanceLocker::supportsService( const OUString& ServiceName )
{
    return cppu::supportsService(this, ServiceName);
}

cpo::uno::Sequence< OUString > OInstanceLocker::getSupportedServiceNames()
{
    return { u"com.sun.star.embed.InstanceLocker"_ustr };
}

// OLockListener


OLockListener::OLockListener( cpo::uno::WeakReference< lang::XComponent > xWrapper,
                    uno::Reference< uno::XInterface > xInstance,
                    sal_Int32 nMode,
                    uno::Reference< embed::XActionsApproval > xApproval )
: m_xInstance(std::move( xInstance ))
, m_xApproval(std::move( xApproval ))
, m_xWrapper(std::move( xWrapper ))
, m_bDisposed( false )
, m_bInitialized( false )
, m_nMode( nMode )
{
}


OLockListener::~OLockListener()
{
}


void OLockListener::Dispose()
{
    std::unique_lock aGuard( m_aMutex );

    if ( m_bDisposed )
        return;

    auto xInstance = std::move(m_xInstance);
    auto xApproval = std::move(m_xApproval);
    auto nMode = m_nMode;
    m_bDisposed = true;
    aGuard.unlock();

    if ( nMode & embed::Actions::PREVENT_CLOSE )
    {
        try
        {
            uno::Reference< util::XCloseBroadcaster > xCloseBroadcaster( xInstance, uno::UNO_QUERY );
            if ( xCloseBroadcaster.is() )
                xCloseBroadcaster->removeCloseListener( static_cast< util::XCloseListener* >( this ) );

            uno::Reference< util::XCloseable > xCloseable( xInstance, uno::UNO_QUERY );
            if ( xCloseable.is() )
                xCloseable->close( true );
        }
        catch( uno::Exception& )
        {}
    }

    if ( nMode & embed::Actions::PREVENT_TERMINATION )
    {
        try
        {
            uno::Reference< frame::XDesktop > xDesktop( xInstance, uno::UNO_QUERY );
            if (xDesktop)
                xDesktop->removeTerminateListener( static_cast< frame::XTerminateListener* >( this ) );
        }
        catch( uno::Exception& )
        {}
    }
}

// XEventListener

void OLockListener::disposing( const lang::EventObject& aEvent )
{
    std::unique_lock aGuard( m_aMutex );

    // object is disposed
    if ( aEvent.Source != m_xInstance )
        return;

    // the object does not listen for anything any more
    m_nMode = 0;

    // dispose the wrapper;
    uno::Reference< lang::XComponent > xComponent( m_xWrapper.get(), uno::UNO_QUERY );
    aGuard.unlock();
    if ( xComponent.is() )
    {
        try { xComponent->dispose(); }
        catch( uno::Exception& ){}
    }
}


// XCloseListener

void OLockListener::queryClosing( const lang::EventObject& aEvent, bool )
{
    // GetsOwnership parameter is always ignored, the user of the service must close the object always
    std::unique_lock aGuard( m_aMutex );
    if ( !(!m_bDisposed && aEvent.Source == m_xInstance && ( m_nMode & embed::Actions::PREVENT_CLOSE )) )
        return;

    try
    {
        uno::Reference< embed::XActionsApproval > xApprove = m_xApproval;

        // unlock the mutex here
        aGuard.unlock();

        if ( xApprove.is() && xApprove->approveAction( embed::Actions::PREVENT_CLOSE ) )
            throw util::CloseVetoException();
    }
    catch( util::CloseVetoException& )
    {
        // rethrow this exception
        throw;
    }
    catch( uno::Exception& )
    {
        // no action should be done
    }
}


void OLockListener::notifyClosing( const lang::EventObject& aEvent )
{
    std::unique_lock aGuard( m_aMutex );

    // object is closed, no reason to listen
    if ( aEvent.Source != m_xInstance )
        return;

    uno::Reference< util::XCloseBroadcaster > xCloseBroadcaster( aEvent.Source, uno::UNO_QUERY );
    if ( !xCloseBroadcaster.is() )
        return;

    xCloseBroadcaster->removeCloseListener( static_cast< util::XCloseListener* >( this ) );
    m_nMode &= ~embed::Actions::PREVENT_CLOSE;
    if ( !m_nMode )
    {
        // dispose the wrapper;
        uno::Reference< lang::XComponent > xComponent( m_xWrapper.get(), uno::UNO_QUERY );
        aGuard.unlock();
        if ( xComponent.is() )
        {
            try { xComponent->dispose(); }
            catch( uno::Exception& ){}
        }
    }
}


// XTerminateListener

void OLockListener::queryTermination( const lang::EventObject& aEvent )
{
    std::unique_lock aGuard( m_aMutex );
    if ( !(aEvent.Source == m_xInstance && ( m_nMode & embed::Actions::PREVENT_TERMINATION )) )
        return;

    try
    {
        uno::Reference< embed::XActionsApproval > xApprove = m_xApproval;

        // unlock the mutex here
        aGuard.unlock();

        if ( xApprove.is() && xApprove->approveAction( embed::Actions::PREVENT_TERMINATION ) )
            throw frame::TerminationVetoException();
    }
    catch( frame::TerminationVetoException& )
    {
        // rethrow this exception
        throw;
    }
    catch( uno::Exception& )
    {
        // no action should be done
    }
}


void OLockListener::notifyTermination( const lang::EventObject& aEvent )
{
    std::unique_lock aGuard( m_aMutex );

    // object is terminated, no reason to listen
    if ( aEvent.Source != m_xInstance )
        return;

    uno::Reference< frame::XDesktop > xDesktop( aEvent.Source, uno::UNO_QUERY );
    if ( !xDesktop.is() )
        return;

    try
    {
        xDesktop->removeTerminateListener( static_cast< frame::XTerminateListener* >( this ) );
        m_nMode &= ~embed::Actions::PREVENT_TERMINATION;
        if ( !m_nMode )
        {
            // dispose the wrapper;
            uno::Reference< lang::XComponent > xComponent( m_xWrapper.get(), uno::UNO_QUERY );
            aGuard.unlock();
            if ( xComponent.is() )
            {
                try { xComponent->dispose(); }
                catch( uno::Exception& ){}
            }
        }
    }
    catch( uno::Exception& )
    {}
}


// XInitialization

void OLockListener::Init()
{
    std::unique_lock aGuard( m_aMutex );

    if ( m_bDisposed || m_bInitialized )
        return;

    try
    {
        if ( m_nMode & embed::Actions::PREVENT_CLOSE )
        {
            uno::Reference< util::XCloseBroadcaster > xCloseBroadcaster( m_xInstance, uno::UNO_QUERY_THROW );
            xCloseBroadcaster->addCloseListener( static_cast< util::XCloseListener* >( this ) );
        }

        if ( m_nMode & embed::Actions::PREVENT_TERMINATION )
        {
            uno::Reference< frame::XDesktop > xDesktop( m_xInstance, uno::UNO_QUERY_THROW );
            xDesktop->addTerminateListener( static_cast< frame::XTerminateListener* >( this ) );
        }
    }
    catch( uno::Exception& )
    {
        // dispose the wrapper;
        uno::Reference< lang::XComponent > xComponent( m_xWrapper.get(), uno::UNO_QUERY );
        aGuard.unlock();
        if ( xComponent.is() )
        {
            try { xComponent->dispose(); }
            catch( uno::Exception& ){}
        }

        throw;
    }

    m_bInitialized = true;
}

extern "C" SAL_DLLPUBLIC_EXPORT css::uno::XInterface *
com_sun_star_comp_embed_InstanceLocker(
    css::uno::XComponentContext *,
    cpo::uno::Sequence<cpo::uno::Any> const &)
{
    return cppu::acquire(new OInstanceLocker());
}

/* vim:set shiftwidth=4 softtabstop=4 expandtab: */