video-pc/mainwindow.cpp

#include "mainwindow.h"

#include <QSplitter>
#include <QStackedWidget>
#include <QTextBrowser>
#include <QThread>

#include <util/asynchelper.h>
#include <util/jsonmapper.h>
#include <util/layoutbuilder.h>

#include "api/roomapi.h"
#include "api/userapi.h"
#include "appevent.h"
#include "capturer/video/wgc/App.h"
#include "network/websocketclient.h"
#include "qboxlayout.h"
#include "qlistwidget.h"
#include "qobject.h"
#include "ui/av_recorder.h"
#include "views/loginwindow.h"

#include "widgets/bubbletip.h"
#include "widgets/chatView/chat1/chatview.h"
#include "widgets/chatView/chatwindow.h"
#include "widgets/framelessbase.h"
#include "widgets/maskoverlay.h"
#include "widgets/userprofilewidget.h"

#include <QDebug>
#include <QFutureWatcher>

#include <api/loginapi.h>

#include "util/qformatter.h"

#include <qtpromise/qpromise.h>
#include <qtpromise/qpromisefuture.h>
#include <qtpromise/qpromisehelpers.h>

#include "AvPlayer2/PlayWidget.h"

#include <QListWidget>

class MainWidget : public QWidget
{
public:
    explicit MainWidget(QWidget *parent = nullptr)
        : QWidget(parent)
        , userProfile(nullptr)
        , chatView(nullptr)
    {
        setupUI();
        initConnect();
    }

    ~MainWidget()
    {
        if (userProfile) {
            delete userProfile;
            userProfile = nullptr;
        }
    }
    void setupUI()
    {
        // 只创建一次布局和静态控件
        userProfile = new UserProfileWidget(this);
        chatView = new ChatWindow();
        chatView->setMinimumWidth(400);

        QWidget *rightWidget = new QWidget;
        QVBoxLayout *vbox = new QVBoxLayout(rightWidget);
        vbox->setContentsMargins(0, 0, 0, 0);
        vbox->addWidget(userProfile, 0);
        vbox->addWidget(chatView, 1);

        // 这里先实现一个房间的出来
        roomListWidget = new QListWidget;

        splitter = new QSplitter(Qt::Horizontal, this);
        // 先插入一个占位控件
        playerContainer = new QWidget(this);
        splitter->addWidget(roomListWidget);
        splitter->addWidget(playerContainer);
        splitter->addWidget(rightWidget);

        splitter->setStretchFactor(0, 10);
        splitter->setStretchFactor(1, 60);
        splitter->setStretchFactor(2, 30);

        QVBoxLayout *mainLayout = new QVBoxLayout(this);
        mainLayout->addWidget(splitter, 1);
        mainLayout->setContentsMargins(0, 0, 0, 0);
        mainLayout->setSpacing(0);

        setLayout(mainLayout);
    }
    void initConnect()
    {
        connect(AppEvent::instance(),
                &AppEvent::connectionStateChanged,
                this,
                [this](bool connected) {
                    if (userProfile) {
                        userProfile->setStatus(connected ? "在线" : "离线");
                    }
                });
    }

    // 动态插入或替换播放器控件
    void setPlayerWidget(QWidget *newPlayer)
    {
        if (playerWidget) {
            playerWidget->setParent(nullptr);
            playerWidget->deleteLater();
        }
        playerWidget = newPlayer;
        playerWidget->setParent(playerContainer);

        // 清空 playerContainer 的布局
        QLayout *oldLayout = playerContainer->layout();
        if (oldLayout) {
            QLayoutItem *item;
            while ((item = oldLayout->takeAt(0)) != nullptr) {
                if (item->widget())
                    item->widget()->setParent(nullptr);
                delete item;
            }
            delete oldLayout;
        }
        QVBoxLayout *vbox = new QVBoxLayout(playerContainer);
        vbox->setContentsMargins(0, 0, 0, 0);
        vbox->addWidget(playerWidget);
    }

    // 角色切换时调用
    void setRole(const QStringList &roleList)
    {
        QWidget *newPlayer = nullptr;
        if (roleList.contains("role.admin")) {
            newPlayer = new AvRecorder(this);
        } else {
            newPlayer = new PlayWidget(this);
        }
        setPlayerWidget(newPlayer);
    }

    QString currentRole;

    QSplitter *splitter = nullptr;
    QWidget *playerContainer = nullptr;
    QWidget *playerWidget = nullptr;
    UserProfileWidget *userProfile = nullptr;
    ChatWindow *chatView = nullptr;
    QListWidget *roomListWidget = nullptr;
};

