95 lines
3.4 KiB
JavaScript
95 lines
3.4 KiB
JavaScript
"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 });
|
|
}
|