@typeberry/lib 0.8.4 → 0.9.0-4a9f363

package/packages/jam/database-fjall/hybrid-states.js

@@ -0,0 +1,218 @@
+import { HashDictionary, SortedSet } from "#@typeberry/collections";
+import { InMemoryValueRefsStore, LeafDb, StateUpdateError, updateLeafs, ValueRefs, } from "#@typeberry/database";
+import { Logger } from "#@typeberry/logger";
+import { SerializedState, StateEntryUpdateAction, serializeStateUpdate, } from "#@typeberry/state-merkleization";
+import { leafComparator } from "#@typeberry/trie";
+import { OK, Result } from "#@typeberry/utils";
+import { FjallRoot, toUint8Array } from "./root.js";
+const logger = Logger.new(import.meta.filename, "db");
+/**
+ * One open fjall keyspace together with its content-addressed `values`
+ * partition.
+ *
+ * Opening the keyspace is the slow part, so the fuzz target opens one session
+ * per run and reuses it for every reset (see `HybridSerializedStates.fromSession`).
+ * The values partition is immutable - the key is the hash of the value - so it
+ * is fine that values pile up across resets, the unreferenced ones just sit
+ * there unused.
+ */
+export class FjallValuesSession {
+    root;
+    values;
+    constructor(root, 
+    /** Shared content-addressed values partition, reused across resets. */
+    values) {
+        this.root = root;
+        this.values = values;
+    }
+    /** Open (or create) the keyspace at `dbPath` and its `values` partition. */
+    static async open(dbPath, options = {}) {
+        if (options.readOnly === true) {
+            throw new Error("FjallValuesSession requires a writable keyspace.");
+        }
+        const root = await FjallRoot.open(dbPath, options);
+        try {
+            const values = await root.writablePartition("values");
+            return new FjallValuesSession(root, values);
+        }
+        catch (e) {
+            await root.close();
+            throw e;
+        }
+    }
+    /** Flush the journal to disk (a no-op for ephemeral keyspaces). */
+    async persist() {
+        await this.root.persist();
+    }
+    /** Size of the keyspace directory on disk, in bytes. */
+    sizeInBytes() {
+        return this.root.sizeInBytes();
+    }
+    /** Release the keyspace handle (skips the sync-all fsync when ephemeral). */
+    async close() {
+        await this.root.close();
+    }
+}
+/**
+ * Hybrid serialized-states db (fjall variant).
+ *
+ * States (leafs) are kept in memory, only the large values go to fjall on disk.
+ * Reads hit fjall directly, which keeps its own bounded block cache. Meant for
+ * long fuzzing, used together with pruning so the heap stays bounded.
+ *
+ * Construction is async, and value writes are flushed explicitly, because fjall
+ * has no transaction primitive.
+ *
+ * Values that no longer belong to any surviving state are removed from fjall,
+ * decided by in-memory refcounting (`ValueRefs`) driven by the importer's
+ * finality signal. Counts are not persisted: this db cannot resume from disk
+ * anyway (the leaf sets live in memory), so values left over by a previous run
+ * are never collected. An instance backed by a shared session (fuzz reset
+ * reuse) only ever prunes values it inserted itself, since its refcounts start
+ * empty - values left behind by earlier resets stay untouched.
+ */
+export class HybridSerializedStates {
+    spec;
+    blake2b;
+    session;
+    ownsSession;
+    static async new({ spec, blake2b, dbPath, ephemeral, cacheSizeBytes, }) {
+        const session = await FjallValuesSession.open(dbPath, { ephemeral, cacheSizeBytes });
+        // This instance owns the session it just opened, so its `close()` closes it.
+        return new HybridSerializedStates(spec, blake2b, session, true);
+    }
+    /**
+     * Wrap an already-open `FjallValuesSession` and reuse its keyspace.
+     *
+     * The new instance starts with its own empty in-memory leaf sets but shares
+     * the values partition on disk. Its `close()` does not close the session, the
+     * session owner closes it once. The fuzz target uses this to keep one keyspace
+     * across resets and only rebuild the in-memory state for each vector.
+     */
+    static fromSession(spec, blake2b, session) {
+        return new HybridSerializedStates(spec, blake2b, session, false);
+    }
+    inMemStates = HashDictionary.new();
+    // A single shared values accessor reused by every `LeafDb` we hand out.
+    valuesDb;
+    /** Shared content-addressed values partition (owned by `session`). */
+    values;
+    refsStore = new InMemoryValueRefsStore();
+    refs = new ValueRefs(this.refsStore);
+    // Queue of not-yet-committed value removals, awaited on close.
+    pendingCleanup = Promise.resolve();
+    constructor(spec, blake2b, session, 
+    /** Whether `close()` should close the underlying session. */
+    ownsSession) {
+        this.spec = spec;
+        this.blake2b = blake2b;
+        this.session = session;
+        this.ownsSession = ownsSession;
+        this.values = session.values;
+        this.valuesDb = { get: (key) => this.readValue(key) };
+    }
+    async insertInitialState(headerHash, entries) {
+        const { values, leafs } = updateLeafs(SortedSet.fromArray(leafComparator, []), this.blake2b, Array.from(entries, (x) => [StateEntryUpdateAction.Insert, x[0], x[1]]));
+        const res = await this.writeValues(values);
+        if (res.isError) {
+            return res;
+        }
+        this.inMemStates.set(headerHash, leafs);
+        this.applyRefs(this.refs.onInitial(values.map((v) => v[0])));
+        return Result.ok(OK);
+    }
+    async updateAndSetState(header, state, update) {
+        const updatedValues = serializeStateUpdate(this.spec, this.blake2b, update);
+        // Clone the leaf set before mutating: the previous state keeps using its own.
+        const newLeafs = SortedSet.fromSortedArray(leafComparator, state.backend.leafs.array);
+        const { values, removed, leafs } = updateLeafs(newLeafs, this.blake2b, updatedValues);
+        const res = await this.writeValues(values);
+        if (res.isError) {
+            // Leave the caller's state untouched: its new leaves would reference
+            // values that never reached disk.
+            return res;
+        }
+        // Re-create the lookup with the shared values accessor only once the new
+        // values are durably written.
+        state.updateBackend(LeafDb.fromLeaves(leafs, this.valuesDb));
+        this.inMemStates.set(header, leafs);
+        this.applyRefs(this.refs.onImport(header, { inserted: values.map((v) => v[0]), removed }));
+        return Result.ok(OK);
+    }
+    async getStateRoot(state) {
+        return state.backend.getStateRoot(this.blake2b);
+    }
+    getState(header) {
+        const leafs = this.inMemStates.get(header);
+        if (leafs === undefined) {
+            return null;
+        }
+        const leafDb = LeafDb.fromLeaves(leafs, this.valuesDb);
+        return SerializedState.new(this.spec, this.blake2b, leafDb);
+    }
+    commitFinalized(headers) {
+        this.applyRefs(this.refs.commitFinalized(headers));
+    }
+    markUnused(header) {
+        // Release the speculative references first (a no-op for finalized states,
+        // whose deltas were already consumed by `commitFinalized`).
+        this.applyRefs(this.refs.releaseUnfinalized(header));
+        this.inMemStates.delete(header);
+    }
+    diskSizeInBytes() {
+        return this.session.sizeInBytes();
+    }
+    /** Apply a refcounting update and remove values that lost their last reference. */
+    applyRefs(update) {
+        this.refsStore.apply(update);
+        if (update.removeValues.length === 0) {
+            return;
+        }
+        // Queued, not awaited: a failed removal only leaks a value. No explicit
+        // persist - removals are flushed together with the next value write.
+        this.pendingCleanup = this.pendingCleanup
+            .then(() => Promise.all(update.removeValues.map((v) => this.values.remove(v.raw))))
+            .catch((e) => {
+            logger.error `Failed to remove unreferenced values: ${e}`;
+        });
+    }
+    async close() {
+        // Finish any queued value removals before tearing down (or releasing) the
+        // session - they operate on the shared `values` partition.
+        await this.pendingCleanup;
+        // Instances backed by a shared session (fuzz reset reuse) keep the keyspace
+        // open for the next reset. The session owner closes it once.
+        if (this.ownsSession) {
+            await this.session.close();
+        }
+    }
+    /** Write new large values to fjall in a single batch, then flush. */
+    async writeValues(values) {
+        if (values.length === 0) {
+            return Result.ok(OK);
+        }
+        try {
+            // Flush queued removals first: a pending cleanup might be about to delete
+            // a content hash we are re-inserting now. Writing behind it keeps the
+            // re-inserted value on disk (removals are only queued at refcount 0, so
+            // none can be queued for a value still referenced).
+            await this.pendingCleanup;
+            const entries = values.map(([hash, val]) => ({ key: hash.raw, value: val.raw }));
+            await this.values.insertBatch(entries);
+            await this.session.persist();
+        }
+        catch (e) {
+            logger.error `${e}`;
+            return Result.error(StateUpdateError.Commit, () => `Failed to commit values: ${e}`);
+        }
+        return Result.ok(OK);
+    }
+    /** Read a value from fjall. */
+    readValue(key) {
+        const val = toUint8Array(this.values.get(key.raw));
+        if (val === null) {
+            throw new Error(`Missing value at key: ${key}`);
+        }
+        return val;
+    }
+}

