@cosmology/ast
Version:
Cosmos TypeScript AST generation
386 lines (385 loc) • 17.2 kB
JavaScript
import * as t from '@babel/types';
import { callExpression, identifier, TypeLong } from '../../../utils';
import { getDefaultTSTypeFromProtoType } from '../../types';
const getPropNames = (field) => {
const messageProp = field.name;
const objProp = field.options?.json_name ?? field.name;
return {
messageProp,
objProp
};
};
export const fromJSON = {
// sender: isSet(object.sender) ? String(object.sender) : ""
string(args) {
const { messageProp, objProp } = getPropNames(args.field);
args.context.addUtil('isSet');
return t.objectProperty(t.identifier(messageProp), t.conditionalExpression(t.callExpression(t.identifier('isSet'), [
t.memberExpression(t.identifier('object'), t.identifier(objProp))
]), t.callExpression(t.identifier('String'), [
t.memberExpression(t.identifier('object'), t.identifier(objProp))
]), getDefaultTSTypeFromProtoType(args.context, args.field, args.isOneOf)));
},
number(args) {
const { messageProp, objProp } = getPropNames(args.field);
args.context.addUtil('isSet');
return t.objectProperty(t.identifier(messageProp), t.conditionalExpression(t.callExpression(t.identifier('isSet'), [
t.memberExpression(t.identifier('object'), t.identifier(objProp))
]), t.callExpression(t.identifier('Number'), [
t.memberExpression(t.identifier('object'), t.identifier(objProp))
]), getDefaultTSTypeFromProtoType(args.context, args.field, args.isOneOf)));
},
double(args) {
return fromJSON.number(args);
},
float(args) {
return fromJSON.number(args);
},
int32(args) {
return fromJSON.number(args);
},
sint32(args) {
return fromJSON.number(args);
},
uint32(args) {
return fromJSON.number(args);
},
fixed32(args) {
return fromJSON.number(args);
},
sfixed32(args) {
return fromJSON.number(args);
},
// disableMacros: isSet(object.disableMacros) ? Boolean(object.disableMacros) : false
bool(args) {
const { messageProp, objProp } = getPropNames(args.field);
args.context.addUtil('isSet');
return t.objectProperty(t.identifier(messageProp), t.conditionalExpression(t.callExpression(t.identifier('isSet'), [
t.memberExpression(t.identifier('object'), t.identifier(objProp))
]), t.callExpression(t.identifier('Boolean'), [
t.memberExpression(t.identifier('object'), t.identifier(objProp))
]), getDefaultTSTypeFromProtoType(args.context, args.field, args.isOneOf)));
},
// int64Value: isSet(object.int64Value) ? Long.fromValue(object.int64Value) : Long.UZERO,
long(args) {
const { messageProp, objProp } = getPropNames(args.field);
args.context.addUtil('isSet');
TypeLong.addUtil(args.context);
return t.objectProperty(t.identifier(messageProp), t.conditionalExpression(t.callExpression(t.identifier('isSet'), [
t.memberExpression(t.identifier('object'), t.identifier(objProp))
]), TypeLong.getFromValueWithArgs(args.context, t.memberExpression(t.identifier('object'), t.identifier(objProp))), getDefaultTSTypeFromProtoType(args.context, args.field, args.isOneOf)));
},
int64(args) {
return fromJSON.long(args);
},
// uint64Value: isSet(object.uint64Value) ? Long.fromString(object.uint64Value) : Long.ZERO,
uint64(args) {
return fromJSON.long(args);
},
sint64(args) {
return fromJSON.long(args);
},
fixed64(args) {
return fromJSON.long(args);
},
sfixed64(args) {
return fromJSON.long(args);
},
// signDoc: isSet(object.signDoc) ? SignDocDirectAux.fromJSON(object.signDoc) : undefined,
type(args) {
const { messageProp, objProp } = getPropNames(args.field);
let name = args.context.getTypeName(args.field);
args.context.addUtil('isSet');
if (!args.context.options.aminoEncoding.useLegacyInlineEncoding &&
args.context.options.interfaces.enabled &&
args.context.options.interfaces?.useGlobalDecoderRegistry &&
args.field.type === 'google.protobuf.Any' &&
args.field.options['(cosmos_proto.accepts_interface)']) {
name = 'GlobalDecoderRegistry';
}
return t.objectProperty(t.identifier(messageProp), t.conditionalExpression(t.callExpression(t.identifier('isSet'), [
t.memberExpression(t.identifier('object'), t.identifier(objProp))
]), t.callExpression(t.memberExpression(t.identifier(name), t.identifier('fromJSON')), [
t.memberExpression(t.identifier('object'), t.identifier(objProp))
]), t.identifier('undefined')));
},
// mode: isSet(object.mode) ? signModeFromJSON(object.mode) : 0,
enum(args) {
const { messageProp, objProp } = getPropNames(args.field);
const setDefaultEnumToUnrecognized = args.context.pluginValue('prototypes.typingsFormat.setDefaultEnumToUnrecognized');
args.context.addUtil('isSet');
const fromJSONFuncName = args.context.getFromEnum(args.field);
return t.objectProperty(t.identifier(messageProp), t.conditionalExpression(t.callExpression(t.identifier('isSet'), [
t.memberExpression(t.identifier('object'), t.identifier(objProp))
]), t.callExpression(t.identifier(fromJSONFuncName), [
t.memberExpression(t.identifier('object'), t.identifier(objProp))
]), args.isOptional ? t.identifier('undefined') : t.numericLiteral(!setDefaultEnumToUnrecognized ? 0 : -1)));
},
// queryData: isSet(object.queryData) ? bytesFromBase64(object.queryData) : new Uint8Array()
bytes(args) {
const { messageProp, objProp } = getPropNames(args.field);
args.context.addUtil('isSet');
args.context.addUtil('bytesFromBase64');
return t.objectProperty(t.identifier(messageProp), t.conditionalExpression(t.callExpression(t.identifier('isSet'), [
t.memberExpression(t.identifier('object'), t.identifier(objProp))
]), t.callExpression(t.identifier('bytesFromBase64'), [
t.memberExpression(t.identifier('object'), t.identifier(objProp))
]), getDefaultTSTypeFromProtoType(args.context, args.field, args.isOneOf)));
},
// period: isSet(object.period) ? Duration.fromJSON(object.period) : undefined,
duration(args) {
const durationFormat = args.context.pluginValue('prototypes.typingsFormat.duration');
switch (durationFormat) {
case 'string':
return fromJSON.durationString(args);
case 'duration':
default:
return fromJSON.type(args);
}
},
// period: isSet(object.period) ? String(object.period) : undefined,
durationString(args) {
const { messageProp, objProp } = getPropNames(args.field);
args.context.addUtil('isSet');
return t.objectProperty(t.identifier(messageProp), t.conditionalExpression(t.callExpression(t.identifier('isSet'), [
t.memberExpression(t.identifier('object'), t.identifier(objProp))
]), t.callExpression(t.identifier('String'), [
t.memberExpression(t.identifier('object'), t.identifier(objProp))
]), t.identifier('undefined')));
},
// periodReset: isSet(object.periodReset) ? fromJsonTimestamp(object.periodReset) : undefined
timestamp(args) {
let timestampFormat = args.context.pluginValue('prototypes.typingsFormat.timestamp');
const env = args.context.pluginValue('env');
if (env == 'default') {
timestampFormat = 'timestamp';
}
switch (timestampFormat) {
case 'timestamp':
return fromJSON.timestampTimestamp(args);
case 'date':
default:
return fromJSON.timestampDate(args);
}
},
timestampTimestamp(args) {
const { messageProp, objProp } = getPropNames(args.field);
args.context.addUtil('isSet');
args.context.addUtil('fromJsonTimestamp');
return t.objectProperty(t.identifier(messageProp), t.conditionalExpression(t.callExpression(t.identifier('isSet'), [
t.memberExpression(t.identifier('object'), t.identifier(objProp))
]), t.callExpression(t.identifier('fromJsonTimestamp'), [
t.memberExpression(t.identifier('object'), t.identifier(objProp))
]), t.identifier('undefined')));
},
timestampDate(args) {
const { messageProp, objProp } = getPropNames(args.field);
args.context.addUtil('isSet');
return t.objectProperty(t.identifier(messageProp), t.conditionalExpression(t.callExpression(t.identifier('isSet'), [
t.memberExpression(t.identifier('object'), t.identifier(objProp))
]), t.newExpression(t.identifier('Date'), [
t.memberExpression(t.identifier('object'), t.identifier(objProp))
]), t.identifier('undefined')));
},
// labels: isObject(object.labels) ? Object.entries(object.labels).reduce<{
// [key: string]: string;
// }>((acc, [key, value]) => {
// acc[key] = String(value);
// return acc;
// }, {}) : {},
// referenceMap: isObject(object.referenceMap) ? Object.entries(object.referenceMap).reduce<{
// [key: Long]: Reference;
// }>((acc, [key, value]) => {
// acc[Number(key)] = Reference.fromJSON(value);
// return acc;
// }, {}) : {},
keyHash(args) {
const { messageProp, objProp } = getPropNames(args.field);
const keyType = args.field.keyType;
const valueType = args.field.parsedType.name;
args.context.addUtil('isObject');
let fromJSON = null;
// valueTypeType: string for identifier
let valueTypeType = valueType;
switch (valueType) {
case 'string':
fromJSON = t.callExpression(t.identifier('String'), [
t.identifier('value')
]);
break;
case 'int32':
case 'uint32':
valueTypeType = 'number';
fromJSON = t.callExpression(t.identifier('Number'), [
t.identifier('value')
]);
break;
case 'int64':
case 'uint64':
case 'sint64':
case 'fixed64':
case 'sfixed64':
TypeLong.addUtil(args.context);
valueTypeType = TypeLong.getPropType(args.context);
fromJSON = TypeLong.getFromValueWithArgs(args.context, t.tsAsExpression(t.identifier('value'), t.tsUnionType([
t.tsTypeReference(TypeLong.getPropIdentifier(args.context)),
t.tsStringKeyword()
])));
break;
default:
fromJSON = t.callExpression(t.memberExpression(t.identifier(valueType), t.identifier('fromJSON')), [
t.identifier('value')
]);
}
let wrapKey = null;
let keyTypeType = null;
switch (keyType) {
case 'string':
wrapKey = (a) => a;
keyTypeType = t.tsStringKeyword();
break;
case 'int64':
case 'uint64':
case 'sint64':
case 'fixed64':
case 'sfixed64':
wrapKey = (a) => t.callExpression(t.identifier('Number'), [
a
]);
TypeLong.addUtil(args.context);
keyTypeType = t.tsTypeReference(TypeLong.getPropIdentifier(args.context));
break;
case 'uint32':
case 'int32':
wrapKey = (a) => t.callExpression(t.identifier('Number'), [
a
]);
keyTypeType = t.tsTypeReference(t.identifier('number'));
break;
default:
throw new Error('keyHash requires new type. Ask maintainers.');
}
return t.objectProperty(t.identifier(messageProp), t.conditionalExpression(t.callExpression(t.identifier('isObject'), [
t.memberExpression(t.identifier('object'), t.identifier(objProp))
]), callExpression(t.memberExpression(t.callExpression(t.memberExpression(t.identifier('Object'), t.identifier('entries')), [
t.memberExpression(t.identifier('object'), t.identifier(objProp))
]), t.identifier('reduce')), [
t.arrowFunctionExpression([
t.identifier('acc'),
t.arrayPattern([
t.identifier('key'),
t.identifier('value')
])
], t.blockStatement([
t.expressionStatement(t.assignmentExpression('=', t.memberExpression(t.identifier('acc'), wrapKey(t.identifier('key')), true), fromJSON)),
t.returnStatement(t.identifier('acc'))
])),
t.objectExpression([])
], t.tsTypeParameterInstantiation([
t.tsTypeLiteral([
t.tsIndexSignature([
identifier('key', t.tsTypeAnnotation(keyTypeType))
], t.tsTypeAnnotation(t.tsTypeReference(t.identifier(valueTypeType))))
])
])), t.objectExpression([])));
},
// codeIds: Array.isArray(object?.codeIds) ? object.codeIds.map((e: any) => Long.fromString(e)) : [],
array(args, expr) {
const { messageProp, objProp } = getPropNames(args.field);
return t.objectProperty(t.identifier(messageProp), t.conditionalExpression(t.callExpression(t.memberExpression(t.identifier('Array'), t.identifier('isArray')), [
t.optionalMemberExpression(t.identifier('object'), t.identifier(objProp), false, true)
]), t.callExpression(t.memberExpression(t.memberExpression(t.identifier('object'), t.identifier(objProp)), t.identifier('map')), [
t.arrowFunctionExpression([
identifier('e', t.tsTypeAnnotation(t.tsAnyKeyword()))
], expr, false)
]), t.arrayExpression([])));
}
};
export const arrayTypes = {
string() {
return t.callExpression(t.identifier('String'), [
t.identifier('e')
]);
},
bool() {
return t.callExpression(t.identifier('Boolean'), [
t.identifier('e')
]);
},
// myBytesArray: Array.isArray(object?.myBytesArray) ? object.myBytesArray.map((e: any) => bytesFromBase64(e)) : [],
bytes(args) {
args.context.addUtil('bytesFromBase64');
return t.callExpression(t.identifier('bytesFromBase64'), [
t.identifier('e')
]);
},
// codeIds: Array.isArray(object?.codeIds) ? object.codeIds.map((e: any) => Long.fromValue(e)) : [],
long(args) {
TypeLong.addUtil(args.context);
return TypeLong.getFromValueWithArgs(args.context, t.identifier('e'));
},
uint64(args) {
return arrayTypes.long(args);
},
int64(args) {
return arrayTypes.long(args);
},
sint64(args) {
return arrayTypes.long(args);
},
fixed64(args) {
return arrayTypes.long(args);
},
sfixed64(args) {
return arrayTypes.long(args);
},
// myUint32Array: Array.isArray(object?.myUint32Array) ? object.myUint32Array.map((e: any) => Number(e)) : [],
number() {
return t.callExpression(t.identifier('Number'), [
t.identifier('e')
]);
},
// myDoubleArray: Array.isArray(object?.myDoubleArray) ? object.myDoubleArray.map((e: any) => Number(e)) : [],
uint32() {
return arrayTypes.number();
},
int32() {
return arrayTypes.number();
},
sint32() {
return arrayTypes.number();
},
fixed32() {
return arrayTypes.number();
},
sfixed32() {
return arrayTypes.number();
},
double() {
return arrayTypes.number();
},
float() {
return arrayTypes.number();
},
// arrayField: Array.isArray(object?.arrayField) ? object.arrayField.map((e: any) => scalarTypeFromJSON(e)) : []
enum(args) {
const fromJSONFuncName = args.context.getFromEnum(args.field);
return t.callExpression(t.identifier(fromJSONFuncName), [
t.identifier('e')
]);
},
// tokenInMaxs: Array.isArray(object?.tokenInMaxs) ? object.tokenInMaxs.map((e: any) => Coin.fromJSON(e)) : []
type(args) {
let name = args.context.getTypeName(args.field);
if (!args.context.options.aminoEncoding.useLegacyInlineEncoding &&
args.context.options.interfaces.enabled &&
args.context.options.interfaces?.useGlobalDecoderRegistry &&
args.field.type === 'google.protobuf.Any' &&
args.field.options['(cosmos_proto.accepts_interface)']) {
name = 'GlobalDecoderRegistry';
}
return t.callExpression(t.memberExpression(t.identifier(name), t.identifier('fromJSON')), [
t.identifier('e')
]);
}
};