Use sqlite for cinema and DKDM recipient lists.

1 files changed, 243 insertions, 0 deletions

diff --git a/src/lib/dkdm_recipient_list.cc b/src/lib/dkdm_recipient_list.cc
new file mode 100644
index 000000000..34179337e
--- /dev/null
+++ b/src/lib/dkdm_recipient_list.cc
@@ -0,0 +1,243 @@
+/*
+    Copyright (C) 2024 Carl Hetherington <cth@carlh.net>
+
+    This file is part of DCP-o-matic.
+
+    DCP-o-matic is free software; you can redistribute it and/or modify
+    it under the terms of the GNU General Public License as published by
+    the Free Software Foundation; either version 2 of the License, or
+    (at your option) any later version.
+
+    DCP-o-matic is distributed in the hope that it will be useful,
+    but WITHOUT ANY WARRANTY; without even the implied warranty of
+    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+    GNU General Public License for more details.
+
+    You should have received a copy of the GNU General Public License
+    along with DCP-o-matic.  If not, see <http://www.gnu.org/licenses/>.
+
+*/
+
+
+#include "config.h"
+#include "dkdm_recipient.h"
+#include "dkdm_recipient_list.h"
+#include "sqlite_statement.h"
+#include "sqlite_transaction.h"
+#include "util.h"
+#include <boost/algorithm/string.hpp>
+
+
+using std::make_pair;
+using std::pair;
+using std::string;
+using std::vector;
+using boost::optional;
+
+
+DKDMRecipientList::DKDMRecipientList()
+	: _dkdm_recipients("dkdm_recipients")
+{
+	setup(Config::instance()->dkdm_recipients_file());
+}
+
+
+DKDMRecipientList::DKDMRecipientList(boost::filesystem::path db_file)
+	: _dkdm_recipients("dkdm_recipients")
+{
+	setup(db_file);
+}
+
+
+
+DKDMRecipientList::~DKDMRecipientList()
+{
+	if (_db) {
+		sqlite3_close(_db);
+	}
+}
+
+
+void
+DKDMRecipientList::read_legacy_file(boost::filesystem::path xml_file)
+{
+	cxml::Document doc("DKDMRecipients");
+	doc.read_file(xml_file);
+
+	read_legacy_document(doc);
+}
+
+
+void
+DKDMRecipientList::read_legacy_string(string const& xml)
+{
+	cxml::Document doc("DKDMRecipients");
+	doc.read_file(xml);
+
+	read_legacy_document(doc);
+}
+
+
+void
+DKDMRecipientList::read_legacy_document(cxml::Document const& doc)
+{
+	for (auto recipient_node: doc.node_children("DKDMRecipient")) {
+		vector<string> emails;
+		for (auto email_node: recipient_node->node_children("Email")) {
+			emails.push_back(email_node->content());
+		}
+
+		optional<dcp::Certificate> certificate;
+		if (auto certificate_string = recipient_node->optional_string_child("Recipient")) {
+			certificate = dcp::Certificate(*certificate_string);
+		}
+
+		DKDMRecipient recipient(
+			recipient_node->string_child("Name"),
+			recipient_node->string_child("Notes"),
+			certificate,
+			emails
+			);
+
+		add_dkdm_recipient(recipient);
+	}
+}
+
+
+void
+DKDMRecipientList::setup(boost::filesystem::path db_file)
+{
+	_dkdm_recipients.add_column("name", "TEXT");
+	_dkdm_recipients.add_column("notes", "TEXT");
+	_dkdm_recipients.add_column("recipient", "TEXT");
+	_dkdm_recipients.add_column("emails", "TEXT");
+
+#ifdef DCPOMATIC_WINDOWS
+	auto rc = sqlite3_open16(db_file.c_str(), &_db);
+#else
+	auto rc = sqlite3_open(db_file.c_str(), &_db);
+#endif
+	if (rc != SQLITE_OK) {
+		throw FileError("Could not open SQLite database", db_file);
+	}
+
+	sqlite3_busy_timeout(_db, 500);
+
+	SQLiteStatement screens(_db, _dkdm_recipients.create());
+	screens.execute();
+}
+
+
+DKDMRecipientList::DKDMRecipientList(DKDMRecipientList&& other)
+	: _dkdm_recipients(std::move(other._dkdm_recipients))
+{
+	_db = other._db;
+	other._db = nullptr;
+}
+
+
+DKDMRecipientList&
+DKDMRecipientList::operator=(DKDMRecipientList&& other)
+{
+	if (this != &other) {
+		_db = other._db;
+		other._db = nullptr;
+	}
+	return *this;
+}
+
+
+DKDMRecipientID
+DKDMRecipientList::add_dkdm_recipient(DKDMRecipient const& dkdm_recipient)
+{
+	SQLiteStatement add_dkdm_recipient(_db, _dkdm_recipients.insert());
+
+	add_dkdm_recipient.bind_text(1, dkdm_recipient.name);
+	add_dkdm_recipient.bind_text(2, dkdm_recipient.notes);
+	add_dkdm_recipient.bind_text(3, dkdm_recipient.recipient ? dkdm_recipient.recipient->certificate(true) : "");
+	add_dkdm_recipient.bind_text(4, join_strings(dkdm_recipient.emails));
+
+	add_dkdm_recipient.execute();
+
+	return sqlite3_last_insert_rowid(_db);
+}
+
+
+void
+DKDMRecipientList::update_dkdm_recipient(DKDMRecipientID id, DKDMRecipient const& dkdm_recipient)
+{
+	SQLiteStatement add_dkdm_recipient(_db, _dkdm_recipients.update("WHERE id=?"));
+
+	add_dkdm_recipient.bind_text(1, dkdm_recipient.name);
+	add_dkdm_recipient.bind_text(2, dkdm_recipient.notes);
+	add_dkdm_recipient.bind_text(3, dkdm_recipient.recipient ? dkdm_recipient.recipient->certificate(true) : "");
+	add_dkdm_recipient.bind_text(4, join_strings(dkdm_recipient.emails));
+	add_dkdm_recipient.bind_int64(5, id.get());
+
+	add_dkdm_recipient.execute();
+}
+
+
+void
+DKDMRecipientList::remove_dkdm_recipient(DKDMRecipientID id)
+{
+	SQLiteStatement statement(_db, "DELETE FROM dkdm_recipients WHERE ID=?");
+	statement.bind_int64(1, id.get());
+	statement.execute();
+}
+
+
+static
+vector<pair<DKDMRecipientID, DKDMRecipient>>
+dkdm_recipients_from_result(SQLiteStatement& statement)
+{
+	vector<pair<DKDMRecipientID, DKDMRecipient>> output;
+
+	statement.execute([&output](SQLiteStatement& statement) {
+		DCPOMATIC_ASSERT(statement.data_count() == 5);
+		DKDMRecipientID const id = statement.column_int64(0);
+		auto const name = statement.column_text(1);
+		auto const notes = statement.column_text(2);
+		auto certificate_string = statement.column_text(3);
+		optional<dcp::Certificate> certificate = certificate_string.empty() ? optional<dcp::Certificate>() : dcp::Certificate(certificate_string);
+		auto const join_with_spaces = statement.column_text(4);
+		vector<string> emails;
+		boost::algorithm::split(emails, join_with_spaces, boost::is_any_of(" "));
+		output.push_back(make_pair(id, DKDMRecipient(name, notes, certificate, { emails })));
+	});
+
+	return output;
+}
+
+
+
+
+vector<std::pair<DKDMRecipientID, DKDMRecipient>>
+DKDMRecipientList::dkdm_recipients() const
+{
+	SQLiteStatement statement(_db, _dkdm_recipients.select("ORDER BY name ASC"));
+	return dkdm_recipients_from_result(statement);
+}
+
+
+boost::optional<DKDMRecipient>
+DKDMRecipientList::dkdm_recipient(DKDMRecipientID id) const
+{
+	SQLiteStatement statement(_db, _dkdm_recipients.select("WHERE id=?"));
+	statement.bind_int64(1, id.get());
+	auto result = dkdm_recipients_from_result(statement);
+	if (result.empty()) {
+		return {};
+	}
+	return result[0].second;
+}
+
+
+void
+DKDMRecipientList::clear()
+{
+	SQLiteStatement sql(_db, "DELETE FROM dkdm_recipients");
+	sql.execute();
+}
+
+