#ifndef NEENGINEMANAGER_H
#define NEENGINEMANAGER_H

#include <QObject>
#include "nertc_audio_device_manager.h"
#include "nertc_engine_defines.h"
#include "nertc_engine_ex.h"
#include "nertc_video_device_manager.h"

//using namespace nertc;
class NEEngineManager : public QObject,
public nertc::IRtcEngineEventHandlerEx//接口类用于向应用程序发送扩展回调通知
{
    Q_OBJECT
    
public:
    NEEngineManager(QObject* parent = nullptr);
    ~NEEngineManager();
    
public:
    nertc::IRtcEngineEx* coreEngine;//IRtcEngineEx 是 NERTC SDK 的扩展接口类。创建一个 IRtcEngine 对象并调用这个对象的方法可以激活 NERTC SDK 的通信功能
    
    int initialize();
    void cleanup();
    int joinChannel(const char* token, const char* channel_name, uint64_t uid);
    int leaveChannel();
    int setupLocalVideo(void* window, nertc::NERtcVideoScalingMode scaling_mode = nertc::kNERtcVideoScaleFit);
    int setupLocalSubVideo(void* hWnd, nertc::NERtcVideoScalingMode scaling_mode = nertc::kNERtcVideoScaleFit);
    int setupRemoteVideo(const uint64_t& uid,
                         void* window,
                         nertc::NERtcVideoScalingMode scaling_mode = nertc::kNERtcVideoScaleFit);
    int setupRemoteSubVideo(const uint64_t& uid,
                            void* hWnd,
                            nertc::NERtcVideoScalingMode scaling_mode = nertc::kNERtcVideoScaleFit);
    
    nertc::IAudioDeviceManager* audioDeviceManager();//接口类提供用于管理音频设备的相关接口。可通过实例化 IAudioDeviceManager 类来获取 IAudioDeviceManager 接口。
    nertc::IVideoDeviceManager* videoDeviceManager();//接口类提供用于管理视频设备的相关接口。可通过实例化 IVideoDeviceManager 类来获取 IVideoDeviceManager 接口。
    void setupVideoDevice();
    
    void onJoinChannel(nertc::channel_id_t cid,
                       nertc::uid_t uid,
                       nertc::NERtcErrorCode result,
                       uint64_t elapsed) override;
    void onLeaveChannel(nertc::NERtcErrorCode result) override;
    void onUserJoined(nertc::uid_t uid, const char* user_name) override;
    void onUserLeft(nertc::uid_t uid, nertc::NERtcSessionLeaveReason reason) override;
    void onUserVideoStart(nertc::uid_t uid, nertc::NERtcVideoProfileType max_profile) override;
    void onUserVideoStop(nertc::uid_t uid) override;
    void onUserSubStreamVideoStart(nertc::uid_t uid, nertc::NERtcVideoProfileType max_profile) override;
    void onUserSubStreamVideoStop(nertc::uid_t uid) override;
    void onScreenCaptureStatus(nertc::NERtcScreenCaptureStatus status) override;
    ////需要加的接口
    
    // Q_SIGNALS:
signals:
    // 加入频道结果
    void signal_onJoinChannel(const unsigned long long cid, const unsigned long long uid, const int result);
    void signal_onLeaveChannel(const int result);
    void signal_onUserJoined(const unsigned long long uid, const char* user_name);
    void signal_onUserLeft(const unsigned long long uid, const int& reason);
    void signal_onUserVideoStart(const unsigned long long uid, const int max_profile);
    void signal_onUserVideoStop(const unsigned long long uid);
    void signal_onUserSubVideoStarted(const unsigned long long uid, const int max_profile);
    void signal_onUserSubVideoStopped(const unsigned long long uid);
    void signal_onScreenCaptureStatus(const int& status);
};

#endif  // NEENGINEMANAGER_H


#include "ne_engine_manager.h"
#include "enginemanager.h"
#include <QCoreApplication>
#include <QDebug>
#include <QDir>
#include <QFile>
#include <QStandardPaths>
#include <QImage>

#define NERtcCheckCoreEngine \
    if (coreEngine == nullptr) { \
        return 0; \
    }

NEEngineManager::NEEngineManager(QObject* parent)
    : QObject(parent) {
    coreEngine = nullptr;
}

NEEngineManager::~NEEngineManager() {
    qDebug() << "NEEngineManager::~NEEngineManager";
}