package/packages/jam/database-fjall/hybrid-states.test.d.ts

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=hybrid-states.test.d.ts.map

package/packages/jam/database-fjall/hybrid-states.test.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"hybrid-states.test.d.ts","sourceRoot":"","sources":["../../../../../packages/jam/database-fjall/hybrid-states.test.ts"],"names":[],"mappings":""}

package/packages/jam/database-fjall/hybrid-states.test.js

@@ -0,0 +1,113 @@
+import assert from "node:assert";
+import * as fs from "node:fs";
+import { afterEach, before, beforeEach, describe, it } from "node:test";
+import { Bytes, BytesBlob } from "#@typeberry/bytes";
+import { tinyChainSpec } from "#@typeberry/config";
+import { Blake2b, HASH_SIZE } from "#@typeberry/hash";
+import { InMemoryState } from "#@typeberry/state";
+import { StateEntries } from "#@typeberry/state-merkleization";
+import { deepEqual, OK, Result } from "#@typeberry/utils";
+import { FjallValuesSession, HybridSerializedStates } from "./hybrid-states.js";
+let blake2b;
+before(async () => {
+    blake2b = await Blake2b.createHasher();
+});
+function createTempDir(suffix = "fjall-hybrid") {
+    return fs.mkdtempSync(`typeberry-${suffix}`);
+}
+describe("Fjall hybrid serialized states", () => {
+    const spec = tinyChainSpec;
+    const headerHash = Bytes.zero(HASH_SIZE).asOpaque();
+    let dbPath = "";
+    beforeEach(() => {
+        dbPath = createTempDir();
+    });
+    afterEach(() => {
+        fs.rmSync(dbPath, { recursive: true });
+    });
+    it("round-trips an initial state through the on-disk values store", async () => {
+        const states = await HybridSerializedStates.new({ spec, blake2b, dbPath });
+        try {
+            const empty = InMemoryState.empty(spec);
+            const serialized = StateEntries.serializeInMemory(spec, blake2b, empty);
+            const expectedRoot = serialized.getRootHash(blake2b);
+            const res = await states.insertInitialState(headerHash, serialized);
+            deepEqual(res, Result.ok(OK));
+            const state = states.getState(headerHash);
+            assert.ok(state !== null);
+            const stateRoot = await states.getStateRoot(state);
+            assert.strictEqual(`${stateRoot}`, `${expectedRoot}`);
+            deepEqual(InMemoryState.copyFrom(spec, state, new Map()), empty);
+        }
+        finally {
+            await states.close();
+        }
+    });
+    it("reads large values back from disk", async () => {
+        const states = await HybridSerializedStates.new({ spec, blake2b, dbPath });
+        try {
+            // > 32 bytes => stored in the values db (not embedded in the leaf).
+            const big1 = BytesBlob.blobFromString("x".repeat(100));
+            const big2 = BytesBlob.blobFromString("y".repeat(100));
+            const key1 = Bytes.fill(HASH_SIZE, 1).asOpaque();
+            const key2 = Bytes.fill(HASH_SIZE, 2).asOpaque();
+            const entries = StateEntries.fromEntriesUnsafe([
+                [key1, big1],
+                [key2, big2],
+            ]);
+            const res = await states.insertInitialState(headerHash, entries);
+            deepEqual(res, Result.ok(OK));
+            const state = states.getState(headerHash);
+            assert.ok(state !== null);
+            assert.strictEqual(`${state.backend.get(key2)}`, `${big2}`);
+            assert.strictEqual(`${state.backend.get(key1)}`, `${big1}`);
+        }
+        finally {
+            await states.close();
+        }
+    });
+    it("shares an open values session across resets without closing it", async () => {
+        // The fuzz reset path opens the values keyspace once per session and reuses
+        // it: each "reset" builds a fresh states instance sharing that session, and
+        // closing a session-backed states must NOT close the shared keyspace.
+        const session = await FjallValuesSession.open(dbPath);
+        try {
+            const big = BytesBlob.blobFromString("z".repeat(100));
+            const key = Bytes.fill(HASH_SIZE, 7).asOpaque();
+            const entries = StateEntries.fromEntriesUnsafe([[key, big]]);
+            // First "reset": write values through a states instance, then close it.
+            const first = HybridSerializedStates.fromSession(spec, blake2b, session);
+            const res = await first.insertInitialState(headerHash, entries);
+            deepEqual(res, Result.ok(OK));
+            await first.close();
+            // Second "reset": a fresh states sharing the same session. Its in-memory
+            // leaf set is independent (empty until it inserts)...
+            const second = HybridSerializedStates.fromSession(spec, blake2b, session);
+            assert.strictEqual(second.getState(headerHash), null);
+            // ...but the on-disk values store is the same one, still open and usable
+            // (a closed keyspace would throw here).
+            await second.insertInitialState(headerHash, entries);
+            const state = second.getState(headerHash);
+            assert.ok(state !== null);
+            assert.strictEqual(`${state.backend.get(key)}`, `${big}`);
+            await second.close();
+        }
+        finally {
+            await session.close();
+        }
+    });
+    it("drops the leaf set on markUnused while values stay on disk", async () => {
+        const states = await HybridSerializedStates.new({ spec, blake2b, dbPath });
+        try {
+            const empty = InMemoryState.empty(spec);
+            const serialized = StateEntries.serializeInMemory(spec, blake2b, empty);
+            await states.insertInitialState(headerHash, serialized);
+            assert.ok(states.getState(headerHash) !== null);
+            states.markUnused(headerHash);
+            assert.strictEqual(states.getState(headerHash), null);
+        }
+        finally {
+            await states.close();
+        }
+    });
+});

