ESGRAF48/source/mainwindow.cpp

#include "mainwindow.h"
#include "ui_mainwindow.h"

#include "version.h"

#include <QJsonObject>
#include <QJsonDocument>
#include <QFile>
#include <QFileDialog>
#include <QDataWidgetMapper>
#include <QCloseEvent>
#include <QMessageBox>
#include <QtPrintSupport/QPrinter>
#include <QtPrintSupport/QPrintDialog>
#include <QTextDocument>
#include <QPainter>

#include <QDebug>

#include <fstream>

MainWindow::MainWindow(QWidget* parent)
    : QMainWindow(parent)
    , ui(new Ui::MainWindow)
    , m_dataModel(this)
{
    ui->setupUi(this);
    setupUi();

    newFile();
}

MainWindow::MainWindow(QWidget* parent, const QString& filename)
    : QMainWindow(parent)
    , ui(new Ui::MainWindow)
    , m_dataModel(this)
{
    ui->setupUi(this);
    setupUi();

    openFile(filename);
}

void MainWindow::setupUi()
{
    ui->metaDataWidget->setModel(&m_dataModel.m_metaData);
    ui->verbEndWidget->setModel(&m_dataModel.m_verbEnd);
    ui->genusWidget->setModel(&m_dataModel.m_genus);
    ui->pluralWidget->setModel(&m_dataModel.m_plural);
    ui->akkusativDativWidget->setAkkusativModel(&m_dataModel.m_akkusativ);
    ui->akkusativDativWidget->setDativModel(&m_dataModel.m_dativ);

    ui->v2SvkWidget->setWFModel(&m_dataModel.m_wfModel);
    ui->v2SvkWidget->setOTModel(&m_dataModel.m_otModel);
    ui->v2SvkWidget->setTPrModel(&m_dataModel.m_tPrModel);
    ui->v2SvkWidget->setTPeModel(&m_dataModel.m_tPeModel);

    ui->lateSkillsWidget->setPassivModel(&m_dataModel.m_passiv);
    ui->lateSkillsWidget->setGenitivModel(&m_dataModel.m_genitiv);
    ui->resultWidget->setModel(&m_dataModel.m_results);

    connect(ui->actionNew, &QAction::triggered, this, &MainWindow::newFile);
    connect(ui->actionOpen, &QAction::triggered, this, qOverload<>(&MainWindow::openFile));
    connect(ui->actionSave, &QAction::triggered, this, qOverload<>(&MainWindow::saveFile));
    connect(ui->actionSave_as, &QAction::triggered, this, &MainWindow::saveFileAs);
    connect(ui->actionPrint, &QAction::triggered, this, &MainWindow::print);
    connect(ui->actionExport_PDF, &QAction::triggered, this, qOverload<>(&MainWindow::savePdf));

    connect(ui->actionAbout, &QAction::triggered, this, &MainWindow::aboutDialog);

    connect(&m_dataModel, &DataModel::modelChanged, this, &MainWindow::dataModelChanged);
}

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

void MainWindow::newFile()
{
    closeFile();

    setWindowModified(false);
    setWindowTitle("untitled[*]");
    m_filename = "";
    m_saveOnClose = false;
}

void MainWindow::openFile()
{
    QString filename =
        QFileDialog::getOpenFileName(this, "Open file", "", "ESGRAF 4-8 (*.esgraf48)");
    if (filename.isEmpty())
    {
        return;
    }

    try
    {
        openFile(filename);
    }
    catch (std::exception& e)
    {
        QString errorMessage = QString("Error loading \"") + filename + "\": " + e.what();
        QMessageBox::critical(this, "Error", errorMessage);
    }
}

void MainWindow::openFile(const QString& filename)
{
    closeFile();

    m_dataModel.read(filename);

    setWindowModified(false);
    setWindowTitle(filename + "[*]");
    m_saveOnClose = false;
    m_filename = filename;

    ui->metaDataWidget->toFirst();
}

void MainWindow::saveFile()
{
    if (m_filename.isEmpty())
    {
        saveFileAs();
    }
    else
    {
        saveFile(m_filename);
    }
}

void MainWindow::saveFileAs()
{
    QFileDialog saveFilenameDialog(this);
    saveFilenameDialog.setAcceptMode(QFileDialog::AcceptSave);
    saveFilenameDialog.setDefaultSuffix("esgraf48");
    saveFilenameDialog.setFileMode(QFileDialog::AnyFile);
    saveFilenameDialog.setNameFilter("ESGRAF 4-8 (*.esgraf48)");
    saveFilenameDialog.setWindowTitle("Save file");

    if (!saveFilenameDialog.exec())
    {
        return;
    }

    saveFile(saveFilenameDialog.selectedFiles().first());
}

void MainWindow::closeFile()
{
    if (m_saveOnClose == true)
    {
        QMessageBox msgBox;
        msgBox.setText("The document has been modified.");
        msgBox.setInformativeText("Do you want to save your changes?");
        msgBox.setStandardButtons(QMessageBox::Save | QMessageBox::Discard);
        msgBox.setDefaultButton(QMessageBox::Save);

        if (msgBox.exec() == QMessageBox::Save)
        {
            saveFile();
        }
    }
}

void MainWindow::print() const
{
    QPrinter printer;
    printer.setPaperSize(QPrinter::A4);
    printer.setPageMargins(20, 20, 20, 20, QPrinter::Millimeter);

    QPrintDialog dialog(&printer);
    if (dialog.exec() != QDialog::Accepted)
    {
        return;
    }

    m_dataModel.printTo(printer);
}

void MainWindow::dataModelChanged()
{
    m_saveOnClose = true;
    setWindowModified(true);
}

void MainWindow::savePdf()
{
    QFileDialog saveFilenameDialog(this);
    saveFilenameDialog.setDefaultSuffix("pdf");
    saveFilenameDialog.setFileMode(QFileDialog::AnyFile);
    saveFilenameDialog.setNameFilter("PDF File (*.pdf)");
    saveFilenameDialog.setWindowTitle("Save file");

    if (!saveFilenameDialog.exec())
    {
        return;
    }

    savePdf(saveFilenameDialog.selectedFiles().first());
}

void MainWindow::closeEvent(QCloseEvent* event)
{
    closeFile();
}

void MainWindow::saveFile(const QString& filename)
{
    try
    {
        m_dataModel.write(filename);
    }
    catch (std::exception& e)
    {
        QString errorMessage = QString("Error saving \"") + filename + "\": " + e.what();
        QMessageBox::critical(this, "Error", errorMessage);
        return;
    }

    qDebug() << "Wrote" << filename;

    setWindowTitle(filename + "[*]");
    setWindowModified(false);
    m_filename = filename;
    m_saveOnClose = false;
}

void MainWindow::savePdf(const QString& filename)
{
    QPrinter printer;
    printer.setOutputFormat(QPrinter::PdfFormat);
    printer.setPaperSize(QPrinter::A4);
    printer.setPageMargins(20, 20, 20, 20, QPrinter::Millimeter);
    printer.setOutputFileName(filename);

    m_dataModel.printTo(printer);
}

void MainWindow::aboutDialog()
{
    QString infoString =
        QString::fromUtf8(ESGRAF48_DESCRIPTION) + " Version " + QString::fromUtf8(ESGRAF48_VERSION);
    QMessageBox::information(this, ESGRAF48_DESCRIPTION, infoString);
}