/*
 * Copyright (C) 2022 The Android Open Source Project
 *
 * 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.
 */

#pragma once

#include <any>
#include <map>
#include <optional>
#include <vector>

#include <aidl/android/hardware/audio/effect/BnFactory.h>

namespace aidl::android::hardware::audio::effect {

class Factory : public BnFactory {
  public:
    Factory();
    /**
     * @brief Get identity of all effects supported by the device, with the optional filter by type
     * and/or by instance UUID.
     *
     * @param in_type Type UUID.
     * @param in_instance Instance UUID.
     * @param out_descriptor List of identities .
     * @return ndk::ScopedAStatus
     */
    ndk::ScopedAStatus queryEffects(
            const std::optional<::aidl::android::media::audio::common::AudioUuid>& in_type,
            const std::optional<::aidl::android::media::audio::common::AudioUuid>& in_instance,
            std::vector<Descriptor::Identity>* out_descriptor) override;

    /**
     * @brief Query list of defined processing, with the optional filter by AudioStreamType
     *
     * @param in_type Type of processing, could be AudioStreamType or AudioSource. Optional.
     * @param _aidl_return List of processing filtered by in_type.
     * @return ndk::ScopedAStatus
     */
    ndk::ScopedAStatus queryProcessing(const std::optional<Processing::Type>& in_type,
                                       std::vector<Processing>* _aidl_return) override;

    /**
     * @brief Create an effect instance for a certain implementation (identified by UUID).
     *
     * @param in_impl_uuid Effect implementation UUID.
     * @param _aidl_return A pointer to created effect instance.
     * @return ndk::ScopedAStatus
     */
    ndk::ScopedAStatus createEffect(
            const ::aidl::android::media::audio::common::AudioUuid& in_impl_uuid,
            std::shared_ptr<::aidl::android::hardware::audio::effect::IEffect>* _aidl_return)
            override;

    /**
     * @brief Destroy an effect instance.
     *
     * @param in_handle Effect instance handle.
     * @return ndk::ScopedAStatus
     */
    ndk::ScopedAStatus destroyEffect(
            const std::shared_ptr<::aidl::android::hardware::audio::effect::IEffect>& in_handle)
            override;

  private:
    ~Factory();
    // List of effect descriptors supported by the devices.
    std::vector<Descriptor::Identity> mIdentityList;

    typedef binder_exception_t (*EffectCreateFunctor)(std::shared_ptr<IEffect>*);
    typedef binder_exception_t (*EffectDestroyFunctor)(const std::shared_ptr<IEffect>&);
    struct effect_interface_s {
        EffectCreateFunctor createEffectFunc;
        EffectDestroyFunctor destroyEffectFunc;
    };

    std::map<aidl::android::media::audio::common::AudioUuid /* implementationUUID */,
             std::pair<std::unique_ptr<void, std::function<void(void*)>> /* dlHandle */,
                       std::unique_ptr<struct effect_interface_s>>>
            mEffectLibMap;
    std::map<std::weak_ptr<IEffect>, aidl::android::media::audio::common::AudioUuid,
             std::owner_less<>>
            mEffectUuidMap;

    ndk::ScopedAStatus destroyEffectImpl(const std::shared_ptr<IEffect>& in_handle);
    void cleanupEffectMap();
};
}  // namespace aidl::android::hardware::audio::effect