explorer.cpp

#include "explorer.hh"
#include <QCoreApplication>
#include <QVBoxLayout>
#include <QClipboard>
#include <QGuiApplication>
#include <QDir>
#include <QFile>
#include <QTextStream>
#include "borderdelegate.hh"
#include "customiconfilesystemmodel.hh"

Explorer::Explorer(MainWindow* window, QWidget *parent)
    : QWidget(parent), window(window)
{
    targetFolderId = 0;

    QVBoxLayout *layout = new QVBoxLayout(this);

    m_listView = new QListView(this);
    layout->addWidget(m_listView);
    layout->setContentsMargins(0, 0, 0, 0);

    m_model = new CustomIconFileSystemModel(window, this);
    m_model->setFilter(QDir::Files);
    m_model->setHeaderData(0, Qt::Horizontal, "File Explorer");

    m_listView->setModel(m_model);
    m_listView->setViewMode(QListView::IconMode);

    if(window->isFullscreenOn()){
        m_listView->setGridSize(QSize(200, 200));
        m_listView->setIconSize(QSize(199, 199));
    }
    else{
        m_listView->setGridSize(QSize(100, 100));
        m_listView->setIconSize(QSize(99, 99));
    }

    m_listView->setSpacing(1);
    m_listView->setResizeMode(QListView::Adjust);
    m_listView->setUniformItemSizes(false);
    m_listView->setMovement(QListView::Static);
    m_listView->setStyleSheet("border: 1px solid transparent;");
    m_listView->installEventFilter(this);
    m_listView->setItemDelegate(new BorderDelegate(this));

    m_emptyLabel = new QLabel("This folder is empty", this);
    m_emptyLabel->setAlignment(Qt::AlignCenter);
    m_emptyLabel->hide();
    setStyleSheet("color: rgb(154, 160, 166);");

    layout->addWidget(m_emptyLabel);

    this->setLayout(layout);
    connect(m_listView, &QListView::clicked, this, &Explorer::onFileClicked);
    connect(m_model, &QFileSystemModel::directoryLoaded, this, &Explorer::onDirectoryLoaded);

    updateModelRootPath();
}

void Explorer::updateModelRootPath()
{
    QString rootPath;

    if (targetFolderId == 0)
    {
        rootPath = QCoreApplication::applicationDirPath() + "/files/library";
    }
    else if (targetFolderId == 1)
    {
        rootPath = QCoreApplication::applicationDirPath() + "/files/images";
    }
    else if (targetFolderId == 2)
    {
        rootPath = QCoreApplication::applicationDirPath() + "/files/gifs";
    }
    else if (targetFolderId == 3)
    {
        rootPath = QCoreApplication::applicationDirPath() + "/files/texts";
    }
    else if (targetFolderId == 4)
    {
        rootPath = QCoreApplication::applicationDirPath() + "/files/videos";
    }

    QDir dir;
    if (!dir.exists(rootPath))
    {
        dir.mkpath(rootPath);
    }

    m_model->setRootPath(rootPath);
    m_listView->setRootIndex(m_model->index(rootPath));

    emit targetFolderChanged(targetFolderId);
}

void Explorer::onFileClicked(const QModelIndex &index)
{
    QString filePath = m_model->filePath(index);
    QString resolvedPath = resolveShortcut(filePath);
    QFileInfo fileInfo(resolvedPath);

    if (fileInfo.suffix().toLower() == "txt") {
        QFile file(resolvedPath);
        if (file.open(QIODevice::ReadOnly)) {
            QTextStream in(&file);
            QString content = in.readAll();
            file.close();

            QClipboard *clipboard = QGuiApplication::clipboard();
            clipboard->setText(content);
        }
    } else {
        QUrl fileUrl = QUrl::fromLocalFile(resolvedPath);

        QMimeData *mimeData = new QMimeData();
        mimeData->setUrls(QList<QUrl>() << fileUrl);

        QClipboard *clipboard = QGuiApplication::clipboard();
        clipboard->setMimeData(mimeData);
    }

    if (window && window->isFullscreenOn()) {
        window->toggleFullscreen();
    }
}

