lms/src/liblms/impl/metadata/AvFormat.cpp

/*
 * 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/>.
 */

#include "metadata/AvFormat.hpp"

#include <algorithm>
#include <iostream>

#include "av/AvInfo.hpp"

#include "utils/Logger.hpp"
#include "utils/String.hpp"


namespace MetaData
{

using MetadataMap = std::map<std::string, std::string>;

template <typename T>
std::optional<T>
findFirstValueOfAs(const MetadataMap& metadataMap, std::initializer_list<std::string> tags)
{
	auto it = std::find_first_of(std::cbegin(metadataMap), std::cend(metadataMap), std::cbegin(tags), std::cend(tags), [](const auto& it, const auto& str) { return it.first == str; });
	if (it == std::cend(metadataMap))
		return std::nullopt;

	return StringUtils::readAs<T>(StringUtils::stringTrim(it->second));
}

template <>
std::optional<std::vector<UUID>>
findFirstValueOfAs(const MetadataMap& metadataMap, std::initializer_list<std::string> tags)
{
	std::optional<std::string> str {findFirstValueOfAs<std::string>(metadataMap, tags)};
	if (!str)
		return std::nullopt;

	std::vector<std::string> strUuids = StringUtils::splitString(*str, "/");
	std::vector<UUID> res;

	for (const std::string strUuid : strUuids)
	{
		std::optional<UUID> uuid {UUID::fromString(strUuid)};
		if (!uuid)
			return std::nullopt;

		res.push_back(std::move(*uuid));
	}

	return res;
}


static
std::optional<Album>
getAlbum(const MetadataMap& metadataMap)
{
	std::optional<Album> res;

	auto album {findFirstValueOfAs<std::string>(metadataMap, {"ALBUM"})};
	if (!album)
		return res;

	auto albumMBID {findFirstValueOfAs<UUID>(metadataMap, {"MUSICBRAINZ ALBUM ID", "MUSICBRAINZ_ALBUMID", "MUSICBRAINZ/ALBUM ID"})};

	return Album{*album, albumMBID};
}

static
std::vector<Artist>
getAlbumArtists(const MetadataMap& metadataMap)
{
	std::vector<Artist> res;

	auto name {findFirstValueOfAs<std::string>(metadataMap, {"ALBUM_ARTIST"})};
	if (!name)
		return res;

	auto mbid {findFirstValueOfAs<UUID>(metadataMap, {"MUSICBRAINZ ALBUM ARTIST ID", "MUSICBRAINZ/ALBUM ARTIST ID"})};

	return {Artist {*name, mbid} };
}

static
std::vector<Artist>
getArtists(const MetadataMap& metadataMap)
{
	std::vector<Artist> artists;

	std::vector<std::string> artistNames;
	if (metadataMap.find("ARTISTS") != metadataMap.end())
	{
		artistNames = StringUtils::splitString(metadataMap.find("ARTISTS")->second, "/;");
	}
	else if (metadataMap.find("ARTIST") != metadataMap.end())
	{
		artistNames = {metadataMap.find("ARTIST")->second};
	}

	auto artistMBIDs {findFirstValueOfAs<std::vector<UUID>>(metadataMap, {"MUSICBRAINZ ARTIST ID", "MUSICBRAINZ_ARTISTID", "MUSICBRAINZ/ARTIST ID"})};

	for (std::size_t i {}; i < artistNames.size(); ++i)
	{
		if (artistMBIDs && artistNames.size() == artistMBIDs->size())
			artists.emplace_back(Artist {artistNames[i], (*artistMBIDs)[i]});
		else
			artists.emplace_back(Artist {artistNames[i], {}});
	}

	return artists;
}

std::optional<Track>
AvFormat::parse(const std::filesystem::path& p, bool debug)
{
	Track track;

	try
	{
		Av::MediaFile mediaFile {p};

		// Stream info
		{
			std::vector<AudioStream> audioStreams;

			for (auto stream : mediaFile.getStreamInfo())
			{
				MetaData::AudioStream audioStream {static_cast<unsigned>(stream.bitrate)};
				track.audioStreams.emplace_back(audioStream);
			}
		}

		track.duration = mediaFile.getDuration();
		track.hasCover = mediaFile.hasAttachedPictures();

		MetaData::Clusters clusters;

		const std::map<std::string, std::string> metadataMap {mediaFile.getMetaData()};

		for (const auto& metadata : metadataMap)
		{
			const std::string& tag {metadata.first};
			const std::string& value {metadata.second};

			if (debug)
				std::cout << "TAG = " << tag << ", VAL = " << value << std::endl;

			if (tag == "TITLE")
				track.title = value;
			else if (tag == "TRACK")
			{
				// Expecting 'Number/Total'
				std::vector<std::string> strings {StringUtils::splitString(value, "/") };

				if (strings.size() > 0)
				{
					track.trackNumber = StringUtils::readAs<std::size_t>(strings[0]);

					if (strings.size() > 1)
						track.totalTrack = StringUtils::readAs<std::size_t>(strings[1]);
				}
			}
			else if (tag == "DISC")
			{
				// Expecting 'Number/Total'
				std::vector<std::string> strings {StringUtils::splitString(value, "/")};

				if (strings.size() > 0)
				{
					track.discNumber = StringUtils::readAs<std::size_t>(strings[0]);

					if (strings.size() > 1)
						track.totalDisc = StringUtils::readAs<std::size_t>(strings[1]);
				}
			}
			else if (tag == "DATE"
					|| tag == "YEAR"
					|| tag == "WM/Year")
			{
				track.year = StringUtils::readAs<int>(value);
			}
			else if (tag == "TDOR"	// Original release time (ID3v2 2.4)
					|| tag == "TORY")	// Original release year
			{
				track.originalYear = StringUtils::readAs<int>(value);
			}
			else if (tag == "ACOUSTID ID")
			{
				track.acoustID = UUID::fromString(value);
			}
			else if (tag == "MUSICBRAINZ RELEASE TRACK ID"
					|| tag == "MUSICBRAINZ_RELEASETRACKID"
					|| tag == "MUSICBRAINZ_TRACKID"
					|| tag == "MUSICBRAINZ/TRACK ID")
			{
				track.musicBrainzTrackID = UUID::fromString(value);
			}
			else if (_clusterTypeNames.find(tag) != _clusterTypeNames.end())
			{
				std::vector<std::string> clusterNames {StringUtils::splitString(value, "/,;")};

				if (!clusterNames.empty())
					track.clusters[tag] = std::set<std::string>{clusterNames.begin(), clusterNames.end()};
			}

		}

		track.artists = getArtists(metadataMap);
		track.album = getAlbum(metadataMap);
		track.albumArtists = getAlbumArtists(metadataMap);
	}
	catch(Av::MediaFileException& e)
	{
		return std::nullopt;
	}

	return track;
}

} // namespace MetaData