RecordControlApplication/TcpController.cpp

#include "TcpController.h"
#include "Channel.h"
#include "Constant.h"
#include "Json.h"
#include "Log.h"
#include "Tool.h"
#include <QDir>
#include <QProcess>

extern const QList<Channel*> channelList;

TcpController::TcpController(QObject* parent)
    : QObject(parent)
{
    // 映射请求路径与处理函数
    routes.insert("set_ipaddr", std::bind(&TcpController::setIpaddr, this, std::placeholders::_1));
    routes.insert("set_video_enc", std::bind(&TcpController::setVideoEnc, this, std::placeholders::_1));
    routes.insert("get_video_enc", std::bind(&TcpController::getVideoEnc, this, std::placeholders::_1));
    routes.insert("get_file_list", std::bind(&TcpController::getFileList, this, std::placeholders::_1));
    routes.insert("delete_file", std::bind(&TcpController::deleteFile, this, std::placeholders::_1));
    routes.insert("set_name", std::bind(&TcpController::setName, this, std::placeholders::_1));
    routes.insert("get_name", std::bind(&TcpController::getName, this, std::placeholders::_1));
    routes.insert("set_record_mode", std::bind(&TcpController::setRecordMode, this, std::placeholders::_1));
    routes.insert("set_playback_mode", std::bind(&TcpController::setPlaybackMode, this, std::placeholders::_1));
    routes.insert("reoot", std::bind(&TcpController::reboot, this, std::placeholders::_1));
}

Routes TcpController::getRoutes()
{
    return this->routes;
}

/**
 * @brief 设置盒子的ip
 * @param socket
 */
void TcpController::setIpaddr(QTcpSocket* socket)
{
    QVariantMap params = this->parseParams(socket);
    QVariantMap netMap = Json::loadFile(Constant::NetConfigPath).toMap();

    if (params.contains("ip"))
        netMap["ip"] = params.value("ip");
    if (params.contains("mask"))
        netMap["mask"] = params.value("mask");
    if (params.contains("gateway"))
        netMap["gateway"] = params.value("gateway");

    Json::saveFile(netMap, Constant::NetConfigPath);
    socket->write("url=set_ipaddr\r\nstatus=success");
    Tool::writeCom(Constant::NetScriptPath);
}

/**
 * @brief 设置盒子的名称
 * @param socket
 */
void TcpController::setName(QTcpSocket* socket)
{
    QVariantMap params = this->parseParams(socket);
    if (params.isEmpty() || !params.contains("name")) {
        Log::error("setName params error, missing param \"name\"");
        socket->write("url=set_name\r\nstatus=error");
        return;
    }
    QVariantMap cfg = Json::loadFile(Constant::ConfigurationPath).toMap();
    cfg["name"] = params.value("name");
    Json::saveFile(cfg, Constant::ConfigurationPath);
    socket->write("url=set_name\r\nstatus=success");
}

/**
 * @brief 获取盒子名称
 */
void TcpController::getName(QTcpSocket* socket)
{
    QVariantMap config = Json::loadFile(Constant::ConfigurationPath).toMap();
    QString name = config["name"].toString();
    QString data = QString("url=get_name\r\nstatus=success\r\nname=%1").arg(name);
    socket->write(data.toStdString().data());
}

/**
 * @brief 设置视频录制参数
 * @param socket
 */
void TcpController::setVideoEnc(QTcpSocket* socket)
{
    QVariantMap params = this->parseParams(socket);
    if (params.isEmpty() || !params.contains("interface")) {
        Log::error("setVideoEncode params error, missing param \"interface\"");
        socket->write("url=set_video_enc\r\nstatus=error");
        return;
    }

    QString interface = params.value("interface").toString();
    Channel* chn = nullptr;
    for (Channel* c : channelList) {
        if (c->channelName == interface)
            chn = c;
    }
    if (chn) {
        // 修改分辨率和帧率
        chn->videoEncoder->setData(params);
        // 更新配置文件
        QVariantMap cfg = Json::loadFile(Constant::ConfigurationPath).toMap();
        QVariantList interfaces = cfg.value("interface").toList();
        for (int i = 0; i < interfaces.length(); i++) {
            QVariantMap item = interfaces.at(i).toMap();
            if (item.value("name").toString() == interface) {
                QVariantMap encV = item.value("encV").toMap();
                if (params.contains("width"))
                    encV["width"] = params.value("width");
                if (params.contains("height"))
                    encV["height"] = params.value("height");
                if (params.contains("framerate"))
                    encV["framerate"] = params.value("framerate");
                item["encV"] = encV;
                interfaces.replace(i, item);
            }
        }
        cfg["interface"] = interfaces;
        Json::saveFile(cfg, Constant::ConfigurationPath);
        socket->write("url=set_video_enc\r\nstatus=success");
    } else {
        Log::error("set video encode params error, no such interface {}", interface.toStdString());
        socket->write("url=set_video_enc\r\nstatus=error");
    }
}

/**
 * @brief 获取视频录制参数
 * @param socket
 */