QString Explorer::resolveShortcut(const QString &filePath)
{
    QFileInfo fileInfo(filePath);

    // Handling Windows shortcuts
    #ifdef Q_OS_WIN
        if (fileInfo.suffix().toLower() == "lnk")
        {
            QString targetPath;
            CoInitialize(NULL);
            IShellLink *pShellLink;
            HRESULT hres = CoCreateInstance(CLSID_ShellLink, NULL, CLSCTX_INPROC_SERVER, IID_IShellLink, (LPVOID *)&pShellLink);

            if (SUCCEEDED(hres))
            {
                IPersistFile *pPersistFile;
                hres = pShellLink->QueryInterface(IID_IPersistFile, (LPVOID *)&pPersistFile);

                if (SUCCEEDED(hres))
                {
                    hres = pPersistFile->Load((LPCOLESTR)filePath.utf16(), STGM_READ);
                    if (SUCCEEDED(hres))
                    {
                        WIN32_FIND_DATA wfd;
                        WCHAR wsz[MAX_PATH];
                        hres = pShellLink->GetPath(wsz, MAX_PATH, (WIN32_FIND_DATA *)&wfd, SLGP_UNCPRIORITY);
                        if (SUCCEEDED(hres))
                        {
                            targetPath = QString::fromUtf16(reinterpret_cast<const char16_t*>(wsz));

                        }
                    }
                    pPersistFile->Release();
                }
                pShellLink->Release();
            }
            CoUninitialize();

            return targetPath.isEmpty() ? filePath : targetPath;
        }
    #endif

    //#ifdef Q_OS_LINUX
    //    if (fileInfo.isSymLink()) {
    //        return fileInfo.symLinkTarget();
    //    }
    //#endif

    return filePath;
}

void Explorer::onAllFilesButtonClicked()
{
    targetFolderId = 0;
    updateModelRootPath();
}

void Explorer::onImagesButtonClicked()
{
    targetFolderId = 1;
    updateModelRootPath();
}

void Explorer::onGifsButtonClicked()
{
    targetFolderId = 2;
    updateModelRootPath();
}

void Explorer::onTextsButtonClicked()
{
    targetFolderId = 3;
    updateModelRootPath();
}

void Explorer::onVideosButtonClicked()
{
    targetFolderId = 4;
    updateModelRootPath();
}

bool Explorer::eventFilter(QObject *obj, QEvent *event)
{
    if (obj == m_listView && event->type() == QEvent::KeyPress)
    {
        QKeyEvent *keyEvent = static_cast<QKeyEvent *>(event);
        if (keyEvent->key() == Qt::Key_Delete)
        {
            QModelIndex currentIndex = m_listView->currentIndex();
            if (currentIndex.isValid())
            {
                QString filePath = m_model->filePath(currentIndex);
                QFileInfo fileInfo(filePath);

                bool isShortcut = false;
                QString targetPath;

                #ifdef Q_OS_WIN
                if (fileInfo.suffix().toLower() == "lnk")
                {
                    isShortcut = true;
                    targetPath = resolveShortcut(filePath);
                }
                #endif

                //#ifdef Q_OS_LINUX
                //    if (fileInfo.isSymLink()) {
                //        isShortcut = true;
                //        targetPath = fileInfo.symLinkTarget();
                //    }
                //#endif

                if (isShortcut)
                {
                    QFile targetFile(targetPath);
                    if (!targetFile.remove())
                    {
                        QMessageBox::warning(this, tr("Delete File"), tr("Could not delete target file. Please check permissions."));
                    }
                }

                QFile file(filePath);
                if (file.remove())
                {
                    m_model->remove(currentIndex);
                }
                else
                {
                    QMessageBox::warning(this, tr("Delete File"), tr("Could not delete file. Please check permissions."));
                }
            }
            return true;
        }
    }

    return QWidget::eventFilter(obj, event);
}

void Explorer::onDirectoryLoaded(const QString &)
{
    if (m_model->rowCount(m_listView->rootIndex()) == 0)
    {
        m_emptyLabel->show();
        m_listView->hide();
    }
    else
    {
        m_emptyLabel->hide();
        m_listView->show();
    }
}