1 files changed, 107 insertions, 0 deletions
diff --git a/src/frontend/qtwidgets/rulesDialog.cpp b/src/frontend/qtwidgets/rulesDialog.cpp
new file mode 100644
index 0000000..cd62342
--- /dev/null
+++ b/src/frontend/qtwidgets/rulesDialog.cpp
@@ -0,0 +1,107 @@
+#include <QPushButton>
+#include <QScrollArea>
+#include <QVBoxLayout>
+
+#include <QDebug>
+
+#include "../../backend/db_sqlite.h"
+#include "lib.h"
+#include "ruleLayout.h"
+#include "rulesDialog.h"
+
+RulesDialog::RulesDialog(const Entry &entry) {
+	BackendDB database;
+
+	this->entry_id = entry.id;
+
+	// load uic
+	ui.setupUi(this);
+
+	// load rules into object member
+	this->rules = database.loadRules(entry_id);
+
+	// display widgets
+	this->drawRules();
+
+	// set connections
+	QObject::connect(ui.add_rule_button, &QPushButton::released, this, &RulesDialog::addRule);
+}
+
+// for maintaining values when adding/removing rules
+void RulesDialog::updateRulesList() {
+	int i;
+
+	for(i = 0; i < ui.rules_layout->children().size(); ++i) {
+		this->rules[i].when = Rule::When(qobject_cast<RuleLayout *>(ui.rules_layout->children()[i])->when_widget->currentIndex());
+		this->rules[i].date = qobject_cast<RuleLayout *>(ui.rules_layout->children()[i])->date_widget->dateTime();
+		this->rules[i].color = qobject_cast<RuleLayout *>(ui.rules_layout->children()[i])->color_widget->text();
+		this->rules[i].highlight = qobject_cast<RuleLayout *>(ui.rules_layout->children()[i])->highlight_widget->text();
+	}
+}
+
+void RulesDialog::drawRules() {
+	RuleLayout *new_layout;
+	int i;
+
+
+	// remove all children from layout
+	recursiveClear(ui.rules_layout);
+
+	// Draw each rule
+	for(i = 0; i < this->rules.size(); ++i) {
+		new_layout = new RuleLayout(rules[i]);
+
+		// connect delete button
+		QObject::connect(new_layout->del_button, &QPushButton::released, this, [=](){ this->deleteRule(i); });
+
+		ui.rules_layout->addLayout(new_layout);
+	}
+}
+
+void RulesDialog::addRule() {
+	Rule new_rule = Rule(0, this->entry_id, Rule::before);
+	RuleLayout *new_layout = new RuleLayout(new_rule);
+
+	// add new rule to the member variable
+	this->rules.append(new_rule);
+
+	// redraw rules
+	this->updateRulesList();
+	this->drawRules();
+}
+
+void RulesDialog::deleteRule(int i) {
+	if(i < 0) return;
+	if(i >= this->rules.size()) return;
+
+	this->updateRulesList();
+	// mark as needing to be deleted from database if id > 0
+	if(this->rules[i].id > 0)
+		this->deleted_rules.append(this->rules[i]);
+	this->rules.removeAt(i);
+	this->drawRules();
+}
+
+void RulesDialog::accept() {
+	int i;
+	BackendDB database;
+
+	this->updateRulesList();
+
+	// update rules in database
+	for(i = 0; i < this->rules.size(); ++i) {
+		// insert - this is a new rule
+		if(this->rules[i].id == 0)
+			database.insertRule(this->rules[i]);
+		// update - this is an existing rule
+		else
+			database.updateRule(this->rules[i]);
+	}
+
+	// delete deleted rules in database
+	for(i = 0; i < this->deleted_rules.size(); ++i) {
+		database.removeRule(this->deleted_rules[i]);
+	}
+
+	QDialog::accept();
+}