import {Db, encodeKey} from "@lodestar/db";
import {Root, Slot} from "@lodestar/types";
import {intToBytes} from "@lodestar/utils";
import {Bucket, getBucketNameByValue} from "../buckets.js";

const bucketId = getBucketNameByValue(Bucket.index_stateArchiveRootIndex);

export function getRootIndex(db: Db, stateRoot: Root): Promise<Uint8Array | null> {
  return db.get(getRootIndexKey(stateRoot), {bucketId});
}

export function storeRootIndex(db: Db, slot: Slot, stateRoot: Root): Promise<void> {
  return db.put(getRootIndexKey(stateRoot), intToBytes(slot, 8, "be"), {bucketId});
}

export function getRootIndexKey(root: Root): Uint8Array {
  return encodeKey(Bucket.index_stateArchiveRootIndex, root);
}