package/packages/jam/database-fjall/index.d.ts

@@ -0,0 +1,5 @@
+export * from "./blocks.js";
+export * from "./hybrid-states.js";
+export * from "./root.js";
+export * from "./states.js";
+//# sourceMappingURL=index.d.ts.map

package/packages/jam/database-fjall/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../../../packages/jam/database-fjall/index.ts"],"names":[],"mappings":"AAAA,cAAc,aAAa,CAAC;AAC5B,cAAc,oBAAoB,CAAC;AACnC,cAAc,WAAW,CAAC;AAC1B,cAAc,aAAa,CAAC"}

package/packages/jam/database-fjall/index.js

@@ -0,0 +1,4 @@
+export * from "./blocks.js";
+export * from "./hybrid-states.js";
+export * from "./root.js";
+export * from "./states.js";

package/packages/jam/database-fjall/root.d.ts

@@ -0,0 +1,64 @@
+import { type Partition, type ReadonlyPartition } from "@fjall-js/fjall";
+export type { Partition, ReadonlyPartition };
+export type FjallPartition = Partition | ReadonlyPartition;
+/**
+ * Normalize a value read from fjall (a Node `Buffer`) into a plain `Uint8Array`.
+ */
+export declare function toUint8Array(value: Buffer | null): Uint8Array | null;
+export type FjallRootOptions = {
+    /** Open a read-only wrapper surface over the shared keyspace. */
+    readOnly?: boolean;
+    /**
+     * When set, we skip explicit durability flushes (`persist`).
+     *
+     * Only safe for throwaway databases, like the fuzz target that wipes on every
+     * reset.
+     */
+    ephemeral?: boolean;
+    /**
+     * Cache size in bytes, shared by all partitions of the keyspace. fjall reads
+     * through this cache, so it bounds how much we keep in memory. When not set,
+     * typeberry uses a conservative default.
+     */
+    cacheSizeBytes?: number;
+};
+/**
+ * Thin wrapper over the fjall keyspace.
+ *
+ * fjall is an LSM-tree: it reads and writes through normal file i/o and keeps
+ * only a bounded block cache in memory, so the resident set stays bounded even
+ * when the store on disk is big.
+ */
+export declare class FjallRoot {
+    private readonly keyspace;
+    /** Path of the underlying keyspace directory, used to report on-disk usage. */
+    private readonly dbPath;
+    private readonly options;
+    private constructor();
+    /** Open (or create) a fjall keyspace at the given path. */
+    static open(dbPath: string, options?: FjallRootOptions): Promise<FjallRoot>;
+    /** Whether this root was opened through fjall's read-only surface. */
+    get readOnly(): boolean;
+    /** Open (or create) a partition under this keyspace. */
+    partition(name: string): Promise<FjallPartition>;
+    /** Open a writable partition, failing early when this root is read-only. */
+    writablePartition(name: string): Promise<Partition>;
+    /**
+     * Flush the journal to disk so prior writes survive a crash.
+     *
+     * Call after a logically-complete unit of work (one block, one state commit).
+     * A no-op for ephemeral databases.
+     */
+    persist(): Promise<void>;
+    /**
+     * Size of the keyspace directory on disk, in bytes.
+     *
+     * Returns `null` when the directory cannot be walked (e.g. not created yet).
+     * A fjall keyspace is a directory of partition and journal files, so we sum
+     * them recursively.
+     */
+    sizeInBytes(): number | null;
+    /** Release this keyspace handle. Call persist() first when durability is needed. */
+    close(): Promise<void>;
+}
+//# sourceMappingURL=root.d.ts.map

