X-Git-Url: http://vcs.maemo.org/git/?a=blobdiff_plain;f=src%2Fimdb%2Fimdb-line-parser.vala;h=7963fda139926b6e11dedfbe5694466a90ed2ff5;hb=066c91421665a1a8bfbe174614534c7e0f72ea86;hp=fd3875aabe2530dc8e059e18cdd0625db0663519;hpb=72ca17691c6d2f2c80f4288d0c7321d5acc4d3bc;p=cinaest

diff --git a/src/imdb/imdb-line-parser.vala b/src/imdb/imdb-line-parser.vala
index fd3875a..7963fda 100644
--- a/src/imdb/imdb-line-parser.vala
+++ b/src/imdb/imdb-line-parser.vala
@@ -93,10 +93,16 @@ class GenreLineParser : LineParser {
 }
 
 class RatingLineParser : LineParser {
+	enum RatingState {
+		HEADER,
+		NONE
+	}
+	RatingState state;
 	Regex re_rating;
 
 	public RatingLineParser (IMDbSqlite _sqlite) {
 		base (_sqlite);
+		state = RatingState.HEADER;
 		try {
 			re_rating = new Regex ("^      .+ +([0-9]+) +([0-9.]+) +(.+)$");
 		} catch (RegexError e) {
@@ -105,6 +111,15 @@ class RatingLineParser : LineParser {
 	}
 
 	public override void parse_line (string line) {
+		if (state == RatingState.HEADER) {
+			if (line == "MOVIE RATINGS REPORT")
+				state = RatingState.NONE;
+			return;
+		}
+
+		if (state != RatingState.NONE)
+			return;
+
 		MatchInfo matchinfo;
 
 		// Skip series episodes
@@ -117,7 +132,7 @@ class RatingLineParser : LineParser {
 		string title;
 		string votes = matchinfo.fetch (1);
 		string rating = matchinfo.fetch (2);
-	        try {
+		try {
 			title = convert(matchinfo.fetch (3), -1, "utf-8", "latin1");
 		} catch (ConvertError e) {
 			return;
@@ -133,3 +148,167 @@ class RatingLineParser : LineParser {
 		sqlite.movie_set_rating (title, (int) (rating.to_double () * 10), votes.to_int ());
 	}
 }
+
+class AkaLineParser : LineParser {
+	enum AkaState {
+		HEADER,
+		NONE,
+		TITLE
+	}
+	AkaState state;
+	string title;
+
+	public AkaLineParser (IMDbSqlite _sqlite) {
+		base (_sqlite);
+		state = AkaState.HEADER;
+		title = null;
+	}
+
+	public override void parse_line (string line) {
+		if (state == AkaState.HEADER) {
+			if (line == "AKA TITLES LIST") title = line;
+			if (line == "===============" && title != null)
+				state = AkaState.NONE;
+			return;
+		}
+
+		if (state == AkaState.NONE) {
+			// Skip empty lines
+			if (line == "")
+				return;
+
+			// Skip series episodes
+			if (line[0] == '"')
+				return;
+
+			// Parse error
+			if (line[0] == ' ')
+				return;
+
+		        try {
+				title = convert (line, -1, "utf-8", "latin1");
+			} catch (ConvertError e) {
+				title = null;
+				return;
+			}
+
+			if (skip_title (title))
+				return;
+
+			state = AkaState.TITLE;
+		}
+
+		if (state == AkaState.TITLE) {
+			// Empty lines mark end of title
+			if (line == "") {
+				state = AkaState.NONE;
+				return;
+			}
+
+			if (line.has_prefix ("   (aka ")) {
+				if (skip_title (title))
+					return;
+
+				char* start = line.offset (8);
+				char* end = ((string) start).str ("))");
+				if (end != null)
+					end[1] = '\0';
+
+				string aka;
+				try {
+					aka = convert ((string) start, -1, "utf-8", "latin1");
+				} catch (ConvertError e) {
+					return;
+				}
+
+				sqlite.add_aka (title, aka);
+			}
+		}
+	}
+}
+
+class PlotLineParser : LineParser {
+	enum PlotState {
+		HEADER,
+		NONE,
+		TITLE
+	}
+	string title;
+	string plot;
+	PlotState state;
+
+	public PlotLineParser (IMDbSqlite _sqlite) {
+		base (_sqlite);
+		state = PlotState.HEADER;
+		title = null;
+	}
+
+	public override void parse_line (string line) {
+		if (state == PlotState.HEADER) {
+			if (line == "PLOT SUMMARIES LIST") title = line;
+			if (line == "===================" && title != null)
+				state = PlotState.NONE;
+			return;
+		}
+
+		// Skip empty lines
+		if (line == "")
+			return;
+
+		if (state == PlotState.NONE) {
+			if (line.has_prefix ("MV: ")) {
+				// Skip series episodes
+				if (line[4] == '"')
+					return;
+
+			        try {
+					title = convert (line.offset (4), -1, "utf-8", "latin1");
+				} catch (ConvertError e) {
+					stderr.printf ("Error converting title to UTF-8\n");
+					title = null;
+					return;
+				}
+
+				if (skip_title (title))
+					return;
+
+				state = PlotState.TITLE;
+				plot = "";
+			}
+			return;
+		}
+
+		if (state == PlotState.TITLE) {
+			if (line.has_prefix ("PL: ")) {
+				if (skip_title (title))
+					return;
+
+				try {
+					if (plot != "")
+						plot += " ";
+					plot += convert (line.offset (4), -1, "utf-8", "latin1");
+				} catch (ConvertError e) {
+					stderr.printf ("Error converting plot for \"%s\" to UTF-8\n", title);
+					plot = "";
+					return;
+				}
+			}
+
+			// BY: tag marks end of plot
+			if (line.has_prefix ("BY: ")) {
+				string author;
+			        try {
+					author = convert (line.offset (4), -1, "utf-8", "latin1");
+				} catch (ConvertError e) {
+					stderr.printf ("Error converting plot author for \"%s\" to UTF-8\n", title);
+					author = null;
+				}
+
+				sqlite.add_plot (title, plot, author);
+
+				state = PlotState.NONE;
+				return;
+			}
+		}
+	}
+}