#include "Tool.h"
#include <QDebug>
#include <QDir>
#include <QFileInfo>
#include <QProcess>

#ifdef Q_OS_WIN
#include <QStorageInfo>
#endif

Tool::Tool()
{
}

/**
 * @brief get file list of
 */
QStringList Tool::getFileList(QString path)
{
    QStringList fileList;
    QDir dir(path);
    if (!dir.exists()) {
        qDebug() << QString("dir %1 do not exsit...").arg(path);
        return fileList;
    }
    dir.setFilter(QDir::Files);
    dir.setSorting(QDir::Name);
    QStringList nameFilters;
    nameFilters << "*.mp4";
    dir.setNameFilters(nameFilters);
    fileList = dir.entryList();
    return fileList;
}

/**
 * @brief remove file or dir
 */
bool Tool::removeFile(QString path)
{
    QFileInfo info(path);
    if (!info.exists()) {
        qDebug() << path << "do not exsit";
        return false;
    }
    if (info.isFile()) {
        QFile file(path);
        file.remove();
        qDebug() << "remove file" << path << "success";
        return true;
    } else if (info.isDir()) {
        QDir dir(path);
        dir.removeRecursively();
        qDebug() << "remove dir" << path << "success";
        return true;
    }
    qDebug() << path << "is not file or dir";
    return false;
}

/**
 * @brief call linux shell tool
 */
#ifdef Q_OS_UNIX
QString Tool::writeCom(QString path)
{
    QProcess proc;
    QStringList argList;
    argList << "-c" << path;
    proc.start("/bin/sh", argList);
    // wait process start
    proc.waitForFinished();
    proc.waitForReadyRead();
    // read data from console
    QByteArray procOutput = proc.readAll();
    proc.close();
    return QString(procOutput);
}
#endif

/**
 * get available storage of disk
 */
int64_t Tool::getAvailableStorage(QString mountedPath)
{
#ifdef Q_OS_UNIX
    // get the available size of disk
    QString mountInfo = writeCom(QString("df %1").arg(mountedPath));
    // parse the string like
    // "Filesystem           1K-blocks      Used Available Use% Mounted on\n/dev/sda             487110880     45112 462630056   0% /root/usb\n"
    QString diskInfo = mountInfo.split("\n")[1];
    QStringList list = diskInfo.split(" ");
    int index = -1;
    int64_t used;
    int64_t available;
    for (int i = 0; i < list.length(); i++) {
        QString s = list.at(i).trimmed();
        if (s.isEmpty())
            continue;
        index += 1;
        if (index == 2)
            used = s.toInt();
        else if (index == 3)
            available = s.toInt();
    }
    qDebug() << QString("%1: used %2 bytes, available %3 bytes")
                    .arg(mountedPath)
                    .arg(used)
                    .arg(available);
    return available;
#endif
#ifdef Q_OS_WIN
    QList<QStorageInfo> list = QStorageInfo::mountedVolumes();
    for (QStorageInfo& info : list) {
        if (info.rootPath() == mountedPath) {
            int64_t available = info.bytesAvailable();
            return available;
        }
    }
    return 0;
#endif
}