AntecedentMorphConfigurator/codegeneratordialog.cpp at main · proostas/AntecedentMorphConfigurator · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
#include "codegeneratordialog.hpp"

#include <QFormLayout>
#include <QComboBox>
#include <QHBoxLayout>
#include <QLineEdit>
#include <QToolButton>
#include <QPlainTextEdit>
#include <QDialogButtonBox>
#include <QPushButton>
#include <QFileDialog>
#include <QStandardPaths>
#include "zmkcodegenerator.hpp"
#include "qmkcodegenerator.hpp"
#include <QFile>
#include <QSettings>

CodeGeneratorDialog::CodeGeneratorDialog(Schema *schema, CodeGenerator::Firmware firmware, QWidget *parent)
    : QDialog{parent},
      m_schema{schema},
      m_firmware{firmware},
      m_layout{new QFormLayout{this}},
      m_outputLayout{new QHBoxLayout},
      m_outputPath{new QLineEdit{this}},
      m_outputPathSelector{new QToolButton{this}},
      m_log{new QPlainTextEdit{this}},
      m_buttonBox{new QDialogButtonBox{this}}
{
    setWindowTitle(QString{"%1 Code Generator"}.arg(m_firmware == CodeGenerator::ZMKFirmware ? "ZMK" : "QMK"));
    resize(600, 500);

    if (m_firmware == CodeGenerator::ZMKFirmware)
        m_generator = std::make_unique<ZmkCodeGenerator>(m_schema);
    else
        m_generator = std::make_unique<QmkCodeGenerator>(m_schema);

    m_outputLayout->addWidget(m_outputPath);
    m_outputPathSelector->setText("...");
    connect(m_outputPathSelector, &QToolButton::clicked, this, [this](){
        QSettings settings;
        auto path = settings.value("paths/zmkConfig").toString();
        QString filePath = QFileDialog::getSaveFileName(
            this, "Destination File Path",
            !path.isEmpty() ? path : QStandardPaths::standardLocations(QStandardPaths::HomeLocation).value(0),
            "All Files (*)");
        if (filePath.isEmpty())
            return;

        m_outputPath->setText(filePath);
    });
    m_outputLayout->addWidget(m_outputPathSelector);
    m_layout->addRow("Output: ", m_outputLayout);

    m_log->setReadOnly(true);
    m_layout->setWidget(1, QFormLayout::SpanningRole, m_log);

    m_buttonBox->setStandardButtons(QDialogButtonBox::StandardButton::Save|QDialogButtonBox::StandardButton::Close);
    m_buttonBox->button(QDialogButtonBox::StandardButton::Save)->setText("Generate");
    connect(m_buttonBox, &QDialogButtonBox::accepted, this, &CodeGeneratorDialog::generate);
    connect(m_buttonBox, &QDialogButtonBox::rejected, this, &CodeGeneratorDialog::reject);
    m_layout->setWidget(2, QFormLayout::SpanningRole, m_buttonBox);
}

void CodeGeneratorDialog::generate()
{
    m_log->clear();
    if (m_outputPath->text().isEmpty()) {
        m_log->appendPlainText("Select the destination to write to");
        return;
    }

    auto verifyResult = m_generator->verify();
    if (!verifyResult.first) {
        m_log->appendPlainText(QString{"Verify failed at %1"}.arg(verifyResult.second));
        return;
    } else {
        m_log->appendPlainText("Verify OK");
    }

    auto prepareResult = m_generator->prepare();
    if (!prepareResult.first) {
        m_log->appendPlainText(QString{"Prepare failed at %1"}.arg(verifyResult.second));
        return;
    } else {
        m_log->appendPlainText("Prepare OK");
    }

    QFile file{m_outputPath->text()};
    if (!file.open(QFile::WriteOnly|QFile::Truncate)) {
        m_log->appendPlainText(QString{"Failed to open file: %1"}.arg(file.errorString()));
        return;
    }

    QTextStream out{&file};
    m_generator->generate(out);
}