@acala-network/chopsticks 0.7.2 → 0.8.0-0

package/lib/cli-options.d.ts

@@ -0,0 +1,40 @@
+export declare const defaultOptions: {
+    endpoint: {
+        desc: string;
+        string: boolean;
+    };
+    block: {
+        desc: string;
+        string: boolean;
+    };
+    'wasm-override': {
+        desc: string;
+        string: boolean;
+    };
+    db: {
+        desc: string;
+        string: boolean;
+    };
+    config: {
+        desc: string;
+        string: boolean;
+    };
+    'runtime-log-level': {
+        desc: string;
+        number: boolean;
+    };
+    'offchain-worker': {
+        desc: string;
+        boolean: boolean;
+    };
+};
+export declare const mockOptions: {
+    'import-storage': {
+        desc: string;
+        string: boolean;
+    };
+    'mock-signature-host': {
+        desc: string;
+        boolean: boolean;
+    };
+};

package/lib/cli-options.js

@@ -0,0 +1,43 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.mockOptions = exports.defaultOptions = void 0;
+exports.defaultOptions = {
+    endpoint: {
+        desc: 'Endpoint to connect to',
+        string: true,
+    },
+    block: {
+        desc: 'Block hash or block number. Default to latest block',
+        string: true,
+    },
+    'wasm-override': {
+        desc: 'Path to wasm override',
+        string: true,
+    },
+    db: {
+        desc: 'Path to database',
+        string: true,
+    },
+    config: {
+        desc: 'Path to config file with default options',
+        string: true,
+    },
+    'runtime-log-level': {
+        desc: 'Runtime maximum log level [off = 0; error = 1; warn = 2; info = 3; debug = 4; trace = 5]',
+        number: true,
+    },
+    'offchain-worker': {
+        desc: 'Enable offchain worker',
+        boolean: true,
+    },
+};
+exports.mockOptions = {
+    'import-storage': {
+        desc: 'Pre-defined JSON/YAML storage file path',
+        string: true,
+    },
+    'mock-signature-host': {
+        desc: 'Mock signature host so any signature starts with 0xdeadbeef and filled by 0xcd is considered valid',
+        boolean: true,
+    },
+};

package/lib/cli.d.ts

@@ -1,2 +1 @@
-import { Config } from './schema';
-export declare const fetchConfig: (path: string) => Promise<Config>;
+export {};

package/lib/cli.js

@@ -3,210 +3,46 @@ var __importDefault = (this && this.__importDefault) || function (mod) {
     return (mod && mod.__esModule) ? mod : { "default": mod };
 };
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.fetchConfig = void 0;
-const node_path_1 = require("node:path");
 const dotenv_1 = require("dotenv");
 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 js_yaml_1 = __importDefault(require("js-yaml"));
 const yargs_1 = __importDefault(require("yargs"));
-const _1 = require(".");
+const chopsticks_core_1 = require("@acala-network/chopsticks-core");
 const schema_1 = require("./schema");
-const decoder_1 = require("./utils/decoder");
-const dry_run_1 = require("./dry-run");
-const dry_run_preimage_1 = require("./dry-run-preimage");
-const utils_1 = require("./utils");
-const shared_1 = require("./rpc/shared");
-const run_block_1 = require("./run-block");
-const try_runtime_1 = require("./try-runtime");
+const cli_options_1 = require("./cli-options");
+const plugins_1 = require("./plugins");
+const _1 = require(".");
 (0, dotenv_1.config)();
