0xweb/dequanto/src/evm/utils/stringifyMappings.ts

Contract package manager and other web3 tools

const stringifyMapping = (mapping: any) => {
    const mappingKey: string[] = [];
    const mappingValue: string[] = [];
    let deepMapping = false;
    mapping.keys
        .filter((mappingChild: any) => mappingChild.length > 0)
        .forEach((mappingChild: any) => {
            if (
                mappingChild.length > 0 &&
                mappingChild[0].type &&
                !mappingKey.includes(mappingChild[0].type)
            ) {
                mappingKey.push(mappingChild[0].type);
            }
            if (mappingChild.length > 1 && !deepMapping) {
                deepMapping = true;
                mappingValue.push(
                    stringifyMapping({
                        name: mapping.name,
                        structs: mapping.structs,
                        keys: mapping.keys.map((items: any) => {
                            items.shift();
                            return items;
                        }),
                        values: mapping.values
                    })
                );
            } else if (mappingChild.length === 1 && !deepMapping) {
                mapping.values.forEach((mappingChild2: any) => {
                    if (mappingChild2.type && !mappingValue.includes(mappingChild2.type)) {
                        mappingValue.push(mappingChild2.type);
                    }
                });
            }
        });
    if (mappingKey.length === 0) {
        mappingKey.push('unknown');
    }
    if (mapping.structs.length > 0 && mappingValue.length === 0) {
        mappingValue.push(mapping.name + 'Struct');
    } else if (mappingValue.length === 0) {
        mappingValue.push('unknown');
    }
    return 'mapping (' + mappingKey.join('|') + ' => ' + mappingValue.join('|') + ')';
};

export default (mappings: any) => {
    let output = '';

    Object.keys(mappings).forEach((key: string, index: number) => {
        const mapping = mappings[key];
        if (mapping.name) {
            output += stringifyMapping(mapping) + ' public ' + mapping.name + ';';
        } else {
            output += stringifyMapping(mapping) + ' mapping' + (index + 1) + ';';
        }
        output += '\n';
    });

    if (Object.keys(mappings).length > 0) {
        output += '\n';
    }

    return output;
};