@ocap/message 1.28.8 → 1.29.0

package/lib/message.cjs

@@ -0,0 +1,500 @@
+const require_rolldown_runtime = require('./_virtual/rolldown_runtime.cjs');
+const require_provider = require('./provider.cjs');
+let debug = require("debug");
+debug = require_rolldown_runtime.__toESM(debug);
+let lodash_camelCase = require("lodash/camelCase");
+lodash_camelCase = require_rolldown_runtime.__toESM(lodash_camelCase);
+let _ocap_util = require("@ocap/util");
+
+//#region src/message.ts
+/**
+* @fileOverview Contains basic helper methods to encode/format/mock a protobuf message
+* @module @ocap/message
+* @requires @ocap/util
+* @requires @ocap/proto
+* @example
+* bun add @ocap/message
+*
+* const { createMessage, fakeMessage, formatMessage } = require('@ocap/message');
+*/
+const jspb = require("google-protobuf");
+const { Any } = require("google-protobuf/google/protobuf/any_pb");
+const { Timestamp } = require("google-protobuf/google/protobuf/timestamp_pb");
+const debug$1 = (0, debug.default)("@ocap/message");
+const getEnumTypes = () => Object.keys(require_provider.enums);
+const scalarTypes = [
+	"bool",
+	"bytes",
+	"string",
+	"double",
+	"float",
+	"sint32",
+	"uint32",
+	"sfixed32",
+	"sint64",
+	"uint64",
+	"sfixed64"
+];
+const fakeValues = {
+	bool: true,
+	sint32: 1,
+	uint32: 2,
+	sfixed32: 3,
+	sint64: 4,
+	uint64: 5,
+	sfixed64: 6,
+	BigUint: () => "1234",
+	BigSint: () => "4567",
+	float: "12.2",
+	double: "12.3",
+	string: "arcblock",
+	bytes: new Uint8Array(),
+	enums: (type) => type ? Object.values(require_provider.enums[type])[0] : void 0
+};
+/**
+* Generated a fake message for a type, the message can be RPC request/response
+*
+* @public
+* @static
+* @param type - Message type string, should be defined in forge-abi or forge-core-protocol
+* @returns object
+*/
+function fakeMessage(type) {
+	if (!type) return;
+	if (fakeValues[type]) return fakeValues[type];
+	const { fields, oneofs } = require_provider.getMessageType(type);
+	if (!fields) return;
+	let selectedFields = fields;
+	if (oneofs?.value && Array.isArray(oneofs.value.oneof)) {
+		const selectedField = oneofs.value.oneof[0];
+		selectedFields = { [selectedField]: fields[selectedField] };
+	}
+	const result = {};
+	Object.keys(selectedFields).forEach((key) => {
+		const { type: subType, keyType, rule } = selectedFields[key];
+		if (rule === "repeated") {
+			result[key] = [1, 2].map(() => fakeMessage(subType));
+			return;
+		}
+		if (keyType) {
+			result[key] = { [fakeValues[keyType]]: fakeMessage(subType) };
+			return;
+		}
+		if (getEnumTypes().includes(subType)) {
+			result[key] = fakeValues.enums(subType);
+			return;
+		}
+		if ([
+			"hash",
+			"appHash",
+			"txHash",
+			"address",
+			"from",
+			"to",
+			"proposer"
+		].includes(key)) {
+			result[key] = "F2D072CBD4954A20F26280730795D91AC1039996CEB6E24A31E9CE548DCB5E55";
+			return;
+		}
+		if (fakeValues[subType]) result[key] = fakeValues[subType];
+		if (subType === "google.protobuf.Timestamp") {
+			result[key] = (/* @__PURE__ */ new Date()).toISOString();
+			return;
+		}
+		if (subType === "google.protobuf.Any") {
+			result[key] = {
+				type: "string",
+				value: "ABCD 1234"
+			};
+			return;
+		}
+		result[key] = fakeMessage(subType);
+	});
+	return result;
+}
+/**
+* Format an message from RPC to UI friendly
+*
+* @public
+* @static
+* @param type - input type
+* @param data - input data
+* @returns object [almost same structure as input]
+*/
+function formatMessage(type, data) {
+	if (!type) return data;
+	if (["json", "vc"].includes(type)) return data;
+	if (typeof data !== "object") return data;
+	if (scalarTypes.includes(type)) return data;
+	const result = {};
+	const { fields } = require_provider.getMessageType(type);
+	if (!fields) {
+		console.log({
+			type,
+			data
+		});
+		throw new Error(`Cannot get fields for type ${type}`);
+	}
+	Object.keys(fields).forEach((key) => {
+		const { type: subType, keyType, rule } = fields[key];
+		let value = data[key];
+		if (rule === "repeated") value = data[(0, lodash_camelCase.default)(`${key}_list`)] || data[key];
+		if (keyType) value = data[(0, lodash_camelCase.default)(`${key}_map`)] || data[key];
+		if (value === void 0) return;
+		if (rule === "repeated") {
+			result[key] = value.map((x) => formatMessage(subType, x));
+			return;
+		}
+		if (keyType) {
+			debug$1("formatMessage.map", {
+				type,
+				subType,
+				keyType,
+				value
+			});
+			result[key] = (value || []).reduce((acc, [k, v]) => {
+				acc[k] = formatMessage(subType, v);
+				return acc;
+			}, {});
+			return;
+		}
+		if (getEnumTypes().includes(subType)) {
+			result[key] = require_provider.messages[subType][value];
+			return;
+		}
+		if (["BigUint", "BigSint"].includes(subType)) {
+			result[key] = decodeBigInt(value);
+			return;
+		}
+		if ((0, _ocap_util.isUint8Array)(value)) {
+			if (["appHash", "blockHash"].includes(key)) result[key] = Buffer.from(value).toString("hex");
+			if ([
+				"signature",
+				"pk",
+				"sk"
+			].includes(key)) result[key] = Buffer.from(value).toString("base64");
+			return;
+		}
+		if (subType === "google.protobuf.Timestamp") {
+			result[key] = decodeTimestamp(value);
+			return;
+		}
+		if (subType === "google.protobuf.Any") {
+			if (value) {
+				const decoded = decodeAny(value);
+				result[key] = {
+					type: decoded.type || value.typeUrl,
+					value: formatMessage(decoded.type || value.typeUrl || "", decoded.value)
+				};
+			}
+			return;
+		}
+		if (value && typeof value === "object") {
+			result[key] = formatMessage(subType, value);
+			return;
+		}
+		result[key] = value;
+	});
+	return result;
+}
+/**
+* Create an protobuf encoded Typed message with specified data, ready to send to rpc server
+*
+* @public
+* @static
+* @param type - message type defined in forge-proto
+* @param params - message content
+* @returns Message instance
+*/
+function createMessage(type, params) {
+	if (!type && !params) {
+		console.log({
+			type,
+			params
+		});
+		return;
+	}
+	const { fn: Message, fields } = require_provider.getMessageType(type);
+	if (!Message || !fields) {
+		console.error({
+			type,
+			params,
+			fields,
+			Message
+		});
+		throw new Error(`Unsupported messageType: ${type}`);
+	}
+	const message = new Message();
+	if (!params) return message;
+	Object.keys(fields).forEach((key) => {
+		const { type: subType, keyType, rule, id } = fields[key];
+		const value = params[key] || params[(0, lodash_camelCase.default)(`${key}_list`)];
+		if (value === void 0) return;
+		if (keyType) {
+			const keys = Object.keys(value);
+			if (keys.length) {
+				const fn$1 = (0, lodash_camelCase.default)(`get_${key}_map`);
+				const map = message[fn$1]();
+				debug$1("createMessage.map", {
+					type,
+					subType,
+					keyType,
+					id,
+					fn: fn$1,
+					keys
+				});
+				keys.forEach((k) => {
+					map.set(k, createMessage(subType, value[k]));
+				});
+				jspb.Message.setField(message, id, map);
+			}
+			return;
+		}
+		const fn = (0, lodash_camelCase.default)(rule === "repeated" ? `add_${key}` : `set_${key}`);
+		if (typeof message[fn] !== "function") throw new Error(`Unexpected field names ${JSON.stringify({
+			type,
+			key,
+			subType,
+			fn,
+			rule
+		})}`);
+		const values = rule === "repeated" ? value : [value];
+		try {
+			values.forEach((v) => {
+				if (getEnumTypes().includes(subType)) {
+					message[fn](v);
+					return;
+				}
+				if (["BigUint", "BigSint"].includes(subType)) {
+					message[fn](encodeBigInt(v, subType));
+					return;
+				}
+				if (subType === "google.protobuf.Timestamp") {
+					debug$1(`createMessage.${subType}`, { v });
+					message[fn](encodeTimestamp(v));
+					return;
+				}
+				if (subType === "google.protobuf.Any") {
+					message[fn](encodeAny(v));
+					return;
+				}
+				if (subType === "bytes") {
+					message[fn]((0, _ocap_util.toUint8Array)(v));
+					return;
+				}
+				const { fn: SubMessage, fields: subFields } = require_provider.getMessageType(subType);
+				if (SubMessage && subFields) {
+					debug$1(`createMessage.${subType}`, {
+						type,
+						subType,
+						key
+					});
+					const subMessage = createMessage(subType, v);
+					message[fn](subMessage);
+					return;
+				}
+				message[fn](v);
+			});
+		} catch (err) {
+			debug$1("createMessage.processField.error", {
+				type,
+				key,
+				subType,
+				fn,
+				rule,
+				values,
+				err
+			});
+			throw err;
+		}
+	});
+	return message;
+}
+/**
+* Decode an google.protobuf.Any%{ typeUrl, value } => { type, value }
+*
+* @public
+* @static
+* @param data - encoded data object
+* @returns Object { type, value }
+*/
+function decodeAny(data) {
+	if (!data) return {
+		type: "Unknown",
+		value: ""
+	};
+	if (data.type && data.value) return data;
+	const { typeUrl, value } = data;
+	if (typeUrl && ["json", "vc"].includes(typeUrl)) return {
+		type: typeUrl,
+		value: JSON.parse(Buffer.from(value, "base64").toString())
+	};
+	const type = require_provider.fromTypeUrl(typeUrl);
+	const { fn: Message } = require_provider.getMessageType(type);
+	if (!Message) return data;
+	const buffer = Buffer.isBuffer(value) ? value : Buffer.from(value, "base64");
+	return {
+		type,
+		value: Message.deserializeBinary(new Uint8Array(buffer)).toObject()
+	};
+}
+/**
+* Encode { type, value } => google.protobuf.Any%{ typeUrl, value }
+* Does nothing on already encoded message
+*
+* @public
+* @static
+* @param data
+* @returns google.protobuf.Any
+*/
+function encodeAny(data) {
+	if (!data) return;
+	const anyMessage = new Any();
+	try {
+		if (data.typeUrl && data.value && !data.type) {
+			anyMessage.setTypeUrl(data.typeUrl);
+			if (data.typeUrl === "fg:x:address") anyMessage.setValue(data.value);
+			else if (["json", "vc"].includes(data.typeUrl)) anyMessage.setValue(new Uint8Array(Buffer.from(JSON.stringify(data.value))));
+			else anyMessage.setValue(new Uint8Array(Buffer.from(data.value, "base64")));
+		} else {
+			const { value: anyValue, type: anyType } = data;
+			const typeUrl = require_provider.toTypeUrl(anyType);
+			anyMessage.setTypeUrl(typeUrl);
+			if (["json", "vc"].includes(typeUrl)) anyMessage.setValue(new Uint8Array(Buffer.from(JSON.stringify(anyValue))));
+			else {
+				const anyValueBinary = createMessage(anyType, anyValue);
+				anyMessage.setValue(anyValueBinary.serializeBinary());
+			}
+		}
+	} catch (err) {
+		console.error("error encode any type", data);
+		throw err;
+	}
+	return anyMessage;
+}
+/**
+* Convert an { seconds, nanos } | date-string to google.protobuf.Timestamp object
+*
+* @public
+* @static
+* @param value
+* @returns instanceof google.protobuf.Timestamp
+*/
+function encodeTimestamp(value) {
+	if (!value) return;
+	const timestamp = new Timestamp();
+	if (typeof value === "string") {
+		const millionSeconds = Date.parse(value);
+		if (Number.isNaN(millionSeconds) === false) {
+			timestamp.setSeconds(Math.floor(millionSeconds / 1e3));
+			timestamp.setNanos(Math.floor(millionSeconds % 1e3 * 1e6));
+		}
+	} else {
+		const { seconds, nanos = 0 } = value;
+		timestamp.setSeconds(seconds);
+		timestamp.setNanos(nanos);
+	}
+	return timestamp;
+}
+/**
+* Decode google.protobuf.Timestamp message to ISO Date String
+*
+* FIXME: node strictly equal because we rounded the `nanos` field
+*
+* @public
+* @static
+* @param data
+* @returns String timestamp
+*/
+function decodeTimestamp(data) {
+	if (data?.seconds) {
+		const date = /* @__PURE__ */ new Date();
+		date.setTime(data.seconds * 1e3 + Math.ceil((data.nanos || 0) / 1e6));
+		return date.toISOString();
+	}
+	return "";
+}
+/**
+* Encode BigUint and BigSint types defined in forge-sdk, double encoding is avoided
+*
+* @public
+* @static
+* @param value - value to encode
+* @param type - type names defined in forge-proto
+* @returns Message
+*/
+function encodeBigInt(value, type) {
+	const { fn: BigIntMessageClass } = require_provider.getMessageType(type);
+	if (!BigIntMessageClass) throw new Error(`Unknown BigInt type: ${type}`);
+	const message = new BigIntMessageClass();
+	if (value && typeof value === "object" && "value" in value && (0, _ocap_util.isUint8Array)(value.value)) {
+		message.setValue(value.value);
+		if (type === "BigSint") message.setMinus(value.minus);
+		return message;
+	}
+	const number = (0, _ocap_util.toBN)(value);
+	const zero = (0, _ocap_util.toBN)(0);
+	message.setValue(new Uint8Array(number.toArray()));
+	if (type === "BigSint") message.setMinus(number.lt(zero));
+	return message;
+}
+/**
+* Convert BigUint and BigSint to string representation of numbers
+*
+* @public
+* @static
+* @link https://stackoverflow.com/questions/23948278/how-to-convert-byte-array-into-a-signed-big-integer-in-javascript
+* @param data - usually from encodeBigInt
+* @returns human readable number
+*/
+function decodeBigInt(data) {
+	const bn = (0, _ocap_util.toBN)((0, _ocap_util.bytesToHex)(data.value));
+	return `${data.minus ? "-" : ""}${bn.toString(10)}`;
+}
+/**
+* Attach an $format method to rpc response
+*
+* @private
+* @param data
+* @param type
+*/
+function attachFormatFn(type, data, key = "$format") {
+	Object.defineProperty(data, key, {
+		writable: false,
+		enumerable: false,
+		configurable: false,
+		value: () => formatMessage(type, data)
+	});
+}
+/**
+* Attach an example method to
+*
+* @private
+* @param data
+* @param type
+*/
+function attachExampleFn(type, host, key) {
+	Object.defineProperty(host, key, {
+		writable: false,
+		enumerable: false,
+		configurable: false,
+		value: () => fakeMessage(type)
+	});
+}
+
+//#endregion
+exports.addProvider = require_provider.addProvider;
+exports.attachExampleFn = attachExampleFn;
+exports.attachFormatFn = attachFormatFn;
+exports.createMessage = createMessage;
+exports.decodeAny = decodeAny;
+exports.decodeBigInt = decodeBigInt;
+exports.decodeTimestamp = decodeTimestamp;
+exports.encodeAny = encodeAny;
+exports.encodeBigInt = encodeBigInt;
+exports.encodeTimestamp = encodeTimestamp;
+exports.fakeMessage = fakeMessage;
+exports.formatMessage = formatMessage;
+exports.fromTypeUrl = require_provider.fromTypeUrl;
+exports.getMessageType = require_provider.getMessageType;
+exports.toTypeUrl = require_provider.toTypeUrl;