-const CONFIGS_BASE_URL = 'https://raw.githubusercontent.com/AcalaNetwork/chopsticks/master/configs/';
-const fetchConfig = async (path) => {
-    let file;
-    if ((0, utils_1.isUrl)(path)) {
-        file = await axios_1.default.get(path).then((x) => x.data);
-    }
-    else {
-        try {
-            file = (0, node_fs_1.readFileSync)(path, 'utf8');
-        }
-        catch (err) {
-            if ((0, node_path_1.basename)(path) === path && ['', '.yml', '.yaml', '.json'].includes((0, node_path_1.extname)(path))) {
-                if ((0, node_path_1.extname)(path) === '') {
-                    path += '.yml';
-                }
-                const url = CONFIGS_BASE_URL + path;
-                shared_1.logger.info(`Loading config file ${url}`);
-                file = await axios_1.default.get(url).then((x) => x.data);
-            }
-            else {
-                throw err;
-            }
-        }
-    }
-    const config = js_yaml_1.default.load(lodash_1.default.template(file, { variable: 'env' })(process.env));
-    return schema_1.configSchema.parse(config);
-};
-exports.fetchConfig = fetchConfig;
 const processArgv = async (argv) => {
     if (argv.config) {
-        argv = { ...(await (0, exports.fetchConfig)(argv.config)), ...argv };
+        Object.assign(argv, lodash_1.default.defaults(argv, await (0, schema_1.fetchConfig)(argv.config)));
     }
     argv.port = argv.port ?? (process.env.PORT ? Number(process.env.PORT) : 8000);
-    return argv;
-};
-const defaultOptions = {
-    endpoint: {
-        desc: 'Endpoint to connect to',
-        string: true,
-    },
-    block: {
-        desc: 'Block hash or block number. Default to latest block',
-        string: true,
-    },
-    'wasm-override': {
-        desc: 'Path to wasm override',
-        string: true,
-    },
-    db: {
-        desc: 'Path to database',
-        string: true,
-    },
-    config: {
-        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,
-    },
-    'offchain-worker': {
-        desc: 'Enable offchain worker',
-        boolean: true,
-    },
-};
-const mockOptions = {
-    'import-storage': {
-        desc: 'Pre-defined JSON/YAML storage file path',
-        string: true,
-    },
-    'mock-signature-host': {
-        desc: 'Mock signature host so any signature starts with 0xdeadbeef and filled by 0xcd is considered valid',
-        boolean: true,
-    },
 };
-(0, yargs_1.default)((0, helpers_1.hideBin)(process.argv))
+const commands = (0, yargs_1.default)((0, helpers_1.hideBin)(process.argv))
     .scriptName('chopsticks')
