filestree.cpp

#include "filestree.h"
#include "ui_filestree.h"

FilesTree::FilesTree(QWidget *parent) :
    QDialog(parent),
    ui(new Ui::FilesTree)
{
    ui->setupUi(this);

    loadLocalDirectory();

    signalConnection = connect(AppSingleton::instance()->client, &WsClient::onTreeChange, this, &FilesTree::onTreeChangeEvent);

    on_syncTree_clicked();
}

FilesTree::~FilesTree()
{
    delete ui;
}

void FilesTree::loadLocalDirectory()
{
    TreeItem *rootItem = new TreeItem({tr("Name"), tr("Type"), tr("Size")});
    QString path = AppSingleton::instance()->getLocalPath();
    QDir dir(path);

    qDebug() << dir.entryList(QStringList(), QDir::Files);
    qDebug() << dir.entryList(QStringList(), QDir::Dirs);

    QQueue<QDir> q;
    QQueue<TreeItem*> qt;

    q.enqueue(dir);
    qt.enqueue(rootItem);

    while (!q.empty()) {
        // this code is mean to work on linux need changes in window and mac
        QDir d = q.dequeue();
        TreeItem *father = qt.dequeue();

        QStringList dirs = d.entryList(QStringList(), QDir::Dirs);
        for (int i = 0; i < dirs.size(); i++) {
            if (dirs.at(i).compare(".", Qt::CaseInsensitive) == 0 || dirs.at(i).compare("..", Qt::CaseInsensitive) == 0) {
                continue;
            }
            qDebug() << dirs.at(i);

            // this code is mean to work on linux need changes in window and mac
            QString path = QString("%1/%2").arg(d.absolutePath(), dirs.at(i));
            QDir nd(path);
            qDebug() << path;

            TreeItem *dirItem = new TreeItem({dirs.at(i), tr(""), tr("")}, father, 0, path, "", FOLDER);
            father->appendChild(dirItem);

            q.enqueue(nd);
            qt.enqueue(dirItem);
        }

        QStringList files = d.entryList(QStringList(), QDir::Files);
        for (int i = 0; i < files.size(); i++) {
            QString path = QString("%1/%2").arg(d.absolutePath(), files.at(i));
            QPair<QString, TreeItemType> type = getFileType(files.at(i));
            TreeItem *dirItem = new TreeItem({files.at(i), type.first, tr("")}, father, 0, path, "", type.second);
            father->appendChild(dirItem);
        }
    }

    ui->treeView_2->setModel(new ItemModel(rootItem));
}

QPair<QString, TreeItemType> FilesTree::getFileType(QString fileName)
{
    QStringList videos = QString("webm mkv flv vob ogv ogg drc mng avi mts m2ts mov qt wmv yuv rm rmvb viv asf amv mp4 m4p m4v mpg mp2 mpeg mpe mpv m2v svi 3gp 3g2 mxf roq nsv flv f4v f4p f4a f4b").split(" ");
    for (int i = 0; i < (int) videos.size(); i++) {
        if (fileName.endsWith(QString(".%1").arg(videos.at(i)), Qt::CaseInsensitive)) {
            return qMakePair("video", VIDEO);
        }
    }

    QStringList images = QString("jpg jpeg jfi jpe jif jfif heif heic png gif svg svgz eps webp tiff tif pnp raw arw cr rw2 nrw k25").split(" ");
    for (int i = 0; i < (int) images.size(); i++) {
        if (fileName.endsWith(QString(".%1").arg(images.at(i)), Qt::CaseInsensitive)) {
            return qMakePair("image", IMAGE);
        }
    }

    return qMakePair("unknow", OTHER);
}

void FilesTree::createParentPaths(TreeItem *treeItem)
{
    TreeItem* c=treeItem;

    QStack<QDir> paths;

    while (c != nullptr) {
        QDir dir(QString("%1%2").arg(AppSingleton::instance()->getLocalPath(), c->getPath()));
        if (dir.exists()) {
            break;
        } else {
            if (c->getType() == FOLDER || c->getType() == ROOT) {
                paths.push(dir);
            }
            c = c->parentItem();
        }
    }

    while (!paths.empty()) {
        QDir dir = paths.pop();
        dir.mkpath(".");
    }
}

