Fix #10 - Added both create_function and close

Author: WebReflection

README.md

@@ -152,6 +152,8 @@ The API in a nutshell is:
   * **query**: a template literal tag to simply query the database (no result returned)
   * **raw**: a template literal tag to represent static parts of the query (not values)
   * **transaction**: a function that returns a template literal tag to perform any statement until `tag.commit()` is awaited and executed.
+  * **close**: a function to force save and close the db at any time.
+  * **create_function**: a function to register custom SQLite functions. **Please note** when used as *Worker* it requires a **pure function** (no outer scope access) and **Function** evaluation. No limitations when used directly through init.
 
 All tags, except the `raw` helper, are *asynchronous*, so that it's possible to *await* their result.
 

cjs/init.js

@@ -40,20 +40,18 @@ const init = (options = {}) => new Promise((resolve, onerror) => {
             let queue = Promise.resolve();
             const {result} = this;
             const db = new Database(result || options.database || new Uint8Array(0));
-            const save = () => {
-              queue = queue.then(() => new Promise((resolve, onerror) => {
-                const uint8array = db.export();
-                assign(store('readwrite').put(uint8array, keyPath).transaction, {
-                  oncomplete() {
-                    resolve();
-                    if (options.update)
-                      options.update(uint8array);
-                  },
-                  onabort: onerror,
-                  onerror
-                });
-              }));
-            };
+            const save = () => (queue = queue.then(() => new Promise((resolve, onerror) => {
+              const uint8array = db.export();
+              assign(store('readwrite').put(uint8array, keyPath).transaction, {
+                oncomplete() {
+                  resolve();
+                  if (options.update)
+                    options.update(uint8array);
+                },
+                onabort: onerror,
+                onerror
+              });
+            })));
             if (!result)
               save();
             const {all, get, query, raw, transaction} = SQLiteTag({
@@ -91,6 +89,11 @@ const init = (options = {}) => new Promise((resolve, onerror) => {
             let t = 0;
             resolve({
               all, get, raw, transaction,
+              create_function: (name, func) => db.create_function(name, func),
+              close: () => {
+                clearTimeout(t);
+                return save().then(() => db.close());
+              },
               query(template) {
                 if (/\b(?:INSERT|DELETE|UPDATE)\b/i.test(template[0])) {
                   clearTimeout(t);

cjs/sqlite-worker.js

@@ -57,6 +57,11 @@ function SQLiteWorker(options = {}) {
         }}
       );
     },
+    close: () => post('close'),
+    create_function: (name, func) => post(
+      'create_function',
+      [name, func.toString()]
+    ),
     all: query('all'),
     get: query('get'),
     query: q,

cjs/worker.js

@@ -12,6 +12,28 @@ addEventListener('message', ({data: {id, action, options}}) => {
       ({message: error}) => postMessage({id, error})
     );
   }
+  else if (action === 'close') {
+    db.then(module => module.close().then(
+      () => {
+        postMessage({id});
+      },
+      ({message: error}) => {
+        postMessage({id, error});
+      }
+    ));
+  }
+  else if (action === 'create_function') {
+    db.then(module => {
+      try {
+        const [name, func] = options;
+        const result = module[action](name, Function('return ' + func)());
+        postMessage({id, result});
+      }
+      catch({message: error}) {
+        postMessage({id, error});
+      }
+    });
+  }
   // action === `all` || `get` || `query`
   else {
     const {template, values} = options;