filebot/source/net/filebot/web/ID3Lookup.java

package net.filebot.web;

import static net.filebot.Logging.*;
import static net.filebot.MediaTypes.*;
import static net.filebot.util.FileUtilities.*;
import static net.filebot.util.StringUtilities.*;

import java.io.File;
import java.util.Collection;
import java.util.LinkedHashMap;
import java.util.Map;

import javax.swing.Icon;

import net.filebot.ResourceManager;
import net.filebot.mediainfo.MediaInfo;
import net.filebot.mediainfo.MediaInfo.StreamKind;

public class ID3Lookup implements MusicIdentificationService {

	@Override
	public String getName() {
		return "ID3 Tags";
	}

	@Override
	public String getIdentifier() {
		return "ID3";
	}

	@Override
	public Icon getIcon() {
		return ResourceManager.getIcon("search.mediainfo");
	}

	@Override
	public Map<File, AudioTrack> lookup(Collection<File> files) throws Exception {
		Map<File, AudioTrack> info = new LinkedHashMap<File, AudioTrack>();

		try (MediaInfo mediaInfo = new MediaInfo()) {
			for (File f : filter(files, AUDIO_FILES, VIDEO_FILES)) {
				try {
					// open or throw exception
					mediaInfo.open(f);

					// artist and song title information is required
					String artist = getString(mediaInfo, "Performer", "Composer");
					String title = getString(mediaInfo, "Title", "Track");

					if (artist != null && title != null) {
						// all other properties are optional
						String album = getString(mediaInfo, "Album");
						String albumArtist = getString(mediaInfo, "Album/Performer");
						String trackTitle = getString(mediaInfo, "Track");
						Integer mediumIndex = null;
						Integer mediumCount = null;
						Integer trackIndex = getInteger(mediaInfo, "Track/Position");
						Integer trackCount = getInteger(mediaInfo, "Track/Position_Total");
						String mbid = null;

						// try to parse 2016-03-10
						String dateString = getString(mediaInfo, "Recorded_Date");
						SimpleDate albumReleaseDate = SimpleDate.parse(dateString);

						// try to parse 2016
						if (albumReleaseDate == null) {
							Integer year = matchInteger(dateString);
							if (year != null) {
								albumReleaseDate = new SimpleDate(year, 1, 1);
							}
						}

						info.put(f, new AudioTrack(artist, title, album, albumArtist, trackTitle, albumReleaseDate, mediumIndex, mediumCount, trackIndex, trackCount, mbid, getIdentifier()));
					}
				} catch (Throwable e) {
					debug.warning(e.getMessage());
				}
			}
		}

		return info;
	}

	private String getString(MediaInfo mediaInfo, String... keys) {
		for (String key : keys) {
			String value = mediaInfo.get(StreamKind.General, 0, key);
			if (value.length() > 0) {
				return value;
			}
		}
		return null;
	}

	private Integer getInteger(MediaInfo mediaInfo, String field) {
		return matchInteger(getString(mediaInfo, field));
	}

}