package hu.procyon.atomizer;

import java.io.IOException;
import java.util.List;
import java.util.Map;

public class MusicUrlFilter implements HrefFilter {

	@Override
	public boolean probeHref(Href url, WebScraper scraper) {
		System.out.println("probing " + url.toString());
		Map<String, List<String>> headers = null;
		try {
			headers = scraper.doHeadRequest(url);
			if (headers == null) {
				return false;
			}
			List<String> contentType = headers.get("Content-Type");
			if (contentType.contains("audio/mpeg") ||
				contentType.contains("audio/x-mpeg") ||
				contentType.contains("audio/mp3") ||
				contentType.contains("audio/x-mp3") ||
				contentType.contains("audio/mpeg3") ||
				contentType.contains("audio/x-mpeg3") ||
				contentType.contains("audio/mpg") ||
				contentType.contains("audio/x-mpg") ||
				contentType.contains("audio/x-mpegaudio") ||
				contentType.contains("audio/ogg") ||
				contentType.contains("application/ogg") ||
				contentType.contains("audio/vorbis") ||
				contentType.contains("audio/x-ogg") ||
				contentType.contains("application/x-ogg")) {
				return true;
			}
		} catch (IOException e) {
			e.printStackTrace(System.err);
		}
		return false;
	}

}
