Add support for a close method for views (holepunchto#54)

index.js

@@ -224,6 +224,7 @@ module.exports = class Autobase extends ReadyResource {
 
     this._hasApply = !!this._handlers.apply
     this._hasOpen = !!this._handlers.open
+    this._hasClose = !!this._handlers.close
 
     this._viewStore = new LinearizedStore(this)
 
@@ -259,6 +260,7 @@ module.exports = class Autobase extends ReadyResource {
   }
 
   async _close () {
+    if (this._hasClose) await this._handlers.close(this.view)
     await this.store.close()
   }
 

test/basic.js

@@ -53,6 +53,35 @@ test('basic - view', async t => {
   t.alike(await base.view.get(0), block)
 })
 
+test('basic - view with close', async t => {
+  const [base] = await create(1, apply, open, close)
+
+  const block = { message: 'hello, world!' }
+  await base.append(block)
+
+  t.is(base.system.digest.writers.length, 1)
+  t.is(base.view.core.indexedLength, 1)
+  t.alike(await base.view.core.get(0), block)
+
+  t.is(base.view.lastBlock, null)
+  await base.close()
+  t.is(base.view.lastBlock.message, 'hello, world!')
+
+  function open (store) {
+    const core = store.get('test', { valueEncoding: 'json' })
+    return {
+      core,
+      lastBlock: null,
+      append: v => core.append(v)
+    }
+  }
+
+  async function close (view) {
+    view.lastBlock = await view.core.get(view.core.length - 1)
+    await view.core.close()
+  }
+})
+
 test('basic - compare views', async t => {
   const bases = await create(2, apply, store => store.get('test', { valueEncoding: 'json' }))
 

test/helpers/index.js

@@ -12,8 +12,8 @@ module.exports = {
   compare
 }
 
-async function create (n, apply, open) {
-  const opts = { apply, open, valueEncoding: 'json' }
+async function create (n, apply, open, close) {
+  const opts = { apply, open, close, valueEncoding: 'json' }
   const bases = [new Autobase(new Corestore(ram, { primaryKey: Buffer.alloc(32).fill(0) }), null, opts)]
   await bases[0].ready()
   if (n === 1) return bases