package/packages/jam/database-fjall/root.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"root.d.ts","sourceRoot":"","sources":["../../../../../packages/jam/database-fjall/root.ts"],"names":[],"mappings":"AAEA,OAAO,EAIL,KAAK,SAAS,EAEd,KAAK,iBAAiB,EACvB,MAAM,iBAAiB,CAAC;AAIzB,YAAY,EAAE,SAAS,EAAE,iBAAiB,EAAE,CAAC;AAC7C,MAAM,MAAM,cAAc,GAAG,SAAS,GAAG,iBAAiB,CAAC;AAE3D;;GAEG;AACH,wBAAgB,YAAY,CAAC,KAAK,EAAE,MAAM,GAAG,IAAI,GAAG,UAAU,GAAG,IAAI,CAKpE;AAED,MAAM,MAAM,gBAAgB,GAAG;IAC7B,iEAAiE;IACjE,QAAQ,CAAC,EAAE,OAAO,CAAC;IACnB;;;;;OAKG;IACH,SAAS,CAAC,EAAE,OAAO,CAAC;IACpB;;;;OAIG;IACH,cAAc,CAAC,EAAE,MAAM,CAAC;CACzB,CAAC;AAEF;;;;;;GAMG;AACH,qBAAa,SAAS;IAElB,OAAO,CAAC,QAAQ,CAAC,QAAQ;IACzB,+EAA+E;IAC/E,OAAO,CAAC,QAAQ,CAAC,MAAM;IACvB,OAAO,CAAC,QAAQ,CAAC,OAAO;IAJ1B,OAAO;IAOP,2DAA2D;WAC9C,IAAI,CAAC,MAAM,EAAE,MAAM,EAAE,OAAO,GAAE,gBAAqB,GAAG,OAAO,CAAC,SAAS,CAAC;IAWrF,sEAAsE;IACtE,IAAI,QAAQ,IAAI,OAAO,CAEtB;IAED,wDAAwD;IAClD,SAAS,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,cAAc,CAAC;IAItD,4EAA4E;IACtE,iBAAiB,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,SAAS,CAAC;IAOzD;;;;;OAKG;IACG,OAAO,IAAI,OAAO,CAAC,IAAI,CAAC;IAU9B;;;;;;OAMG;IACH,WAAW,IAAI,MAAM,GAAG,IAAI;IAQ5B,oFAAoF;IAC9E,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC;CAG7B"}

