WIP

2020-02-13 13:00:22 +01:00
parent e274c0ca89
commit 876fb12fe4
192 changed files with 1046 additions and 676 deletions
@@ -0,0 +1,146 @@
+/*
+ * 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 "database/ScanSettings.hpp"
+
+#include <Wt/Dbo/WtSqlTraits.h>
+
+#include "utils/Logger.hpp"
+#include "utils/String.hpp"
+
+#include "database/Cluster.hpp"
+#include "database/Session.hpp"
+
+namespace {
+
+const std::set<std::string> defaultClusterTypeNames =
+{
+	"GENRE",
+	"ALBUMGROUPING",
+	"MOOD",
+	"ALBUMMOOD",
+};
+
+}
+
+namespace Database {
+
+void
+ScanSettings::init(Session& session)
+{
+	session.checkUniqueLocked();
+
+	pointer settings {get(session)};
+	if (settings)
+		return;
+
+	settings = session.getDboSession().add(std::make_unique<ScanSettings>());
+	settings.modify()->setClusterTypes(session, defaultClusterTypeNames );
+}
+
+ScanSettings::pointer
+ScanSettings::get(Session& session)
+{
+	session.checkSharedLocked();
+
+	return session.getDboSession().find<ScanSettings>();
+}
+
+std::set<std::filesystem::path>
+ScanSettings::getAudioFileExtensions() const
+{
+	auto extensions = StringUtils::splitString(_audioFileExtensions, " ");
+	return std::set<std::filesystem::path>(std::cbegin(extensions), std::cend(extensions));
+}
+
+void
+ScanSettings::addAudioFileExtension(const std::filesystem::path& ext)
+{
+	_audioFileExtensions += " " + ext.string();
+}
+
+std::vector<ClusterType::pointer>
+ScanSettings::getClusterTypes() const
+{
+	return std::vector<ClusterType::pointer>(std::cbegin(_clusterTypes), std::cend(_clusterTypes));
+}
+
+void
+ScanSettings::setMediaDirectory(const std::filesystem::path& p)
+{
+	_mediaDirectory = StringUtils::stringTrimEnd(p.string(), "/\\");
+}
+
+template <typename It>
+std::set<std::string> getNames(It begin, It end)
+{
+	std::set<std::string> names;
+	std::transform(begin, end, std::inserter(names, std::cbegin(names)),
+		[](const ClusterType::pointer& clusterType)
+		{
+			return clusterType->getName();
+		});
+
+	return names;
+}
+
+void
+ScanSettings::setClusterTypes(Session& session, const std::set<std::string>& clusterTypeNames)
+{
+	session.checkUniqueLocked();
+
+	bool needRescan {};
+
+	// Create any missing cluster type
+	for (const std::string& 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);
+
+			needRescan = true;
+		}
+	}
+
+	// Delete no longer existing cluster types
+	for (ClusterType::pointer& clusterType : _clusterTypes)
+	{
+		if (std::none_of(clusterTypeNames.begin(), clusterTypeNames.end(),
+			[clusterType](const std::string& name) { return name == clusterType->getName(); }))
+		{
+			LMS_LOG(DB, INFO) << "Deleting cluster type " << clusterType->getName();
+			clusterType.remove();
+		}
+	}
+
+	if (needRescan)
+		_scanVersion += 1;
+}
+
+void
+ScanSettings::incScanVersion()
+{
+	_scanVersion += 1;
+}
+
+} // namespace Database
+