package dk.thoerup.spejdernetscraper;

import org.jsoup.Jsoup;
import org.jsoup.nodes.Document;
import org.jsoup.nodes.Element;
import org.jsoup.select.Elements;

public class IMDBSeriesScraper {
	
	public String fechTitle(String sid, int season, int episode) throws Exception {
		
		String epMatch = "Ep" + episode;
		
		String url = "http://www.imdb.com/title/" + sid + "/episodes?season=" + season;
		
		Document page = Jsoup.connect(url)
				.get();
		
		Element episodesContent = page.getElementById("episodes_content");
		
		Elements episodesList = episodesContent.getElementsByClass("list_item");
		
		for (int i=0; i<episodesList.size(); i++) {
			Element curEp = episodesList.get(i);
			
			Element image = curEp.getElementsByClass("image").first();
			
			Element anchor = image.child(0);
			
			String title = anchor.attr("title");
			String content = anchor.text().trim();
			
			if ( content.endsWith(epMatch)) {
				return title;	
			}			
		}
		
		
		throw new Exception("Episode not found !");
	}

}