@atomiqlabs/chain-starknet 1.0.0-beta.0

package/src/starknet/btcrelay/StarknetBtcRelay.ts

@@ -0,0 +1,415 @@
+import {Buffer} from "buffer";
+import {StarknetBtcHeader} from "./headers/StarknetBtcHeader";
+import {BigIntBufferUtils, BitcoinRpc, BtcBlock, BtcRelay, StatePredictorUtils} from "@atomiqlabs/base";
+import {
+    bigNumberishToBuffer,
+    bufferToU32Array,
+    toHex,
+    u32ReverseEndianness
+} from "../../utils/Utils";
+import {StarknetContractBase} from "../contract/StarknetContractBase";
+import {StarknetBtcStoredHeader} from "./headers/StarknetBtcStoredHeader";
+import {StarknetTx} from "../base/modules/StarknetTransactions";
+import {StarknetSigner} from "../wallet/StarknetSigner";
+import {BtcRelayAbi} from "./BtcRelayAbi";
+import {BigNumberish, constants, hash, Provider} from "starknet";
+import {StarknetFees} from "../base/modules/StarknetFees";
+import {StarknetRetryPolicy} from "../base/StarknetBase";
+import {StarknetAction} from "../base/StarknetAction";
+
+function serializeBlockHeader(e: BtcBlock): StarknetBtcHeader {
+    return new StarknetBtcHeader({
+        reversed_version: u32ReverseEndianness(e.getVersion()),
+        previous_blockhash: bufferToU32Array(Buffer.from(e.getPrevBlockhash(), "hex").reverse()),
+        merkle_root: bufferToU32Array(Buffer.from(e.getMerkleRoot(), "hex").reverse()),
+        reversed_timestamp: u32ReverseEndianness(e.getTimestamp()),
+        nbits: u32ReverseEndianness(e.getNbits()),
+        nonce: u32ReverseEndianness(e.getNonce()),
+        hash: Buffer.from(e.getHash(), "hex").reverse()
+    });
+}
+
+const GAS_PER_BLOCKHEADER = 750;
+const GAS_PER_BLOCKHEADER_FORK = 750;
+
+const btcRelayAddreses = {
+    [constants.StarknetChainId.SN_SEPOLIA]: "0x032afcea912ba13f6a1878fe38af23eaec3e6b4c7db31a3571550d3cf80d3e31",
+    [constants.StarknetChainId.SN_MAIN]: "0x05cc69b09e8c53520f9e328f6eca72cf02fe46ce290b757d42414e2238001603"
+};
+
+function serializeCalldata(headers: StarknetBtcHeader[], storedHeader: StarknetBtcStoredHeader, span: BigNumberish[]) {
+    span.push(toHex(headers.length));
+    headers.forEach(header => {
+        span.push(...header.serialize());
+    });
+    span.push(...storedHeader.serialize());
+    return span;
+}
+
+export class StarknetBtcRelay<B extends BtcBlock>
+    extends StarknetContractBase<typeof BtcRelayAbi>
+    implements BtcRelay<StarknetBtcStoredHeader, StarknetTx, B, StarknetSigner> {
+
+    public SaveMainHeaders(signer: string, mainHeaders: StarknetBtcHeader[], storedHeader: StarknetBtcStoredHeader): StarknetAction {
+
+        return new StarknetAction(signer, this,
+            {
+                contractAddress: this.contract.address,
+                entrypoint: "submit_main_blockheaders",
+                calldata: serializeCalldata(mainHeaders, storedHeader, [])
+            },
+            {l1: GAS_PER_BLOCKHEADER * mainHeaders.length, l2: 0}
+        )
+    }
+
+    public SaveShortForkHeaders(signer: string, forkHeaders: StarknetBtcHeader[], storedHeader: StarknetBtcStoredHeader): StarknetAction {
+        return new StarknetAction(signer, this,
+            {
+                contractAddress: this.contract.address,
+                entrypoint: "submit_short_fork_blockheaders",
+                calldata: serializeCalldata(forkHeaders, storedHeader, [])
+            },
+            {l1: GAS_PER_BLOCKHEADER * forkHeaders.length, l2: 0}
+        )
+    }
+
+    public SaveLongForkHeaders(signer: string, forkId: number, forkHeaders: StarknetBtcHeader[], storedHeader: StarknetBtcStoredHeader, totalForkHeaders: number = 100): StarknetAction {
+        return new StarknetAction(signer, this,
+            {
+                contractAddress: this.contract.address,
+                entrypoint: "submit_fork_blockheaders",
+                calldata: serializeCalldata(forkHeaders, storedHeader, [toHex(forkId)])
+            },
+            {l1: (GAS_PER_BLOCKHEADER * forkHeaders.length) + (GAS_PER_BLOCKHEADER_FORK * totalForkHeaders), l2: 0}
+        )
+    }
+
+    bitcoinRpc: BitcoinRpc<B>;
+
+    readonly maxHeadersPerTx: number = 100;
+    readonly maxForkHeadersPerTx: number = 100;
+    readonly maxShortForkHeadersPerTx: number = 100;
+
+    constructor(
+        chainId: constants.StarknetChainId,
+        provider: Provider,
+        bitcoinRpc: BitcoinRpc<B>,
+        contractAddress: string = btcRelayAddreses[chainId],
+        retryPolicy?: StarknetRetryPolicy,
+        solanaFeeEstimator: StarknetFees = new StarknetFees(provider)
+    ) {
+        super(chainId, provider, contractAddress, BtcRelayAbi, retryPolicy, solanaFeeEstimator);
+        this.bitcoinRpc = bitcoinRpc;
+    }
+
+    /**
+     * Computes subsequent commited headers as they will appear on the blockchain when transactions
+     *  are submitted & confirmed
+     *
+     * @param initialStoredHeader
+     * @param syncedHeaders
+     * @private
+     */
+    private computeCommitedHeaders(initialStoredHeader: StarknetBtcStoredHeader, syncedHeaders: StarknetBtcHeader[]) {
+        const computedCommitedHeaders = [initialStoredHeader];
+        for(let blockHeader of syncedHeaders) {
+            computedCommitedHeaders.push(computedCommitedHeaders[computedCommitedHeaders.length-1].computeNext(blockHeader));
+        }
+        return computedCommitedHeaders;
+    }
+
+    /**
+     * A common logic for submitting blockheaders in a transaction
+     *
+     * @param signer
+     * @param headers headers to sync to the btc relay
+     * @param storedHeader current latest stored block header for a given fork
+     * @param tipWork work of the current tip in a given fork
+     * @param forkId forkId to submit to, forkId=0 means main chain, forkId=-1 means short fork
+     * @param feeRate feeRate for the transaction
+     * @private
+     */
+    private async _saveHeaders(
+        signer: string,
+        headers: BtcBlock[],
+        storedHeader: StarknetBtcStoredHeader,
+        tipWork: Buffer,
+        forkId: number,
+        feeRate: string
+    ) {
+        const blockHeaderObj = headers.map(serializeBlockHeader);
+        let starknetAction: StarknetAction;
+        switch(forkId) {
+            case -1:
+                starknetAction = this.SaveShortForkHeaders(signer, blockHeaderObj, storedHeader);
+                break;
+            case 0:
+                starknetAction = this.SaveMainHeaders(signer, blockHeaderObj, storedHeader);
+                break;
+            default:
+                starknetAction = this.SaveLongForkHeaders(signer, forkId, blockHeaderObj, storedHeader);
+                break;
+        }
+
+        const tx = await starknetAction.tx(feeRate);
+
+        const computedCommitedHeaders = this.computeCommitedHeaders(storedHeader, blockHeaderObj);
+        const lastStoredHeader = computedCommitedHeaders[computedCommitedHeaders.length-1];
+        if(forkId!==0 && StatePredictorUtils.gtBuffer(lastStoredHeader.getBlockHash(), tipWork)) {
+            //Fork's work is higher than main chain's work, this fork will become a main chain
+            forkId = 0;
+        }
+
+        return {
+            forkId: forkId,
+            lastStoredHeader,
+            tx,
+            computedCommitedHeaders
+        }
+    }
+
+    private getBlock(commitHash?: BigNumberish, blockHash?: Buffer): Promise<[StarknetBtcStoredHeader, bigint] | null> {
+        const keys = [commitHash == null ? null : toHex(commitHash)];
+        if (blockHash != null) {
+            const starknetBlockHash = hash.computePoseidonHashOnElements(bufferToU32Array(Buffer.from([...blockHash]).reverse()));
+            keys.push(starknetBlockHash);
+        }
+        return this.Events.findInContractEvents(
+            ["btc_relay::events::StoreHeader", "btc_relay::events::StoreForkHeader"],
+            keys,
+            (event) => {
+                return Promise.resolve([StarknetBtcStoredHeader.fromSerializedFeltArray(event.data), BigInt(event.params.commit_hash)]);
+            }
+        );
+    }
+
+    private async getBlockHeight(): Promise<number> {
+        return Number(await this.contract.get_blockheight());
+    }
+
+    /**
+     * Returns data about current main chain tip stored in the btc relay
+     */
+    public async getTipData(): Promise<{ commitHash: string; blockhash: string, chainWork: Buffer, blockheight: number }> {
+        const commitHash = await this.contract.get_tip_commit_hash();
+        if(commitHash==null || BigInt(commitHash)===BigInt(0)) return null;
+
+        const result = await this.getBlock(commitHash);
+        if(result==null) return null;
+
+        const [storedBlockHeader] = result;
+
+        return {
+            blockheight: storedBlockHeader.getBlockheight(),
+            commitHash: bigNumberishToBuffer(commitHash, 32).toString("hex"),
+            blockhash: storedBlockHeader.getBlockHash().toString("hex"),
+            chainWork: storedBlockHeader.getChainWork()
+        };
+    }
+
+    /**
+     * Retrieves blockheader with a specific blockhash, returns null if requiredBlockheight is provided and
+     *  btc relay contract is not synced up to the desired blockheight
+     *
+     * @param blockData
+     * @param requiredBlockheight
+     */
+    public async retrieveLogAndBlockheight(blockData: {blockhash: string}, requiredBlockheight?: number): Promise<{
+        header: StarknetBtcStoredHeader,
+        height: number
+    } | null> {
+        //TODO: we can fetch the blockheight and events in parallel
+        const blockHeight = await this.getBlockHeight();
+        if(requiredBlockheight!=null && blockHeight < requiredBlockheight) {
+            return null;
+        }
+        const result = await this.getBlock(null, Buffer.from(blockData.blockhash, "hex"));
+        if(result==null) return null;
+
+        const [storedBlockHeader, commitHash] = result;
+
+        //Check if block is part of the main chain
+        const chainCommitment = await this.contract.get_commit_hash(storedBlockHeader.block_height);
+        if(BigInt(chainCommitment)!==BigInt(commitHash)) return null;
+
+        this.logger.debug("retrieveLogAndBlockheight(): block found," +
+            " commit hash: "+toHex(commitHash)+" blockhash: "+blockData.blockhash+" current btc relay height: "+blockHeight);
+
+        return {header: storedBlockHeader, height: blockHeight};
+    }
+
+    /**
+     * Retrieves blockheader data by blockheader's commit hash,
+     *
+     * @param commitmentHashStr
+     * @param blockData
+     */
+    public async retrieveLogByCommitHash(commitmentHashStr: string, blockData: {blockhash: string}): Promise<StarknetBtcStoredHeader> {
+        const result = await this.getBlock(commitmentHashStr, Buffer.from(blockData.blockhash, "hex"));
+        if(result==null) return null;
+
+        const [storedBlockHeader, commitHash] = result;
+
+        //Check if block is part of the main chain
+        const chainCommitment = await this.contract.get_commit_hash(storedBlockHeader.block_height);
+        if(BigInt(chainCommitment)!==BigInt(commitHash)) return null;
+
+        this.logger.debug("retrieveLogByCommitHash(): block found," +
+            " commit hash: "+commitmentHashStr+" blockhash: "+blockData.blockhash+" height: "+storedBlockHeader.block_height);
+
+        return storedBlockHeader;
+    }
+
+    /**
+     * Retrieves latest known stored blockheader & blockheader from bitcoin RPC that is in the main chain
+     */
+    public async retrieveLatestKnownBlockLog(): Promise<{
+        resultStoredHeader: StarknetBtcStoredHeader,
+        resultBitcoinHeader: B
+    }> {
+        const data = await this.Events.findInContractEvents(
+            ["btc_relay::events::StoreHeader", "btc_relay::events::StoreForkHeader"],
+            null,
+            async (event) => {
+                const storedHeader = StarknetBtcStoredHeader.fromSerializedFeltArray(event.data);
+
+                const blockHashHex = storedHeader.getBlockHash().toString("hex");
+                const commitHash = event.params.commit_hash;
+
+                const [isInBtcMainChain, btcRelayCommitHash] = await Promise.all([
+                    this.bitcoinRpc.isInMainChain(blockHashHex).catch(() => false),
+                    this.contract.get_commit_hash(storedHeader.block_height)
+                ]);
+
+                if(!isInBtcMainChain) return null;
+                if(BigInt(commitHash)!==BigInt(btcRelayCommitHash)) return null;
+
+                return {
+                    resultStoredHeader: storedHeader,
+                    resultBitcoinHeader: await this.bitcoinRpc.getBlockHeader(blockHashHex),
+                    commitHash: commitHash
+                }
+            }
+        )
+
+        if(data!=null) this.logger.debug("retrieveLatestKnownBlockLog(): block found," +
+            " commit hash: "+toHex(data.commitHash)+" blockhash: "+data.resultBitcoinHeader.getHash()+
+            " height: "+data.resultStoredHeader.getBlockheight());
+
+        return data;
+    }
+
+    /**
+     * Saves blockheaders as a bitcoin main chain to the btc relay
+     *
+     * @param signer
+     * @param mainHeaders
+     * @param storedHeader
+     * @param feeRate
+     */
+    public saveMainHeaders(signer: string, mainHeaders: BtcBlock[], storedHeader: StarknetBtcStoredHeader, feeRate?: string) {
+        this.logger.debug("saveMainHeaders(): submitting main blockheaders, count: "+mainHeaders.length);
+        return this._saveHeaders(signer, mainHeaders, storedHeader, null, 0, feeRate);
+    }
+
+    /**
+     * Creates a new long fork and submits the headers to it
+     *
+     * @param signer
+     * @param forkHeaders
+     * @param storedHeader
+     * @param tipWork
+     * @param feeRate
+     */
+    public async saveNewForkHeaders(signer: string, forkHeaders: BtcBlock[], storedHeader: StarknetBtcStoredHeader, tipWork: Buffer, feeRate?: string) {
+        let forkId: number = Math.floor(Math.random() * 0xFFFFFFFFFFFF);
+
+        this.logger.debug("saveNewForkHeaders(): submitting new fork & blockheaders," +
+            " count: "+forkHeaders.length+" forkId: 0x"+forkId.toString(16));
+
+        return await this._saveHeaders(signer, forkHeaders, storedHeader, tipWork, forkId, feeRate);
+    }
+
+    /**
+     * Continues submitting blockheaders to a given fork
+     *
+     * @param signer
+     * @param forkHeaders
+     * @param storedHeader
+     * @param forkId
+     * @param tipWork
+     * @param feeRate
+     */
+    public saveForkHeaders(signer: string, forkHeaders: BtcBlock[], storedHeader: StarknetBtcStoredHeader, forkId: number, tipWork: Buffer, feeRate?: string) {
+        this.logger.debug("saveForkHeaders(): submitting blockheaders to existing fork," +
+            " count: "+forkHeaders.length+" forkId: 0x"+forkId.toString(16));
+
+        return this._saveHeaders(signer, forkHeaders, storedHeader, tipWork, forkId, feeRate);
+    }
+
+    /**
+     * Submits short fork with given blockheaders
+     *
+     * @param signer
+     * @param forkHeaders
+     * @param storedHeader
+     * @param tipWork
+     * @param feeRate
+     */
+    public saveShortForkHeaders(signer: string, forkHeaders: BtcBlock[], storedHeader: StarknetBtcStoredHeader, tipWork: Buffer, feeRate?: string) {
+        this.logger.debug("saveShortForkHeaders(): submitting short fork blockheaders," +
+            " count: "+forkHeaders.length);
+
+        return this._saveHeaders(signer, forkHeaders, storedHeader, tipWork, -1, feeRate);
+    }
+
+    /**
+     * Estimate required synchronization fee (worst case) to synchronize btc relay to the required blockheight
+     *
+     * @param requiredBlockheight
+     * @param feeRate
+     */
+    public async estimateSynchronizeFee(requiredBlockheight: number, feeRate?: string): Promise<bigint> {
+        const tipData = await this.getTipData();
+        const currBlockheight = tipData.blockheight;
+
+        const blockheightDelta = requiredBlockheight-currBlockheight;
+
+        if(blockheightDelta<=0) return 0n;
+
+        const synchronizationFee = BigInt(blockheightDelta) * await this.getFeePerBlock(feeRate);
+        this.logger.debug("estimateSynchronizeFee(): required blockheight: "+requiredBlockheight+
+            " blockheight delta: "+blockheightDelta+" fee: "+synchronizationFee.toString(10));
+
+        return synchronizationFee;
+    }
+
+    /**
+     * Returns fee required (in SOL) to synchronize a single block to btc relay
+     *
+     * @param feeRate
+     */
+    public async getFeePerBlock(feeRate?: string): Promise<bigint> {
+        feeRate ??= await this.Fees.getFeeRate();
+        return StarknetFees.getGasFee(GAS_PER_BLOCKHEADER, feeRate);
+    }
+
+    /**
+     * Gets fee rate required for submitting blockheaders to the main chain
+     */
+    public getMainFeeRate(signer: string | null): Promise<string> {
+        return this.Fees.getFeeRate();
+    }
+
+    /**
+     * Gets fee rate required for submitting blockheaders to the specific fork
+     */
+    public getForkFeeRate(signer: string, forkId: number): Promise<string> {
+        return this.Fees.getFeeRate();
+    }
+
+    saveInitialHeader(signer: string, header: B, epochStart: number, pastBlocksTimestamps: number[], feeRate?: string): Promise<StarknetTx> {
+        throw new Error("Not supported, starknet contract is initialized with constructor!");
+    }
+
+}

