filemanager.cpp

#include "filemanager.h"

#include <QFileDialog>
#include <QRegularExpression>
#include <QStandardPaths>

FileManager::FileManager(QWidget *parent) :
    QTabWidget(parent)
{
    setTabsClosable(true);
    setMovable(true);

    createBackgroundImage();

    connect(this,   SIGNAL(tabCloseRequested(int)), this,   SLOT(closeFile(int)));
    connect(this,   SIGNAL(currentChanged(int)),    this,   SLOT(changeTab(int)));

    createHome();
}

bool FileManager::createHome()
{
    QSettings settings;
    settings.beginGroup("Paths");

    QDir projects(QDir::homePath() + "/" + tr("PropellerIDE Projects"));

    if (!projects.exists())
    {
        if (!QDir().mkdir(projects.path()))
        {
            qCWarning(logfilemanager) << "failed to create projects folder:" << projects.path();
            return false;
        }
    }

    settings.setValue("homeDirectory", projects.path());

    if (settings.value("lastDirectory").toString().isEmpty())
        settings.setValue("lastDirectory", projects.path());

    settings.endGroup();

    return true;
}

QString FileManager::getDirectory()
{
    QSettings settings;
    settings.beginGroup("Paths");

    return settings.value("lastDirectory", 
                settings.value("homeDirectory", QDir::homePath()).toString()).toString();
}

void FileManager::setDirectory(QString dir)
{
    QSettings settings;
    settings.beginGroup("Paths");
    settings.setValue("lastDirectory", dir);
}


// THIS IS A HACK OMG SUCH A HACK
void FileManager::setLanguage(Language * language)
{
    this->language = language;
}

int FileManager::newFile()
{
    // removes the background image (need to move this elsewhere)
    setStyleSheet("");

    Editor * editor = new Editor(language, QWidget::window());
    editor->setAttribute(Qt::WA_DeleteOnClose);
    editor->installEventFilter(QWidget::window());
    editor->saveContent();

    int index = addTab(editor,tr("Untitled"));

    setTabToolTip(index,"");

    connect(editor,SIGNAL(textChanged()),this,SLOT(fileChanged()));
    connect(editor,SIGNAL(undoAvailable(bool)),this,SLOT(setUndo(bool)));
    connect(editor,SIGNAL(redoAvailable(bool)),this,SLOT(setRedo(bool)));
    connect(editor,SIGNAL(copyAvailable(bool)),this,SLOT(setCopy(bool)));

    emit closeAvailable(true);
    setCurrentIndex(index);

    return index;
}

void FileManager::setUndo(bool available)
{
    if (sender() == currentWidget())
        emit undoAvailable(available);
}

void FileManager::setRedo(bool available)
{
    if (sender() == currentWidget())
        emit redoAvailable(available);
}

void FileManager::setCopy(bool available)
{
    if (sender() == currentWidget())
        emit copyAvailable(available);
}

void FileManager::open()
{
    QStringList fileNames = QFileDialog::getOpenFileNames(this,
                tr("Open File"), getDirectory(), "Spin Files (*.spin);;All Files (*)");

    setDirectory(QDir(fileNames[0]).path());

    for (int i = 0; i < fileNames.size(); i++)
        if (!fileNames.at(i).isEmpty())
            openFile(fileNames.at(i));

}


int FileManager::isFileOpen(const QString & fileName)
{
    // check if file already open before opening another instance
    for (int i = 0; i < count(); i++)
    {
        if (fileName == tabToolTip(i))
        {
            setCurrentIndex(i);
            emit sendMessage(tr("File already open: %1").arg(fileName));
            return 1;
        }
    }
    return 0;
}

int FileManager::isFileEmpty(int index)
{
    if (count() && (tabToolTip(index).isEmpty()
                    && getEditor(index)->toPlainText().isEmpty()
                    && !getEditor(index)->contentChanged()) )
        return 1;
    else
        return 0;
}

