/*
 * Copyright (C) 2023 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 "ConfigManager.h"
#include "EvsCameraBase.h"
#include "EvsGlDisplay.h"

#include <aidl/android/frameworks/automotive/display/ICarDisplayProxy.h>
#include <aidl/android/hardware/automotive/evs/BnEvsEnumerator.h>
#include <aidl/android/hardware/automotive/evs/CameraDesc.h>
#include <aidl/android/hardware/automotive/evs/DeviceStatusType.h>
#include <aidl/android/hardware/automotive/evs/IEvsCamera.h>
#include <aidl/android/hardware/automotive/evs/IEvsEnumeratorStatusCallback.h>
#include <aidl/android/hardware/automotive/evs/Stream.h>
#include <android-base/thread_annotations.h>
#include <utils/Thread.h>

#include <atomic>
#include <mutex>
#include <optional>
#include <thread>
#include <unordered_map>

namespace aidl::android::hardware::automotive::evs::implementation {

class EvsEnumerator final : public ::aidl::android::hardware::automotive::evs::BnEvsEnumerator {
  public:
    // Methods from ::aidl::android::hardware::automotive::evs::IEvsEnumerator
    ndk::ScopedAStatus isHardware(bool* flag) override;
    ndk::ScopedAStatus openCamera(const std::string& cameraId, const evs::Stream& streamConfig,
                                  std::shared_ptr<evs::IEvsCamera>* obj) override;
    ndk::ScopedAStatus closeCamera(const std::shared_ptr<evs::IEvsCamera>& obj) override;
    ndk::ScopedAStatus getCameraList(std::vector<evs::CameraDesc>* _aidl_return) override;
    ndk::ScopedAStatus getStreamList(const evs::CameraDesc& desc,
                                     std::vector<evs::Stream>* _aidl_return) override;
    ndk::ScopedAStatus openDisplay(int32_t displayId,
                                   std::shared_ptr<evs::IEvsDisplay>* obj) override;
    ndk::ScopedAStatus closeDisplay(const std::shared_ptr<evs::IEvsDisplay>& obj) override;
    ndk::ScopedAStatus getDisplayIdList(std::vector<uint8_t>* list) override;
    ndk::ScopedAStatus getDisplayState(evs::DisplayState* state) override;
    ndk::ScopedAStatus getDisplayStateById(int32_t displayId, evs::DisplayState* state) override;
    ndk::ScopedAStatus registerStatusCallback(
            const std::shared_ptr<evs::IEvsEnumeratorStatusCallback>& callback) override;
    ndk::ScopedAStatus openUltrasonicsArray(
            const std::string& id, std::shared_ptr<evs::IEvsUltrasonicsArray>* obj) override;
    ndk::ScopedAStatus closeUltrasonicsArray(
            const std::shared_ptr<evs::IEvsUltrasonicsArray>& obj) override;
    ndk::ScopedAStatus getUltrasonicsArrayList(
            std::vector<evs::UltrasonicsArrayDesc>* list) override;

    // Implementation details
    EvsEnumerator(const std::shared_ptr<
                  ::aidl::android::frameworks::automotive::display::ICarDisplayProxy>&
                          proxyService);

    void notifyDeviceStatusChange(const std::string_view& deviceName, evs::DeviceStatusType type);

  private:
    struct CameraRecord {
        evs::CameraDesc desc;
        std::weak_ptr<EvsCameraBase> activeInstance;

        CameraRecord(const char* cameraId) : desc() { desc.id = cameraId; }
    };

    class ActiveDisplays {
      public:
        struct DisplayInfo {
            int32_t id{-1};
            std::weak_ptr<EvsGlDisplay> displayWeak;
            uintptr_t internalDisplayRawAddr;
        };

        std::optional<DisplayInfo> popDisplay(int32_t id);

        std::optional<DisplayInfo> popDisplay(const std::shared_ptr<IEvsDisplay>& display);

        std::unordered_map<int32_t, DisplayInfo> getAllDisplays();

        bool tryInsert(int32_t id, const std::shared_ptr<EvsGlDisplay>& display);

      private:
        std::mutex mMutex;
        std::unordered_map<int32_t, DisplayInfo> mIdToDisplay GUARDED_BY(mMutex);
        std::unordered_map<uintptr_t, int32_t> mDisplayToId GUARDED_BY(mMutex);
    };

    bool checkPermission();
    void closeCamera_impl(const std::shared_ptr<evs::IEvsCamera>& pCamera,
                          const std::string& cameraId);
    ndk::ScopedAStatus getDisplayStateImpl(std::optional<int32_t> displayId,
                                           evs::DisplayState* state);

    static bool qualifyCaptureDevice(const char* deviceName);
    static CameraRecord* findCameraById(const std::string& cameraId);
    static void enumerateCameras();
    static bool addCaptureDevice(const std::string& deviceName);
    static bool removeCaptureDevice(const std::string& deviceName);
    // Enumerate available displays and return an id of the internal display
    static uint64_t enumerateDisplays();

    static ActiveDisplays& mutableActiveDisplays();

    // NOTE:  All members values are static so that all clients operate on the same state
    //        That is to say, this is effectively a singleton despite the fact that HIDL
    //        constructs a new instance for each client.
    //        Because our server has a single thread in the thread pool, these values are
    //        never accessed concurrently despite potentially having multiple instance objects
    //        using them.
    static std::unordered_map<std::string, CameraRecord> sCameraList;
    // Object destructs if client dies.
    static std::mutex sLock;                               // Mutex on shared camera device list.
    static std::condition_variable sCameraSignal;          // Signal on camera device addition.
    static std::unique_ptr<ConfigManager> sConfigManager;  // ConfigManager
    static std::shared_ptr<::aidl::android::frameworks::automotive::display::ICarDisplayProxy>
            sDisplayProxy;
    static std::unordered_map<uint8_t, uint64_t> sDisplayPortList;

    uint64_t mInternalDisplayId;
    std::shared_ptr<evs::IEvsEnumeratorStatusCallback> mCallback;
};

}  // namespace aidl::android::hardware::automotive::evs::implementation