@cosmology/ast
Version:
Cosmos TypeScript AST generation
102 lines (101 loc) • 4.69 kB
JavaScript
;
var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
if (k2 === undefined) k2 = k;
var desc = Object.getOwnPropertyDescriptor(m, k);
if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
desc = { enumerable: true, get: function() { return m[k]; } };
}
Object.defineProperty(o, k2, desc);
}) : (function(o, m, k, k2) {
if (k2 === undefined) k2 = k;
o[k2] = m[k];
}));
var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
Object.defineProperty(o, "default", { enumerable: true, value: v });
}) : function(o, v) {
o["default"] = v;
});
var __importStar = (this && this.__importStar) || function (mod) {
if (mod && mod.__esModule) return mod;
var result = {};
if (mod != null) for (var k in mod) if (k !== "default" && Object.prototype.hasOwnProperty.call(mod, k)) __createBinding(result, mod, k);
__setModuleDefault(result, mod);
return result;
};
var __exportStar = (this && this.__exportStar) || function(m, exports) {
for (var p in m) if (p !== "default" && !Object.prototype.hasOwnProperty.call(exports, p)) __createBinding(exports, m, p);
};
Object.defineProperty(exports, "__esModule", { value: true });
exports.createRegisterType = exports.registerTypeUrlMethod = void 0;
const t = __importStar(require("@babel/types"));
const utils_1 = require("../../../utils");
const utils_2 = require("./utils");
__exportStar(require("./utils"), exports);
const registerTypeUrlMethod = (args) => {
const { context, name, proto } = args;
const isImplementsInterface = Boolean(proto.options?.["(cosmos_proto.implements_interface)"]);
const methodName = "registerTypeUrl";
const typeNames = new Set();
let registerStats = Object.keys(proto.fields ?? {}).reduce((statements, fieldName) => {
const field = proto.fields[fieldName];
if (field.parsedType.type === "Type") {
switch (field.type) {
case "google.protobuf.Duration":
case "Duration":
case "google.protobuf.Timestamp":
case "Timestamp":
return statements;
case "Any":
case "google.protobuf.Any":
const lookupInterface = field.options?.["(cosmos_proto.accepts_interface)"];
const acceptedTypes = (0, utils_1.getAcceptedInterfacesTypes)(context, lookupInterface);
const acceptedTypeNames = acceptedTypes.map((acceptedType) => acceptedType.readAs);
const typeName = context.getTypeName(field);
acceptedTypeNames.push(typeName);
return acceptedTypeNames.reduce((states, acceptedTypeName) => {
if (typeNames.has(acceptedTypeName)) {
return states;
}
typeNames.add(acceptedTypeName);
return [
...states,
(0, exports.createRegisterType)(acceptedTypeName),
].filter(Boolean);
}, []);
}
const typeName = context.getTypeName(field);
if (typeNames.has(typeName)) {
return statements;
}
typeNames.add(typeName);
return [...statements, (0, exports.createRegisterType)(typeName)].filter(Boolean);
}
return statements;
}, []);
if (isImplementsInterface) {
registerStats = [
(0, utils_2.createIfGlobalDecoderRegistryRegisterExistingTypeUrl)(context, name, proto),
(0, utils_2.createRegisterObject)(context, name, proto),
(0, utils_2.createRegisterAminoProtoMapping)(context, name, proto),
...registerStats,
].filter(Boolean);
}
else {
if (registerStats.length) {
registerStats = [
(0, utils_2.createIfGlobalDecoderRegistryRegisterExistingTypeUrl)(context, name, proto),
...registerStats,
].filter(Boolean);
}
}
const method = (0, utils_1.objectMethod)("method", t.identifier(methodName), [], t.blockStatement(registerStats ?? []));
return method;
};
exports.registerTypeUrlMethod = registerTypeUrlMethod;
const createRegisterType = (typeName) => {
if (typeName === "Any") {
return;
}
return t.expressionStatement(t.callExpression(t.memberExpression(t.identifier(typeName), t.identifier("registerTypeUrl")), []));
};
exports.createRegisterType = createRegisterType;