void TcpController::getVideoEnc(QTcpSocket* socket)
{
    QVariantMap params = this->parseParams(socket);

    if (params.contains("interface")) {
        QString interface = params.value("interface").toString();

        QVariantMap cfg = Json::loadFile(Constant::ConfigurationPath).toMap();
        QVariantList interfaces = cfg.value("interface").toList();

        QVariantMap targetInterfaceCfg;
        for (int i = 0; i < interfaces.length(); i++) {
            QVariantMap item = interfaces.at(i).toMap();
            if (item.value("name").toString() == interface) {
                targetInterfaceCfg = item;
            }
        }
        QString width;
        QString height;
        QString framerate;
        QVariantMap encodeV = targetInterfaceCfg.value("encV").toMap();
        width = encodeV.value("width").toString();
        height = encodeV.value("height").toString();
        framerate = encodeV.value("framerate").toString();
        QString data = QString("url=get_video_enc\r\nstatus=success\r\nwidth=%1\r\nheight=%2\r\nframerate=%3")
                           .arg(width)
                           .arg(height)
                           .arg(framerate);
        socket->write(data.toStdString().data());
    } else {
        Log::error("getVideoEnc params error, missing param \"interface\"");
        socket->write("url=get_video_enc\r\nstatus=error");
        return;
    }
}

/**
 * @brief 获取文件列表
 * @param socket
 */
void TcpController::getFileList(QTcpSocket* socket)
{
    QVariantMap params = this->parseParams(socket);
    if (params.isEmpty() || !params.contains("interface")) {
        Log::error("getFileList params error, missing param \"interface\"");
        socket->write("url=get_file_list\r\nstatus=error");
        return;
    }

    QString interface = params.value("interface").toString();
    QString videoPath = QString("%1/%2").arg(Constant::VideoPath).arg(interface);

    QStringList videoList = Tool::getFileList(videoPath);

    if (videoList.isEmpty()) {
        Log::error("getFileList error, video dir is empty ");
        socket->write("url=get_file_list\r\nstatus=error");
        return;
    }
    QString data;
    data = "url=get_file_list\r\nstatus=success\r\n" + videoList.join("\r\n");
    socket->write(data.toStdString().data());
}

/**
 * @brief 删除文件
 * @param socket
 */
void TcpController::deleteFile(QTcpSocket* socket)
{
    QVariantMap params = this->parseParams(socket);
    if (!params.contains("interface") || !params.contains("filename")) {
        Log::error("deleteFile params error, missing params \"interface\" or \"filename\"");
        socket->write("url=delete_file\r\nstatus=error");
        return;
    }

    QString interface = params.value("interface").toString();
    QString fileName = params.value("filename").toString();
    // xxx.mp4  ==> xxx.jpg
    QString snapName = fileName.split(".")[0] + ".jpg";

    QString filePath = QString("%1/%2/%3").arg(Constant::VideoPath).arg(interface).arg(fileName);
    QFile video(filePath);
    if (video.exists()) {
        video.remove();
        Log::info("remove video {}", filePath.toStdString());
        socket->write("url=delete_file\r\nstatus=success");
    } else {
        Log::error("error, file: {} dont exist", filePath.toStdString());
        socket->write("url=delete_file\r\nstatus=error");
    }
}

/**
 * @brief 设置录制模式，一路录制/两路录制，只修改配置文件，重启之后生效
 * @param socket
 */
void TcpController::setRecordMode(QTcpSocket* socket)
{
    QVariantMap params = this->parseParams(socket);
    if (!params.contains("recordMode")) {
        Log::error("setRecordMode params error, missing param \"recordMode\"");
        socket->write("url=set_record_mode\r\nstatus=error");
        return;
    }
    int recordMode = params.value("recordMode").toInt();
    if (recordMode != Constant::OneChannelRecord
        && recordMode != Constant::TwoChannelRecord
        && recordMode != Constant::NoChannelRecord) {
        Log::error("setRecordMode params error, unkown param \"recordMode\": {}", recordMode);
        socket->write("url=set_record_mode\r\nstatus=error");
        return;
    }
    QVariantMap cfg = Json::loadFile(Constant::ConfigurationPath).toMap();
    int mode = cfg["recordMode"].toInt();
    if (recordMode == mode) {
        Log::info("record mode has not changed, there is nothing need to do");
        socket->write("url=set_record_mode\r\nstatus=success");
        return;
    }
    // 更新配置文件
    cfg["recordMode"] = recordMode;
    Json::saveFile(cfg, Constant::ConfigurationPath);
    socket->write("url=set_record_mode\r\nstatus=success");
}

/**
 * @brief 设置回放模式， 一路回放/两路同时回放，只修改配置文件，重启之后生效
 * @param socket
 */
void TcpController::setPlaybackMode(QTcpSocket* socket)
{
    QVariantMap params = this->parseParams(socket);
    if (!params.contains("playbackMode")) {
        Log::error("setPlaybackMode params error, missing param \"playbackmode\"");
        socket->write("url=set_playback_mode\r\nstatus=error");
        return;
    }
    int playbackMode = params.value("playbackMode").toInt();
    if (playbackMode != Constant::OneChannelPlayback
        && playbackMode != Constant::TwoChannelPlayback) {
        Log::error("setPlayBack params error, unkown param \"playbackMode\": {}", playbackMode);
        socket->write("url=set_playback_mode\r\nstatus=error");
    }

    QVariantMap cfg = Json::loadFile(Constant::ConfigurationPath).toMap();
    cfg["playbackMode"] = playbackMode;
    Json::saveFile(cfg, Constant::ConfigurationPath);
    socket->write("url=set_playback_mode\r\nstatus=success");
}

/**
 * @brief 重启程序
 * @param socket
 */
void TcpController::reboot(QTcpSocket* socket)
{
    // 先停止所有的录制
    for (Channel* chn : channelList) {
        chn->stopRecord();
    }
    socket->write("url=reboot\r\nstatus=success");
    socket->flush();

    // 退出程序，等待后台运行的shell脚本重启程序
    exit(0);
}