void FilesTree::createChildrenPaths(TreeItem *treeItem)
{
    QQueue<TreeItem*> q;
    q.enqueue(treeItem);

    while (!q.empty()) {
        TreeItem *it = q.dequeue();
        QDir dir(QString("%1%2").arg(AppSingleton::instance()->getLocalPath(), it->getPath()));
        if (!dir.exists()) {
            dir.mkpath(".");
        }

        for (int i = 0; i < it->childCount(); i++) {
            TreeItem *child = it->child(i);
            if (child->getType() == FOLDER) {
                q.enqueue(child);
            }
        }
    }
}

QList<TreeItem *> FilesTree::listOfCopyItems(TreeItem *treeItem)
{
    QList<TreeItem*> res;

    if (treeItem->getType() == FOLDER) {
        QQueue<TreeItem*> q;
        q.enqueue(treeItem);

        while (!q.empty()) {
            TreeItem *it = q.dequeue();
            for (int i = 0; i < it->childCount(); i++) {
                TreeItem *child = it->child(i);
                if (child->getType() == FOLDER) {
                    q.enqueue(child);
                } else if (child->getType() == FOLDER || child->getType() == IMAGE || child->getType() == OTHER) {
                    res.append(child);
                }
            }
        }
    } else if (treeItem->getType() == VIDEO || treeItem->getType() == IMAGE || treeItem->getType() == OTHER) {
        qDebug() << treeItem;
        res.append(treeItem);
    }

    return res;
}

void FilesTree::onTreeChangeEvent(QJsonDocument json)
{
    qDebug() << "Tree change event";

    TreeItem *rootItem = new TreeItem({tr("Name"), tr("Type"), tr("Size")});
    QQueue<QJsonObject> q;
    QQueue<TreeItem*> qt;

    q.enqueue(json.object().value("root").toObject());
    qt.enqueue(rootItem);

    while (!q.empty()) {
        QJsonObject it = q.dequeue();
        TreeItem* father = qt.dequeue();

        QJsonArray dirs = it.value("dirs").toArray();
        if (!dirs.empty()) {
            for (int i = 0; i < dirs.size(); i++) {
                QJsonObject d = dirs.at(i).toObject();
                QString name = d.value("name").toString();
                QString path = QString("%1/%2").arg(father->getPath(), name);
                TreeItem *dirItem = new TreeItem({name, tr(""), tr("")}, father, 0, path, d.value("hash").toString(), FOLDER);
                father->appendChild(dirItem);

                q.enqueue(d);
                qt.enqueue(dirItem);
            }
        }

        QJsonArray files = it.value("files").toArray();
        if (!files.empty()) {
            for (int i = 0; i < files.size(); i++) {
                QJsonObject f = files.at(i).toObject();
                QString name = f.value("name").toString();
                QString path = QString("%1/%2").arg(father->getPath(), name);
                int size = f.value("size").toInt();
                QPair<QString, TreeItemType> type = getFileType(name);
                TreeItem *fileItem = new TreeItem({name, type.first, f.value("size").toInteger()}, father, size, path, f.value("hash").toString(), type.second);
                father->appendChild(fileItem);
            }
        }
    }

    ui->treeView->setModel(new ItemModel(rootItem));
}

void FilesTree::on_syncTree_clicked()
{
    AppSingleton::instance()->client->getTree();
}

void FilesTree::on_copyButton_clicked()
{
    QModelIndexList selectionList = ui->treeView->selectionModel()->selectedIndexes();
    if (!selectionList.empty()) {
        QModelIndex item = selectionList.at(0);
        TreeItem *treeItem = static_cast<TreeItem*>(item.internalPointer());
        qDebug() << treeItem->getPath();

        createParentPaths(treeItem);
        if (treeItem->getType() == FOLDER) {
            createChildrenPaths(treeItem);
        }

        // add copy files to list
        QList<TreeItem*> files = listOfCopyItems(treeItem);
        for (int i = 0; i < files.size(); i++) {
            // todo: check here if the files was already copied
            AppSingleton::instance()->copyThread->appendCopy(files.at(i)->getPath(), files.at(i)->getHash(), files.at(i)->getSize());
        }

        AppSingleton::instance()->copyThread->reanudateCopies();
    }
}

void FilesTree::on_removeButton_clicked()
{
    // todo
}


void FilesTree::on_FilesTree_rejected()
{
    disconnect(signalConnection);
}


void FilesTree::on_FilesTree_accepted()
{
    disconnect(signalConnection);
}