@ocap/contract 1.28.8 → 1.29.0

package/README.md

@@ -1,6 +1,5 @@
 ![nft](https://www.arcblock.io/.netlify/functions/badge/?text=OCAP+Contract)
 
-[![styled with prettier](https://img.shields.io/badge/styled_with-prettier-ff69b4.svg)](https://github.com/prettier/prettier)
 [![docs](https://img.shields.io/badge/powered%20by-arcblock-green.svg)](https://docs.arcblock.io)
 [![Gitter](https://badges.gitter.im/ArcBlock/community.svg)](https://gitter.im/ArcBlock/community?utm_source=badge&utm_medium=badge&utm_campaign=pr-badge)
 

package/esm/index.d.mts

@@ -0,0 +1,45 @@
+//#region src/index.d.ts
+interface CallArg {
+  type: string;
+  name: string;
+}
+interface SupportedCall {
+  args: CallArg[];
+}
+interface CompiledCall {
+  call: string;
+  args: Record<string, string>;
+}
+interface Quota {
+  value: string;
+  tokens?: Record<string, string>;
+}
+interface QuotaInput {
+  value?: string;
+  tokens?: Array<{
+    address: string;
+    value: string;
+  }>;
+}
+declare const SUPPORTED_CALLS: Record<string, SupportedCall>;
+/**
+ * Try to compile contract source code and return call set
+ *
+ * @param source
+ * @return list of parsed contract call and its args
+ * @throws Error for invalid contract or args
+ */
+declare const compile: (source: string) => CompiledCall[];
+/**
+ * Validate the compiled contract source
+ *
+ * @param compiled contract call list
+ * @param quota { value, tokens: { [address]: value }}
+ * @param ensureZeroSum
+ * @return boolean
+ */
+declare const validate: (compiled: CompiledCall[], quota?: Quota, ensureZeroSum?: boolean) => boolean;
+declare const merge: (compiled: CompiledCall[]) => CompiledCall[];
+declare const getQuota: (input?: QuotaInput) => Quota;
+//#endregion
+export { SUPPORTED_CALLS, compile, getQuota, merge, validate };

package/esm/index.mjs

@@ -0,0 +1,170 @@
+import { isValid, toTypeInfo } from "@arcblock/did";
+import { types } from "@ocap/mcrypto";
+import { BN } from "@ocap/util";
+import * as esprima from "esprima";
+import get from "lodash/get.js";
+import set from "lodash/set.js";
+
+//#region src/index.ts
+const isCallExpression = (x) => x.type === "ExpressionStatement" && x.expression && x.expression.type === "CallExpression";
+const ZERO = new BN("0");
+const TOKEN_RECEIVER_ROLE_TYPES = [
+	types.RoleType.ROLE_APPLICATION,
+	types.RoleType.ROLE_ACCOUNT,
+	types.RoleType.ROLE_BLOCKLET
+];
+const SUPPORTED_CALLS = {
+	transfer: { args: [{
+		type: "Literal",
+		name: "to"
+	}, {
+		type: "Literal",
+		name: "amount"
+	}] },
+	transferToken: { args: [
+		{
+			type: "Literal",
+			name: "tokenAddress"
+		},
+		{
+			type: "Literal",
+			name: "to"
+		},
+		{
+			type: "Literal",
+			name: "amount"
+		}
+	] }
+};
+/**
+* Try to compile contract source code and return call set
+*
+* @param source
+* @return list of parsed contract call and its args
+* @throws Error for invalid contract or args
+*/
+const compile = (source) => {
+	if (typeof source !== "string" || !source) throw new Error("contract source must be a non-empty string");
+	try {
+		const { body } = esprima.parseScript(source);
+		if (body.every((x) => isCallExpression(x)) === false) throw new Error("only function call expressions are allowed");
+		const result = [];
+		const expressions = body.map((x) => x.expression);
+		const allowedCalls = Object.keys(SUPPORTED_CALLS);
+		for (const expression of expressions) {
+			const name = get(expression, "callee.name");
+			if (allowedCalls.includes(name) === false) throw new Error(`unrecognized call ${name}`);
+			const expectedArgs = SUPPORTED_CALLS[name].args;
+			const actualArgs = expression.arguments;
+			const expectedArgCount = expectedArgs.length;
+			const actualArgCount = actualArgs.length;
+			if (actualArgCount !== expectedArgCount) throw new Error(`invalid number of arguments for ${name}, expected ${expectedArgCount}, but found ${actualArgCount}`);
+			for (let i = 0; i < expectedArgs.length; i++) {
+				const expectedArg = expectedArgs[i];
+				const actualArg = actualArgs[i];
+				if (expectedArg.type !== actualArg.type) throw new Error(`invalid #${i + 1} argument for ${name}, expected ${expectedArg.type}, but found ${actualArg.type}`);
+			}
+			result.push({
+				call: name,
+				args: expectedArgs.reduce((acc, x, i) => {
+					acc[x.name] = actualArgs[i].value;
+					return acc;
+				}, {})
+			});
+		}
+		if (result.length === 0) throw new Error("compiled contract result is empty");
+		return result;
+	} catch (err) {
+		throw new Error(`Invalid contract source: ${err.message}`);
+	}
+};
+const validateDid = (to, allowedRoleTypes) => {
+	if (isValid(to) === false) throw new Error("invalid did");
+	const type = toTypeInfo(to);
+	if (typeof type.role !== "number" || allowedRoleTypes.includes(type.role) === false) throw new Error("invalid did role type");
+	return true;
+};
+const validateAmount = (amount) => {
+	let bn = null;
+	try {
+		bn = new BN(amount);
+	} catch (err) {
+		throw new Error(`invalid amount: ${err.message}`);
+	}
+	if (bn.toString() !== amount) throw new Error("invalid amount: not valid big number");
+	return bn;
+};
+/**
+* Validate the compiled contract source
+*
+* @param compiled contract call list
+* @param quota { value, tokens: { [address]: value }}
+* @param ensureZeroSum
+* @return boolean
+*/
+const validate = (compiled, quota, ensureZeroSum = false) => {
+	if (!Array.isArray(compiled) || compiled.length === 0) throw new Error("no contract calls to validate");
+	if (compiled.length > 30) throw new Error("too much contract calls found, max is 15");
+	let remaining = new BN(quota ? quota.value : "0");
+	const tokens = quota?.tokens ?? {};
+	const tokenRemaining = Object.keys(tokens).reduce((acc, x) => {
+		acc[x] = new BN(tokens[x]);
+		return acc;
+	}, {});
+	for (const item of compiled) {
+		const { call, args } = item;
+		if (call === "transfer") {
+			validateDid(args.to, TOKEN_RECEIVER_ROLE_TYPES);
+			const consumption = validateAmount(args.amount);
+			if (consumption.lte(ZERO)) throw new Error("transfer amount must be greater than 0");
+			if (quota) {
+				if (remaining.sub(consumption).isNeg()) throw new Error("transfer exceeded quota");
+				remaining = remaining.sub(consumption);
+			}
+		} else if (call === "transferToken") {
+			validateDid(args.tokenAddress, [types.RoleType.ROLE_TOKEN]);
+			validateDid(args.to, TOKEN_RECEIVER_ROLE_TYPES);
+			const token = args.tokenAddress;
+			const consumption = validateAmount(args.amount);
+			if (consumption.lte(ZERO)) throw new Error("token transfer amount must be greater than 0");
+			if (quota) {
+				if (!tokenRemaining[token] || tokenRemaining[token].sub(consumption).isNeg()) throw new Error(`transferToken for ${args.tokenAddress} exceeded quota`);
+				tokenRemaining[token] = tokenRemaining[token].sub(consumption);
+			}
+		}
+	}
+	if (ensureZeroSum) {
+		if (remaining.isZero() === false) throw new Error("transfer violates zero sum");
+		const invalidToken = Object.keys(tokenRemaining).find((x) => tokenRemaining[x].isZero() === false);
+		if (invalidToken) throw new Error(`transferToken for ${invalidToken} violates zero sum`);
+	}
+	return true;
+};
+const merge = (compiled) => {
+	const mergeAndSum = (items, uniqKey, sumBy) => {
+		const result = {};
+		const keys = Array.isArray(uniqKey) ? uniqKey : [uniqKey];
+		keys.sort();
+		for (const item of items) {
+			const key = keys.map((x) => get(item, x)).join("-");
+			if (result[key] === void 0) result[key] = item;
+			else {
+				const old = new BN(get(result[key], sumBy));
+				const delta = new BN(get(item, sumBy));
+				set(result[key], sumBy, old.add(delta).toString());
+			}
+		}
+		return Object.values(result);
+	};
+	return [...mergeAndSum(compiled.filter((x) => x.call === "transfer"), "args.to", "args.amount"), ...mergeAndSum(compiled.filter((x) => x.call === "transferToken"), ["args.tokenAddress", "args.to"], "args.amount")];
+};
+const getQuota = (input = {}) => ({
+	value: input.value || "0",
+	tokens: (input.tokens || []).reduce((acc, x) => {
+		acc[x.address] = x.value;
+		return acc;
+	}, {})
+});
+
+//#endregion
+export { SUPPORTED_CALLS, compile, getQuota, merge, validate };

package/lib/_virtual/rolldown_runtime.cjs

@@ -0,0 +1,29 @@
+//#region rolldown:runtime
+var __create = Object.create;
+var __defProp = Object.defineProperty;
+var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
+var __getOwnPropNames = Object.getOwnPropertyNames;
+var __getProtoOf = Object.getPrototypeOf;
+var __hasOwnProp = Object.prototype.hasOwnProperty;
+var __copyProps = (to, from, except, desc) => {
+	if (from && typeof from === "object" || typeof from === "function") {
+		for (var keys = __getOwnPropNames(from), i = 0, n = keys.length, key; i < n; i++) {
+			key = keys[i];
+			if (!__hasOwnProp.call(to, key) && key !== except) {
+				__defProp(to, key, {
+					get: ((k) => from[k]).bind(null, key),
+					enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable
+				});
+			}
+		}
+	}
+	return to;
+};
+var __toESM = (mod, isNodeMode, target) => (target = mod != null ? __create(__getProtoOf(mod)) : {}, __copyProps(isNodeMode || !mod || !mod.__esModule ? __defProp(target, "default", {
+	value: mod,
+	enumerable: true
+}) : target, mod));
+
+//#endregion
+
+exports.__toESM = __toESM;

package/lib/index.cjs

@@ -0,0 +1,178 @@
+const require_rolldown_runtime = require('./_virtual/rolldown_runtime.cjs');
+let _arcblock_did = require("@arcblock/did");
+let _ocap_mcrypto = require("@ocap/mcrypto");
+let _ocap_util = require("@ocap/util");
+let esprima = require("esprima");
+esprima = require_rolldown_runtime.__toESM(esprima);
+let lodash_get = require("lodash/get");
+lodash_get = require_rolldown_runtime.__toESM(lodash_get);
+let lodash_set = require("lodash/set");
+lodash_set = require_rolldown_runtime.__toESM(lodash_set);
+
+//#region src/index.ts
+const isCallExpression = (x) => x.type === "ExpressionStatement" && x.expression && x.expression.type === "CallExpression";
+const ZERO = new _ocap_util.BN("0");
+const TOKEN_RECEIVER_ROLE_TYPES = [
+	_ocap_mcrypto.types.RoleType.ROLE_APPLICATION,
+	_ocap_mcrypto.types.RoleType.ROLE_ACCOUNT,
+	_ocap_mcrypto.types.RoleType.ROLE_BLOCKLET
+];
+const SUPPORTED_CALLS = {
+	transfer: { args: [{
+		type: "Literal",
+		name: "to"
+	}, {
+		type: "Literal",
+		name: "amount"
+	}] },
+	transferToken: { args: [
+		{
+			type: "Literal",
+			name: "tokenAddress"
+		},
+		{
+			type: "Literal",
+			name: "to"
+		},
+		{
+			type: "Literal",
+			name: "amount"
+		}
+	] }
+};
+/**
+* Try to compile contract source code and return call set
+*
+* @param source
+* @return list of parsed contract call and its args
+* @throws Error for invalid contract or args
+*/
+const compile = (source) => {
+	if (typeof source !== "string" || !source) throw new Error("contract source must be a non-empty string");
+	try {
+		const { body } = esprima.parseScript(source);
+		if (body.every((x) => isCallExpression(x)) === false) throw new Error("only function call expressions are allowed");
+		const result = [];
+		const expressions = body.map((x) => x.expression);
+		const allowedCalls = Object.keys(SUPPORTED_CALLS);
+		for (const expression of expressions) {
+			const name = (0, lodash_get.default)(expression, "callee.name");
+			if (allowedCalls.includes(name) === false) throw new Error(`unrecognized call ${name}`);
+			const expectedArgs = SUPPORTED_CALLS[name].args;
+			const actualArgs = expression.arguments;
+			const expectedArgCount = expectedArgs.length;
+			const actualArgCount = actualArgs.length;
+			if (actualArgCount !== expectedArgCount) throw new Error(`invalid number of arguments for ${name}, expected ${expectedArgCount}, but found ${actualArgCount}`);
+			for (let i = 0; i < expectedArgs.length; i++) {
+				const expectedArg = expectedArgs[i];
+				const actualArg = actualArgs[i];
+				if (expectedArg.type !== actualArg.type) throw new Error(`invalid #${i + 1} argument for ${name}, expected ${expectedArg.type}, but found ${actualArg.type}`);
+			}
+			result.push({
+				call: name,
+				args: expectedArgs.reduce((acc, x, i) => {
+					acc[x.name] = actualArgs[i].value;
+					return acc;
+				}, {})
+			});
+		}
+		if (result.length === 0) throw new Error("compiled contract result is empty");
+		return result;
+	} catch (err) {
+		throw new Error(`Invalid contract source: ${err.message}`);
+	}
+};
+const validateDid = (to, allowedRoleTypes) => {
+	if ((0, _arcblock_did.isValid)(to) === false) throw new Error("invalid did");
+	const type = (0, _arcblock_did.toTypeInfo)(to);
+	if (typeof type.role !== "number" || allowedRoleTypes.includes(type.role) === false) throw new Error("invalid did role type");
+	return true;
+};
+const validateAmount = (amount) => {
+	let bn = null;
+	try {
+		bn = new _ocap_util.BN(amount);
+	} catch (err) {
+		throw new Error(`invalid amount: ${err.message}`);
+	}
+	if (bn.toString() !== amount) throw new Error("invalid amount: not valid big number");
+	return bn;
+};
+/**
+* Validate the compiled contract source
+*
+* @param compiled contract call list
+* @param quota { value, tokens: { [address]: value }}
+* @param ensureZeroSum
+* @return boolean
+*/
+const validate = (compiled, quota, ensureZeroSum = false) => {
+	if (!Array.isArray(compiled) || compiled.length === 0) throw new Error("no contract calls to validate");
+	if (compiled.length > 30) throw new Error("too much contract calls found, max is 15");
+	let remaining = new _ocap_util.BN(quota ? quota.value : "0");
+	const tokens = quota?.tokens ?? {};
+	const tokenRemaining = Object.keys(tokens).reduce((acc, x) => {
+		acc[x] = new _ocap_util.BN(tokens[x]);
+		return acc;
+	}, {});
+	for (const item of compiled) {
+		const { call, args } = item;
+		if (call === "transfer") {
+			validateDid(args.to, TOKEN_RECEIVER_ROLE_TYPES);
+			const consumption = validateAmount(args.amount);
+			if (consumption.lte(ZERO)) throw new Error("transfer amount must be greater than 0");
+			if (quota) {
+				if (remaining.sub(consumption).isNeg()) throw new Error("transfer exceeded quota");
+				remaining = remaining.sub(consumption);
+			}
+		} else if (call === "transferToken") {
+			validateDid(args.tokenAddress, [_ocap_mcrypto.types.RoleType.ROLE_TOKEN]);
+			validateDid(args.to, TOKEN_RECEIVER_ROLE_TYPES);
+			const token = args.tokenAddress;
+			const consumption = validateAmount(args.amount);
+			if (consumption.lte(ZERO)) throw new Error("token transfer amount must be greater than 0");
+			if (quota) {
+				if (!tokenRemaining[token] || tokenRemaining[token].sub(consumption).isNeg()) throw new Error(`transferToken for ${args.tokenAddress} exceeded quota`);
+				tokenRemaining[token] = tokenRemaining[token].sub(consumption);
+			}
+		}
+	}
+	if (ensureZeroSum) {
+		if (remaining.isZero() === false) throw new Error("transfer violates zero sum");
+		const invalidToken = Object.keys(tokenRemaining).find((x) => tokenRemaining[x].isZero() === false);
+		if (invalidToken) throw new Error(`transferToken for ${invalidToken} violates zero sum`);
+	}
+	return true;
+};
+const merge = (compiled) => {
+	const mergeAndSum = (items, uniqKey, sumBy) => {
+		const result = {};
+		const keys = Array.isArray(uniqKey) ? uniqKey : [uniqKey];
+		keys.sort();
+		for (const item of items) {
+			const key = keys.map((x) => (0, lodash_get.default)(item, x)).join("-");
+			if (result[key] === void 0) result[key] = item;
+			else {
+				const old = new _ocap_util.BN((0, lodash_get.default)(result[key], sumBy));
+				const delta = new _ocap_util.BN((0, lodash_get.default)(item, sumBy));
+				(0, lodash_set.default)(result[key], sumBy, old.add(delta).toString());
+			}
+		}
+		return Object.values(result);
+	};
+	return [...mergeAndSum(compiled.filter((x) => x.call === "transfer"), "args.to", "args.amount"), ...mergeAndSum(compiled.filter((x) => x.call === "transferToken"), ["args.tokenAddress", "args.to"], "args.amount")];
+};
+const getQuota = (input = {}) => ({
+	value: input.value || "0",
+	tokens: (input.tokens || []).reduce((acc, x) => {
+		acc[x.address] = x.value;
+		return acc;
+	}, {})
+});
+
+//#endregion
+exports.SUPPORTED_CALLS = SUPPORTED_CALLS;
+exports.compile = compile;
+exports.getQuota = getQuota;
+exports.merge = merge;
+exports.validate = validate;

package/lib/index.d.cts

@@ -0,0 +1,45 @@
+//#region src/index.d.ts
+interface CallArg {
+  type: string;
+  name: string;
+}
+interface SupportedCall {
+  args: CallArg[];
+}
+interface CompiledCall {
+  call: string;
+  args: Record<string, string>;
+}
+interface Quota {
+  value: string;
+  tokens?: Record<string, string>;
+}
+interface QuotaInput {
+  value?: string;
+  tokens?: Array<{
+    address: string;
+    value: string;
+  }>;
+}
+declare const SUPPORTED_CALLS: Record<string, SupportedCall>;
+/**
+ * Try to compile contract source code and return call set
+ *
+ * @param source
+ * @return list of parsed contract call and its args
+ * @throws Error for invalid contract or args
+ */
+declare const compile: (source: string) => CompiledCall[];
+/**
+ * Validate the compiled contract source
+ *
+ * @param compiled contract call list
+ * @param quota { value, tokens: { [address]: value }}
+ * @param ensureZeroSum
+ * @return boolean
+ */
+declare const validate: (compiled: CompiledCall[], quota?: Quota, ensureZeroSum?: boolean) => boolean;
+declare const merge: (compiled: CompiledCall[]) => CompiledCall[];
+declare const getQuota: (input?: QuotaInput) => Quota;
+//#endregion
+export { SUPPORTED_CALLS, compile, getQuota, merge, validate };

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "@ocap/contract",
   "description": "Utility to compile/validate and run ocap contract",
-  "version": "1.28.8",
+  "version": "1.29.0",
   "author": {
     "name": "wangshijun",
     "email": "shijun@arcblock.io",
@@ -17,15 +17,39 @@
   "contributors": [
     "wangshijun <shijun@arcblock.io> (https://github.com/wangshijun)"
   ],
+  "type": "module",
+  "main": "./lib/index.cjs",
+  "module": "./esm/index.mjs",
+  "types": "./esm/index.d.mts",
+  "exports": {
+    ".": {
+      "types": "./esm/index.d.mts",
+      "import": "./esm/index.mjs",
+      "default": "./lib/index.cjs"
+    },
+    "./lib/*.js": {
+      "types": "./esm/*.d.mts",
+      "import": "./esm/*.mjs",
+      "default": "./lib/*.cjs"
+    },
+    "./lib/*": {
+      "types": "./esm/*.d.mts",
+      "import": "./esm/*.mjs",
+      "default": "./lib/*.cjs"
+    }
+  },
   "dependencies": {
-    "@arcblock/did": "1.28.8",
-    "@ocap/mcrypto": "1.28.8",
-    "@ocap/util": "1.28.8",
-    "debug": "^4.3.6",
+    "@arcblock/did": "1.29.0",
+    "@ocap/mcrypto": "1.29.0",
+    "@ocap/util": "1.29.0",
+    "debug": "^4.4.3",
     "esprima": "^4.0.1",
-    "lodash": "^4.17.21"
+    "lodash": "^4.17.23"
   },
   "devDependencies": {
+    "@types/esprima": "^4.0.6",
+    "@types/estree": "^1.0.6",
+    "@types/lodash": "^4.17.16",
     "remark-cli": "^10.0.1",
     "remark-preset-github": "^4.0.4"
   },
@@ -47,21 +71,23 @@
     "nodejs"
   ],
   "license": "Apache-2.0",
-  "main": "./lib/index.js",
   "files": [
-    "lib"
+    "lib",
+    "esm"
   ],
   "repository": {
     "type": "git",
     "url": "https://github.com/ArcBlock/blockchain/tree/master/core/contract"
   },
   "scripts": {
+    "build": "tsdown",
+    "prebuild": "rm -rf lib esm",
     "lint": "biome check",
     "lint:fix": "biome check --write",
     "docs": "bun run gen-dts && bun run gen-docs && bun run cleanup-docs && bun run format-docs",
     "cleanup-docs": "node ../../scripts/cleanup-docs.js docs/README.md $npm_package_name",
-    "gen-dts": "j2d lib/index.js",
-    "gen-docs": "jsdoc2md lib/index.js > docs/README.md",
+    "gen-dts": "j2d lib/index.cjs",
+    "gen-docs": "jsdoc2md lib/index.cjs > docs/README.md",
     "format-docs": "remark . -o",
     "test": "bun test",
     "coverage": "npm run test -- --coverage"

package/lib/index.d.ts

@@ -1,31 +0,0 @@
-// Generate by [js2dts@0.3.3](https://github.com/whxaxes/js2dts#readme)
-
-declare const _Lib: _Lib.T105;
-declare namespace _Lib {
-  export interface T100 {
-    [key: string]: any;
-  }
-  export interface T101 {
-    value: any;
-    tokens: any;
-  }
-  export interface T102 {
-    type: string;
-    name: string;
-  }
-  export interface T103 {
-    args: _Lib.T102[];
-  }
-  export interface T104 {
-    transfer: _Lib.T103;
-    transferToken: _Lib.T103;
-  }
-  export interface T105 {
-    compile: (source: string, ...args: any[]) => any[];
-    validate: (compiled: any[], quota: any, ensureZeroSum?: boolean) => boolean;
-    merge: (compiled: any) => any[];
-    getQuota: (input?: _Lib.T100) => _Lib.T101;
-    SUPPORTED_CALLS: _Lib.T104;
-  }
-}
-export = _Lib;

package/lib/index.js

@@ -1,260 +0,0 @@
-const get = require('lodash/get');
-const set = require('lodash/set');
-const esprima = require('esprima');
-const { types } = require('@ocap/mcrypto');
-const { BN } = require('@ocap/util');
-const { isValid, toTypeInfo } = require('@arcblock/did');
-
-const isCallExpression = (x) =>
-  x.type === 'ExpressionStatement' && x.expression && x.expression.type === 'CallExpression';
-
-const ZERO = new BN('0');
-
-const TOKEN_RECEIVER_ROLE_TYPES = [
-  types.RoleType.ROLE_APPLICATION,
-  types.RoleType.ROLE_ACCOUNT,
-  types.RoleType.ROLE_BLOCKLET,
-];
-
-const SUPPORTED_CALLS = {
-  // used to transfer primary token to someone
-  transfer: {
-    args: [
-      {
-        type: 'Literal',
-        name: 'to',
-      },
-      {
-        type: 'Literal',
-        name: 'amount',
-      },
-    ],
-  },
-  // used to transfer secondary token to someone
-  transferToken: {
-    args: [
-      {
-        type: 'Literal',
-        name: 'tokenAddress',
-      },
-      {
-        type: 'Literal',
-        name: 'to',
-      },
-      {
-        type: 'Literal',
-        name: 'amount',
-      },
-    ],
-  },
-};
-
-/**
- * Try to compile contract source code and return call set
- *
- * @param {string} source
- * @return {array} list of parsed contract call and its args
- * @throws {Error} for invalid contract or args
- */
-const compile = (source) => {
-  if (typeof source !== 'string' || !source) {
-    throw new Error('contract source must be a non-empty string');
-  }
-
-  try {
-    const { body } = esprima.parseScript(source);
-    if (body.every((x) => isCallExpression(x)) === false) {
-      throw new Error('only function call expressions are allowed');
-    }
-
-    const result = [];
-
-    const expressions = body.map((x) => x.expression);
-    const allowedCalls = Object.keys(SUPPORTED_CALLS);
-    for (const expression of expressions) {
-      const name = get(expression, 'callee.name');
-      if (allowedCalls.includes(name) === false) {
-        throw new Error(`unrecognized call ${name}`);
-      }
-
-      const expectedArgs = SUPPORTED_CALLS[name].args;
-      const actualArgs = expression.arguments;
-      const expectedArgCount = expectedArgs.length;
-      const actualArgCount = actualArgs.length;
-      if (actualArgCount !== expectedArgCount) {
-        throw new Error(
-          `invalid number of arguments for ${name}, expected ${expectedArgCount}, but found ${actualArgCount}`
-        );
-      }
-
-      for (let i = 0; i < expectedArgs.length; i++) {
-        const expectedArg = expectedArgs[i];
-        const actualArg = actualArgs[i];
-        if (expectedArg.type !== actualArg.type) {
-          throw new Error(
-            `invalid #${i + 1} argument for ${name}, expected ${expectedArg.type}, but found ${actualArg.type}`
-          );
-        }
-      }
-
-      result.push({
-        call: name,
-        args: expectedArgs.reduce((acc, x, i) => {
-          acc[x.name] = actualArgs[i].value;
-          return acc;
-        }, {}),
-      });
-    }
-
-    if (result.length === 0) {
-      throw new Error('compiled contract result is empty');
-    }
-
-    return result;
-  } catch (err) {
-    throw new Error(`Invalid contract source: ${err.message}`);
-  }
-};
-
-const validateDid = (to, allowedRoleTypes) => {
-  if (isValid(to) === false) {
-    throw new Error('invalid did');
-  }
-
-  const type = toTypeInfo(to);
-  if (allowedRoleTypes.includes(type.role) === false) {
-    throw new Error('invalid did role type');
-  }
-
-  return true;
-};
-
-const validateAmount = (amount) => {
-  let bn = null;
-  try {
-    bn = new BN(amount);
-  } catch (err) {
-    throw new Error(`invalid amount: ${err.message}`);
-  }
-
-  if (bn.toString() !== amount) {
-    throw new Error('invalid amount: not valid big number');
-  }
-
-  return bn;
-};
-
-/**
- * Validate the compiled contract source
- *
- * @param {Array} compiled contract call list
- * @param {object} quota { value, tokens: { [address]: value }}
- * @return {boolean}
- */
-const validate = (compiled, quota, ensureZeroSum = false) => {
-  if (!Array.isArray(compiled) || compiled.length === 0) {
-    throw new Error('no contract calls to validate');
-  }
-
-  // Since QLDB's max batch size is 40, each contract call will update 1 account
-  // We need to set a smaller hard limit to avoid QLDB errors
-  if (compiled.length > 30) {
-    throw new Error('too much contract calls found, max is 15');
-  }
-
-  // to track consumption
-  let remaining = new BN(quota ? quota.value : '0');
-  const tokenRemaining = Object.keys(quota?.tokens ? quota.tokens : {}).reduce((acc, x) => {
-    acc[x] = new BN(quota.tokens[x]);
-    return acc;
-  }, {});
-
-  for (const item of compiled) {
-    const { call, args } = item;
-    if (call === 'transfer') {
-      validateDid(args.to, TOKEN_RECEIVER_ROLE_TYPES);
-      const consumption = validateAmount(args.amount);
-      if (consumption.lte(ZERO)) {
-        throw new Error('transfer amount must be greater than 0');
-      }
-      if (quota) {
-        if (remaining.sub(consumption).isNeg()) {
-          throw new Error('transfer exceeded quota');
-        }
-        remaining = remaining.sub(consumption);
-      }
-    } else if (call === 'transferToken') {
-      validateDid(args.tokenAddress, [types.RoleType.ROLE_TOKEN]);
-      validateDid(args.to, TOKEN_RECEIVER_ROLE_TYPES);
-
-      const token = args.tokenAddress;
-      const consumption = validateAmount(args.amount);
-      if (consumption.lte(ZERO)) {
-        throw new Error('token transfer amount must be greater than 0');
-      }
-      if (quota) {
-        if (!tokenRemaining[token] || tokenRemaining[token].sub(consumption).isNeg()) {
-          throw new Error(`transferToken for ${args.tokenAddress} exceeded quota`);
-        }
-
-        tokenRemaining[token] = tokenRemaining[token].sub(consumption);
-      }
-    }
-  }
-
-  if (ensureZeroSum) {
-    if (remaining.isZero() === false) {
-      throw new Error('transfer violates zero sum');
-    }
-    const invalidToken = Object.keys(tokenRemaining).find((x) => tokenRemaining[x].isZero() === false);
-    if (invalidToken) {
-      throw new Error(`transferToken for ${invalidToken} violates zero sum`);
-    }
-  }
-
-  return true;
-};
-
-// merge duplicate receiver transfers if they exist
-const merge = (compiled) => {
-  const mergeAndSum = (items, uniqKey, sumBy) => {
-    const result = {};
-    const keys = Array.isArray(uniqKey) ? uniqKey : [uniqKey];
-    keys.sort();
-    for (const item of items) {
-      const key = keys.map((x) => get(item, x)).join('-');
-      if (result[key] === undefined) {
-        result[key] = item;
-      } else {
-        const old = new BN(get(result[key], sumBy));
-        const delta = new BN(get(item, sumBy));
-        set(result[key], sumBy, old.add(delta).toString());
-      }
-    }
-
-    return Object.values(result);
-  };
-
-  return [
-    ...mergeAndSum(
-      compiled.filter((x) => x.call === 'transfer'),
-      'args.to',
-      'args.amount'
-    ),
-    ...mergeAndSum(
-      compiled.filter((x) => x.call === 'transferToken'),
-      ['args.tokenAddress', 'args.to'],
-      'args.amount'
-    ),
-  ];
-};
-
-const getQuota = (input = {}) => ({
-  value: input.value || '0',
-  tokens: (input.tokens || []).reduce((acc, x) => {
-    acc[x.address] = x.value;
-    return acc;
-  }, {}),
-});
-
-module.exports = { compile, validate, merge, getQuota, SUPPORTED_CALLS };