level.ts

import { MemoryLevel } from 'memory-level'

import type { BatchDBOp, DB } from '@ethereumjs/util'
import type { AbstractLevel } from 'abstract-level'

const ENCODING_OPTS = { keyEncoding: 'view', valueEncoding: 'view' }

export class LevelDB implements DB {
  readonly _leveldb: AbstractLevel<
    string | Uint8Array | Uint8Array,
    string | Uint8Array,
    string | Uint8Array
  >

  constructor(
    leveldb?: AbstractLevel<
      string | Uint8Array | Uint8Array,
      string | Uint8Array,
      string | Uint8Array
    > | null
  ) {
    this._leveldb = leveldb ?? new MemoryLevel(ENCODING_OPTS)
  }

  async get(key: Uint8Array): Promise<Uint8Array | undefined> {
    let value
    try {
      value = await this._leveldb.get(key, ENCODING_OPTS)
    } catch (error: any) {
      // https://github.com/Level/abstract-level/blob/915ad1317694d0ce8c580b5ab85d81e1e78a3137/abstract-level.js#L309
      // This should be `true` if the error came from LevelDB
      // so we can check for `NOT true` to identify any non-404 errors
      if (error.notFound !== true) {
        throw error
      }
    }
    return value
  }

  async put(key: Uint8Array, val: Uint8Array): Promise<void> {
    await this._leveldb.put(key, val, ENCODING_OPTS)
  }

  async del(key: Uint8Array): Promise<void> {
    await this._leveldb.del(key, ENCODING_OPTS)
  }

  async batch(opStack: BatchDBOp[]): Promise<void> {
    await this._leveldb.batch(opStack, ENCODING_OPTS)
  }

  shallowCopy(): DB {
    return new LevelDB(this._leveldb)
  }
}