#include "AutoSaveDialog.hpp"

// Qt Headers
#include <QSettings>
#include <QDir>
#include <QFileDialog>
#include <QMessageBox>

// Roblox Studio Headers
#include "AuthoringSettings.hpp"
#include "RobloxDocManager.hpp"
#include "UpdateUIManager.hpp"
#include "RobloxMainWindow.hpp"
#include "RobloxIDEDoc.hpp"
#include "RobloxApplicationManager.hpp"

LOGVARIABLE(AutoSave, 0)

AutoSaveDialog::AutoSaveDialog(QWidget* Parent)
    : QDialog(Parent, Qt::WindowTitleHint | Qt::WindowSystemMenuHint)
{
    setAttribute(Qt::WA_DeleteOnClose, false);

    m_UI.setupUi(this);
    this->setStyleSheet("background-color: white;");

#ifdef Q_WS_WIN32

    QString text =
        tr("<font size=4><b>An auto-save recovery file was detected!</b></font><br><br>"
           "This means that " AYA_PROJECT_NAME " Studio was not shut down properly and <u><i>may not have saved your data</i></u>!<br><br>"
           "Do you want to load the auto-save recovery file?<br>"
           "<ul>"
           "    <li>"
           "        <font size=4><b><i>Open</i></b></font> will open the file.  You can choose which file to load if there is more than one "
           "            recovery file.  Once an auto-save is loaded, it will not be deleted until the file is re-saved "
           "            with a different filename.<br>"
           "    </li>"
           "    <li>"
           "        <font size=4><b><i>Ignore</i></b></font> will continue and leave the files alone.  However, the next time " AYA_PROJECT_NAME " Studio "
           "            is run, this message box will pop up again.<br>"
           "    </li>"
           "    <li>"
           "        <font size=4><b><i>Delete</i></b></font> will ask to delete all auto-save files.<br>"
           "    </li>"
           "</ul>"
           "<br><br>");

#else

    QString text =
        tr("<font size=3><font size=4><b>An auto-save recovery file was detected!</b></font><br><br>"
           "This means that " AYA_PROJECT_NAME " Studio was not shut down properly and <u><i>may not have saved your data</i></u>!<br><br>"
           "Do you want to load the auto-save recovery file?<br>"
           "<ul>"
           "    <li>"
           "        <font size=4><b><i>Open</i></b></font> will open the file.  You can choose which file to load if there is more than one "
           "            recovery file.  Once an auto-save is loaded, it will not be deleted until the file is re-saved "
           "            with a different filename.<br>"
           "    </li>"
           "    <li>"
           "        <font size=4><b><i>Ignore</i></b></font> will continue and leave the files alone.  However, the next time " AYA_PROJECT_NAME " Studio "
           "            is run, this message box will pop up again.<br>"
           "    </li>"
           "    <li>"
           "        <font size=4><b><i>Delete</i></b></font> will ask to delete all auto-save files.<br>"
           "    </li>"
           "</ul>"
           "<br><br></font>");

#endif

    m_UI.label->setText(text);

    connect(m_UI.openButton, SIGNAL(clicked()), this, SLOT(onOpen()));
    connect(m_UI.ignoreButton, SIGNAL(clicked()), this, SLOT(onIgnore()));
    connect(m_UI.deleteButton, SIGNAL(clicked()), this, SLOT(onDelete()));
}

AutoSaveDialog::~AutoSaveDialog() {}

/**
 * Checks to see if there are any auto-save files.
 */
bool AutoSaveDialog::checkForAutoSaveFiles()
{
    // don't show if we've loaded a place
    if (RobloxDocManager::Instance().getPlayDoc())
        return false;

    // don't check for auto-saves if there are other instances running
    if (RobloxApplicationManager::instance().getApplicationCount() != 1)
        return false;

    return !getFileList().empty();
}

/**
 * Gets the list of auto-save files in the auto-save directory.
 *  Filters out files that do not have "_autosave_" in their name.
 */
QStringList AutoSaveDialog::getFileList()
{
    QStringList list;

    FASTLOGS(FLog::AutoSave, "Getting list of AutoSave files in dir %s", AuthoringSettings::singleton().autoSaveDir.absolutePath().toStdString());

    const QDir dir(AuthoringSettings::singleton().autoSaveDir);
    const QStringList entries = dir.entryList(QStringList() << "*.rbxl");
    if (entries.empty())
        return list;

    for (int i = 0; i < entries.size(); i++)
    {
        FASTLOGS(FLog::AutoSave, "Found file %s", entries[i].toStdString());
        if (entries[i].contains("_AUTOSAVE_", Qt::CaseInsensitive))
        {
            FASTLOG(FLog::AutoSave, "Pattern matches, add to list");
            list.append(entries[i]);
        }
    }

    return list;
}

/**
 * Callback for closing the dialog.
 *  Calls onIgnore().
 */
void AutoSaveDialog::closeEvent(QCloseEvent* event)
{
    QDialog::closeEvent(event);
    onIgnore();
}

/**
 * Callback for user clicking the open button.
 *  Attempts to open the file if there is just one file or pops a file open
 *  dialog if there are more.  Failing to open the file will delete the file.
 */
void AutoSaveDialog::onOpen()
{
    const QDir dir(AuthoringSettings::singleton().autoSaveDir);
    const QStringList entries = getFileList();

    QString fileName;

    if (entries.size() == 1)
    {
        // if only one file, try to load it
        fileName = dir.filePath(entries[0]);
    }
    else
    {
        fileName = QFileDialog::getOpenFileName(this, tr("Open Roblox File"), dir.absolutePath(), tr("Roblox Places Files (*.rbxl)"));
    }

    FASTLOGS(FLog::AutoSave, "AutoSave open: file %s", fileName.toStdString());

    if (!fileName.isEmpty())
    {
        accept();

        bool openResult = UpdateUIManager::Instance().getMainWindow().handleFileOpen(fileName, IRobloxDoc::IDE);
        if (openResult)
        {
            FASTLOG(FLog::AutoSave, "File opened successfully");
            RobloxDocManager::Instance().getPlayDoc()->setAutoSaveLoad();
        }
        else
        {
            FASTLOG(FLog::AutoSave, "Bad file, remove");
            // kill the bad file, it's useless to us
            QFile::remove(fileName);
            if (!getFileList().empty())
                reject();
        }
    }
}

/**
 * Callback for user clicking the ignore button.
 *  Exits the dialog.
 */
void AutoSaveDialog::onIgnore()
{
    FASTLOG(FLog::AutoSave, "AutoSave ignore");
    accept();
}

/**
 * Callback for user clicking the delete button.
 *  Deletes all files after confirmation.
 */
void AutoSaveDialog::onDelete()
{
    FASTLOG(FLog::AutoSave, "AutoSave delete");
    const QDir dir(AuthoringSettings::singleton().autoSaveDir);
    const QStringList entries = getFileList();
    const QString confirmMessage = QString(tr("Do you want to delete all auto-save files?  There are %1 file(s).")).arg(entries.size());
    const int confirmationResult = QMessageBox::question(this, tr("Confirm Auto-Save Delete"), confirmMessage, QMessageBox::Yes | QMessageBox::No);

    if (confirmationResult == QMessageBox::Yes)
    {
        for (int i = 0; i < entries.size(); ++i)
        {
            FASTLOGS(FLog::AutoSave, "Removing %s", entries[i].toStdString());
            QFile::remove(dir.filePath(entries[i]));
        }
        accept();
    }
}