QString FileManager::reformatText(QString text)
{
    QRegularExpression newlines("\r\n|\r|\n",QRegularExpression::DotMatchesEverythingOption);
    text.replace(newlines,"\n");
    return text;
}

int FileManager::openFile(const QString & fileName)
{
    qCDebug(logfilemanager) << "opening" << fileName;

    if (fileName.isEmpty())
        return 1;

    QFile file(fileName);
    if (!file.open(QFile::ReadOnly))
    {
        QMessageBox::warning(this, tr("Warning"),
                             tr("Cannot read file %1:\n%2.")
                             .arg(fileName)
                             .arg(file.errorString()));
        return 1;
    }

    if (isFileOpen(fileName))
        return 1;

    int index;
    if (isFileEmpty(currentIndex()))
        index = currentIndex();
    else
        index = newFile();

    QTextStream in(&file);
    in.setAutoDetectUnicode(true);
    in.setCodec("UTF-8");
    getEditor(index)->setPlainText(reformatText(in.readAll()));

    setTabToolTip(index,QFileInfo(fileName).canonicalFilePath());
    setTabText(index,QFileInfo(fileName).fileName());
    getEditor(index)->saveContent();
    fileChanged();

    emit fileUpdated(index);
    emit sendMessage(tr("File opened successfully: %1").arg(fileName));

    return 0;
}


void FileManager::newFromFile()
{
    QString fileName = QFileDialog::getOpenFileName(this,
                tr("New From File..."), getDirectory(), "Spin Files (*.spin);;All Files (*)");

    if (!fileName.isEmpty())
        newFromFile(fileName);
}


int FileManager::newFromFile(const QString & fileName)
{
    qCDebug(logfilemanager) << "creating file from" << fileName;

    if (fileName.isEmpty())
        return 1;

    QFile file(fileName);
    if (!file.open(QFile::ReadOnly))
    {
        QMessageBox::warning(this, tr("Warning"),
                             tr("Cannot read file %1:\n%2.")
                             .arg(fileName)
                             .arg(file.errorString()));
        return 1;
    }

    int index = newFile();

    QTextStream in(&file);
    in.setAutoDetectUnicode(true);
    in.setCodec("UTF-8");
    getEditor(index)->setPlainText(reformatText(in.readAll()));

    fileChanged();

    emit fileUpdated(index);

    return 0;
}


void FileManager::save()
{
    save(currentIndex());
}

void FileManager::save(int index)
{
    QString fileName = tabToolTip(index);

    if (fileName.isEmpty())
        saveAs(index);
    else
        saveFile(fileName, index);
}

void FileManager::saveAs()
{
    saveAs(currentIndex());
}


void FileManager::saveAs(int index)
{
    QString fileName = tabToolTip(index);

    QString lastDir = QDir(fileName).path();
    if (fileName.isEmpty())
    {
        fileName = tr("Untitled.spin");
        lastDir = getDirectory();
    }

    fileName = QFileDialog::getSaveFileName(this,
            tr("Save File As..."), 
            lastDir,
            tr("Spin Files (*.spin)"));

    if (fileName.isEmpty())
        return;

    saveFile(fileName, index);
}


void FileManager::saveAll()
{
    for (int i = 0; i < count(); i++)
        save(i);
}


void FileManager::saveFile(const QString & fileName, int index)
{
    qCDebug(logfilemanager) << "saving" << fileName;

    QFile file(fileName);
    if (!file.open(QFile::WriteOnly | QFile::Text))
    {
        QMessageBox::warning(this, tr("Recent Files"),
                    tr("Cannot write file %1:\n%2.")
                    .arg(fileName)
                    .arg(file.errorString()));
        return;
    }

    QTextStream os(&file);
    QApplication::setOverrideCursor(Qt::WaitCursor);
    os.setCodec("UTF-8");
    os << getEditor(index)->toPlainText();
    os.flush();
    QApplication::restoreOverrideCursor();

    setTabToolTip(index,QFileInfo(fileName).canonicalFilePath());
    setTabText(index,QFileInfo(fileName).fileName());
    getEditor(index)->saveContent();
    fileChanged();
    emit fileUpdated(index);
    emit sendMessage(tr("File saved successfully: %1").arg(fileName));
}


