@cero-base/core 0.2.0 → 0.4.1

package/src/database/dispatch.js

@@ -0,0 +1,156 @@
+import { SINGLE, COLLECTION, ACTION, COUNTERS } from '../lib/constants.js'
+import { toId, toKey } from '../lib/utils.js'
+import { Identity } from '../identity/index.js'
+
+// Built-in collections wired into every cero spec.
+export const BUILTINS = [
+  { name: 'members', verb: 'member' },
+  { name: 'devices', verb: 'device' },
+  { name: 'invites', verb: 'invite' },
+  { name: 'handles', verb: 'handle' }
+]
+
+export function makeDispatcher(spec, ns, routes) {
+  const dispatcher = new spec.dispatch.Router()
+
+  const countersCol = `@${ns}/${COUNTERS}`
+  const member = (view, id) => view.get(`@${ns}/members`, { id })
+  const device = (view, id) => view.get(`@${ns}/devices`, { id })
+
+  async function insert(view, name, col, op) {
+    if (name !== COUNTERS) {
+      const existing = op.id != null ? await view.get(col, op) : null
+      if (existing && existing.index != null) {
+        op.index = existing.index
+      } else {
+        const counter = (await view.get(countersCol, { name })) ?? { name, value: 0 }
+        op.index = counter.value + 1
+        await view.insert(countersCol, { name, value: op.index })
+      }
+    }
+    await view.insert(col, op)
+  }
+
+  const upsert = (name, col, kind) => async (op, ctx) => {
+    const d = ctx.key && (await device(ctx.view, toId(ctx.key)))
+    op.memberId = d?.memberId || null
+    if (kind === COLLECTION) await insert(ctx.view, name, col, op)
+    else await ctx.view.insert(col, op)
+  }
+  const remove = (col) => async (op, ctx) => ctx.view.delete(col, op)
+  const add = (verb, fn) => dispatcher.add(`@${ns}/${verb}`, fn)
+
+  add('add-writer', async (op, ctx) => {
+    if (!Identity.verify(op.master, op.writer, op.sig)) return
+    await ctx.host.addWriter(op.writer, { isIndexer: op.isIndexer !== false })
+    const ts = Date.now()
+    await insert(ctx.view, 'devices', `@${ns}/devices`, {
+      id: toId(op.writer),
+      memberId: toId(op.master),
+      name: op.name || null,
+      isMobile: op.isMobile === true,
+      createdAt: ts,
+      updatedAt: ts
+    })
+  })
+
+  add('del-writer', async (op, ctx) => {
+    if (!Identity.verify(op.master, op.writer, op.sig)) return
+    await ctx.host.removeWriter(op.writer)
+    await ctx.view.delete(`@${ns}/devices`, { id: toId(op.writer) })
+  })
+
+  add('claim-writer', async (op, ctx) => {
+    if (!Identity.verify(op.identity, op.writer, op.sig)) return
+    const memberId = toId(op.identity)
+    if (!(await member(ctx.view, memberId))) return
+    await ctx.host.addWriter(op.writer, { isIndexer: true })
+    const ts = Date.now()
+    await insert(ctx.view, 'devices', `@${ns}/devices`, {
+      id: toId(op.writer),
+      memberId,
+      name: op.name || null,
+      isMobile: op.isMobile === true,
+      createdAt: ts,
+      updatedAt: ts
+    })
+  })
+
+  add('add-member', async (op, ctx) => {
+    await insert(ctx.view, 'members', `@${ns}/members`, op)
+    const deviceId = toId(op.key)
+    const existingDevice = await device(ctx.view, deviceId)
+    const ts = Date.now()
+    await insert(ctx.view, 'devices', `@${ns}/devices`, {
+      id: deviceId,
+      memberId: op.id,
+      name: existingDevice?.name ?? null,
+      isMobile: existingDevice?.isMobile ?? false,
+      createdAt: existingDevice?.createdAt ?? ts,
+      updatedAt: ts
+    })
+  })
+
+  add('set-member', async (op, ctx) => {
+    const existing = await member(ctx.view, op.id)
+    if (!existing) return
+    await insert(ctx.view, 'members', `@${ns}/members`, { ...existing, ...op })
+  })
+
+  add('del-member', async (op, ctx) => {
+    const existing = await member(ctx.view, op.id)
+    if (!existing) return
+    if (existing.key) await ctx.host.removeWriter(existing.key)
+    await ctx.view.delete(`@${ns}/members`, op)
+  })
+
+  add('del-device', async (op, ctx) => {
+    const existing = await device(ctx.view, op.id)
+    if (!existing) return
+    await ctx.host.removeWriter(toKey(existing.id))
+    await ctx.view.delete(`@${ns}/devices`, op)
+  })
+
+  for (const b of BUILTINS) {
+    const col = `@${ns}/${b.name}`
+    if (b.verb === 'member') continue
+    if (b.verb === 'device') {
+      add(`add-${b.verb}`, async (op, ctx) => {
+        await insert(ctx.view, b.name, col, { ...op, memberId: op.memberId ?? null })
+      })
+      add(`set-${b.verb}`, async (op, ctx) => {
+        const existing = await device(ctx.view, op.id)
+        await insert(ctx.view, b.name, col, { ...op, memberId: existing?.memberId ?? null })
+      })
+      continue
+    }
+    add(`add-${b.verb}`, upsert(b.name, col, COLLECTION))
+    add(`set-${b.verb}`, upsert(b.name, col, COLLECTION))
+    add(`del-${b.verb}`, remove(col))
+  }
+
+  for (const [name, info] of Object.entries(spec.meta?.refs || {})) {
+    if (info.builtin) continue
+    if (info.kind === 'handle') continue
+    if (info.kind === ACTION) {
+      add(name, routes[name] || (async () => {}))
+      continue
+    }
+    const col = `@${ns}/${name}`
+    const kind = info.kind === SINGLE ? SINGLE : COLLECTION
+    add(`set-${name}`, upsert(name, col, kind))
+    if (info.kind === SINGLE) continue
+    add(`add-${name}`, upsert(name, col, COLLECTION))
+    add(`del-${name}`, remove(col))
+  }
+
+  return {
+    dispatcher,
+    async apply(nodes, view, host) {
+      for (const node of nodes) {
+        await dispatcher.dispatch(node.value, { view, host, key: node.key })
+      }
+      await view.flush()
+    }
+  }
+}