MainWindow::MainWindow(QWidget *parent)
    : TMainWindow(parent)
{
    setWindowTitle("test");
    setCentralWidget(new QWidget(this));
    // 创建主布局
    layout = new QVBoxLayout(centralWidget());
    layout->setContentsMargins(0, 0, 0, 0);

    // 创建堆叠部件
    stackedWidget = new QStackedWidget(this);
    layout->addWidget(stackedWidget);

    // 创建登录页面
    loginWidget = new LoginWidget(this);
    stackedWidget->addWidget(loginWidget);

    mainWidget = nullptr;

    connect(loginWidget, &LoginWidget::loginRequested, this, &MainWindow::onLoginSuccess);
    resize(1280, 600);

    // 显示登录页面
    stackedWidget->setCurrentWidget(loginWidget);
}

MainWindow::~MainWindow() {}

void MainWindow::createMainWindow()
{
    if (mainWidget) {
        delete mainWidget;
        mainWidget = nullptr;
    }
    // 主窗口
    mainWidget = new MainWidget(this);
    stackedWidget->addWidget(mainWidget);

    // 退出处理
    connect(mainWidget->userProfile, &UserProfileWidget::logoutClicked, this, [this]() {
        AppEvent *appEvent = AppEvent::instance();
        appEvent->setJwtToken(QString()); // 清理token
        appEvent->setRefreshTime(0);      // 清理token刷新时间
        appEvent->setRoles({});           // 清理角色
        appEvent->setUserId({});          // 清理用户id

        if (mainWidget->userProfile) {
            stackedWidget->setCurrentWidget(loginWidget);
        }
    });

    connect(mainWidget->roomListWidget,
            &QListWidget::itemClicked,
            this,
            [this](QListWidgetItem *item) {
                if (item) {
                    const QString id = item->data(Qt::UserRole + 100).value<QString>();

                    // 重新进入房间
                    mainWidget->chatView->initWebsocket(id);
                    if (!mainWidget->playerWidget) {
                        return;
                    }
                    // 推流配置
                    if (AvRecorder *avRecorder = qobject_cast<AvRecorder *>(
                            mainWidget->playerWidget)) {
                        SettingsPage::Param param;
                        param.liveUrl = "rtmp://192.168.3.76:1935/stream/V1";
                        param.liveName = id.toStdString();
                        avRecorder->setSettings(param);
                    }
                    // 拉取视频流程
                    if (PlayWidget *playWidget = qobject_cast<PlayWidget *>(
                            mainWidget->playerWidget)) {
                        playWidget->startToPlay("rtmp://192.168.3.76:1935/stream/V1/" + id);
                    }
                }
            });
}

void MainWindow::onLoginSuccess(const QString &username, const QString &password)
{
    // 准备登录参数
    AuthApi::LoginParams loginParams;
    loginParams.username = username;
    loginParams.password = password;

    QFuture<HttpResponse> loginApiFuture = AuthApi::loginApi(loginParams);

    // 使用fromQFuture将QFuture转换为QPromise
    QtPromise::QPromise<HttpResponse> loginApiPromise = QtPromise::resolve(loginApiFuture);

    MaskOverlay::instance()->show(stackedWidget, 0, MaskOverlay::ActiveWindow);

    loginApiPromise
        .then([this](const HttpResponse &response) {
            qDebug() << response.code << response.data << response.message;
            if (response.code != 0) {
                BubbleTip::showTip(this, response.message, BubbleTip::Top, 3000);
                return;
            }
            QString token;
            if (response.data.isObject()) {
                QJsonObject dataObj = response.data.toObject();
                qDebug() << dataObj;
                token = dataObj["token"].toString();
                AppEvent::instance()->setJwtToken(token);
                authLogin(); // 获取登录信息
            }

            if (!token.isEmpty()) {
                qDebug() << "登录成功，获取到的令牌：" << token;
                // 存储令牌
            }
        })
        .finally([]() { MaskOverlay::instance()->hide(); });
}
void MainWindow::authLogin()
{
    // 创建两个异步操作的Promise
    QFuture<HttpResponse> userInfoFuture = getUserInfoApi();
    QFuture<HttpResponse> accessCodesFuture = AuthApi::getAccessCodesApi();

    // 使用fromQFuture将QFuture转换为QPromise
    QtPromise::QPromise<HttpResponse> userInfoPromise = QtPromise::resolve(userInfoFuture);
    QtPromise::QPromise<HttpResponse> accessCodesPromise = QtPromise::resolve(accessCodesFuture);

    // 使用QtPromise::all等待所有Promise完成
    QtPromise::all(QVector<QtPromise::QPromise<HttpResponse>>{userInfoPromise, accessCodesPromise})
        .then([this](const QVector<HttpResponse> &results) {
            // 处理用户信息

            // 用户信息: 200 QJsonValue(object, QJsonObject({"id":"e52e6b47-f4d3-43a6-b481-570ffcc1c9db","role":[],"username":"admin"})) "success"
            // 访问代码: 203 QJsonValue(null) "Error transferring http://127.0.0.1:8888/api/auth/codes - server replied: Not Found"
            const HttpResponse &userInfo = results[0];
            qDebug() << "用户信息:" << userInfo.code << userInfo.data << userInfo.message;
            UserInfo user = JsonMapper::fromJson<UserInfo>(userInfo.data);

            // 处理访问代码
            const HttpResponse &accessCodes = results[1];
            qDebug() << "访问代码:" << accessCodes.code << accessCodes.data << accessCodes.message;

            // 处理成功的逻辑
            if (userInfo.code == 0 && accessCodes.code == 0) {
                qDebug() << "所有请求成功完成";
                // 这里可以添加后续处理逻辑
            }
            qDebug() << "当前角色:" << user.roleName;
            qDebug() << "当前ID:" << user.id;
            AppEvent::instance()->setRoles(user.roleName);
            AppEvent::instance()->setUserId(user.id);
            AppEvent::instance()->setUserName(user.username);
            createMainWindow();

            checkRoom();
            // 创建主界面部件
            mainWidget->userProfile->setUsername(user.username);

            mainWidget->setRole(user.roleName);

            stackedWidget->addWidget(mainWidget);
            stackedWidget->setCurrentWidget(mainWidget);
        })
        .fail([this](const std::exception &e) {
            // 处理错误
            AppEvent::instance()->setRoles({});
            AppEvent::instance()->setUserId({});
            qWarning() << "登录后处理失败:" << e.what();
            BubbleTip::showTip(this,
                               QString("请求失败: %1").arg(e.what()),
                               BubbleTip::Top,
                               3000,
                               BubbleTip::Type::Error);
        });
}