int NEEngineManager::initialize() {
    nertc::NERtcEngineContext engine_context;
    memset(&engine_context, 0, sizeof(nertc::NERtcEngineContext));
    engine_context.app_key = Config::sharedInstance().appKey.c_str();
    
    QString app_dir = QStandardPaths::writableLocation(QStandardPaths::AppDataLocation);
    QByteArray byte_array_dir = app_dir.toUtf8();

    engine_context.log_dir_path = byte_array_dir.data();
    engine_context.log_level = nertc::kNERtcLogLevelInfo;
    engine_context.event_handler = this;
    engine_context.video_use_exnternal_render = false;

    qDebug() << "initialize log_level=" << engine_context.log_level << ", log_dir_path=" << engine_context.log_dir_path;

    if (coreEngine == nullptr) {
        coreEngine = (nertc::IRtcEngineEx*)createNERtcEngine();
    }

    int ret = coreEngine->initialize(engine_context);
    if (nertc::kNERtcNoError != ret) {
        qDebug() << "ERROR: Failed to initialize NERtc Engine";
    }

    return ret;
}

void NEEngineManager::cleanup() {
    if (coreEngine != nullptr) {
        coreEngine->release();
        void* engine = (void*)coreEngine;
        destroyNERtcEngine(engine);
        coreEngine = nullptr;
    }
}

int NEEngineManager::joinChannel(const char* token, const char* channel_name, uint64_t uid) {
    NERtcCheckCoreEngine
    int ret = coreEngine->joinChannel(token, channel_name, uid);
    return ret;
}

int NEEngineManager::leaveChannel() {
    NERtcCheckCoreEngine
    int ret = coreEngine->leaveChannel();
    return ret;
}

int NEEngineManager::setupLocalVideo(void* window, nertc::NERtcVideoScalingMode scaling_mode) {
    NERtcCheckCoreEngine
    nertc::NERtcVideoCanvas canvas;
    canvas.cb = nullptr;
    canvas.user_data = nullptr;
    canvas.window = window;
    canvas.scaling_mode = scaling_mode;

    int ret = coreEngine->setupLocalVideoCanvas(&canvas);
    if (ret) {
        qDebug() << "ERROR: Can not setup local video canvas! ERROR CODE:" << ret;
    }

    return ret;
}

int NEEngineManager::setupLocalSubVideo(void* hWnd, nertc::NERtcVideoScalingMode scaling_mode) {
    NERtcCheckCoreEngine
    nertc::NERtcVideoCanvas canvas;
    canvas.cb = nullptr;
    canvas.user_data = nullptr;
    canvas.window = hWnd;
    canvas.scaling_mode = scaling_mode;

    int ret = coreEngine->setupLocalSubStreamVideoCanvas(&canvas);
    if (ret) {
        qDebug() << "ERROR: Can not setup local sub video canvas! ERROR CODE:" << ret;
    }

    return ret;
}

int NEEngineManager::setupRemoteVideo(const uint64_t& uid, void* window, nertc::NERtcVideoScalingMode scaling_mode) {
    NERtcCheckCoreEngine
    nertc::NERtcVideoCanvas canvas;
    canvas.cb = nullptr;
    canvas.user_data = nullptr;
    canvas.window = window;
    canvas.scaling_mode = scaling_mode;

    int ret = coreEngine->setupRemoteVideoCanvas(uid, &canvas);
    if (ret) {
        qDebug() << "ERROR: Can not setup remote video canvas! ERROR CODE:" << ret;
    }

    return ret;
}

int NEEngineManager::setupRemoteSubVideo(const uint64_t& uid, void* hWnd, nertc::NERtcVideoScalingMode scaling_mode) {
    NERtcCheckCoreEngine
    nertc::NERtcVideoCanvas canvas;
    canvas.cb = nullptr;
    canvas.user_data = nullptr;
    canvas.window = hWnd;
    canvas.scaling_mode = scaling_mode;

    int ret = coreEngine->setupRemoteSubStreamVideoCanvas(uid, &canvas);
    if (ret) {
        qDebug() << "ERROR: Can not setup remote sub video canvas! ERROR CODE:" << ret;
    }

    return ret;
}

