"use strict"; // Copyright 2021-2024 Buf Technologies, Inc. // // Licensed under the Apache License, Version 2.0 (the "License"); // you may not use this file except in compliance with the License. // You may obtain a copy of the License at // // http://www.apache.org/licenses/LICENSE-2.0 // // Unless required by applicable law or agreed to in writing, software // distributed under the License is distributed on an "AS IS" BASIS, // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. // See the License for the specific language governing permissions and // limitations under the License. Object.defineProperty(exports, "__esModule", { value: true }); exports.makeEnum = exports.makeEnumType = exports.setEnumType = exports.getEnumType = void 0; const assert_js_1 = require("./assert.js"); const enumTypeSymbol = Symbol("@bufbuild/protobuf/enum-type"); /** * Get reflection information from a generated enum. * If this function is called on something other than a generated * enum, it raises an error. */ function getEnumType(enumObject) { // eslint-disable-next-line @typescript-eslint/no-unsafe-member-access,@typescript-eslint/no-unsafe-assignment,@typescript-eslint/no-explicit-any const t = enumObject[enumTypeSymbol]; (0, assert_js_1.assert)(t, "missing enum type on enum object"); return t; // eslint-disable-line @typescript-eslint/no-unsafe-return } exports.getEnumType = getEnumType; /** * Sets reflection information on a generated enum. */ function setEnumType(enumObject, typeName, values, opt) { // eslint-disable-next-line @typescript-eslint/no-unsafe-member-access, @typescript-eslint/no-explicit-any enumObject[enumTypeSymbol] = makeEnumType(typeName, values.map((v) => ({ no: v.no, name: v.name, localName: enumObject[v.no], })), opt); } exports.setEnumType = setEnumType; /** * Create a new EnumType with the given values. */ function makeEnumType(typeName, values, // eslint-disable-next-line @typescript-eslint/no-unused-vars _opt) { const names = Object.create(null); const numbers = Object.create(null); const normalValues = []; for (const value of values) { // We do not surface options at this time // const value: EnumValueInfo = {...v, options: v.options ?? emptyReadonlyObject}; const n = normalizeEnumValue(value); normalValues.push(n); names[value.name] = n; numbers[value.no] = n; } return { typeName, values: normalValues, // We do not surface options at this time // options: opt?.options ?? Object.create(null), findName(name) { return names[name]; }, findNumber(no) { return numbers[no]; }, }; } exports.makeEnumType = makeEnumType; /** * Create a new enum object with the given values. * Sets reflection information. */ function makeEnum(typeName, values, opt) { const enumObject = {}; for (const value of values) { const n = normalizeEnumValue(value); enumObject[n.localName] = n.no; enumObject[n.no] = n.localName; } setEnumType(enumObject, typeName, values, opt); return enumObject; } exports.makeEnum = makeEnum; function normalizeEnumValue(value) { if ("localName" in value) { return value; } return Object.assign(Object.assign({}, value), { localName: value.name }); }