package/packages/jam/database-fjall/root.js

@@ -0,0 +1,105 @@
+import * as fs from "node:fs";
+import * as path from "node:path";
+import { open, openReadonly, } from "@fjall-js/fjall";
+const DEFAULT_CACHE_SIZE_BYTES = 64 * 1024 * 1024;
+/**
+ * Normalize a value read from fjall (a Node `Buffer`) into a plain `Uint8Array`.
+ */
+export function toUint8Array(value) {
+    if (value === null) {
+        return null;
+    }
+    return new Uint8Array(value.buffer, value.byteOffset, value.byteLength);
+}
+/**
+ * Thin wrapper over the fjall keyspace.
+ *
+ * fjall is an LSM-tree: it reads and writes through normal file i/o and keeps
+ * only a bounded block cache in memory, so the resident set stays bounded even
+ * when the store on disk is big.
+ */
+export class FjallRoot {
+    keyspace;
+    dbPath;
+    options;
+    constructor(keyspace, 
+    /** Path of the underlying keyspace directory, used to report on-disk usage. */
+    dbPath, options) {
+        this.keyspace = keyspace;
+        this.dbPath = dbPath;
+        this.options = options;
+    }
+    /** Open (or create) a fjall keyspace at the given path. */
+    static async open(dbPath, options = {}) {
+        // fjall-js 0.3 shares one engine per path. Readers must use the read-only
+        // wrapper surface; durability is driven explicitly through persist().
+        const config = {
+            path: dbPath,
+            cacheSizeBytes: options.cacheSizeBytes ?? DEFAULT_CACHE_SIZE_BYTES,
+        };
+        const keyspace = options.readOnly === true ? await openReadonly(config) : await open(config);
+        return new FjallRoot(keyspace, dbPath, options);
+    }
+    /** Whether this root was opened through fjall's read-only surface. */
+    get readOnly() {
+        return this.options.readOnly === true;
+    }
+    /** Open (or create) a partition under this keyspace. */
+    async partition(name) {
+        return this.keyspace.partition(name);
+    }
+    /** Open a writable partition, failing early when this root is read-only. */
+    async writablePartition(name) {
+        if (this.readOnly) {
+            throw new Error(`Cannot open writable fjall partition '${name}' from a read-only keyspace.`);
+        }
+        return (await this.keyspace.partition(name));
+    }
+    /**
+     * Flush the journal to disk so prior writes survive a crash.
+     *
+     * Call after a logically-complete unit of work (one block, one state commit).
+     * A no-op for ephemeral databases.
+     */
+    async persist() {
+        if (this.options.ephemeral === true) {
+            return;
+        }
+        if (this.readOnly) {
+            throw new Error("Cannot persist a read-only fjall keyspace.");
+        }
+        await this.keyspace.persist();
+    }
+    /**
+     * Size of the keyspace directory on disk, in bytes.
+     *
+     * Returns `null` when the directory cannot be walked (e.g. not created yet).
+     * A fjall keyspace is a directory of partition and journal files, so we sum
+     * them recursively.
+     */
+    sizeInBytes() {
+        try {
+            return dirSizeInBytes(this.dbPath);
+        }
+        catch {
+            return null;
+        }
+    }
+    /** Release this keyspace handle. Call persist() first when durability is needed. */
+    async close() {
+        await this.keyspace.close();
+    }
+}
+function dirSizeInBytes(dir) {
+    let total = 0;
+    for (const entry of fs.readdirSync(dir, { withFileTypes: true })) {
+        const full = path.join(dir, entry.name);
+        if (entry.isDirectory()) {
+            total += dirSizeInBytes(full);
+        }
+        else if (entry.isFile()) {
+            total += fs.statSync(full).size;
+        }
+    }
+    return total;
+}

