IMDb SQLite database: add get_cast method
[cinaest] / src / imdb / imdb-sqlite.vala
index 10065c0..ff9edcb 100644 (file)
@@ -63,32 +63,57 @@ class IMDbSqlite : Object {
                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;
@@ -108,12 +133,22 @@ class IMDbSqlite : Object {
                        }
                }
 
-               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;
        }
@@ -126,66 +161,180 @@ class IMDbSqlite : Object {
                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;
 
-               string sql = "INSERT INTO Akas(Aka, TitleID) VALUES (\"%s\", %d);".printf (aka, rowid);
-               int rc;
-               rc = db.exec (sql, callback, null);
-               if (rc != Sqlite.OK) {
-                       stderr.printf ("SQL error: %d, %s\n", rc, db.errmsg ());
+               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;
 
-               string sql;
-               if (author != null) {
-                       sql = "INSERT INTO Plots(rowid, Plot, Author) VALUES (%d, \"%s\", \"%s\");".printf (rowid, plot.replace ("\"", """), author.replace ("\"", """));
-               } else {
-                       sql = "INSERT INTO Plots(rowid, Plot) VALUES (%d, \"%s\");".printf (rowid, plot.replace ("\"", """));
+               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 person_statement = null;
+       public int add_person (string name) {
                int rc;
-               rc = db.exec (sql, callback, null);
-               if (rc != Sqlite.OK) {
-                       stderr.printf ("SQL error: %d, %s\n", rc, db.errmsg ());
-                       stderr.printf ("offending SQL: %s\n", sql);
+
+               if (person_statement == null) {
+                       string sql = "INSERT INTO People(Name) VALUES (?);";
+                       rc = db.prepare_v2 (sql, -1, out person_statement);
+                       if (rc != Sqlite.OK) {
+                               error ("Failed to prepare statement \"" + sql + "\": " + db.errmsg ());
+                       }
+               }
+
+               assert (person_statement.bind_text (1, name) == Sqlite.OK);
+
+               rc = person_statement.step ();
+               if (rc != Sqlite.DONE) {
+                       if (rc == Sqlite.CONSTRAINT)
+                               stderr.printf ("Skipped duplicate person \"%s\"\n", name);
+                       else
+                               stderr.printf ("Failed to insert person \"%s\": %d, %s\n", name, rc, db.errmsg ());
+                       person_statement.reset ();
                        return 1;
                }
+               person_statement.reset ();
 
                return 0;
        }
 
-       public bool movie_exists (string title, out int rowid = null) {
-               string sql = "SELECT rowid FROM Movies WHERE Title=\"%s\"".printf (title);
-               Statement stmt;
+       Statement actor_statement;
+       public int add_actor (string name, string title, string? info, string? character, int number = 0) {
+               int actorid;
+               int titleid;
                int rc;
 
-               rc = db.prepare_v2 (sql, -1, out stmt);
-               if (rc != Sqlite.OK) {
-                       stderr.printf ("SQL error: %d, %s\n", rc, db.errmsg ());
-                       return false;
+               if (!person_exists (name, out actorid))
+                       return 1;
+               if (!movie_exists (title, out titleid))
+                       return 1;
+
+               if (actor_statement == null) {
+                       string sql = "INSERT INTO Actors(ActorID, TitleID, Info, Character, Number) VALUES (?, ?, ?, ?, ?);";
+                       assert (db.prepare_v2 (sql, -1, out actor_statement) == Sqlite.OK);
+               }
+
+               assert (actor_statement.bind_int (1, actorid) == Sqlite.OK);
+               assert (actor_statement.bind_int (2, titleid) == Sqlite.OK);
+               if (info != null)
+                       assert (actor_statement.bind_text (3, info) == Sqlite.OK);
+               else
+                       assert (actor_statement.bind_null (3) == Sqlite.OK);
+
+               if (character != null)
+                       assert (actor_statement.bind_text (4, character) == Sqlite.OK);
+               else
+                       assert (actor_statement.bind_null (4) == Sqlite.OK);
+               assert (actor_statement.bind_int (5, number) == Sqlite.OK);
+
+               rc = actor_statement.step ();
+               if (rc != Sqlite.DONE) {
+                       stderr.printf ("Failed to insert actor \"%s\": %d, %s\n", name, rc, db.errmsg ());
+                       actor_statement.reset ();
+                       return 1;
+               }
+               actor_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 = stmt.step ();
+                       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;
+       }
+
+       Statement person_exists_statement = null;
+       public bool person_exists (string name, out int rowid = null) {
+               if (person_exists_statement == null) {
+                       string sql = "SELECT rowid FROM People WHERE Name=?";
+                       assert (db.prepare_v2 (sql, -1, out person_exists_statement) == Sqlite.OK);
+               }
+
+               assert (person_exists_statement.bind_text (1, name) == Sqlite.OK);
+
+               int rc = Sqlite.OK;
+               do {
+                       rc = person_exists_statement.step ();
                        if (rc == Sqlite.ROW) {
                                if (&rowid != null) {
-                                       rowid = stmt.column_int (0);
+                                       rowid = person_exists_statement.column_int (0);
                                }
+                               person_exists_statement.reset ();
+
                                return true;
                        }
                } while (rc == Sqlite.ROW);
+               person_exists_statement.reset ();
 
                return false;
        }
@@ -213,8 +362,25 @@ class IMDbSqlite : Object {
                return count;
        }
 
-       public bool has_plots () {
-               return (count ("Plots") > 0);
+       public bool has_table (string table) {
+               string sql = "SELECT * FROM sqlite_master WHERE type=\"table\" AND name=\"%s\"".printf (table);
+               Statement stmt;
+               int rc;
+
+               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 false;
+               }
+
+               do {
+                       rc = stmt.step ();
+                       if (rc == Sqlite.ROW)
+                               return true;
+               } while (rc == Sqlite.ROW);
+
+               return false;
        }
 
        public string get_plot (string title) {
@@ -240,6 +406,36 @@ class IMDbSqlite : Object {
                return "";
        }
 
+       public List<Role> get_cast (string title) {
+               int rowid;
+               string sql = "SELECT Name,Character FROM Actors,People WHERE TitleID IN (SELECT rowid FROM Movies WHERE Title=\"%s\") AND ActorID=People.rowid AND Number>0 ORDER BY NUMBER LIMIT 3".printf (title);
+               Statement stmt;
+               int rc;
+               int count = 0;
+               var cast = new List<Role> ();
+
+               if (!movie_exists (title, out rowid))
+                       return 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) {
+                               var role = new Role ();
+                               role.actor_name = stmt.column_text (0);
+                               role.character = stmt.column_text (1);
+                               cast.append (role);
+                       }
+               } while (rc == Sqlite.ROW);
+
+               return cast;
+       }
+
        public int clear () {
                int rc;
 
@@ -251,7 +447,13 @@ class IMDbSqlite : Object {
                        "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)",
+                       "CREATE TABLE Plots (Plot TEXT NOT NULL, Author TEXT);" +
+                       "DROP TABLE IF EXISTS People;" +
+                       "CREATE TABLE People (Name TEXT PRIMARY KEY COLLATE NOCASE);" +
+                       "DROP TABLE IF EXISTS Actors;" +
+                       "CREATE TABLE Actors (ActorID INTEGER NOT NULL, TitleID INTEGER NOT NULL, Info TEXT, Character TEXT, Number INTEGER NOT NULL DEFAULT 0);" +
+                       "CREATE INDEX MovieCast ON Actors(TitleID);" +
+                       "CREATE INDEX ActorFilmography ON Actors(ActorID);",
                        callback, null);
                if (rc != Sqlite.OK) {
                        stderr.printf ("SQL error: %d, %s\n", rc, db.errmsg ());