#ifndef BACKEND_DB_SQLITE_H
#define BACKEND_DB_SQLITE_H

#include <QString>
#include <QStringList>
#include <qsqldatabase.h>

#include "../entry.h"
#include "../group.h"
#include "../rule.h"

class BackendDB : QSqlDatabase {
	public:
		BackendDB();
		QList<Group *> loadGroups();
		QList<Entry *> loadEntries();
		QList<Entry *> loadEntries(int parent_id);
		QList<Rule *> loadRules();
		int insertGroup(const Group &new_group);
		int insertEntry(const Entry &new_entry);
		int insertRule(int new_rule); // param datatype TBD
		void updateGroup(const Group &group);
		void updateEntry(const Entry &entry);
		void updateRule(int rule); // param datatype TBD
		int removeGroup(const Group &group);
		int removeEntry(const Entry &entry);
		void removeRule(int rule); // param datatype TBD
		void cleanHidden();

	private:
		const QStringList create_table_queries = {
			"CREATE TABLE groups ("
				"id INTEGER PRIMARY KEY AUTOINCREMENT UNIQUE NOT NULL,"
				"name VARCHAR(255) NOT NULL,"
				"column TINYINT(1) DEFAULT FALSE,"
				"link VARCHAR(255) NOT NULL,"
				"hidden TINYINT(1) DEFAULT FALSE"
				")",
			"CREATE TABLE entries ("
				"id INTEGER PRIMARY KEY AUTOINCREMENT UNIQUE NOT NULL,"
				"parent_id REFERENCES groups (id),"
				"description VARCHAR(255) NOT NULL,"
				"due_date TEXT DEFAULT NULL,"
				"alt_due_date VARCHAR(255) DEFAULT NULL,"
				"link VARCHAR(255) DEFAULT NULL,"
				"color VARCHAR(255) DEFAULT NULL,"
				"highlight VARCHAR(255) DEFAULT NULL,"
				"done TINYINT(1) DEFAULT FALSE,"
				"hidden TINYINT(1) DEFAULT FALSE"
				")",
			"CREATE TABLE rules ("
				"id INTEGER PRIMARY KEY AUTOINCREMENT UNIQUE NOT NULL,"
				"entry_id REFERENCES entries (id),"
				"before_after TINYINT(1) DEFAULT TRUE,"
				"date TEXT NOT NULL,"
				"color VARCHAR(255) DEFAULT NULL,"
				"highlight VARCHAR(255) DEFAULT NULL"
				")"
		};

		QString getDBPath();
		QSqlDatabase openDB();
};

#endif