void FileManager::closeFile()
{
    closeFile(currentIndex());
}

void FileManager::closeAll()
{
    setCurrentIndex(0);
    while (count() > 0)
    {
        if (!closeFile(0))
            return;
    }
}

bool FileManager::saveAndClose(int index)
{
    QMessageBox dialog;
    dialog.setText(tr("Your code has been modified."));
    dialog.setInformativeText(tr("Do you want to save your changes?"));
    dialog.setStandardButtons(QMessageBox::Save | QMessageBox::Discard | QMessageBox::Cancel);
    dialog.setDefaultButton(QMessageBox::Cancel);

    switch (dialog.exec())
    {
        case QMessageBox::Save:
            save(index);
            return true;
            break;
        case QMessageBox::Discard:
            return true;
            break;
        case QMessageBox::Cancel:
        default:
            break;
    }
    return false;
}

bool FileManager::closeFile(int index)
{
    if (getEditor(index)->contentChanged())
    {
        if (!saveAndClose(index))
        {
            return false;
        }
    }

    if (count() > 0 && index >= 0 && index < count())
    {
        getEditor(index)->disconnect();
        getEditor(index)->close();
        removeTab(index);
    }

    if (count() == 0)
    {
        createBackgroundImage();
        emit closeAvailable(false);
    }

    return true;
}

void FileManager::createBackgroundImage()
{
    setStyleSheet("background-image: url(:/icons/propellerhat.png);"
                  "background-repeat: no-repeat;"
                  "background-position: center;");
}

void FileManager::nextTab()
{
    int n = currentIndex();
    n++;
    if (n > count()-1)
        n = 0;
    setCurrentIndex(n);
}

void FileManager::previousTab()
{
    int n = currentIndex();
    n--;
    if (n < 0)
        n = count()-1;
    setCurrentIndex(n);
}

// this function is needed to set focus after changing the index
void FileManager::changeTab(int index)
{
    if(index < 0) return;

    Editor * editor = getEditor(currentIndex());
    editor->setFocus();
    
    emit undoAvailable(editor->getUndo());
    emit redoAvailable(editor->getRedo());
    emit copyAvailable(editor->getCopy());
    emit fileUpdated(index);
}

Editor * FileManager::getEditor(int num)
{
    return (Editor *)widget(num);
}

// this needs some rethinking a lot
void FileManager::fileChanged()
{
    int index = currentIndex();
    QString file = tabToolTip(index);
    QString name = QFileInfo(file).fileName();

    if (file.isEmpty())
        name = tr("Untitled");

    if (getEditor(index)->contentChanged())
    {
        name += '*';
        emit saveAvailable(true);
    }
    else
    {
        emit saveAvailable(false);
    }

    setTabText(index, name);
}

// does anybody have a good idea of how to
// replace this hacky structure with something
// more flexible?
// -------------------------------------------
void FileManager::cut()
{
    if (count() > 0)
        getEditor(currentIndex())->cut();
}

void FileManager::copy()
{
    if (count() > 0)
        getEditor(currentIndex())->copy();
}

void FileManager::paste()
{
    if (count() > 0)
        getEditor(currentIndex())->paste();
}

void FileManager::undo()
{
    if (count() > 0)
        getEditor(currentIndex())->undo();
}

void FileManager::redo()
{
    if (count() > 0)
        getEditor(currentIndex())->redo();
}

void FileManager::selectAll()
{
    if (count() > 0)
        getEditor(currentIndex())->selectAll();
}
// -------------------------------------------