Database db;
List<string> genres;
- public delegate void ReceiveMovieFunction (string title, int year, int rating, int genres);
+ public delegate void ReceiveMovieFunction (string title, string? aka, int year, int rating, int genres);
public IMDbSqlite (string filename) {
int rc;
return 0;
}
+ Statement add_statement = null;
public int add_movie (string title, int year) {
- string sql = "INSERT INTO Movies(Title, Year) VALUES (\"%s\", %d);".printf (title, year);
int rc;
- rc = db.exec (sql, callback, null);
- if (rc != Sqlite.OK) {
- stderr.printf ("Failed to insert movie \"%s\" (%d): %d, %s\n", title, year, rc, db.errmsg ());
+ if (add_statement == null) {
+ string sql = "INSERT INTO Movies(Title, Year) VALUES (?, ?);";
+ assert (db.prepare_v2 (sql, -1, out add_statement) == Sqlite.OK);
+ }
+
+ assert (add_statement.bind_text (1, title) == Sqlite.OK);
+ assert (add_statement.bind_int (2, year) == Sqlite.OK);
+
+ rc = add_statement.step ();
+ if (rc != Sqlite.DONE) {
+ if (rc == Sqlite.CONSTRAINT)
+ stderr.printf ("Skipped duplicate \"%s\" (%d)\n", title, year);
+ else
+ stderr.printf ("Failed to insert movie \"%s\" (%d): %d, %s\n", title, year, rc, db.errmsg ());
+ add_statement.reset ();
return 1;
}
+ add_statement.reset ();
return 0;
}
+ Statement rating_statement = null;
public int movie_set_rating (string title, int rating, int votes) {
- var sql = "UPDATE Movies SET Rating=%d, Votes=%d WHERE Title=\"%s\";".printf (rating, votes, title);
int rc;
- rc = db.exec (sql, callback, null);
- if (rc != Sqlite.OK) {
- stderr.printf ("SQL error: %d, %s\n", rc, db.errmsg ());
+ if (rating_statement == null) {
+ var sql = "UPDATE Movies SET Rating=?, Votes=? WHERE Title=?;";
+ assert (db.prepare_v2 (sql, -1, out rating_statement) == Sqlite.OK);
+ }
+
+ assert (rating_statement.bind_int (1, rating) == Sqlite.OK);
+ assert (rating_statement.bind_int (2, votes) == Sqlite.OK);
+ assert (rating_statement.bind_text (3, title) == Sqlite.OK);
+
+ rc = rating_statement.step ();
+ if (rc != Sqlite.DONE) {
+ stderr.printf ("Failed to set rating %d(%d) for \"%s\": %d, %s\n", rating, votes, title, rc, db.errmsg ());
+ rating_statement.reset ();
return 1;
}
+ rating_statement.reset ();
return 0;
}
+ Statement genre_statement = null;
public int movie_add_genre (string title, string genre) {
string sql;
int bit;
}
}
- sql = "UPDATE Movies SET Genres=Genres|%d WHERE Title=\"%s\";".printf (bit, title);
- rc = db.exec (sql, callback, null);
- if (rc != Sqlite.OK) {
- stderr.printf ("SQL error: %d, %s\n", rc, db.errmsg ());
+ if (genre_statement == null) {
+ sql = "UPDATE Movies SET Genres=Genres|? WHERE Title=?;";
+ assert (db.prepare_v2 (sql, -1, out genre_statement) == Sqlite.OK);
+ }
+
+ assert (genre_statement.bind_int (1, bit) == Sqlite.OK);
+ assert (genre_statement.bind_text (2, title) == Sqlite.OK);
+
+
+ rc = genre_statement.step ();
+ if (rc != Sqlite.DONE) {
+ stderr.printf ("Failed to add genre %s to \"%s\": %d, %s\n", genre, title, rc, db.errmsg ());
+ rating_statement.reset ();
return 1;
}
+ genre_statement.reset ();
return 0;
}
return 0;
}
+ Statement aka_statement = null;
+ public int add_aka (string title, string aka) {
+ int rc;
+ int rowid;
+
+ if (!movie_exists (title, out rowid))
+ return 1;
+
+ if (aka_statement == null) {
+ string sql = "INSERT INTO Akas(Aka, TitleID) VALUES (?, ?);";
+ assert (db.prepare_v2 (sql, -1, out aka_statement) == Sqlite.OK);
+ }
+
+ assert (aka_statement.bind_text (1, aka) == Sqlite.OK);
+ assert (aka_statement.bind_int (2, rowid) == Sqlite.OK);
+
+ rc = aka_statement.step ();
+ if (rc != Sqlite.DONE) {
+ stderr.printf ("Failed to add aka \"%s\" to \"%s\": %d, %s\n", aka, title, rc, db.errmsg ());
+ aka_statement.reset ();
+ return 1;
+ }
+ aka_statement.reset ();
+
+ return 0;
+ }
+
+ Statement plot_statement = null;
+ public int add_plot (string title, string plot, string? author) {
+ int rc;
+ int rowid;
+
+ if (!movie_exists (title, out rowid))
+ return 1;
+
+ if (plot_statement == null) {
+ var sql = "INSERT INTO Plots(rowid, Plot, Author) VALUES (?, ?, ?);";
+ assert (db.prepare_v2 (sql, -1, out plot_statement) == Sqlite.OK);
+ }
+
+ assert (plot_statement.bind_int (1, rowid) == Sqlite.OK);
+ assert (plot_statement.bind_text (2, plot.replace ("\"", """)) == Sqlite.OK);
+ if (author != null)
+ assert (plot_statement.bind_text (3, author) == Sqlite.OK);
+ else
+ assert (plot_statement.bind_null (3) == Sqlite.OK);
+
+ rc = plot_statement.step ();
+ if (rc != Sqlite.DONE) {
+ stderr.printf ("Failed to add plot for \"%s\": %d, %s\n", title, rc, db.errmsg ());
+ plot_statement.reset ();
+ return 1;
+ }
+ plot_statement.reset ();
+
+ return 0;
+ }
+
+ Statement exists_statement = null;
+ public bool movie_exists (string title, out int rowid = null) {
+ if (exists_statement == null) {
+ string sql = "SELECT rowid FROM Movies WHERE Title=?";
+ assert (db.prepare_v2 (sql, -1, out exists_statement) == Sqlite.OK);
+ }
+
+ assert (exists_statement.bind_text (1, title) == Sqlite.OK);
+
+ int rc = Sqlite.OK;
+ do {
+ rc = exists_statement.step ();
+ if (rc == Sqlite.ROW) {
+ if (&rowid != null) {
+ rowid = exists_statement.column_int (0);
+ }
+ exists_statement.reset ();
+ return true;
+ }
+ } while (rc == Sqlite.ROW);
+ exists_statement.reset ();
+
+ return false;
+ }
+
+ public int count (string table) {
+ string sql = "SELECT count(*) FROM %s".printf (table);
+ Statement stmt;
+ int rc;
+ int count = 0;
+
+ rc = db.prepare_v2 (sql, -1, out stmt);
+ if (rc != Sqlite.OK) {
+ stderr.printf ("SQL error: %d, %s\n", rc, db.errmsg ());
+ db.progress_handler (0, null);
+ return 0;
+ }
+
+ do {
+ rc = stmt.step ();
+ if (rc == Sqlite.ROW) {
+ count = stmt.column_int (0);
+ }
+ } while (rc == Sqlite.ROW);
+
+ return count;
+ }
+
+ public bool has_plots () {
+ return (count ("Plots") > 0);
+ }
+
+ public string get_plot (string title) {
+ string sql = "SELECT Plot FROM Plots WHERE rowid in (SELECT rowid FROM Movies WHERE Title=\"%s\")".printf (title);
+ Statement stmt;
+ int rc;
+ int count = 0;
+
+ rc = db.prepare_v2 (sql, -1, out stmt);
+ if (rc != Sqlite.OK) {
+ stderr.printf ("SQL error: %d, %s\n", rc, db.errmsg ());
+ db.progress_handler (0, null);
+ return "";
+ }
+
+ do {
+ rc = stmt.step ();
+ if (rc == Sqlite.ROW) {
+ return stmt.column_text (0).replace (""", "\"");
+ }
+ } while (rc == Sqlite.ROW);
+
+ return "";
+ }
+
public int clear () {
int rc;
- rc = db.exec ("DROP TABLE IF EXISTS Movies; CREATE TABLE Movies (Title TEXT NOT NULL, Year INTEGER, Rating INTEGER, Votes INTEGER, Genres INTEGER NOT NULL DEFAULT 0); DROP TABLE IF EXISTS Genres; CREATE TABLE Genres (Bit INTEGER PRIMARY KEY, Genre TEXT NOT NULL);", callback, null);
+ rc = db.exec (
+ "DROP TABLE IF EXISTS Movies;" +
+ "CREATE TABLE Movies (Title TEXT PRIMARY KEY COLLATE NOCASE, Year INTEGER, Rating INTEGER, Votes INTEGER NOT NULL DEFAULT 0, Genres INTEGER NOT NULL DEFAULT 0);" +
+ "DROP TABLE IF EXISTS Genres;" +
+ "CREATE TABLE Genres (Bit INTEGER PRIMARY KEY, Genre TEXT NOT NULL);" +
+ "DROP TABLE IF EXISTS Akas;" +
+ "CREATE TABLE Akas (Aka TEXT NOT NULL COLLATE NOCASE, TitleID INTEGER NOT NULL);" +
+ "DROP TABLE IF EXISTS Plots;" +
+ "CREATE TABLE Plots (Plot TEXT NOT NULL, Author TEXT)",
+ callback, null);
if (rc != Sqlite.OK) {
stderr.printf ("SQL error: %d, %s\n", rc, db.errmsg ());
return 1;
return 0;
}
- public int create_title_index () {
+ public int create_aka_index () {
int rc;
- rc = db.exec ("CREATE INDEX MovieTitles ON Movies(Title);", callback, null);
+ rc = db.exec ("CREATE INDEX AkasAka ON Akas(Aka);", callback, null);
if (rc != Sqlite.OK) {
stderr.printf ("SQL error: %d, %s\n", rc, db.errmsg ());
return 1;
public async int query (MovieFilter filter, ReceiveMovieFunction receive_movie, int limit, Cancellable? cancellable) {
var sql = "SELECT Title, Year, Rating, Genres FROM Movies";
var sep = " WHERE ";
+ string match = null;
Statement stmt;
int rc;
db.progress_handler (1000, progress_handler);
if (filter.title != null && filter.title != "") {
- if ("*" in filter.title)
- sql += sep + "Title GLOB \"%s (*)\"".printf (filter.title);
- else
- sql += sep + "Title LIKE \"%s%%\"".printf (filter.title);
+ if ("*" in filter.title) {
+ match = "GLOB \"%s (*)\"".printf (filter.title);
+ } else {
+ match = "LIKE \"%s%%\"".printf (filter.title);
+ }
+ sql += sep + "(Title %s OR rowid IN (SELECT TitleID FROM Akas WHERE Aka %s))".printf (match, match);
sep = " AND ";
}
if (filter.year_min > 0) {
if (filter.genres.field != 0) {
sql += sep + "Genres&%d = %d".printf (filter.genres.field, filter.genres.field);
}
- sql += " ORDER BY Votes DESC LIMIT %d;".printf (limit);
+ sql += " ORDER BY Votes DESC LIMIT %d;".printf (limit + 1);
stdout.printf("SQL: \"%s\"\n", sql);
if (rc != Sqlite.OK) {
stderr.printf ("SQL error: %d, %s\n", rc, db.errmsg ());
db.progress_handler (0, null);
- return 1;
+ return 0;
}
+ int n = 0;
do {
+ if (++n > limit)
+ break;
Idle.add (query.callback);
yield;
rc = stmt.step ();
if (rc == Sqlite.ROW) {
int year = stmt.column_int (1);
string title = stmt.column_text (0);
- int rating = stmt.column_int (2);
+ int rating;
+ if (stmt.column_type (2) != Sqlite.NULL)
+ rating = stmt.column_int (2);
+ else
+ rating = -1;
int genres = stmt.column_int (3);
- receive_movie (strip_year (title, year), year, rating, genres);
+ string aka = null;
+ if (match != null && !(filter.matches_title (strip_year (title, year)))) {
+ aka = movie_aka (title, match);
+ if (aka != null)
+ aka = strip_year (aka, year);
+ }
+ receive_movie (strip_year (title, year), aka, year, rating, genres);
}
} while (rc == Sqlite.ROW);
db.progress_handler (0, null);
- return 0;
+ return n;
+ }
+
+ private string? movie_aka (string title, string match) {
+ string sql = "SELECT Aka FROM Akas WHERE (TitleID = (SELECT rowid FROM Movies WHERE Title = \"%s\") AND Aka %s) LIMIT 1;".printf (title, match);
+ Statement stmt;
+ int rc;
+ string aka = null;
+
+ rc = db.prepare_v2 (sql, -1, out stmt);
+ if (rc != Sqlite.OK) {
+ stderr.printf ("SQL error: %d, %s\n", rc, db.errmsg ());
+ return null;
+ }
+
+ do {
+ rc = stmt.step ();
+ if (rc == Sqlite.ROW) {
+ aka = stmt.column_text (0);
+ }
+ } while (rc == Sqlite.ROW);
+
+ return aka;
}
private int progress_handler () {