#include "ImageGrabberI.h"
#include <functional>
#include <algorithm>
#include <dirent.h>
#include <basic/debug/Debug.h>
#include <cstring>
#include <opencv2/opencv.hpp>
#include <QtCore/QDateTime>
#include <basic/util/opencv/CvUtil.h>
#include <basic/util/app/AppPreference.hpp>

//获取目录下所有文件的名称
std::vector<std::string> forEachFile(const std::string &dir_name) {
    std::vector<std::string> v;
    auto dir = opendir(dir_name.data());
    struct dirent *ent;
    int len = 0;
    if (dir) {
        while ((ent = readdir(dir)) != NULL) {
            std::string p = std::string(ent->d_name);
            len = strlen(p.data());
            if (len == 25) {
                v.emplace_back(p);
            }
        }
        closedir(dir);
    }
    return v;
}

//将视频中的指定帧存为图片
cv::Mat Video2Imag(std::string VideoName, int msec) {
    cv::VideoCapture m;
    cv::Mat img = cv::Mat();

    m.open(VideoName);
    m.set(CV_CAP_PROP_POS_MSEC, msec);
    if (!m.isOpened()) {
        ERR("视频读入错误");
        return img;
    }

    m >> img;

    return img;
}

ImageGrabberI::ImageGrabberI() : fdfsClient("./client.conf") {

}

RecordVideo::ByteSequence
ImageGrabberI::grabImage(const Ice::Int index, const std::string &time, const Ice::Current &) {
    ::RecordVideo::ByteSequence buffer;
    QDateTime dt = QDateTime::fromString(time.data(), "yyyy-MM-dd hh:mm:ss");
    QDateTime dtCurrent = QDateTime::currentDateTime();
    qint64 diff = dtCurrent.toMSecsSinceEpoch() - dt.toMSecsSinceEpoch();
    if (diff > appPref.getLongData("user.timeout") || diff < appPref.getLongData("user.interval.duration")) {
        return buffer;
    }
    std::string path = appPref.getStringData("user.loop.absolute.path");
    std::string absPath;

    if (path.empty()) {
        ERR("路径为空!");
        return buffer;
    }

    char last = path.back();
    if (last != '/') {
        absPath = path + "/";
    } else {
        absPath = path;
    }

    std::string definitePath = absPath + QString::number(index).toStdString() + "/";

    vec.clear();
    vec = forEachFile(definitePath);
    std::sort(vec.begin(), vec.end());

    if (vec.empty()) {
        ERR("没有找到录制视频");
        return buffer;
    }

    for (std::string &s : vec) {
        std::string temp;
        temp = s.substr(0, 21);

        QDateTime dtFile = QDateTime::fromString(temp.data(), "yyyyMMdd_hh:mm:ss:zzz");
        qint64 sub = dt.toMSecsSinceEpoch() - dtFile.toMSecsSinceEpoch();
        if (sub <= appPref.getLongData("user.interval.duration") && sub > 0) {
            std::string fullPath = definitePath + s;
            cv::Mat img = Video2Imag(fullPath, sub);
            CvUtil::cvMat2Buffer(img, buffer);
            return buffer;
        }
    }

    ERR("没有该时段的录制视频");
    return buffer;
}

std::string ImageGrabberI::grabImageUrl(const Ice::Int index, const std::string &time, const Ice::Current &) {
    std::string strHttpImgUrl;
    ::RecordVideo::ByteSequence buffer = grabImage(index, time);

    if (buffer.empty()) {
        return strHttpImgUrl;
    }

    std::string strImgUrl;
    fdfsClient.uploadFile(buffer, strImgUrl, "jpg");
    strHttpImgUrl = appPref.getStringData("user.ip.port") + "/" + strImgUrl;
    return strHttpImgUrl;
}