src/ui/widgets/project_tree.cpp - master - SerZok/LIDE

src/ui/widgets/project_tree.cpp
294 строки9 KB
SerZok
fix [EditTabs]: Исправление TreeProject при перетаскивании вкладок
05 апр 2026, 17:11
05 апр 2026, 17:116668497
#include "project_tree.h"
#include <QMenu>
#include <QFileDialog>
#include <QMessageBox>
#include <QDesktopServices>
#include <QKeyEvent>
#include <QHeaderView>
#include <QInputDialog>
#include <QStandardPaths>
#include <QDockWidget>

// ==================[ ProjectTree ]==================
ProjectTree::ProjectTree(QWidget* parent)
    : QTreeView(parent)
    , m_settings(Settings::instance())
    , m_model(new QFileSystemModel(this))
    , m_delegate(new ProjectTreeDelegate(this))
    , m_newFileAction(nullptr)
    , m_newFolderAction(nullptr)
    , m_openFileAction(nullptr)
{
    m_model->setFilter(QDir::AllEntries | QDir::NoDotAndDotDot | QDir::AllDirs);
    m_model->setNameFilters(m_settings->projectExcludeFiltersList());
    m_model->setNameFilterDisables(false);
    m_model->setReadOnly(false);

    setModel(m_model);
    setItemDelegate(m_delegate);

    setAnimated(true);
    setHeaderHidden(true);
    setExpandsOnDoubleClick(false);
    setContextMenuPolicy(Qt::CustomContextMenu);
    setEditTriggers(QAbstractItemView::EditKeyPressed);

    header()->hideSection(1); // Size
    header()->hideSection(2); // Type
    header()->hideSection(3); // Date

    connect(this, &QTreeView::doubleClicked, this, &ProjectTree::onItemActivated);
    connect(this, &QWidget::customContextMenuRequested, this, &ProjectTree::onContextMenuRequested);
    connect(m_settings, &Settings::projectFilterChanged, this, [this]() { m_model->setNameFilters(m_settings->projectExcludeFiltersList()); });

    restoreState();
}

ProjectTree::~ProjectTree()
{
    saveState();
}


bool ProjectTree::openProject(const QString& path)
{
    QFileInfo info(path);
    if (!info.exists() || !info.isDir()) {
        return false;
    }

    m_rootPath = QDir::cleanPath(path);
    m_model->setRootPath(m_rootPath);

    QModelIndex rootIndex = m_model->index(m_rootPath);
    setRootIndex(rootIndex);
    selectionModel()->setCurrentIndex(
        rootIndex,
        QItemSelectionModel::ClearAndSelect
    );
    expandToDepth(1);
    scrollTo(rootIndex, QAbstractItemView::PositionAtTop);

    emit projectOpened(m_rootPath);
    return true;
}

QString ProjectTree::rootPath() const
{
    return m_rootPath;
}

void ProjectTree::onItemActivated(const QModelIndex& index)
{
    if (!index.isValid()) return;

    if (!m_model->isDir(index)) {
        QString filePath = m_model->filePath(index);
        emit fileActivated(filePath);
    }
    else {
        setExpanded(index, !isExpanded(index));
    }
}