package/packages/jam/database-fjall/states.d.ts

@@ -0,0 +1,31 @@
+import type { HeaderHash, StateRootHash } from "#@typeberry/block";
+import type { ChainSpec } from "#@typeberry/config";
+import { type InitStatesDb, LeafDb, type StatesDb, StateUpdateError } from "#@typeberry/database";
+import type { Blake2b } from "#@typeberry/hash";
+import type { ServicesUpdate, State } from "#@typeberry/state";
+import type { StateEntries } from "#@typeberry/state-merkleization";
+import { SerializedState } from "#@typeberry/state-merkleization";
+import { OK, Result } from "#@typeberry/utils";
+import { type FjallRoot } from "./root.js";
+/** fjall-backed full on-disk serialized state storage. */
+export declare class FjallStates implements StatesDb<SerializedState<LeafDb>>, InitStatesDb<StateEntries> {
+    private readonly spec;
+    private readonly blake2b;
+    private readonly root;
+    private readonly states;
+    private readonly values;
+    static open(spec: ChainSpec, blake2b: Blake2b, root: FjallRoot): Promise<FjallStates>;
+    private readonly valuesDb;
+    private constructor();
+    insertInitialState(headerHash: HeaderHash, entries: StateEntries): Promise<Result<OK, StateUpdateError>>;
+    updateAndSetState(headerHash: HeaderHash, state: SerializedState<LeafDb>, update: Partial<State & ServicesUpdate>): Promise<Result<OK, StateUpdateError>>;
+    getStateRoot(state: SerializedState<LeafDb>): Promise<StateRootHash>;
+    getState(headerHash: HeaderHash): SerializedState<LeafDb> | null;
+    commitFinalized(_headers: HeaderHash[]): void;
+    markUnused(headerHash: HeaderHash): void;
+    diskSizeInBytes(): number | null;
+    close(): Promise<void>;
+    private updateAndCommit;
+    private readValue;
+}
+//# sourceMappingURL=states.d.ts.map

