@acala-network/chopsticks 0.5.9 → 0.6.0

package/lib/blockchain/block-builder.js

@@ -2,7 +2,7 @@
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.dryRunInherents = exports.dryRunExtrinsic = exports.buildBlock = exports.newHeader = void 0;
 const block_1 = require("./block");
-const util_crypto_1 = require("@polkadot/util-crypto");
+const storage_layer_1 = require("./storage-layer");
 const util_1 = require("@polkadot/util");
 const utils_1 = require("../utils");
 const logger_1 = require("../logger");
@@ -90,12 +90,16 @@ const newHeader = async (head) => {
     return header;
 };
 exports.newHeader = newHeader;
-const initNewBlock = async (head, header, inherents) => {
+const initNewBlock = async (head, header, inherents, storageLayer) => {
     const blockNumber = header.number.toNumber();
     const hash = `0x${Math.round(Math.random() * 100000000)
         .toString(16)
         .padEnd(64, '0')}`;
-    const newBlock = new block_1.Block(head.chain, blockNumber, hash, head, { header, extrinsics: [], storage: head.storage });
+    const newBlock = new block_1.Block(head.chain, blockNumber, hash, head, {
+        header,
+        extrinsics: [],
+        storage: storageLayer ?? head.storage,
+    });
     {
         // initialize block
         const { storageDiff } = await newBlock.call('Core_initialize_block', [header.toHex()]);
@@ -125,18 +129,11 @@ const initNewBlock = async (head, header, inherents) => {
 const buildBlock = async (head, inherents, extrinsics, ump, onApplyExtrinsicError) => {
     const registry = await head.registry;
     const header = await (0, exports.newHeader)(head);
-    const { block: newBlock } = await initNewBlock(head, header, inherents);
-    logger.info({
-        number: newBlock.number,
-        extrinsicsCount: extrinsics.length,
-        tempHash: newBlock.hash,
-    }, `Try building block #${newBlock.number.toLocaleString()}`);
-    const pendingExtrinsics = [];
-    const includedExtrinsic = [];
+    let layer;
     // apply ump via storage override hack
     if (Object.keys(ump).length > 0) {
         const meta = await head.meta;
-        const layer = head.pushStorageLayer();
+        layer = new storage_layer_1.StorageLayer(head.storage);
         for (const [paraId, upwardMessages] of Object.entries(ump)) {
             const queueSize = meta.registry.createType('(u32, u32)', [
                 upwardMessages.length,
@@ -148,15 +145,20 @@ const buildBlock = async (head, inherents, extrinsics, ump, onApplyExtrinsicErro
                 [(0, utils_1.compactHex)(meta.query.ump.relayDispatchQueues(paraId)), messages.toHex()],
                 [(0, utils_1.compactHex)(meta.query.ump.relayDispatchQueueSize(paraId)), queueSize.toHex()],
             ]);
+            logger.trace({ paraId, upwardMessages: (0, logger_1.truncate)(upwardMessages) }, 'Pushed UMP');
         }
-        logger.trace({
-            number: newBlock.number,
-            tempHash: newBlock.hash,
-            ump,
-        }, 'Upward messages');
         const needsDispatch = meta.registry.createType('Vec<u32>', Object.keys(ump));
         layer.set((0, utils_1.compactHex)(meta.query.ump.needsDispatch()), needsDispatch.toHex());
     }
+    const { block: newBlock } = await initNewBlock(head, header, inherents, layer);
+    logger.info({
+        number: newBlock.number,
+        extrinsicsCount: extrinsics.length,
+        umpCount: Object.keys(ump).length,
+        tempHash: newBlock.hash,
+    }, `Try building block #${newBlock.number.toLocaleString()}`);
+    const pendingExtrinsics = [];
+    const includedExtrinsic = [];
     // apply extrinsics
     for (const extrinsic of extrinsics) {
         try {
@@ -194,11 +196,12 @@ const buildBlock = async (head, inherents, extrinsics, ump, onApplyExtrinsicErro
         storageDiff,
     });
     logger.info({
-        hash: finalBlock.hash,
-        extrinsics: includedExtrinsic.map((x) => (0, util_crypto_1.blake2AsHex)(x, 256)),
-        pendingExtrinsics: pendingExtrinsics.length,
         number: newBlock.number,
-    }, `Block built #${newBlock.number.toLocaleString()} hash ${finalBlock.hash}`);
+        hash: finalBlock.hash,
+        extrinsics: (0, logger_1.truncate)(includedExtrinsic),
+        pendingExtrinsicsCount: pendingExtrinsics.length,
+        ump: (0, logger_1.truncate)(ump),
+    }, 'Block built');
     return [finalBlock, pendingExtrinsics];
 };
 exports.buildBlock = buildBlock;

package/lib/blockchain/block.d.ts

@@ -8,6 +8,7 @@ import type { RuntimeVersion } from '../executor';
 export type TaskCallResponse = {
     result: HexString;
     storageDiff: [HexString, HexString | null][];
+    runtimeLogs: string[];
 };
 export declare class Block {
     #private;

package/lib/blockchain/block.js

@@ -7,6 +7,7 @@ const util_1 = require("@polkadot/types-known/util");
 const util_2 = require("@polkadot/util");
 const storage_layer_1 = require("./storage-layer");
 const utils_1 = require("../utils");
+const logger_1 = require("../logger");
 const executor_1 = require("../executor");
 class Block {
     number;
@@ -172,9 +173,14 @@ class Block {
             storage,
             mockSignatureHost: this.#chain.mockSignatureHost,
             allowUnresolvedImports: this.#chain.allowUnresolvedImports,
+            runtimeLogLevel: this.#chain.runtimeLogLevel,
         }, (0, executor_1.taskHandler)(this));
-        if (response.Call)
+        if (response.Call) {
+            for (const log of response.Call.runtimeLogs) {
+                logger_1.defaultLogger.info(`RuntimeLogs:\n${log}`);
+            }
             return response.Call;
+        }
         if (response.Error)
             throw Error(response.Error);
         throw Error('Unexpected response');

package/lib/blockchain/index.d.ts

@@ -18,6 +18,7 @@ export interface Options {
     };
     mockSignatureHost?: boolean;
     allowUnresolvedImports?: boolean;
+    runtimeLogLevel?: number;
     registeredTypes: RegisteredTypes;
 }
 export declare class Blockchain {
@@ -27,9 +28,10 @@ export declare class Blockchain {
     readonly db: DataSource | undefined;
     readonly mockSignatureHost: boolean;
     readonly allowUnresolvedImports: boolean;
+    readonly runtimeLogLevel: number;
     readonly registeredTypes: RegisteredTypes;
     readonly headState: HeadState;
-    constructor({ api, buildBlockMode, inherentProvider, db, header, mockSignatureHost, allowUnresolvedImports, registeredTypes, }: Options);
+    constructor({ api, buildBlockMode, inherentProvider, db, header, mockSignatureHost, allowUnresolvedImports, runtimeLogLevel, registeredTypes, }: Options);
     get head(): Block;
     get txPool(): TxPool;
     getBlockAt(number?: number): Promise<Block | undefined>;

package/lib/blockchain/index.js

@@ -16,6 +16,7 @@ class Blockchain {
     db;
     mockSignatureHost;
     allowUnresolvedImports;
+    runtimeLogLevel;
     registeredTypes;
     #txpool;
     #inherentProvider;
@@ -24,11 +25,12 @@ class Blockchain {
     #blocksByHash = {};
     #loadingBlocks = {};
     headState;
-    constructor({ api, buildBlockMode, inherentProvider, db, header, mockSignatureHost = false, allowUnresolvedImports = false, registeredTypes = {}, }) {
+    constructor({ api, buildBlockMode, inherentProvider, db, header, mockSignatureHost = false, allowUnresolvedImports = false, runtimeLogLevel = 0, registeredTypes = {}, }) {
         this.api = api;
         this.db = db;
         this.mockSignatureHost = mockSignatureHost;
         this.allowUnresolvedImports = allowUnresolvedImports;
+        this.runtimeLogLevel = runtimeLogLevel;
         this.registeredTypes = registeredTypes;
         this.#head = new block_1.Block(this, header.number, header.hash);
         this.#registerBlock(this.#head);

package/lib/blockchain/txpool.d.ts

@@ -28,6 +28,10 @@ export declare class TxPool {
     readonly event: EventEmitter;
     constructor(chain: Blockchain, inherentProvider: InherentProvider, mode?: BuildBlockMode);
     get pendingExtrinsics(): HexString[];
+    get ump(): Record<number, HexString[]>;
+    get dmp(): DownwardMessage[];
+    get hrmp(): Record<number, HorizontalMessage[]>;
+    clear(): void;
     pendingExtrinsicsBy(address: string): HexString[];
     submitExtrinsic(extrinsic: HexString): Promise<void>;
     submitUpwardMessages(id: number, ump: HexString[]): void;

package/lib/blockchain/txpool.js

@@ -8,6 +8,8 @@ const node_stream_1 = require("node:stream");
 const lodash_1 = __importDefault(require("lodash"));
 const utils_1 = require("../utils");
 const block_builder_1 = require("./block-builder");
+const logger_1 = require("../logger");
+const logger = logger_1.defaultLogger.child({ name: 'txpool' });
 exports.APPLY_EXTRINSIC_ERROR = 'TxPool::ApplyExtrinsicError';
 var BuildBlockMode;
 (function (BuildBlockMode) {
@@ -34,10 +36,30 @@ class TxPool {
     get pendingExtrinsics() {
         return this.#pool.map(({ extrinsic }) => extrinsic);
     }
+    get ump() {
+        return this.#ump;
+    }
+    get dmp() {
+        return this.#dmp;
+    }
+    get hrmp() {
+        return this.#hrmp;
+    }
+    clear() {
+        this.#pool.length = 0;
+        for (const id of Object.keys(this.#ump)) {
+            delete this.#ump[id];
+        }
+        this.#dmp.length = 0;
+        for (const id of Object.keys(this.#hrmp)) {
+            delete this.#hrmp[id];
+        }
+    }
     pendingExtrinsicsBy(address) {
         return this.#pool.filter(({ signer }) => signer === address).map(({ extrinsic }) => extrinsic);
     }
     async submitExtrinsic(extrinsic) {
+        logger.debug({ extrinsic: (0, logger_1.truncate)(extrinsic) }, 'submit extrinsic');
         this.#pool.push({ extrinsic, signer: await this.#getSigner(extrinsic) });
         this.#maybeBuildBlock();
     }
@@ -47,6 +69,7 @@ class TxPool {
         return tx.signer.toString();
     }
     submitUpwardMessages(id, ump) {
+        logger.debug({ id, ump: (0, logger_1.truncate)(ump) }, 'submit upward messages');
         if (!this.#ump[id]) {
             this.#ump[id] = [];
         }
@@ -54,10 +77,12 @@ class TxPool {
         this.#maybeBuildBlock();
     }
     submitDownwardMessages(dmp) {
+        logger.debug({ dmp: (0, logger_1.truncate)(dmp) }, 'submit downward messages');
         this.#dmp.push(...dmp);
         this.#maybeBuildBlock();
     }
     submitHorizontalMessages(id, hrmp) {
+        logger.debug({ id, hrmp: (0, logger_1.truncate)(hrmp) }, 'submit horizontal messages');
         if (!this.#hrmp[id]) {
             this.#hrmp[id] = [];
         }
@@ -136,6 +161,7 @@ class TxPool {
             throw new Error('Unreachable');
         }
         const { params, deferred } = pending;
+        logger.trace({ params }, 'build block');
         const head = this.#chain.head;
         const inherents = await this.#inherentProvider.createInherents(head, params);
         const [newBlock, pendingExtrinsics] = await (0, block_builder_1.buildBlock)(head, inherents, params.transactions, params.upwardMessages, (extrinsic, error) => {

package/lib/cli.js

@@ -5,7 +5,9 @@ var __importDefault = (this && this.__importDefault) || function (mod) {
 Object.defineProperty(exports, "__esModule", { value: true });
 const helpers_1 = require("yargs/helpers");
 const node_fs_1 = require("node:fs");
+const lodash_1 = __importDefault(require("lodash"));
 const axios_1 = __importDefault(require("axios"));
+const dotenv_1 = __importDefault(require("dotenv"));
 const js_yaml_1 = __importDefault(require("js-yaml"));
 const yargs_1 = __importDefault(require("yargs"));
 const _1 = require(".");
@@ -15,6 +17,8 @@ const dry_run_1 = require("./dry-run");
 const dry_run_preimage_1 = require("./dry-run-preimage");
 const utils_1 = require("./utils");
 const run_block_1 = require("./run-block");
+const try_runtime_1 = require("./try-runtime");
+dotenv_1.default.config();
 const processConfig = async (path) => {
     let file;
     if ((0, utils_1.isUrl)(path)) {
@@ -23,7 +27,7 @@ const processConfig = async (path) => {
     else {
         file = (0, node_fs_1.readFileSync)(path, 'utf8');
     }
-    const config = js_yaml_1.default.load(file);
+    const config = js_yaml_1.default.load(lodash_1.default.template(file, { variable: 'env' })(process.env));
     return schema_1.configSchema.parse(config);
 };
 const processArgv = async (argv) => {
@@ -54,19 +58,23 @@ const defaultOptions = {
         desc: 'Path to config file',
         string: true,
     },
+    'runtime-log-level': {
+        desc: 'Runtime maximum log level [off = 0; error = 1; warn = 2; info = 3; debug = 4; trace = 5]',
+        number: true,
+    },
 };
 (0, yargs_1.default)((0, helpers_1.hideBin)(process.argv))
     .scriptName('chopsticks')
     .command('run-block', 'Replay a block', (yargs) => yargs.options({
     ...defaultOptions,
-    port: {
-        desc: 'Port to listen on',
-        number: true,
-    },
     'output-path': {
         desc: 'File path to print output',
         string: true,
     },
+    'import-storage': {
+        desc: 'Pre-defined JSON/YAML storage file path',
+        string: true,
+    },
     html: {
         desc: 'Generate html with storage diff',
     },
@@ -75,6 +83,26 @@ const defaultOptions = {
     },
 }), async (argv) => {
     await (0, run_block_1.runBlock)(await processArgv(argv));
+})
+    .command('try-runtime', 'Runs runtime upgrade', (yargs) => yargs.options({
+    ...defaultOptions,
+    'wasm-override': {
+        desc: 'Path to WASM built with feature `try-runtime` enabled',
+        string: true,
+        required: true,
+    },
+    'output-path': {
+        desc: 'File path to print output',
+        string: true,
+    },
+    html: {
+        desc: 'Generate html with storage diff',
+    },
+    open: {
+        desc: 'Open generated html',
+    },
+}), async (argv) => {
+    await (0, try_runtime_1.tryRuntime)(await processArgv(argv));
 })
     .command('dry-run', 'Dry run an extrinsic', (yargs) => yargs.options({
     ...defaultOptions,

package/lib/dry-run-preimage.js

@@ -67,10 +67,14 @@ const dryRunPreimage = async (argv) => {
         storage: [],
         mockSignatureHost: false,
         allowUnresolvedImports: false,
+        runtimeLogLevel: argv['runtime-log-level'] || 0,
     }, (0, executor_1.taskHandler)(block));
     if (result.Error) {
         throw new Error(result.Error);
     }
+    for (const logs of result.Call.runtimeLogs) {
+        logger_1.defaultLogger.info(`RuntimeLogs:\n${logs}`);
+    }
     const filePath = await (0, generate_html_diff_1.generateHtmlDiffPreviewFile)(block, result.Call.storageDiff, hash);
     console.log(`Generated preview ${filePath}`);
     if (argv['open']) {
@@ -85,9 +89,7 @@ const dryRunPreimage = async (argv) => {
         if (outcome.isErr) {
             throw new Error(outcome.asErr.toString());
         }
-        else {
-            logger_1.defaultLogger.info(outcome.toHuman(), 'dry_run_outcome');
-        }
+        logger_1.defaultLogger.info(outcome.toHuman(), 'dry_run_outcome');
         const filePath = await (0, generate_html_diff_1.generateHtmlDiffPreviewFile)(context.chain.head, storageDiff, (0, util_crypto_1.blake2AsHex)(argv['extrinsic'], 256));
         console.log(`Generated preview ${filePath}`);
         if (argv['open']) {

package/lib/dry-run.js

@@ -14,9 +14,7 @@ const dryRun = async (argv) => {
     if (outcome.isErr) {
         throw new Error(outcome.asErr.toString());
     }
-    else {
-        logger_1.defaultLogger.info(outcome.toHuman(), 'dry_run_outcome');
-    }
+    logger_1.defaultLogger.info(outcome.toHuman(), 'dry_run_outcome');
     if (argv['html']) {
         const filePath = await (0, generate_html_diff_1.generateHtmlDiffPreviewFile)(context.chain.head, storageDiff, (0, util_crypto_1.blake2AsHex)(argv['extrinsic'], 256));
         console.log(`Generated preview ${filePath}`);

package/lib/executor.d.ts

@@ -30,6 +30,7 @@ export declare const runTask: (task: {
     storage: [HexString, HexString | null][];
     mockSignatureHost: boolean;
     allowUnresolvedImports: boolean;
+    runtimeLogLevel: number;
 }, callback?: JsCallback) => Promise<any>;
 export declare const taskHandler: (block: Block) => JsCallback;
 export declare const emptyTaskHandler: {

package/lib/executor.js

@@ -79,6 +79,7 @@ exports.getAuraSlotDuration = lodash_1.default.memoize(async (wasm, registry) =>
         storage: [],
         mockSignatureHost: false,
         allowUnresolvedImports: false,
+        runtimeLogLevel: 0,
     });
     if (!result.Call)
         throw new Error(result.Error);

package/lib/genesis-provider.js

@@ -127,6 +127,7 @@ class GenesisProvider {
                     storage: [],
                     mockSignatureHost: false,
                     allowUnresolvedImports: true,
+                    runtimeLogLevel: 0,
                 }, this._jsCallback);
             }
             case 'chain_getHeader':

package/lib/run-block.js

@@ -2,12 +2,13 @@
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.runBlock = void 0;
 const node_fs_1 = require("node:fs");
+const logger_1 = require("./logger");
 const generate_html_diff_1 = require("./utils/generate-html-diff");
 const open_html_1 = require("./utils/open-html");
 const executor_1 = require("./executor");
 const setup_1 = require("./setup");
 const runBlock = async (argv) => {
-    const context = await (0, setup_1.setup)(argv);
+    const context = await (0, setup_1.setup)(argv, true);
     const header = await context.chain.head.header;
     const wasm = await context.chain.head.wasm;
     const block = context.chain.head;
@@ -25,10 +26,14 @@ const runBlock = async (argv) => {
         storage: [],
         mockSignatureHost: false,
         allowUnresolvedImports: false,
+        runtimeLogLevel: argv['runtime-log-level'] || 0,
     }, (0, executor_1.taskHandler)(parent));
     if (result.Error) {
         throw new Error(result.Error);
     }
+    for (const logs of result.Call.runtimeLogs) {
+        logger_1.defaultLogger.info(`RuntimeLogs:\n${logs}`);
+    }
     if (argv['html']) {
         const filePath = await (0, generate_html_diff_1.generateHtmlDiffPreviewFile)(parent, result.Call.storageDiff, block.hash);
         console.log(`Generated preview ${filePath}`);

package/lib/schema/index.d.ts

@@ -64,7 +64,7 @@ export type Genesis = z.infer<typeof genesisSchema>;
 export declare const configSchema: z.ZodObject<{
     port: z.ZodOptional<z.ZodNumber>;
     endpoint: z.ZodOptional<z.ZodString>;
-    block: z.ZodOptional<z.ZodUnion<[z.ZodString, z.ZodNumber]>>;
+    block: z.ZodOptional<z.ZodUnion<[z.ZodString, z.ZodNumber, z.ZodNull]>>;
     'build-block-mode': z.ZodOptional<z.ZodNativeEnum<typeof BuildBlockMode>>;
     'import-storage': z.ZodOptional<z.ZodAny>;
     'mock-signature-host': z.ZodOptional<z.ZodBoolean>;
@@ -132,10 +132,16 @@ export declare const configSchema: z.ZodObject<{
     }>]>>;
     timestamp: z.ZodOptional<z.ZodNumber>;
     'registered-types': z.ZodOptional<z.ZodAny>;
+    'runtime-log-level': z.ZodOptional<z.ZodNumber>;
 }, "strict", z.ZodTypeAny, {
-    timestamp?: number | undefined;
-    block?: string | number | undefined;
+    port?: number | undefined;
+    endpoint?: string | undefined;
+    block?: string | number | null | undefined;
+    'build-block-mode'?: BuildBlockMode | undefined;
+    'import-storage'?: any;
+    'mock-signature-host'?: boolean | undefined;
     db?: string | undefined;
+    'wasm-override'?: string | undefined;
     genesis?: string | {
         name: string;
         properties: {
@@ -150,17 +156,18 @@ export declare const configSchema: z.ZodObject<{
             };
         };
     } | undefined;
+    timestamp?: number | undefined;
+    'registered-types'?: any;
+    'runtime-log-level'?: number | undefined;
+}, {
     port?: number | undefined;
     endpoint?: string | undefined;
+    block?: string | number | null | undefined;
     'build-block-mode'?: BuildBlockMode | undefined;
     'import-storage'?: any;
     'mock-signature-host'?: boolean | undefined;
-    'wasm-override'?: string | undefined;
-    'registered-types'?: any;
-}, {
-    timestamp?: number | undefined;
-    block?: string | number | undefined;
     db?: string | undefined;
+    'wasm-override'?: string | undefined;
     genesis?: string | {
         name: string;
         properties: {
@@ -175,12 +182,8 @@ export declare const configSchema: z.ZodObject<{
             };
         };
     } | undefined;
-    port?: number | undefined;
-    endpoint?: string | undefined;
-    'build-block-mode'?: BuildBlockMode | undefined;
-    'import-storage'?: any;
-    'mock-signature-host'?: boolean | undefined;
-    'wasm-override'?: string | undefined;
+    timestamp?: number | undefined;
     'registered-types'?: any;
+    'runtime-log-level'?: number | undefined;
 }>;
 export type Config = z.infer<typeof configSchema>;

package/lib/schema/index.js

@@ -17,7 +17,7 @@ exports.configSchema = zod_1.z
     .object({
     port: zod_1.z.number().optional(),
     endpoint: zod_1.z.string().optional(),
-    block: zod_1.z.union([zod_1.z.string().length(66).startsWith('0x'), zod_1.z.number()]).optional(),
+    block: zod_1.z.union([zod_1.z.string().length(66).startsWith('0x'), zod_1.z.number(), zod_1.z.null()]).optional(),
     'build-block-mode': zod_1.z.nativeEnum(txpool_1.BuildBlockMode).optional(),
     'import-storage': zod_1.z.any().optional(),
     'mock-signature-host': zod_1.z.boolean().optional(),
@@ -26,5 +26,6 @@ exports.configSchema = zod_1.z
     genesis: zod_1.z.union([zod_1.z.string(), exports.genesisSchema]).optional(),
     timestamp: zod_1.z.number().optional(),
     'registered-types': zod_1.z.any().optional(),
+    'runtime-log-level': zod_1.z.number().min(0).max(5).optional(),
 })
     .strict();

package/lib/server.js

@@ -50,7 +50,7 @@ const createWS = async (port) => {
 const createServer = async (handler, port) => {
     let wss;
     let listenPort;
-    for (let i = 0; i < 5; i++) {
+    for (let i = 0; i < 10; i++) {
         const preferPort = (port ?? 0) > 0 ? (port ?? 0) + i : 0;
         logger.debug('Try starting on port %d', preferPort);
         const [maybeWss, maybeListenPort] = await createWS(preferPort);
@@ -137,7 +137,7 @@ const createServer = async (handler, port) => {
                 send({
                     id: req.id,
                     jsonrpc: '2.0',
-                    result: resp || null,
+                    result: resp ?? null,
                 });
             }
             catch (e) {

package/lib/setup.d.ts

@@ -3,7 +3,7 @@ import { ProviderInterface } from '@polkadot/rpc-provider/types';
 import { Api } from './api';
 import { Blockchain } from './blockchain';
 import { Config } from './schema';
-export declare const setup: (argv: Config) => Promise<{
+export declare const setup: (argv: Config, runBlock?: boolean) => Promise<{
     chain: Blockchain;
     api: Api;
     ws: ProviderInterface;

package/lib/setup.js

@@ -11,7 +11,7 @@ const logger_1 = require("./logger");
 const import_storage_1 = require("./utils/import-storage");
 const db_1 = require("./db");
 const time_travel_1 = require("./utils/time-travel");
-const setup = async (argv) => {
+const setup = async (argv, runBlock = false) => {
     let provider;
     if (argv.genesis) {
         if (typeof argv.genesis === 'string') {
@@ -62,14 +62,23 @@ const setup = async (argv) => {
         },
         mockSignatureHost: argv['mock-signature-host'],
         allowUnresolvedImports: argv['allow-unresolved-imports'],
+        runtimeLogLevel: argv['runtime-log-level'],
         registeredTypes: argv['registered-types'],
     });
     if (argv.timestamp)
         await (0, time_travel_1.timeTravel)(chain, argv.timestamp);
+    let at;
+    if (runBlock) {
+        // in case of run block we need to apply wasm-override and import-storage to parent block
+        const block = await chain.head.parentBlock;
+        if (!block)
+            throw new Error('Cannot find parent block');
+        at = block.hash;
+    }
     // override wasm before importing storage, in case new pallets have been
     // added that have storage imports
-    await (0, import_storage_1.overrideWasm)(chain, argv['wasm-override']);
-    await (0, import_storage_1.importStorage)(chain, argv['import-storage']);
+    await (0, import_storage_1.overrideWasm)(chain, argv['wasm-override'], at);
+    await (0, import_storage_1.importStorage)(chain, argv['import-storage'], at);
     return { chain, api, ws: provider };
 };
 exports.setup = setup;

package/lib/try-runtime.d.ts

@@ -0,0 +1,2 @@
+import { Config } from './schema';
+export declare const tryRuntime: (argv: Config) => Promise<never>;

package/lib/try-runtime.js

@@ -0,0 +1,36 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.tryRuntime = void 0;
+const node_fs_1 = require("node:fs");
+const generate_html_diff_1 = require("./utils/generate-html-diff");
+const open_html_1 = require("./utils/open-html");
+const setup_1 = require("./setup");
+const tryRuntime = async (argv) => {
+    const context = await (0, setup_1.setup)(argv);
+    const block = context.chain.head;
+    const registry = await block.registry;
+    registry.register({
+        UpgradeCheckSelect: {
+            _enum: {
+                None: null,
+            },
+        },
+    });
+    const select_none = registry.createType('UpgradeCheckSelect', 'None');
+    const result = await block.call('TryRuntime_on_runtime_upgrade', [select_none.toHex()], []);
+    if (argv['html']) {
+        const filePath = await (0, generate_html_diff_1.generateHtmlDiffPreviewFile)(block, result.storageDiff, block.hash);
+        console.log(`Generated preview ${filePath}`);
+        if (argv['open']) {
+            (0, open_html_1.openHtml)(filePath);
+        }
+    }
+    else if (argv['output-path']) {
+        (0, node_fs_1.writeFileSync)(argv['output-path'], JSON.stringify(result, null, 2));
+    }
+    else {
+        console.dir(result, { depth: null, colors: false });
+    }
+    process.exit(0);
+};
+exports.tryRuntime = tryRuntime;

package/lib/utils/import-storage.d.ts

@@ -1,4 +1,5 @@
+import { HexString } from '@polkadot/util/types';
 import { Blockchain } from '../blockchain';
 import { StorageValues } from './set-storage';
-export declare const importStorage: (chain: Blockchain, storage?: string | StorageValues) => Promise<void>;
-export declare const overrideWasm: (chain: Blockchain, wasmPath?: string) => Promise<void>;
+export declare const importStorage: (chain: Blockchain, storage?: string | StorageValues, at?: HexString) => Promise<void>;
+export declare const overrideWasm: (chain: Blockchain, wasmPath?: string, at?: HexString) => Promise<void>;

package/lib/utils/import-storage.js

@@ -8,7 +8,7 @@ const node_fs_1 = require("node:fs");
 const js_yaml_1 = __importDefault(require("js-yaml"));
 const set_storage_1 = require("./set-storage");
 const logger_1 = require("../logger");
-const importStorage = async (chain, storage) => {
+const importStorage = async (chain, storage, at) => {
     if (storage == null) {
         return;
     }
@@ -21,11 +21,11 @@ const importStorage = async (chain, storage) => {
     else {
         storageValue = storage;
     }
-    const blockHash = await (0, set_storage_1.setStorage)(chain, storageValue);
+    const blockHash = await (0, set_storage_1.setStorage)(chain, storageValue, at);
     logger_1.defaultLogger.trace({ blockHash, storage }, 'ImportStorage');
 };
 exports.importStorage = importStorage;
-const overrideWasm = async (chain, wasmPath) => {
+const overrideWasm = async (chain, wasmPath, at) => {
     if (wasmPath == null) {
         return;
     }
@@ -38,6 +38,14 @@ const overrideWasm = async (chain, wasmPath) => {
     else {
         wasmHex = '0x' + wasm.toString('hex');
     }
-    chain.head.setWasm(wasmHex);
+    if (at) {
+        const block = await chain.getBlock(at);
+        if (!block)
+            throw new Error(`Cannot find block ${at}`);
+        block.setWasm(wasmHex);
+    }
+    else {
+        chain.head.setWasm(wasmHex);
+    }
 };
 exports.overrideWasm = overrideWasm;

package/package.json

@@ -1,50 +1,50 @@
 {
   "name": "@acala-network/chopsticks",
-  "version": "0.5.9",
+  "version": "0.6.0",
   "author": "Bryan Chen <xlchen1291@gmail.com>",
   "license": "Apache-2.0",
   "bin": "./chopsticks.js",
   "scripts": {
     "clean": "rm -rf lib tsconfig.tsbuildinfo",
-    "build": "yarn clean && tsc -p ./tsconfig.json",
+    "build": "tsc -p ./tsconfig.json",
     "script:start": "cd ../..; ts-node --transpile-only packages/chopsticks/src/cli.ts",
-    "script:run": "LOG_LEVEL=trace ts-node-dev --transpile-only --inspect --notify=false src/cli.ts -- dev --config=../../configs/dev.yml",
-    "dev:karura": "ts-node-dev --transpile-only --inspect --notify=false src/cli.ts -- dev --config=../../configs/karura.yml",
-    "dev:acala": "ts-node-dev --transpile-only --inspect --notify=false src/cli.ts -- dev --config=../../configs/acala.yml",
-    "dev:moonriver": "ts-node-dev --transpile-only --inspect --notify=false src/cli.ts -- dev --config=../../configs/moonriver.yml",
-    "dev:moonbeam": "ts-node-dev --transpile-only --inspect --notify=false src/cli.ts -- dev --config=../../configs/moonbeam.yml"
+    "script:run": "cd ../..; LOG_LEVEL=trace ts-node-dev --transpile-only --inspect --notify=false packages/chopsticks/src/cli.ts -- dev --config=configs/dev.yml",
+    "dev:karura": "cd ../..; ts-node-dev --transpile-only --inspect --notify=false packages/chopsticks/src/cli.ts -- dev --config=configs/karura.yml",
+    "dev:acala": "cd ../..; ts-node-dev --transpile-only --inspect --notify=false packages/chopsticks/src/cli.ts -- dev --config=configs/acala.yml",
+    "dev:moonriver": "cd ../..; ts-node-dev --transpile-only --inspect --notify=false packages/chopsticks/src/cli.ts -- dev --config=configs/moonriver.yml",
+    "dev:moonbeam": "cd ../..; ts-node-dev --transpile-only --inspect --notify=false packages/chopsticks/src/cli.ts -- dev --config=configs/moonbeam.yml"
   },
   "dependencies": {
-    "@acala-network/chopsticks-executor": "0.5.9",
-    "@polkadot/api": "^9.14.2",
-    "@polkadot/rpc-provider": "^9.14.2",
-    "@polkadot/types": "^9.14.2",
-    "@polkadot/types-codec": "^9.14.2",
-    "@polkadot/types-known": "^9.14.2",
-    "@polkadot/util": "^10.4.2",
-    "@polkadot/util-crypto": "^10.4.2",
-    "axios": "^1.3.3",
+    "@acala-network/chopsticks-executor": "0.6.0",
+    "@polkadot/api": "^10.2.1",
+    "@polkadot/rpc-provider": "^10.2.1",
+    "@polkadot/types": "^10.2.1",
+    "@polkadot/types-codec": "^10.2.1",
+    "@polkadot/types-known": "^10.2.1",
+    "@polkadot/util": "^11.1.2",
+    "@polkadot/util-crypto": "^11.1.2",
+    "axios": "^1.3.4",
     "js-yaml": "^4.1.0",
     "jsondiffpatch": "^0.4.1",
     "lodash": "^4.17.21",
     "pino": "^8.11.0",
-    "pino-pretty": "^9.4.0",
+    "pino-pretty": "^10.0.0",
     "reflect-metadata": "^0.1.13",
-    "sqlite3": "^5.1.4",
+    "sqlite3": "^5.1.6",
     "typeorm": "^0.3.12",
-    "ws": "^8.12.1",
+    "ws": "^8.13.0",
     "yargs": "^17.7.1",
-    "zod": "^3.20.6"
+    "zod": "^3.21.4"
   },
   "devDependencies": {
     "@types/js-yaml": "^4.0.5",
-    "@types/lodash": "^4.14.191",
-    "@types/node": "^18.14.2",
+    "@types/lodash": "^4.14.192",
+    "@types/node": "^18.15.10",
     "@types/ws": "^8.5.4",
-    "@types/yargs": "^17.0.22",
+    "@types/yargs": "^17.0.24",
     "ts-node": "^10.9.1",
     "ts-node-dev": "^2.0.0",
-    "typescript": "^4.9.5"
+    "typescript": "^5.0.2"
   },
   "files": [
     "lib",