package/lib/message.d.cts

@@ -0,0 +1,144 @@
+import { addProvider, fromTypeUrl, getMessageType, toTypeUrl } from "./provider.cjs";
+
+//#region src/message.d.ts
+
+/**
+ * @fileOverview Contains basic helper methods to encode/format/mock a protobuf message
+ * @module @ocap/message
+ * @requires @ocap/util
+ * @requires @ocap/proto
+ * @example
+ * bun add @ocap/message
+ *
+ * const { createMessage, fakeMessage, formatMessage } = require('@ocap/message');
+ */
+declare const Any: any;
+declare const Timestamp: any;
+interface AnyData {
+  type?: string;
+  typeUrl?: string;
+  value?: unknown;
+}
+interface TimestampData {
+  seconds?: number;
+  nanos?: number;
+}
+interface BigIntData {
+  value?: Uint8Array;
+  minus?: boolean;
+}
+interface ProtobufMessage {
+  toObject(): Record<string, unknown>;
+  serializeBinary(): Uint8Array;
+  [key: string]: unknown;
+}
+/**
+ * Generated a fake message for a type, the message can be RPC request/response
+ *
+ * @public
+ * @static
+ * @param type - Message type string, should be defined in forge-abi or forge-core-protocol
+ * @returns object
+ */
+declare function fakeMessage(type?: string): unknown;
+/**
+ * Format an message from RPC to UI friendly
+ *
+ * @public
+ * @static
+ * @param type - input type
+ * @param data - input data
+ * @returns object [almost same structure as input]
+ */
+declare function formatMessage(type: string, data: Record<string, unknown>): Record<string, unknown>;
+/**
+ * Create an protobuf encoded Typed message with specified data, ready to send to rpc server
+ *
+ * @public
+ * @static
+ * @param type - message type defined in forge-proto
+ * @param params - message content
+ * @returns Message instance
+ */
+declare function createMessage(type: string, params?: Record<string, unknown>): ProtobufMessage | undefined;
+/**
+ * Decode an google.protobuf.Any%{ typeUrl, value } => { type, value }
+ *
+ * @public
+ * @static
+ * @param data - encoded data object
+ * @returns Object { type, value }
+ */
+declare function decodeAny(data: AnyData | null): AnyData;
+/**
+ * Encode { type, value } => google.protobuf.Any%{ typeUrl, value }
+ * Does nothing on already encoded message
+ *
+ * @public
+ * @static
+ * @param data
+ * @returns google.protobuf.Any
+ */
+declare function encodeAny(data: AnyData | undefined): typeof Any | undefined;
+/**
+ * Convert an { seconds, nanos } | date-string to google.protobuf.Timestamp object
+ *
+ * @public
+ * @static
+ * @param value
+ * @returns instanceof google.protobuf.Timestamp
+ */
+declare function encodeTimestamp(value: string | TimestampData | undefined): typeof Timestamp | undefined;
+/**
+ * Decode google.protobuf.Timestamp message to ISO Date String
+ *
+ * FIXME: node strictly equal because we rounded the `nanos` field
+ *
+ * @public
+ * @static
+ * @param data
+ * @returns String timestamp
+ */
+declare function decodeTimestamp(data: TimestampData): string;
+interface BigIntProtobufMessage extends ProtobufMessage {
+  setValue(v: Uint8Array): void;
+  setMinus(v: boolean): void;
+}
+/**
+ * Encode BigUint and BigSint types defined in forge-sdk, double encoding is avoided
+ *
+ * @public
+ * @static
+ * @param value - value to encode
+ * @param type - type names defined in forge-proto
+ * @returns Message
+ */
+declare function encodeBigInt(value: unknown, type: string): BigIntProtobufMessage;
+/**
+ * Convert BigUint and BigSint to string representation of numbers
+ *
+ * @public
+ * @static
+ * @link https://stackoverflow.com/questions/23948278/how-to-convert-byte-array-into-a-signed-big-integer-in-javascript
+ * @param data - usually from encodeBigInt
+ * @returns human readable number
+ */
+declare function decodeBigInt(data: BigIntData): string;
+/**
+ * Attach an $format method to rpc response
+ *
+ * @private
+ * @param data
+ * @param type
+ */
+declare function attachFormatFn(type: string, data: Record<string, unknown>, key?: string): void;
+/**
+ * Attach an example method to
+ *
+ * @private
+ * @param data
+ * @param type
+ */
+declare function attachExampleFn(type: string, host: Record<string, unknown>, key: string): void;
+//#endregion
+export { AnyData, addProvider, attachExampleFn, attachFormatFn, createMessage, decodeAny, decodeBigInt, decodeTimestamp, encodeAny, encodeBigInt, encodeTimestamp, fakeMessage, formatMessage, fromTypeUrl, getMessageType, toTypeUrl };