package/packages/jam/database-fjall/states.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"states.d.ts","sourceRoot":"","sources":["../../../../../packages/jam/database-fjall/states.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,UAAU,EAAE,aAAa,EAAE,MAAM,kBAAkB,CAAC;AAGlE,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,mBAAmB,CAAC;AACnD,OAAO,EACL,KAAK,YAAY,EACjB,MAAM,EACN,KAAK,QAAQ,EACb,gBAAgB,EAGjB,MAAM,qBAAqB,CAAC;AAC7B,OAAO,KAAK,EAAE,OAAO,EAAiB,MAAM,iBAAiB,CAAC;AAE9D,OAAO,KAAK,EAAE,cAAc,EAAE,KAAK,EAAE,MAAM,kBAAkB,CAAC;AAC9D,OAAO,KAAK,EAAE,YAAY,EAAY,MAAM,gCAAgC,CAAC;AAC7E,OAAO,EAAE,eAAe,EAAgD,MAAM,gCAAgC,CAAC;AAE/G,OAAO,EAAE,EAAE,EAAE,MAAM,EAAkB,MAAM,kBAAkB,CAAC;AAC9D,OAAO,EAAuB,KAAK,SAAS,EAAgC,MAAM,WAAW,CAAC;AAI9F,0DAA0D;AAC1D,qBAAa,WAAY,YAAW,QAAQ,CAAC,eAAe,CAAC,MAAM,CAAC,CAAC,EAAE,YAAY,CAAC,YAAY,CAAC;IAS7F,OAAO,CAAC,QAAQ,CAAC,IAAI;IACrB,OAAO,CAAC,QAAQ,CAAC,OAAO;IACxB,OAAO,CAAC,QAAQ,CAAC,IAAI;IACrB,OAAO,CAAC,QAAQ,CAAC,MAAM;IACvB,OAAO,CAAC,QAAQ,CAAC,MAAM;WAZZ,IAAI,CAAC,IAAI,EAAE,SAAS,EAAE,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,SAAS,GAAG,OAAO,CAAC,WAAW,CAAC;IAK3F,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAAW;IAEpC,OAAO;IAUD,kBAAkB,CAAC,UAAU,EAAE,UAAU,EAAE,OAAO,EAAE,YAAY,GAAG,OAAO,CAAC,MAAM,CAAC,EAAE,EAAE,gBAAgB,CAAC,CAAC;IAQxG,iBAAiB,CACrB,UAAU,EAAE,UAAU,EACtB,KAAK,EAAE,eAAe,CAAC,MAAM,CAAC,EAC9B,MAAM,EAAE,OAAO,CAAC,KAAK,GAAG,cAAc,CAAC,GACtC,OAAO,CAAC,MAAM,CAAC,EAAE,EAAE,gBAAgB,CAAC,CAAC;IAUlC,YAAY,CAAC,KAAK,EAAE,eAAe,CAAC,MAAM,CAAC,GAAG,OAAO,CAAC,aAAa,CAAC;IAI1E,QAAQ,CAAC,UAAU,EAAE,UAAU,GAAG,eAAe,CAAC,MAAM,CAAC,GAAG,IAAI;IAahE,eAAe,CAAC,QAAQ,EAAE,UAAU,EAAE,GAAG,IAAI;IAK7C,UAAU,CAAC,UAAU,EAAE,UAAU,GAAG,IAAI;IAMxC,eAAe,IAAI,MAAM,GAAG,IAAI;IAI1B,KAAK;YAEG,eAAe;IAyB7B,OAAO,CAAC,SAAS;CAOlB"}

package/packages/jam/database-fjall/states.js

