/*=============================================================================

  Library: CppMicroServices

  Copyright (c) The CppMicroServices developers. See the COPYRIGHT
  file at the top-level directory of this distribution and at
  https://github.com/CppMicroServices/CppMicroServices/COPYRIGHT .

  Licensed 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

  Unless required by applicable law or agreed to in writing, software
  distributed under the License is distributed on an "AS IS" BASIS,
  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  See the License for the specific language governing permissions and
  limitations under the License.

=============================================================================*/

#ifndef CPPMICROSERVICES_TRACKEDSERVICE_H
#define CPPMICROSERVICES_TRACKEDSERVICE_H

#include "cppmicroservices/ServiceEvent.h"
#include "cppmicroservices/detail/BundleAbstractTracked.h"
#include "cppmicroservices/detail/TrackedServiceListener.h"

#include "cppmicroservices/detail/CounterLatch.h"
#include "cppmicroservices/detail/ScopeGuard.h"

namespace cppmicroservices {

namespace detail {

/**
 * This class is not intended to be used directly. It is exported to support
 * the CppMicroServices bundle system.
 */
template<class S, class TTT>
class TrackedService
  : public TrackedServiceListener
  , public BundleAbstractTracked<ServiceReference<S>, TTT, ServiceEvent>
{

public:
  using T = typename TTT::TrackedType;
  using TrackedParamType = typename TTT::TrackedParamType;

  TrackedService(ServiceTracker<S, T>* serviceTracker,
                 ServiceTrackerCustomizer<S, T>* customizer);

  /**
   * Method connected to service events for the
   * <code>ServiceTracker</code> class. This method must NOT be
   * synchronized to avoid deadlock potential.
   *
   * @param event <code>ServiceEvent</code> object from the framework.
   */
  void ServiceChanged(const ServiceEvent& event) override;

  void WaitOnCustomizersToFinish();

private:
  using Superclass =
    BundleAbstractTracked<ServiceReference<S>, TTT, ServiceEvent>;

  ServiceTracker<S, T>* serviceTracker;
  ServiceTrackerCustomizer<S, T>* customizer;

  CounterLatch latch;

  /**
   * Increment the tracking count and tell the tracker there was a
   * modification.
   *
   * @GuardedBy this
   */
  void Modified() override;

  /**
   * Call the specific customizer adding method. This method must not be
   * called while synchronized on this object.
   *
   * @param item Item to be tracked.
   * @param related Action related object.
   * @return Customized object for the tracked item or <code>null</code>
   *         if the item is not to be tracked.
   */
  std::shared_ptr<TrackedParamType> CustomizerAdding(
    ServiceReference<S> item,
    const ServiceEvent& related) override;

  /**
   * Call the specific customizer modified method. This method must not be
   * called while synchronized on this object.
   *
   * @param item Tracked item.
   * @param related Action related object.
   * @param object Customized object for the tracked item.
   */
  void CustomizerModified(
    ServiceReference<S> item,
    const ServiceEvent& related,
    const std::shared_ptr<TrackedParamType>& object) override;

  /**
   * Call the specific customizer removed method. This method must not be
   * called while synchronized on this object.
   *
   * @param item Tracked item.
   * @param related Action related object.
   * @param object Customized object for the tracked item.
   */
  void CustomizerRemoved(
    ServiceReference<S> item,
    const ServiceEvent& related,
    const std::shared_ptr<TrackedParamType>& object) override;
};

} // namespace detail

} // namespace cppmicroservices

#include "cppmicroservices/detail/TrackedService.tpp"

#endif // CPPMICROSERVICES_TRACKEDSERVICE_H