void MainWindow::checkRoom()
{
    // 创建两个异步操作的Promise
    RoomInfo roomInfo;
    // 如果是管理也就是教师 直接过滤当前教师的选项
    if (AppEvent::instance()->hasRole("role.admin")) {
        roomInfo.ownerId = AppEvent::instance()->userId();
        mainWidget->roomListWidget->hide();

    } else {
        mainWidget->roomListWidget->show();
    }
    QFuture<HttpResponse> getRoomListFuture = getRoomListApi(roomInfo);

    // 使用fromQFuture将QFuture转换为QPromise
    QtPromise::QPromise<HttpResponse> userInfoPromise = QtPromise::resolve(getRoomListFuture);

    // 使用QtPromise::all等待所有Promise完成
    QtPromise::all(QVector<QtPromise::QPromise<HttpResponse>>{userInfoPromise})
        .then([this](const QVector<HttpResponse> &results) {
            const HttpResponse &getRoomList = results[0];
            qDebug() << "访问房间:" << getRoomList.data << getRoomList.rawData
                     << getRoomList.message;
            // JsonObject({"data":[{"createdAt":1752512296000,"createdId":null,"description":"","id":"019809df-75db-73db-adbb-2ca949a9329a","maxUsers":100,"name":"默认房间","ownerId":"0196e8ac-c737-7f93-ba6a-6562f2c8854d","status":1,"updatedAt":1752512296000}],"total":1})

            RoomListData roomListData(getRoomList.data.toObject());
            QList<RoomInfo> RoomInfos = roomListData.getRooms();
            qDebug() << RoomInfos.size();

            for (int var = 0; var < RoomInfos.size(); ++var) {
                const RoomInfo room = RoomInfos.at(var);
                if (room.name) {
                    QListWidgetItem *item = new QListWidgetItem(room.name.value_or(""),
                                                                mainWidget->roomListWidget);
                    item->setData(Qt::UserRole + 100, room.id.value_or(""));
                    mainWidget->roomListWidget->addItem(item);
                }
            }
            if (RoomInfos.size() > 0) {
                mainWidget->chatView->initWebsocket(RoomInfos[0].id.value());

                if (AvRecorder *avRecorder = qobject_cast<AvRecorder *>(mainWidget->playerWidget)) {
                    SettingsPage::Param param;
                    param.liveUrl = "rtmp://192.168.3.76:1935/stream/V1";
                    param.liveName = RoomInfos[0].id.value_or("").toStdString();
                    avRecorder->setSettings(param);
                }
            }
        })
        .fail([this](const std::exception &e) {
            // 处理错误
            qWarning() << "[异常]获取房间失败:" << e.what();
            BubbleTip::showTip(this,
                               QString("请求失败: %1").arg(e.what()),
                               BubbleTip::Top,
                               3000,
                               BubbleTip::Type::Error);
        });
}