diff options
Diffstat (limited to 'src/plugins/lms/rygel-lms-database.vala')
| -rw-r--r-- | src/plugins/lms/rygel-lms-database.vala | 226 |
1 files changed, 226 insertions, 0 deletions
diff --git a/src/plugins/lms/rygel-lms-database.vala b/src/plugins/lms/rygel-lms-database.vala new file mode 100644 index 00000000..2108ae60 --- /dev/null +++ b/src/plugins/lms/rygel-lms-database.vala @@ -0,0 +1,226 @@ +/* + * Copyright (C) 2009,2011 Jens Georg <mail@jensge.org>, + * (C) 2013 Intel Corporation. + * + * Author: Jussi Kukkonen <jussi.kukkonen@intel.com> + * + * This file is part of Rygel. + * + * Rygel is free software; you can redistribute it and/or modify + * it under the terms of the GNU Lesser General Public License as published by + * the Free Software Foundation; either version 2 of the License, or + * (at your option) any later version. + * + * Rygel is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + * GNU Lesser General Public License for more details. + * + * You should have received a copy of the GNU Lesser General Public License + * along with this program; if not, write to the Free Software Foundation, + * Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA. + */ + +using Rygel; +using Gee; +using Sqlite; + +public errordomain Rygel.LMS.DatabaseError { + OPEN, + PREPARE, + BIND, + STEP, + NOT_FOUND +} + +namespace Rygel.LMS { + extern static int utf8_collate_str (uint8[] a, uint8[] b); +} + +public class Rygel.LMS.Database { + private Sqlite.Database db; + + /** + * Function to implement the custom SQL function 'contains' + */ + private static void utf8_contains (Sqlite.Context context, + Sqlite.Value[] args) + requires (args.length == 2) { + if (args[0].to_text () == null || + args[1].to_text () == null) { + context.result_int (0); + + return; + } + + var pattern = Regex.escape_string (args[1].to_text ()); + if (Regex.match_simple (pattern, + args[0].to_text (), + RegexCompileFlags.CASELESS)) { + context.result_int (1); + } else { + context.result_int (0); + } + } + + /** + * Function to implement the custom SQLite collation 'CASEFOLD'. + * + * Uses utf8 case-fold to compare the strings. + */ + private static int utf8_collate (int alen, void* a, int blen, void* b) { + // unowned to prevent array copy + unowned uint8[] _a = (uint8[]) a; + _a.length = alen; + + unowned uint8[] _b = (uint8[]) b; + _b.length = blen; + + return LMS.utf8_collate_str (_a, _b); + } + + public Database (string db_path) throws DatabaseError { + Sqlite.Database.open (db_path, out this.db); + if (this.db.errcode () != Sqlite.OK) { + throw new DatabaseError.OPEN ("Failed to open '%s': %d", + db_path, + this.db.errcode); + } + + this.db.create_function ("contains", + 2, + Sqlite.UTF8, + null, + LMS.Database.utf8_contains, + null, + null); + + this.db.create_collation ("CASEFOLD", + Sqlite.UTF8, + LMS.Database.utf8_collate); + } + + public Statement prepare (string query_string) throws DatabaseError { + Statement statement; + + var err = this.db.prepare_v2 (query_string, -1, out statement); + if (err != Sqlite.OK) + throw new DatabaseError.PREPARE ("Unable to create statement '%s': %d", + query_string, + err); + return statement; + } + + + public Statement prepare_and_init (string query, + GLib.Value[]? arguments) + throws DatabaseError { + + Statement statement; + + var err = this.db.prepare_v2 (query, -1, out statement); + if (err != Sqlite.OK) + throw new DatabaseError.PREPARE ("Unable to create statement '%s': %d", + query, + err); + + for (var i = 1; i <= arguments.length; ++i) { + int sqlite_err; + unowned GLib.Value current_value = arguments[i - 1]; + + if (current_value.holds (typeof (int))) { + sqlite_err = statement.bind_int (i, current_value.get_int ()); + if (sqlite_err != Sqlite.OK) + throw new DatabaseError.BIND("Unable to bind value %d", + sqlite_err); + } else if (current_value.holds (typeof (int64))) { + sqlite_err = statement.bind_int64 (i, current_value.get_int64 ()); + if (sqlite_err != Sqlite.OK) + throw new DatabaseError.BIND("Unable to bind value %d", + sqlite_err); + } else if (current_value.holds (typeof (uint64))) { + sqlite_err = statement.bind_int64 (i, (int64) current_value.get_uint64 ()); + if (sqlite_err != Sqlite.OK) + throw new DatabaseError.BIND("Unable to bind value %d", + sqlite_err); + } else if (current_value.holds (typeof (long))) { + sqlite_err = statement.bind_int64 (i, current_value.get_long ()); + if (sqlite_err != Sqlite.OK) + throw new DatabaseError.BIND("Unable to bind value %d", + sqlite_err); + } else if (current_value.holds (typeof (uint))) { + sqlite_err = statement.bind_int64 (i, current_value.get_uint ()); + if (sqlite_err != Sqlite.OK) + throw new DatabaseError.BIND("Unable to bind value %d", + sqlite_err); + } else if (current_value.holds (typeof (string))) { + sqlite_err = statement.bind_text (i, current_value.get_string ()); + if (sqlite_err != Sqlite.OK) + throw new DatabaseError.BIND("Unable to bind value %d", + sqlite_err); + } else if (current_value.holds (typeof (void *))) { + if (current_value.peek_pointer () == null) { + sqlite_err = statement.bind_null (i); + if (sqlite_err != Sqlite.OK) + throw new DatabaseError.BIND("Unable to bind value %d", + sqlite_err); + } else { + assert_not_reached (); + } + } else { + var type = current_value.type (); + warning (_("Unsupported type %s"), type.name ()); + assert_not_reached (); + } + } + + return statement; + } + + public static void find_object(string id, Statement stmt) throws DatabaseError { + + (void) stmt.reset(); + + int integer_id = int.parse(id); + int sqlite_err = stmt.bind_int(1, integer_id); + if (sqlite_err != Sqlite.OK) + throw new DatabaseError.BIND("Unable to bind id %d", sqlite_err); + + sqlite_err = stmt.step(); + if (sqlite_err != Sqlite.ROW) + throw new DatabaseError.STEP("Unable to find id %s", id); + } + + public static void get_children_init (Statement stmt, + uint offset, uint max_count, string sort_criteria) throws DatabaseError { + + int sqlite_err; + + (void) stmt.reset(); + + sqlite_err = stmt.bind_int(1, (int) max_count); + if (sqlite_err != Sqlite.OK) + throw new DatabaseError.BIND("Unable to bind max_count %d", + sqlite_err); + + sqlite_err = stmt.bind_int(2, (int) offset); + if (sqlite_err != Sqlite.OK) + throw new DatabaseError.BIND("Unable to bind offset %d", + sqlite_err); + } + + public static bool get_children_step(Statement stmt) throws DatabaseError { + + bool retval; + int sqlite_err; + + sqlite_err = stmt.step(); + retval = sqlite_err == Sqlite.ROW; + + if (!retval && (sqlite_err != Sqlite.DONE)) + throw new DatabaseError.STEP("Error iterating through rows %d", + sqlite_err); + + return retval; + } +} |