Erster Commit
This commit is contained in:
+94
@@ -0,0 +1,94 @@
|
||||
"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 });
|
||||
}
|
||||
Reference in New Issue
Block a user