package/src/starknet/btcrelay/headers/StarknetBtcHeader.ts

@@ -0,0 +1,101 @@
+import {BtcHeader} from "@atomiqlabs/base";
+import {Buffer} from "buffer";
+import {BigNumberish} from "starknet";
+import {toHex, u32ArrayToBuffer, u32ReverseEndianness} from "../../../utils/Utils";
+import * as createHash from "create-hash";
+
+export type StarknetBtcHeaderType = {
+    reversed_version: BigNumberish;
+    previous_blockhash: BigNumberish[];
+    merkle_root: BigNumberish[];
+    reversed_timestamp: BigNumberish;
+    nbits: BigNumberish;
+    nonce: BigNumberish;
+    hash?: Buffer
+}
+
+export class StarknetBtcHeader implements BtcHeader {
+
+    reversed_version: number;
+    previous_blockhash: number[];
+    merkle_root: number[];
+    reversed_timestamp: number;
+    nbits: number;
+    nonce: number;
+    hash?: Buffer;
+
+    constructor(obj: StarknetBtcHeaderType) {
+        this.reversed_version = Number(obj.reversed_version);
+        this.previous_blockhash = obj.previous_blockhash.map(val => Number(val));
+        this.merkle_root = obj.merkle_root.map(val => Number(val));
+        this.reversed_timestamp = Number(obj.reversed_timestamp);
+        this.nbits = Number(obj.nbits);
+        this.nonce = Number(obj.nonce);
+        this.hash = obj.hash;
+    }
+
+    getMerkleRoot(): Buffer {
+        return u32ArrayToBuffer(this.merkle_root);
+    }
+
+    getNbits(): number {
+        return u32ReverseEndianness(this.nbits);
+    }
+
+    getNonce(): number {
+        return u32ReverseEndianness(this.nonce);
+    }
+
+    getReversedPrevBlockhash(): Buffer {
+        return u32ArrayToBuffer(this.previous_blockhash);
+    }
+
+    getTimestamp(): number {
+        return u32ReverseEndianness(this.reversed_timestamp);
+    }
+
+    getVersion(): number {
+        return u32ReverseEndianness(this.reversed_version);
+    }
+
+    getHash(): Buffer {
+        if(this.hash!=null) return this.hash;
+        const buffer = Buffer.alloc(80);
+        buffer.writeUInt32BE(this.reversed_version);
+        u32ArrayToBuffer(this.previous_blockhash).copy(buffer, 4);
+        u32ArrayToBuffer(this.merkle_root).copy(buffer, 36);
+        buffer.writeUInt32BE(this.reversed_timestamp, 68);
+        buffer.writeUInt32BE(this.nbits, 72);
+        buffer.writeUInt32BE(this.nonce, 76);
+        return createHash("sha256").update(createHash("sha256").update(buffer).digest()).digest();
+    }
+
+    serialize(): BigNumberish[] {
+        return [
+            this.reversed_version,
+            ...this.previous_blockhash,
+            ...this.merkle_root,
+            this.reversed_timestamp,
+            this.nbits,
+            this.nonce
+        ];
+    }
+
+    static fromSerializedFeltArray(span: BigNumberish[]): StarknetBtcHeader {
+        const reversed_version = toHex(span.shift());
+        const previous_blockhash = span.splice(0, 8).map(toHex);
+        const merkle_root = span.splice(0, 8).map(toHex);
+        const reversed_timestamp = toHex(span.shift());
+        const nbits = toHex(span.shift());
+        const nonce = toHex(span.shift());
+        return new StarknetBtcHeader({
+            reversed_version,
+            previous_blockhash,
+            merkle_root,
+            reversed_timestamp,
+            nbits,
+            nonce
+        });
+    }
+
+}