+    .middleware(processArgv, true)
     .command('*', 'Dev mode, fork off a chain', (yargs) => yargs.options({
-    ...defaultOptions,
-    ...mockOptions,
+    ...cli_options_1.defaultOptions,
+    ...cli_options_1.mockOptions,
     port: {
         desc: 'Port to listen on',
         number: true,
     },
     'build-block-mode': {
         desc: 'Build block mode. Default to Batch',
-        enum: [_1.BuildBlockMode.Batch, _1.BuildBlockMode.Manual, _1.BuildBlockMode.Instant],
+        enum: [chopsticks_core_1.BuildBlockMode.Batch, chopsticks_core_1.BuildBlockMode.Manual, chopsticks_core_1.BuildBlockMode.Instant],
     },
     'allow-unresolved-imports': {
         desc: 'Allow wasm unresolved imports',
         boolean: true,
     },
-}), async (argv) => {
-    await (0, _1.setupWithServer)(await processArgv(argv));
-})
-    .command('run-block', 'Replay a block', (yargs) => yargs.options({
-    ...defaultOptions,
-    ...mockOptions,
-    '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, 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,
-    extrinsic: {
-        desc: 'Extrinsic or call to dry run. If you pass call here then address is required to fake signature',
-        string: true,
-    },
-    address: {
-        desc: 'Address to fake sign extrinsic',
-        string: true,
-    },
-    preimage: {
-        desc: 'Preimage to dry run',
-        string: true,
-    },
-    at: {
-        desc: 'Block hash to dry run',
-        string: true,
-    },
-    'output-path': {
-        desc: 'File path to print output',
-        string: true,
-    },
-    html: {
-        desc: 'Generate html with storage diff',
-    },
-    open: {
-        desc: 'Open generated html',
+    'max-memory-block-count': {
+        desc: 'Max memory block count',
+        number: true,
     },
 }), async (argv) => {
-    const config = await processArgv(argv);
-    if (config.preimage) {
-        await (0, dry_run_preimage_1.dryRunPreimage)(config);
-    }
-    else {
-        await (0, dry_run_1.dryRun)(config);
-    }
-})
-    .command('decode-key <key>', 'Deocde a key', (yargs) => yargs
-    .positional('key', {
-    desc: 'Key to decode',
-    type: 'string',
-})
-    .options({
-    ...defaultOptions,
-}), async (argv) => {
-    const context = await (0, _1.setup)(await processArgv(argv));
-    const { storage, decodedKey } = (0, decoder_1.decodeKey)(await context.chain.head.meta, context.chain.head, argv.key);
-    if (storage && decodedKey) {
-        console.log(`${storage.section}.${storage.method}`, decodedKey.args.map((x) => JSON.stringify(x.toHuman())).join(', '));
-    }
-    else {
-        console.log('Unknown');
-    }
-    process.exit(0);
+    await (0, _1.setupWithServer)(argv);
 })
     .command('xcm', 'XCM setup with relaychain and parachains', (yargs) => yargs
     .options({
@@ -225,16 +61,16 @@ const mockOptions = {
     .alias('parachain', 'p'), async (argv) => {
     const parachains = [];
     for (const config of argv.parachain) {
-        const { chain } = await (0, _1.setupWithServer)(await (0, exports.fetchConfig)(config));
+        const { chain } = await (0, _1.setupWithServer)(await (0, schema_1.fetchConfig)(config));
         parachains.push(chain);
     }
     if (parachains.length > 1) {
-        await (0, _1.connectParachains)(parachains);
+        await (0, chopsticks_core_1.connectParachains)(parachains);
     }
     if (argv.relaychain) {
-        const { chain: relaychain } = await (0, _1.setupWithServer)(await (0, exports.fetchConfig)(argv.relaychain));
+        const { chain: relaychain } = await (0, _1.setupWithServer)(await (0, schema_1.fetchConfig)(argv.relaychain));
         for (const parachain of parachains) {
-            await (0, _1.connectVertical)(relaychain, parachain);
+            await (0, chopsticks_core_1.connectVertical)(relaychain, parachain);
         }
     }
 })
@@ -249,4 +85,5 @@ const mockOptions = {
     .alias('import-storage', 's')
     .alias('wasm-override', 'w')
     .usage('Usage: $0 <command> [options]')
-    .example('$0', '-c acala').argv;
+    .example('$0', '-c acala');
+(0, plugins_1.pluginExtendCli)(commands).then(() => commands.parse());

package/lib/context.d.ts

@@ -0,0 +1,5 @@
+import './utils/tunnel';
+import { Config } from './schema';
+export declare const setupContext: (argv: Config, overrideParent?: boolean) => Promise<{
+    chain: import("@acala-network/chopsticks-core").Blockchain;
+}>;

package/lib/context.js

@@ -0,0 +1,37 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.setupContext = void 0;
+require("./utils/tunnel");
+const override_1 = require("./utils/override");
+const chopsticks_core_1 = require("@acala-network/chopsticks-core");
+const setupContext = async (argv, overrideParent = false) => {
+    const chain = await (0, chopsticks_core_1.setup)({
+        endpoint: argv.endpoint,
+        block: argv.block,
+        genesis: argv.genesis,
+        buildBlockMode: argv['build-block-mode'],
+        db: argv.db,
+        mockSignatureHost: argv['mock-signature-host'],
+        allowUnresolvedImports: argv['allow-unresolved-imports'],
+        runtimeLogLevel: argv['runtime-log-level'],
+        registeredTypes: argv['registered-types'],
+        offchainWorker: argv['offchain-worker'],
+        maxMemoryBlockCount: argv['max-memory-block-count'],
+    });
+    if (argv.timestamp)
+        await (0, chopsticks_core_1.timeTravel)(chain, argv.timestamp);
+    let at;
+    if (overrideParent) {
+        // 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, override_1.overrideStorage)(chain, argv['import-storage'], at);
+    await (0, override_1.overrideWasm)(chain, argv['wasm-override'], at);
+    return { chain };
+};
+exports.setupContext = setupContext;

package/lib/index.d.ts

@@ -1,8 +1,3 @@
-export { Api } from './api';
-export { Blockchain } from './blockchain';
-export { BuildBlockMode } from './blockchain/txpool';
-export { connectParachains, connectVertical } from './xcm';
-export { setup } from './setup';
+export * from '@acala-network/chopsticks-core';
 export { setupWithServer } from './setup-with-server';
-export { fetchConfig } from './cli';
-export * from './blockchain/inherent';
+export { fetchConfig } from './schema';

package/lib/index.js

@@ -14,20 +14,9 @@ var __exportStar = (this && this.__exportStar) || function(m, exports) {
     for (var p in m) if (p !== "default" && !Object.prototype.hasOwnProperty.call(exports, p)) __createBinding(exports, m, p);
 };
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.fetchConfig = exports.setupWithServer = exports.setup = exports.connectVertical = exports.connectParachains = exports.BuildBlockMode = exports.Blockchain = exports.Api = void 0;
-var api_1 = require("./api");
-Object.defineProperty(exports, "Api", { enumerable: true, get: function () { return api_1.Api; } });
-var blockchain_1 = require("./blockchain");
-Object.defineProperty(exports, "Blockchain", { enumerable: true, get: function () { return blockchain_1.Blockchain; } });
-var txpool_1 = require("./blockchain/txpool");
-Object.defineProperty(exports, "BuildBlockMode", { enumerable: true, get: function () { return txpool_1.BuildBlockMode; } });
-var xcm_1 = require("./xcm");
-Object.defineProperty(exports, "connectParachains", { enumerable: true, get: function () { return xcm_1.connectParachains; } });
-Object.defineProperty(exports, "connectVertical", { enumerable: true, get: function () { return xcm_1.connectVertical; } });
-var setup_1 = require("./setup");
-Object.defineProperty(exports, "setup", { enumerable: true, get: function () { return setup_1.setup; } });
+exports.fetchConfig = exports.setupWithServer = void 0;
+__exportStar(require("@acala-network/chopsticks-core"), exports);
 var setup_with_server_1 = require("./setup-with-server");
 Object.defineProperty(exports, "setupWithServer", { enumerable: true, get: function () { return setup_with_server_1.setupWithServer; } });
-var cli_1 = require("./cli");
-Object.defineProperty(exports, "fetchConfig", { enumerable: true, get: function () { return cli_1.fetchConfig; } });
-__exportStar(require("./blockchain/inherent"), exports);
+var schema_1 = require("./schema");
+Object.defineProperty(exports, "fetchConfig", { enumerable: true, get: function () { return schema_1.fetchConfig; } });

package/lib/logger.d.ts

@@ -1,7 +1 @@
-export declare const defaultLogger: import("pino").Logger<{
-    level: string;
-    transport: {
-        target: string;
-    };
-}>;
-export declare const truncate: (val: any) => any;
+export { defaultLogger, truncate } from '@acala-network/chopsticks-core';

package/lib/logger.js

@@ -1,41 +1,6 @@
 "use strict";
-var __importDefault = (this && this.__importDefault) || function (mod) {
-    return (mod && mod.__esModule) ? mod : { "default": mod };
-};
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.truncate = exports.defaultLogger = void 0;
-const pino_1 = __importDefault(require("pino"));
-exports.defaultLogger = (0, pino_1.default)({
-    level: process.env.LOG_LEVEL || 'info',
-    transport: {
-        target: 'pino-pretty',
-    },
-});
-const innerTruncate = (level = 0) => (val) => {
-    const verboseLog = !!process.env.VERBOSE_LOG;
-    const levelLimit = verboseLog ? 10 : 5;
-    if (val == null) {
-        return val;
-    }
-    if (level > levelLimit) {
-        return '( Too Deep )';
-    }
-    switch (typeof val) {
-        case 'string':
-            if (val.length > 66 && !verboseLog) {
-                return val.slice(0, 34) + '…' + val.slice(-32);
-            }
-            else {
-                return val;
-            }
-        case 'object':
-            if (Array.isArray(val)) {
-                return val.map(innerTruncate(level + 1));
-            }
-            return Object.fromEntries(Object.entries(val.toJSON ? val.toJSON() : val).map(([k, v]) => [k, innerTruncate(level + 1)(v)]));
-        default:
-            return val;
-    }
-};
-const truncate = (val) => innerTruncate(0)(val);
-exports.truncate = truncate;
+var chopsticks_core_1 = require("@acala-network/chopsticks-core");
+Object.defineProperty(exports, "defaultLogger", { enumerable: true, get: function () { return chopsticks_core_1.defaultLogger; } });
+Object.defineProperty(exports, "truncate", { enumerable: true, get: function () { return chopsticks_core_1.truncate; } });

package/lib/plugins/decode-key/index.d.ts

@@ -0,0 +1,2 @@
+import type yargs from 'yargs';
+export declare const cli: (y: yargs.Argv) => void;

package/lib/plugins/decode-key/index.js

@@ -0,0 +1,27 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.cli = void 0;
+const decoder_1 = require("../../utils/decoder");
+const cli_options_1 = require("../../cli-options");
+const context_1 = require("../../context");
+const cli = (y) => {
+    y.command('decode-key <key>', 'Deocde a key', (yargs) => yargs
+        .positional('key', {
+        desc: 'Key to decode',
+        type: 'string',
+    })
+        .options({
+        ...cli_options_1.defaultOptions,
+    }), async (argv) => {
+        const context = await (0, context_1.setupContext)(argv);
+        const { storage, decodedKey } = (0, decoder_1.decodeKey)(await context.chain.head.meta, context.chain.head, argv.key);
+        if (storage && decodedKey) {
+            console.log(`${storage.section}.${storage.method}`, decodedKey.args.map((x) => JSON.stringify(x.toHuman())).join(', '));
+        }
+        else {
+            console.log('Unknown');
+        }
+        process.exit(0);
+    });
+};
+exports.cli = cli;

package/lib/plugins/dry-run/cli.d.ts

@@ -0,0 +1,2 @@
+import type yargs from 'yargs';
+export declare const cli: (y: yargs.Argv) => void;

package/lib/plugins/dry-run/cli.js

@@ -0,0 +1,46 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.cli = void 0;
+const cli_options_1 = require("../../cli-options");
+const dry_run_extrinsic_1 = require("./dry-run-extrinsic");
+const dry_run_preimage_1 = require("./dry-run-preimage");
+const cli = (y) => {
+    y.command('dry-run', 'Dry run an extrinsic', (yargs) => yargs.options({
+        ...cli_options_1.defaultOptions,
+        ...cli_options_1.mockOptions,
+        extrinsic: {
+            desc: 'Extrinsic or call to dry run. If you pass call here then address is required to fake signature',
+            string: true,
+        },
+        address: {
+            desc: 'Address to fake sign extrinsic',
+            string: true,
+        },
+        preimage: {
+            desc: 'Preimage to dry run',
+            string: true,
+        },
+        at: {
+            desc: 'Block hash to dry run',
+            string: 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) => {
+        if (argv.preimage) {
+            await (0, dry_run_preimage_1.dryRunPreimage)(argv);
+        }
+        else {
+            await (0, dry_run_extrinsic_1.dryRunExtrinsic)(argv);
+        }
+    });
+};
+exports.cli = cli;

package/lib/plugins/dry-run/dry-run-extrinsic.d.ts

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

package/lib/{dry-run.js → plugins/dry-run/dry-run-extrinsic.js}

@@ -1,14 +1,14 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.dryRun = void 0;
+exports.dryRunExtrinsic = void 0;
 const util_crypto_1 = require("@polkadot/util-crypto");
 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 setup_1 = require("./setup");
-const dryRun = async (argv) => {
-    const context = await (0, setup_1.setup)(argv);
+const logger_1 = require("../../logger");
+const generate_html_diff_1 = require("../../utils/generate-html-diff");
+const open_html_1 = require("../../utils/open-html");
+const context_1 = require("../../context");
+const dryRunExtrinsic = async (argv) => {
+    const context = await (0, context_1.setupContext)(argv);
     const input = argv['address'] ? { call: argv['extrinsic'], address: argv['address'] } : argv['extrinsic'];
     const { outcome, storageDiff } = await context.chain.dryRunExtrinsic(input, argv['at']);
     if (outcome.isErr) {
@@ -30,4 +30,4 @@ const dryRun = async (argv) => {
     }
     process.exit(0);
 };
-exports.dryRun = dryRun;
+exports.dryRunExtrinsic = dryRunExtrinsic;

package/lib/{dry-run-preimage.d.ts → plugins/dry-run/dry-run-preimage.d.ts}

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

package/lib/{dry-run-preimage.js → plugins/dry-run/dry-run-preimage.js}

@@ -3,22 +3,20 @@ Object.defineProperty(exports, "__esModule", { value: true });
 exports.dryRunPreimage = void 0;
 const util_crypto_1 = require("@polkadot/util-crypto");
 const util_1 = require("@polkadot/util");
-const logger_1 = require("./logger");
-const generate_html_diff_1 = require("./utils/generate-html-diff");
-const block_builder_1 = require("./blockchain/block-builder");
-const open_html_1 = require("./utils/open-html");
-const executor_1 = require("./executor");
-const set_storage_1 = require("./utils/set-storage");
-const setup_1 = require("./setup");
+const logger_1 = require("../../logger");
+const generate_html_diff_1 = require("../../utils/generate-html-diff");
+const chopsticks_core_1 = require("@acala-network/chopsticks-core");
+const open_html_1 = require("../../utils/open-html");
+const context_1 = require("../../context");
 const dryRunPreimage = async (argv) => {
-    const context = await (0, setup_1.setup)(argv);
+    const context = await (0, context_1.setupContext)(argv);
     const extrinsic = argv['preimage'];
     const block = context.chain.head;
     const registry = await block.registry;
-    const header = await (0, block_builder_1.newHeader)(block);
+    const header = await (0, chopsticks_core_1.newHeader)(block);
     const data = (0, util_1.hexToU8a)(extrinsic);
     const hash = (0, util_crypto_1.blake2AsHex)(data, 256);
-    await (0, set_storage_1.setStorage)(context.chain, {
+    await (0, chopsticks_core_1.setStorage)(context.chain, {
         Preimage: {
             PreimageFor: [[[[hash, data.byteLength]], extrinsic]],
             StatusFor: [
@@ -61,13 +59,13 @@ const dryRunPreimage = async (argv) => {
     }
     calls.push(['BlockBuilder_finalize_block', []]);
     logger_1.defaultLogger.info({ preimage: registry.createType('Call', data).toHuman() }, 'Dry run preimage');
-    const result = await (0, executor_1.runTask)({
+    const result = await (0, chopsticks_core_1.runTask)({
         wasm: await block.wasm,
         calls,
         mockSignatureHost: false,
         allowUnresolvedImports: false,
         runtimeLogLevel: argv['runtime-log-level'] || 0,
-    }, (0, executor_1.taskHandler)(block));
+    }, (0, chopsticks_core_1.taskHandler)(block));
     if (result.Error) {
         throw new Error(result.Error);
     }

package/lib/plugins/dry-run/index.d.ts

@@ -0,0 +1,2 @@
+export * from './cli';
+export * from './rpc';