From 43c5e992b7780e7ced837259f117565b53c2782a Mon Sep 17 00:00:00 2001
From: Reinhard Pointner <rednoah@users.noreply.github.com>
Date: Fri, 20 Jul 2012 11:49:07 +0000
Subject: [PATCH] * index original title as well as engish AKA title

---
 .../sourceforge/filebot/web/IMDbClient.java   | 32 ++++++++++++++++---
 1 file changed, 27 insertions(+), 5 deletions(-)

diff --git a/source/net/sourceforge/filebot/web/IMDbClient.java b/source/net/sourceforge/filebot/web/IMDbClient.java
index ea972254..d95760b4 100644
--- a/source/net/sourceforge/filebot/web/IMDbClient.java
+++ b/source/net/sourceforge/filebot/web/IMDbClient.java
@@ -22,12 +22,12 @@ import java.util.regex.Pattern;
 
 import javax.swing.Icon;
 
+import net.sourceforge.filebot.ResourceManager;
+
 import org.w3c.dom.Document;
 import org.w3c.dom.Node;
 import org.xml.sax.SAXException;
 
-import net.sourceforge.filebot.ResourceManager;
-
 
 public class IMDbClient implements MovieIdentificationService {
 	
@@ -83,7 +83,7 @@ public class IMDbClient implements MovieIdentificationService {
 		
 		// we might have been redirected to the movie page
 		if (results.isEmpty()) {
-			Movie movie = scrapeMovie(dom);
+			Movie movie = scrapeMovie(dom, locale);
 			if (movie != null) {
 				results.add(movie);
 			}
@@ -93,7 +93,7 @@ public class IMDbClient implements MovieIdentificationService {
 	}
 	
 	
-	protected Movie scrapeMovie(Document dom) {
+	protected Movie scrapeMovie(Document dom, Locale locale) {
 		try {
 			String header = selectString("//H1", dom).toUpperCase();
 			if (header.contains("(VG)") || header.contains("(V)")) // ignore video games and videos
@@ -102,6 +102,28 @@ public class IMDbClient implements MovieIdentificationService {
 			String name = selectString("//H1/A/text()", dom).replaceAll("\\s+", " ").trim();
 			String year = new Scanner(selectString("//H1/A/following::A/text()", dom)).useDelimiter("\\D+").next();
 			String url = selectString("//H1/A/@href", dom);
+			
+			// try to get localized name
+			if (locale != null && locale != Locale.ROOT) {
+				try {
+					String languageName = locale.getDisplayLanguage(Locale.ENGLISH).toLowerCase();
+					List<Node> akaRows = selectNodes("//*[@name='akas']//following::TABLE[1]//TR", dom);
+					
+					for (Node aka : akaRows) {
+						List<Node> columns = getChildren("TD", aka);
+						String akaTitle = getTextContent(columns.get(0));
+						String languageDesc = getTextContent(columns.get(1)).toLowerCase();
+						
+						if (languageName.length() > 0 && languageDesc.contains(languageName)) {
+							name = akaTitle;
+							break;
+						}
+					}
+				} catch (Exception e) {
+					e.printStackTrace();
+				}
+			}
+			
 			return new Movie(name, Pattern.matches("\\d{4}", year) ? Integer.parseInt(year) : -1, getImdbId(url));
 		} catch (Exception e) {
 			// ignore, we probably got redirected to an error page
@@ -113,7 +135,7 @@ public class IMDbClient implements MovieIdentificationService {
 	@Override
 	public Movie getMovieDescriptor(int imdbid, Locale locale) throws Exception {
 		try {
-			return scrapeMovie(parsePage(new URL("http", host, String.format("/title/tt%07d/releaseinfo", imdbid))));
+			return scrapeMovie(parsePage(new URL("http", host, String.format("/title/tt%07d/releaseinfo", imdbid))), locale);
 		} catch (FileNotFoundException e) {
 			return null; // illegal imdbid
 		}