void ProjectTree::onContextMenuRequested(const QPoint& pos)
{
    QModelIndex index = indexAt(pos);
    QMenu menu(this);

    if (index.isValid()) {
        QString path = m_model->filePath(index);
        bool isDir = m_model->isDir(index);

        if (isDir) {
            QAction* newFileAction = menu.addAction(tr("Новый файл..."));
            QAction* newFolderAction = menu.addAction(tr("Новая папка..."));

            connect(newFileAction, &QAction::triggered, [this, path]() {
                QString fileName = QInputDialog::getText(this, tr("Новый файл"), tr("Название файла:"));
                if (!fileName.isEmpty()) {
                    QFile file(path + "/" + fileName);
                    file.open(QIODevice::WriteOnly);
                    file.close();
                }
                });

            connect(newFolderAction, &QAction::triggered, [this, path]() {
                QString dirName = QInputDialog::getText(this, tr("Новая папка"), tr("Название папки:"));
                if (!dirName.isEmpty()) {
                    QDir().mkdir(path + "/" + dirName);
                }
                });

            menu.addSeparator();
        }
        else {
            QAction* openFileAction = menu.addAction(tr("Открыть"));
            connect(openFileAction, &QAction::triggered, [this, index]() {
                onItemActivated(index);
                });
        }

        QAction* renameAction = menu.addAction(tr("Переименовать"));
        QAction* deleteAction = menu.addAction(tr("Удалить"));
        menu.addSeparator();
        QAction* showInFolderAction = menu.addAction(tr("Показать в проводнике"));

        connect(renameAction, &QAction::triggered, [this, index]() {
            if (index.isValid() && (m_model->flags(index) & Qt::ItemIsEditable)) {
                edit(index);
            }
            });

        connect(deleteAction, &QAction::triggered, [this, index, path]() {
            if (QMessageBox::question(this, tr("Подтверждение"),
                tr("Удалить %1?").arg(path)) == QMessageBox::Yes) {
                m_model->remove(index);
            }
            });

        connect(showInFolderAction, &QAction::triggered, [path]() {
            QDesktopServices::openUrl(QUrl::fromLocalFile(QFileInfo(path).path()));
            });
    }
    else {
        QAction* newFileAction = menu.addAction(tr("Новый файл..."));
        QAction* newFolderAction = menu.addAction(tr("Новая папка..."));

        connect(newFileAction, &QAction::triggered, [this]() {
            QString fileName = QInputDialog::getText(this, tr("Новый файл"), tr("Название файла:"));
            if (!fileName.isEmpty()) {
                QFile file(m_rootPath + "/" + fileName);
                file.open(QIODevice::WriteOnly);
                file.close();
            }
            });

        connect(newFolderAction, &QAction::triggered, [this]() {
            QString dirName = QInputDialog::getText(this, tr("Новая папка"), tr("Название папки:"));
            if (!dirName.isEmpty()) {
                QDir().mkdir(m_rootPath + "/" + dirName);
            }
            });

        QAction* showInFolderAction = menu.addAction(tr("Показать в проводнике"));
        connect(showInFolderAction, &QAction::triggered, [this]() {
            QDesktopServices::openUrl(QUrl::fromLocalFile(m_rootPath));
            });
    }

    menu.exec(viewport()->mapToGlobal(pos));
}

void ProjectTree::keyPressEvent(QKeyEvent* event)
{
    if (event->key() == Qt::Key_Delete) {
        QModelIndex index = currentIndex();
        if (index.isValid()) {
            QString path = m_model->filePath(index);
            if (QMessageBox::question(this, tr("Подтверждение"),
                tr("Удалить %1?").arg(path)) == QMessageBox::Yes) {
                m_model->remove(index);
            }
        }
    }

    QTreeView::keyPressEvent(event);
}

QModelIndex ProjectTree::findFileIndex(const QString& path)
{
    QString normalized = QFileInfo(path).absoluteFilePath();
    return m_model->index(normalized);
}

void ProjectTree::expandToPath(const QString& path)
{
    QModelIndex index = m_model->index(path);
    if (!index.isValid()) return;

    // Разворачиваем всех родителей
    QModelIndex parent = index.parent();
    while (parent.isValid()) {
        expand(parent);
        parent = parent.parent();
    }
}

void ProjectTree::onFileLoaded(const QString& path)
{
    m_currentFile = path;
    m_delegate->setCurrentFile(path);

    QModelIndex index = findFileIndex(path);
    if (index.isValid()) {
        expandToPath(path);
        selectionModel()->setCurrentIndex(
            index,
            QItemSelectionModel::ClearAndSelect
        );
        scrollTo(index, QAbstractItemView::PositionAtCenter);
        update(index);
    }
}

void ProjectTree::onFileSaved(const QString& path)
{
    m_modifiedFiles.remove(path);
    m_delegate->setModifiedFiles(m_modifiedFiles);

    QModelIndex index = findFileIndex(path);
    if (index.isValid()) {
        update(index);
    }
}

void ProjectTree::onFileModifiedChanged(const QString& path, bool modified)
{
    if (modified) {
        m_modifiedFiles.insert(path);
    }
    else {
        m_modifiedFiles.remove(path);
    }
    m_delegate->setModifiedFiles(m_modifiedFiles);

    QModelIndex index = findFileIndex(path);
    if (index.isValid()) {
        update(index);
    }
}

void ProjectTree::onFileClosed(const QString& path)
{
    if (m_currentFile == path) {
        m_currentFile.clear();
        m_delegate->setCurrentFile("");
    }
    m_modifiedFiles.remove(path);
    m_delegate->setModifiedFiles(m_modifiedFiles);

    QModelIndex index = findFileIndex(path);
    if (index.isValid()) {
        update(index);
        if (currentIndex() == index) {
            clearSelection();
        }
    }
}

void ProjectTree::saveState()
{
    m_settings->setLastProjectPath(m_rootPath);
}

void ProjectTree::restoreState()
{
    QString lastPath = m_settings->lastProjectPath();
    if (!lastPath.isEmpty() && QDir(lastPath).exists()) {
        openProject(lastPath);
        return;
    }

    openProject(m_settings->projectDefaultPath());
}