int-cli/src/tool/restore_storage.js

INT is the new generation of bottom-up created system of IoT and blockchain

#!/usr/bin/env node
"use strict";
Object.defineProperty(exports, "__esModule", { value: true });
const process = require("process");
const fs = require("fs-extra");
const client_1 = require("../client");
const core_1 = require("../core");
const logger = client_1.initLogger({ loggerOptions: { console: true } });
client_1.initUnhandledRejection(logger);
async function main() {
    const commandTip = `Usage: node restore_storage.js restore --dateDir [data dir] --height [block height] --output [output path]`;
    let command = client_1.parseCommand(process.argv);
    if (!command || !command.command) {
        console.log(commandTip);
    }
    if (command.command === 'restore') {
        const dataDir = command.options.get('dataDir');
        if (!dataDir) {
            console.log(commandTip);
            return;
        }
        const output = command.options.get('output');
        if (!output) {
            console.log(commandTip);
            return;
        }
        const chainCreator = core_1.initChainCreator({ logger });
        const ccir = await chainCreator.createChainInstance(dataDir, { readonly: true, initComponents: true });
        if (ccir.err) {
            chainCreator.logger.error(`create chain instance from ${dataDir} failed `, core_1.stringifyErrorCode(ccir.err));
            return;
        }
        let height = parseInt(command.options.get('height'));
        let headerRet = await ccir.chain.headerStorage.getHeader(height);
        if (headerRet.err) {
            console.log(`get header error ${headerRet.err}, exit.`);
            return;
        }
        console.log(`recovering storage for Block ${headerRet.header.hash}...`);
        const csr = await ccir.chain.storageManager.createStorage('temp', headerRet.header.hash);
        if (csr.err) {
            console.log(`restore storage from redo log failed ${core_1.stringifyErrorCode(csr.err)}`);
            return;
        }
        fs.ensureDirSync(output);
        await csr.storage.uninit();
        fs.copyFileSync(csr.storage.filePath, output);
        console.log(`restore complete.`);
    }
    else {
        console.log(commandTip);
    }
}
if (require.main === module) {
    main();
    process.exit();
}