package/src/starknet/btcrelay/headers/StarknetBtcStoredHeader.ts

@@ -0,0 +1,142 @@
+import {BtcStoredHeader, StatePredictorUtils} from "@atomiqlabs/base";
+import {StarknetBtcHeader, StarknetBtcHeaderType} from "./StarknetBtcHeader";
+import {Buffer} from "buffer";
+import {BigNumberish, cairo, Uint256} from "starknet";
+import {bigNumberishToBuffer, bufferToU32Array, u32ArrayToBuffer, isUint256, toHex} from "../../../utils/Utils";
+
+export type StarknetBtcStoredHeaderType = {
+    blockheader: StarknetBtcHeader | StarknetBtcHeaderType,
+    block_hash: BigNumberish[],
+    chain_work: BigNumberish | Uint256,
+    block_height: BigNumberish,
+    last_diff_adjustment: BigNumberish,
+    prev_block_timestamps: BigNumberish[]
+}
+
+export class StarknetBtcStoredHeader implements BtcStoredHeader<StarknetBtcHeader> {
+
+    blockheader: StarknetBtcHeader;
+    block_hash: number[];
+    chain_work: Uint256;
+    block_height: number;
+    last_diff_adjustment: number;
+    prev_block_timestamps: number[];
+
+    constructor(obj: StarknetBtcStoredHeaderType) {
+        this.blockheader = obj.blockheader instanceof StarknetBtcHeader ? obj.blockheader : new StarknetBtcHeader(obj.blockheader);
+        this.block_hash = obj.block_hash.map(val => Number(val));
+        this.chain_work = isUint256(obj.chain_work) ? obj.chain_work : cairo.uint256(obj.chain_work);
+        this.block_height = Number(obj.block_height);
+        this.last_diff_adjustment = Number(obj.last_diff_adjustment);
+        this.prev_block_timestamps = obj.prev_block_timestamps.map(val => Number(val));
+    }
+
+    getBlockheight(): number {
+        return this.block_height;
+    }
+
+    getChainWork(): Buffer {
+        return bigNumberishToBuffer(this.chain_work, 32);
+    }
+
+    getHeader(): StarknetBtcHeader {
+        return this.blockheader;
+    }
+
+    getLastDiffAdjustment(): number {
+        return this.last_diff_adjustment;
+    }
+
+    getPrevBlockTimestamps(): number[] {
+        return this.prev_block_timestamps;
+    }
+
+    getBlockHash(): Buffer {
+        return u32ArrayToBuffer(this.block_hash).reverse();
+    }
+
+    /**
+     * Computes prevBlockTimestamps for a next block, shifting the old block timestamps to the left & appending
+     *  this block's timestamp to the end
+     *
+     * @private
+     */
+    private computeNextBlockTimestamps(): number[] {
+        const prevBlockTimestamps = [...this.prev_block_timestamps];
+        for(let i=1;i<10;i++) {
+            prevBlockTimestamps[i-1] = prevBlockTimestamps[i];
+        }
+        prevBlockTimestamps[9] = this.blockheader.getTimestamp();
+        return prevBlockTimestamps;
+    }
+
+    /**
+     * Computes total chain work after a new header with "nbits" is added to the chain
+     *
+     * @param nbits
+     * @private
+     */
+    private computeNextChainWork(nbits: number): Buffer {
+        const chainWork = [...this.getChainWork()];
+        StatePredictorUtils.addInPlace(chainWork, [...StatePredictorUtils.getChainwork(nbits)]);
+        return Buffer.from(chainWork);
+    }
+
+    /**
+     * Computes lastDiffAdjustment, this changes only once every DIFF_ADJUSTMENT_PERIOD blocks
+     *
+     * @param headerTimestamp
+     * @private
+     */
+    private computeNextLastDiffAdjustment(headerTimestamp: number) {
+        const blockheight = this.block_height+1;
+
+        let lastDiffAdjustment = this.last_diff_adjustment;
+        if(blockheight % StatePredictorUtils.DIFF_ADJUSTMENT_PERIOD === 0) {
+            lastDiffAdjustment = headerTimestamp;
+        }
+
+        return lastDiffAdjustment;
+    }
+
+    computeNext(header: StarknetBtcHeader): StarknetBtcStoredHeader {
+        return new StarknetBtcStoredHeader({
+            chain_work: "0x"+this.computeNextChainWork(header.getNbits()).toString("hex"),
+            prev_block_timestamps: this.computeNextBlockTimestamps(),
+            block_height: this.block_height+1,
+            last_diff_adjustment: this.computeNextLastDiffAdjustment(header.getTimestamp()),
+            block_hash: bufferToU32Array(header.getHash()),
+            blockheader: header
+        });
+    }
+
+    serialize(): BigNumberish[] {
+        return [
+            ...this.blockheader.serialize(),
+            ...this.block_hash,
+            this.chain_work.low,
+            this.chain_work.high,
+            this.block_height,
+            this.last_diff_adjustment,
+            ...this.prev_block_timestamps
+        ]
+    }
+
+    static fromSerializedFeltArray(span: BigNumberish[]): StarknetBtcStoredHeader {
+        const blockheader = StarknetBtcHeader.fromSerializedFeltArray(span);
+        const block_hash = span.splice(0, 8).map(toHex);
+        const chain_work = {low: span.shift(), high: span.shift()};
+        const block_height = toHex(span.shift());
+        const last_diff_adjustment = toHex(span.shift());
+        const prev_block_timestamps = span.splice(0, 10).map(toHex);
+        return new StarknetBtcStoredHeader({
+            blockheader,
+            block_hash,
+            chain_work,
+            block_height,
+            last_diff_adjustment,
+            prev_block_timestamps
+        });
+    }
+
+}

package/src/starknet/contract/StarknetContractBase.ts

@@ -0,0 +1,29 @@
+import {StarknetBase, StarknetRetryPolicy} from "../base/StarknetBase";
+import {constants, Contract, Provider, TypedContractV2} from "starknet";
+import {StarknetFees} from "../base/modules/StarknetFees";
+import {Abi} from "abi-wan-kanabi";
+import {StarknetContractEvents} from "./modules/StarknetContractEvents";
+
+/**
+ * Base class providing program specific utilities
+ */
+export class StarknetContractBase<T extends Abi> extends StarknetBase {
+
+    contract: TypedContractV2<T>;
+
+    public readonly Events: StarknetContractEvents<T>;
+
+    constructor(
+        chainId: constants.StarknetChainId,
+        provider: Provider,
+        contractAddress: string,
+        contractAbi: T,
+        retryPolicy?: StarknetRetryPolicy,
+        solanaFeeEstimator: StarknetFees = new StarknetFees(provider)
+    ) {
+        super(chainId, provider, retryPolicy, solanaFeeEstimator);
+        this.contract = new Contract(contractAbi, contractAddress, provider).typedv2(contractAbi);
+        this.Events = new StarknetContractEvents(this, contractAbi);
+    }
+
+}