Implement Database.pragma()

Author: penberg

index.js

@@ -134,7 +134,12 @@ class Database {
   }
 
   pragma(source, options) {
-    throw new Error("not implemented");
+    if (options == null) options = {};
+    if (typeof source !== 'string') throw new TypeError('Expected first argument to be a string');
+    if (typeof options !== 'object') throw new TypeError('Expected second argument to be an options object');
+    const simple = options['simple'];
+    const stmt = this.prepare(`PRAGMA ${source}`, this, true);
+    return simple ? stmt.pluck().get() : stmt.all();
   }
 
   backup(filename, options) {

integration-tests/tests/async.test.js

@@ -236,6 +236,12 @@ test.serial("Database.transaction()", async (t) => {
   t.is(stmt.get(5).name, "Junior");
 });
 
+test.serial("Database.pragma()", async (t) => {
+  const db = t.context.db;
+  await db.pragma("cache_size = 2000");
+  t.deepEqual(await db.pragma("cache_size"), [{ "cache_size": 2000 }]);
+});
+
 test.serial("errors", async (t) => {
   const db = t.context.db;
 

integration-tests/tests/sync.test.js

@@ -251,6 +251,12 @@ test.serial("values", async (t) => {
   t.deepEqual(stmt.get(9007199254740991n), [9007199254740991]);
 });
 
+test.serial("Database.pragma()", async (t) => {
+  const db = t.context.db;
+  db.pragma("cache_size = 2000");
+  t.deepEqual(db.pragma("cache_size"), [{ "cache_size": 2000 }]);
+});
+
 test.serial("errors", async (t) => {
   const db = t.context.db;
 

promise.js

@@ -132,7 +132,13 @@ class Database {
   }
 
   pragma(source, options) {
-    throw new Error("not implemented");
+    if (options == null) options = {};
+    if (typeof source !== 'string') throw new TypeError('Expected first argument to be a string');
+    if (typeof options !== 'object') throw new TypeError('Expected second argument to be an options object');
+    const simple = options['simple'];
+    return this.prepare(`PRAGMA ${source}`, this, true).then(async (stmt) => {
+      return simple ? await stmt.pluck().get() : await stmt.all();
+    });
   }
 
   backup(filename, options) {