Centralized scan settings into a dedicated ScanSettings table

src/database/ScanSettings.cpp

@@ -0,0 +1,111 @@
+/*
+ * Copyright (C) 2018 Emeric Poupon
+ *
+ * This file is part of LMS.
+ *
+ * LMS 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 3 of the License, or
+ * (at your option) any later version.
+ *
+ * LMS 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 LMS.  If not, see <http://www.gnu.org/licenses/>.
+ */
+
+#include "ScanSettings.hpp"
+
+#include <Wt/Dbo/WtSqlTraits.h>
+
+#include "utils/Logger.hpp"
+#include "utils/Utils.hpp"
+
+#include "Cluster.hpp"
+
+namespace {
+
+std::set<std::string> defaultClusterTypeNames =
+{
+	"GENRE",
+	"ALBUMGROUPING",
+	"MOOD",
+	"ALBUMMOOD",
+	"COMMENT:SONGS-DB_OCCASION",
+};
+
+}
+
+namespace Database {
+
+
+ScanSettings::pointer
+ScanSettings::get(Wt::Dbo::Session& session)
+{
+	ScanSettings::pointer settings = session.find<ScanSettings>();
+	if (!settings)
+	{
+		settings = session.add(std::make_unique<ScanSettings>());
+		settings.modify()->setClusterTypes(defaultClusterTypeNames);
+	}
+
+	return settings;
+}
+
+std::set<boost::filesystem::path>
+ScanSettings::getAudioFileExtensions() const
+{
+	auto extensions = splitString(_audioFileExtensions, " ");
+	return std::set<boost::filesystem::path>(extensions.begin(), extensions.end());
+}
+
+std::vector<ClusterType::pointer>
+ScanSettings::getClusterTypes() const
+{
+	return std::vector<ClusterType::pointer>(_clusterTypes.begin(), _clusterTypes.end());
+}
+
+void
+ScanSettings::setMediaDirectory(boost::filesystem::path p)
+{
+	_mediaDirectory = stringTrimEnd(p.string(), "/\\");
+}
+
+void
+ScanSettings::setClusterTypes(const std::set<std::string>& clusterTypeNames)
+{
+	assert(session());
+
+	// Backup the old list
+	std::vector<ClusterType::pointer> oldClusterTypes(_clusterTypes.begin(), _clusterTypes.end());
+
+	_clusterTypes.clear();
+	// Create any missing cluster type
+	for (auto clusterTypeName : clusterTypeNames)
+	{
+		auto clusterType = ClusterType::getByName(*session(), clusterTypeName);
+		if (!clusterType)
+		{
+			LMS_LOG(DB, INFO) << "Creating cluster type " << clusterTypeName;
+			clusterType = ClusterType::create(*session(), clusterTypeName);
+		}
+		_clusterTypes.insert(clusterType);
+	}
+
+	// Delete no longer existing cluster types
+	for (auto oldClusterType : oldClusterTypes)
+	{
+		if (std::none_of(clusterTypeNames.begin(), clusterTypeNames.end(),
+			[oldClusterType](const std::string& name) { return name == oldClusterType->getName(); }))
+		{
+			LMS_LOG(DB, INFO) << "Deleting cluster type " << oldClusterType->getName();
+			oldClusterType.remove();
+		}
+	}
+}
+
+} // namespace Database
+