nertc::IAudioDeviceManager* NEEngineManager::audioDeviceManager() {
    NERtcCheckCoreEngine
    void* manager = nullptr;

    int errcode = coreEngine->queryInterface(nertc::kNERtcIIDAudioDeviceManager, &manager);//获取设备管理员对象的指针
    if (errcode != nertc::kNERtcNoError) {
        qDebug() << "ERROR: NEEngineManager::audioDeviceManager queryInterface error:" << errcode;
    }

    return (nertc::IAudioDeviceManager*)manager;
}

nertc::IVideoDeviceManager* NEEngineManager::videoDeviceManager() {
    NERtcCheckCoreEngine
    void* manager = nullptr;

    int errcode = coreEngine->queryInterface(nertc::kNERtcIIDVideoDeviceManager, &manager);
    if (errcode != nertc::kNERtcNoError) {
        qDebug() << "ERROR: NEEngineManager::videoDeviceManager queryInterface error:" << errcode;
    }

    return (nertc::IVideoDeviceManager*)manager;
}

void NEEngineManager::setupVideoDevice()
{
    if (!videoDeviceManager()) {
        return;
    }
    //IDeviceCollection 设备相关方法。此接口类获取设备相关的信息。
    nertc::IDeviceCollection * collection = videoDeviceManager()->enumerateCaptureDevices();//获取系统中所有的视频采集设备列表。该方法返回一个 IDeviceCollection 对象，包含系统中所有的音频采集设备。通过IDeviceCollection 对象，App 可以枚举视频采集设备。
    if (!collection) {
        //方法调用失败：NULL 。
        return;
    }
    
    if(collection->getCount() <= 0) {
        return;
    }
    
    char deviceID[kNERtcMaxDeviceIDLength] = {0};
    char deviceName[kNERtcMaxDeviceNameLength] = {0};
    nertc::NERtcErrorCode errCode = (nertc::NERtcErrorCode)collection->getDevice(0, deviceName, deviceID);//获取指定 index 的设备信息。
    if (errCode != nertc::kNERtcNoError) {
        return;
    }
    
    videoDeviceManager()->setDevice(deviceID);
}

//信令回调
void NEEngineManager::onJoinChannel(nertc::channel_id_t cid,
    nertc::uid_t uid,
    nertc::NERtcErrorCode result,
    uint64_t elapsed) {
    qDebug() << "NEEngineManager::onJoinChannel";
    emit signal_onJoinChannel(cid, uid, (int)result);
}

void NEEngineManager::onLeaveChannel(nertc::NERtcErrorCode result) {
    qDebug() << "NEEngineManager::onLeaveChannel";
    emit signal_onLeaveChannel((int)result);
}

void NEEngineManager::onUserJoined(nertc::uid_t uid, const char* user_name) {
    qDebug() << "[NEEngineManager] user joined: " << uid << ", user_name: " << user_name;
    emit signal_onUserJoined(uid, user_name);
}

void NEEngineManager::onUserLeft(nertc::uid_t uid, nertc::NERtcSessionLeaveReason reason) {
    qDebug() << "[NEEngineManager] user left: " << uid << ", reason: " << reason;
    emit signal_onUserLeft(uid, (int)reason);
}

void NEEngineManager::onUserVideoStart(nertc::uid_t uid, nertc::NERtcVideoProfileType max_profile) {
    qDebug() << "[NEEngineManager] user video start: " << uid << ", max_profile: " << max_profile;
    emit signal_onUserVideoStart(uid, (int)max_profile);
}

void NEEngineManager::onUserVideoStop(nertc::uid_t uid) {
    qDebug() << "[NEEngineManager] user video stop: " << uid;
    emit signal_onUserVideoStop(uid);
}

void NEEngineManager::onUserSubStreamVideoStart(nertc::uid_t uid, nertc::NERtcVideoProfileType max_profile) {
    qDebug() << "NEEngineManager::onUserSubStreamVideoStart";
    Q_EMIT signal_onUserSubVideoStarted(uid, max_profile);
}

void NEEngineManager::onUserSubStreamVideoStop(nertc::uid_t uid) {
    qDebug() << "NEEngineManager::onUserSubStreamVideoStop";
    Q_EMIT signal_onUserSubVideoStopped(uid);
}

void NEEngineManager::onScreenCaptureStatus(nertc::NERtcScreenCaptureStatus status) {
    qDebug() << "NEEngineManager::onScreenCaptureStatus: " << status;
    Q_EMIT signal_onScreenCaptureStatus(status);
}