lms/src/tools/metadata/LmsMetadata.cpp

/*
 * Copyright (C) 2019 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 <chrono>
#include <optional>
#include <stdexcept>
#include <stdlib.h>
#include <iostream>

#include <Wt/WDate.h>

#include "metadata/AvFormatParser.hpp"
#include "metadata/TagLibParser.hpp"
#include "utils/StreamLogger.hpp"

std::ostream& operator<<(std::ostream& os, const MetaData::Artist& artist)
{
	os << artist.name;

	if (artist.musicBrainzArtistID)
		os << " (" << artist.musicBrainzArtistID->getAsString() << ")";

	if (artist.sortName)
		os << " '" << *artist.sortName << "'";

	return os;
}

std::ostream& operator<<(std::ostream& os, const MetaData::Album& album)
{
	os << album.name;

	if (album.musicBrainzAlbumID)
		os << " (" << album.musicBrainzAlbumID->getAsString() << ")";

	return os;
}



void parse(MetaData::IParser& parser, const std::filesystem::path& file)
{
	using namespace MetaData;

	parser.setClusterTypeNames( {"MOOD", "GENRE"} );

	const auto start {std::chrono::steady_clock::now()};
	std::optional<Track> track {parser.parse(file, true)};
	if (!track)
	{
		std::cerr << "Parsing failed" << std::endl;
		return;
	}
	const auto end {std::chrono::steady_clock::now()};

	std::cout << "Parsing time: " << std::chrono::duration_cast<std::chrono::microseconds>(end - start).count() / 1000. << "ms" << std::endl;

	std::cout << "Track metadata:" << std::endl;

	for (const Artist& artist : track->artists)
		std::cout << "Artist: " << artist << std::endl;

	for (const Artist& artist: track->albumArtists)
		std::cout << "Album artist: " << artist << std::endl;

	if (track->album)
		std::cout << "Album: " << *track->album << std::endl;

	std::cout << "Title: " << track->title << std::endl;

	if (track->musicBrainzTrackID)
		std::cout << "MB TrackID = " << track->musicBrainzTrackID->getAsString() << std::endl;

	if (track->musicBrainzRecordID)
		std::cout << "MB RecordID = " << track->musicBrainzRecordID->getAsString() << std::endl;

	for (const auto& cluster : track->clusters)
	{
		std::cout << "Cluster: " << cluster.first << std::endl;
		for (const auto& name : cluster.second)
		{
			std::cout << "\t" << name << std::endl;
		}
	}

	std::cout << "Duration: " << track->duration.count() / 1000 << "s" << std::endl;

	if (track->trackNumber)
		std::cout << "Track: " << *track->trackNumber << std::endl;

	if (track->totalTrack)
		std::cout << "TotalTrack: " << *track->totalTrack << std::endl;

	if (track->discNumber)
		std::cout << "Disc: " << *track->discNumber << std::endl;

	if (track->totalDisc)
		std::cout << "TotalDisc: " << *track->totalDisc << std::endl;

	if (track->year)
		std::cout << "Year: " << *track->year << std::endl;

	if (track->originalYear)
		std::cout << "Original year: " << *track->originalYear << std::endl;

	std::cout << "HasCover = " << std::boolalpha << track->hasCover << std::endl;

	for (const auto& audioStream : track->audioStreams)
		std::cout << "Audio stream: " << audioStream.bitRate << " bps" << std::endl;

	if (track->trackReplayGain)
		std::cout << "Track replay gain: " << *track->trackReplayGain << std::endl;

	if (track->albumReplayGain)
		std::cout << "Album replay gain: " << *track->albumReplayGain << std::endl;

	if (track->acoustID)
		std::cout << "AcoustID: " << track->acoustID->getAsString() << std::endl;

	if (!track->copyright.empty())
		std::cout << "Copyright: " << track->copyright << std::endl;

	if (!track->copyrightURL.empty())
		std::cout << "CopyrightURL: " << track->copyrightURL << std::endl;

	std::cout << std::endl;
}

int main(int argc, char *argv[])
{
	if (argc == 1)
	{
		std::cerr << "Usage: <file> [<file> ...]" << std::endl;
		return EXIT_FAILURE;
	}

	try
	{
		// log to stdout
		ServiceProvider<Logger>::create<StreamLogger>(std::cout);

		for (std::size_t  i {}; i < static_cast<std::size_t>(argc - 1); ++i)
		{
			std::filesystem::path file {argv[i + 1]};

			std::cout << "Parsing file '" << file << "'" << std::endl;

			{
				std::cout << "Using av:" << std::endl;
				MetaData::AvFormatParser parser;
				parse(parser, file);
			}

			{
				std::cout << "Using TagLib:" << std::endl;
				MetaData::TagLibParser parser;
				parse(parser, file);
			}
		}

	}
	catch (std::exception& e)
	{
		std::cerr << "Caught exception: " << e.what() << std::endl;
		return EXIT_FAILURE;
	}

	return EXIT_SUCCESS;
}