UNPKG

@neo-one/smart-contract-compiler

Version:

NEO•ONE TypeScript smart contract compiler.

39 lines (37 loc) 1.73 kB
"use strict"; Object.defineProperty(exports, "__esModule", { value: true }); exports.MapStorageSet = void 0; const tslib_1 = require("tslib"); const ts_utils_1 = require("@neo-one/ts-utils"); const typescript_1 = tslib_1.__importDefault(require("typescript")); const constants_1 = require("../../../constants"); const BuiltinInstanceMemberCall_1 = require("../../BuiltinInstanceMemberCall"); class MapStorageSet extends BuiltinInstanceMemberCall_1.BuiltinInstanceMemberCall { canCall(_sb, _func, node) { return ts_utils_1.tsUtils.argumented.getArguments(node).length === 2; } emitCall(sb, func, node, optionsIn, visited) { const options = sb.pushValueOptions(optionsIn); if (!visited) { sb.visit(ts_utils_1.tsUtils.expression.getExpression(func), options); } if (ts_utils_1.tsUtils.argumented.getArguments(node).length < 2) { return; } const key = ts_utils_1.tsUtils.argumented.getArguments(node)[0]; const keyType = sb.context.analysis.getType(key); if (optionsIn.pushValue) { sb.emitOp(node, 'DUP'); } sb.visit(key, options); if (key.kind === typescript_1.default.SyntaxKind.NumericLiteral) { sb.emitHelper(key, options, sb.helpers.unwrapNumber); sb.emitHelper(key, options, sb.helpers.coerceToInt); sb.emitHelper(key, options, sb.helpers.wrapNumber); } sb.visit(ts_utils_1.tsUtils.argumented.getArguments(node)[1], options); sb.emitHelper(node, optionsIn, sb.helpers.setStructuredStorage({ type: constants_1.Types.MapStorage, keyType })); } } exports.MapStorageSet = MapStorageSet; //# sourceMappingURL=set.js.map