@@ -0,0 +1,99 @@
+import { BytesBlob } from "#@typeberry/bytes";
+import { SortedSet } from "#@typeberry/collections";
+import { LeafDb, StateUpdateError, updateLeafs, } from "#@typeberry/database";
+import { Logger } from "#@typeberry/logger";
+import { SerializedState, StateEntryUpdateAction, serializeStateUpdate } from "#@typeberry/state-merkleization";
+import { leafComparator } from "#@typeberry/trie";
+import { OK, Result, resultToString } from "#@typeberry/utils";
+import { toUint8Array } from "./root.js";
+const logger = Logger.new(import.meta.filename, "db");
+/** fjall-backed full on-disk serialized state storage. */
+export class FjallStates {
+    spec;
+    blake2b;
+    root;
+    states;
+    values;
+    static async open(spec, blake2b, root) {
+        const [states, values] = await Promise.all([root.partition("states"), root.partition("values")]);
+        return new FjallStates(spec, blake2b, root, states, values);
+    }
+    valuesDb;
+    constructor(spec, blake2b, root, states, values) {
+        this.spec = spec;
+        this.blake2b = blake2b;
+        this.root = root;
+        this.states = states;
+        this.values = values;
+        this.valuesDb = { get: (key) => this.readValue(key) };
+    }
+    async insertInitialState(headerHash, entries) {
+        return await this.updateAndCommit(headerHash, SortedSet.fromArray(leafComparator, []), Array.from(entries, (x) => [StateEntryUpdateAction.Insert, x[0], x[1]]));
+    }
+    async updateAndSetState(headerHash, state, update) {
+        const updatedValues = serializeStateUpdate(this.spec, this.blake2b, update);
+        const newLeafs = SortedSet.fromSortedArray(leafComparator, state.backend.leafs.array);
+        const res = await this.updateAndCommit(headerHash, newLeafs, updatedValues);
+        if (res.isOk) {
+            state.updateBackend(LeafDb.fromLeaves(newLeafs, this.valuesDb));
+        }
+        return res;
+    }
+    async getStateRoot(state) {
+        return state.backend.getStateRoot(this.blake2b);
+    }
+    getState(headerHash) {
+        const leafNodes = toUint8Array(this.states.get(headerHash.raw));
+        if (leafNodes === null) {
+            return null;
+        }
+        const leafDbResult = LeafDb.fromLeavesBlob(BytesBlob.blobFrom(leafNodes), this.valuesDb);
+        if (leafDbResult.isError) {
+            throw new Error(`Inconsistent DB. Invalid leaf nodes for ${headerHash}: ${resultToString(leafDbResult)}`);
+        }
+        return SerializedState.new(this.spec, this.blake2b, leafDbResult.ok);
+    }
+    commitFinalized(_headers) {
+        // Values are never pruned here. This db survives restarts, so refcounting
+        // would need counts persisted (and crash-consistent) alongside the values.
+    }
+    markUnused(headerHash) {
+        void writable(this.states, this.root)
+            .remove(headerHash.raw)
+            .catch((e) => logger.warn `Failed to prune state ${headerHash}: ${e}`);
+    }
+    diskSizeInBytes() {
+        return this.root.sizeInBytes();
+    }
+    async close() { }
+    async updateAndCommit(headerHash, leafs, data) {
+        const { values } = updateLeafs(leafs, this.blake2b, data);
+        const stateLeafs = BytesBlob.blobFromParts(leafs.array.map((x) => x.node.raw));
+        try {
+            // Preserve dependency order: values first, then leaves that may reference them.
+            if (values.length > 0) {
+                await writable(this.values, this.root).insertBatch(values.map(([hash, val]) => ({ key: hash.raw, value: val.raw })));
+            }
+            await writable(this.states, this.root).insert(headerHash.raw, stateLeafs.raw);
+            await this.root.persist();
+        }
+        catch (e) {
+            logger.error `${e}`;
+            return Result.error(StateUpdateError.Commit, () => `Failed to commit state update: ${e}`);
+        }
+        return Result.ok(OK);
+    }
+    readValue(key) {
+        const val = toUint8Array(this.values.get(key.raw));
+        if (val === null) {
+            throw new Error(`Missing required value: ${key} in the DB`);
+        }
+        return val;
+    }
+}
+function writable(partition, root) {
+    if (root.readOnly) {
+        throw new Error("Cannot write through a read-only fjall partition.");
+    }
+    return partition;
+}

package/packages/jam/database-fjall/states.test.d.ts

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=states.test.d.ts.map

package/packages/jam/database-fjall/states.test.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"states.test.d.ts","sourceRoot":"","sources":["../../../../../packages/jam/database-fjall/states.test.ts"],"names":[],"mappings":""}