[DB] Some code refactoring

src/database/DatabaseUpdater.hpp

@@ -0,0 +1,129 @@
+/*
+ * Copyright (C) 2013 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/>.
+ */
+
+#ifndef DB_UPDATER_HPP
+#define DB_UPDATER_HPP
+
+#include <Wt/WIOService>
+#include <Wt/WSignal>
+
+#include <mutex>
+
+#include <boost/asio/deadline_timer.hpp>
+
+#include "metadata/AvFormat.hpp"
+
+#include "database/DatabaseHandler.hpp"
+
+namespace Database {
+
+class Updater
+{
+	public:
+		struct Stats
+		{
+			std::size_t	nbSkipped = 0;		// no change since last scan
+			std::size_t	nbScanned = 0;
+			std::size_t	nbScanErrors = 0;	// cannot scan file
+			std::size_t	nbNotImported = 0;	// Not imported (criteria not filled)
+			std::size_t	nbAdded = 0;
+			std::size_t	nbRemoved = 0;
+			std::size_t	nbModified = 0;
+
+			std::size_t nbChanges() const { return nbAdded + nbRemoved + nbModified;}
+		};
+
+		static Updater& instance();
+
+		void setConnectionPool(Wt::Dbo::SqlConnectionPool& connectionPool);
+
+		void setAudioExtensions(const std::vector<std::string>&	extensions);
+		void setVideoExtensions(const std::vector<std::string>&	extensions);
+
+		void start();
+		void stop();
+		void restart();
+
+		Wt::Signal<Stats>& changed() { return _sigChanged; }
+
+		std::mutex&	getMutex(void) { return _mutex; }
+
+	private:
+
+		Updater();
+
+		struct RootDirectory
+		{
+			Database::MediaDirectory::Type type;
+			boost::filesystem::path path;
+
+			RootDirectory(Database::MediaDirectory::Type t, boost::filesystem::path p) : type(t), path(p) {}
+		};
+
+		// Job handling
+		void processNextJob();
+		void scheduleScan(boost::posix_time::time_duration duration);
+		void scheduleScan(boost::posix_time::ptime time);
+
+		// Update database (scheduled callback)
+		void process(boost::system::error_code ec);
+
+		// Check if a file exists and is still in a root directory
+		static bool checkFile(const boost::filesystem::path& p,
+				const std::vector<boost::filesystem::path>& rootDirectories,
+				const std::vector<boost::filesystem::path>& extensions);
+
+
+		void processRootDirectory(  RootDirectory rootDirectory, Stats& stats);
+
+		// Helpers
+		Database::Artist::pointer getArtist( const boost::filesystem::path& file, const std::string& name, const std::string& MBID);
+		Database::Release::pointer getRelease( const boost::filesystem::path& file, const std::string& name, const std::string& MBID);
+		std::vector<Database::Genre::pointer> getGenres( const std::list<std::string>& names);
+		void updateFileExtensions();
+
+		// Audio
+		void checkAudioFiles( Stats& stats );
+		void checkDuplicatedAudioFiles( Stats& stats );
+		void processAudioFile( const boost::filesystem::path& file, Stats& stats);
+
+		// Video
+		void checkVideoFiles( Stats& stats );
+		void processVideoFile( const boost::filesystem::path& file, Stats& stats);
+
+		bool			_running;
+		Wt::WIOService		_ioService;
+		Wt::Signal<Stats>	_sigChanged;
+		std::mutex		_mutex;
+
+		boost::asio::deadline_timer _scheduleTimer;
+
+		Database::Handler*	_db = nullptr;
+
+		std::vector<boost::filesystem::path>	_audioFileExtensions;
+		std::vector<boost::filesystem::path>	_videoFileExtensions;
+
+		MetaData::AvFormat	_metadataParser;
+
+
+}; // class Updater
+
+} // Database
+
+#endif