@bufbuild/protoplugin 2.0.0-alpha.1 → 2.0.0-alpha.3

package/dist/cjs/create-es-plugin.d.ts

@@ -1,6 +1,6 @@
-import type { SupportedEdition } from "@bufbuild/protobuf";
-import type { Schema } from "./ecmascript/schema.js";
-import type { FileInfo } from "./ecmascript/generated-file.js";
+import { type SupportedEdition } from "@bufbuild/protobuf";
+import type { Schema } from "./schema.js";
+import type { FileInfo } from "./generated-file.js";
 import type { Plugin } from "./plugin.js";
 interface PluginInit {
     /**

package/dist/cjs/create-es-plugin.js

@@ -15,11 +15,11 @@
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.createEcmaScriptPlugin = void 0;
 const protobuf_1 = require("@bufbuild/protobuf");
-const reflect_1 = require("@bufbuild/protobuf/reflect");
+const protobuf_2 = require("@bufbuild/protobuf");
 const wkt_1 = require("@bufbuild/protobuf/wkt");
-const schema_js_1 = require("./ecmascript/schema.js");
-const transpile_js_1 = require("./ecmascript/transpile.js");
-const parameter_js_1 = require("./ecmascript/parameter.js");
+const schema_js_1 = require("./schema.js");
+const transpile_js_1 = require("./transpile.js");
+const parameter_js_1 = require("./parameter.js");
 /**
  * Create a new code generator plugin for ECMAScript.
  * The plugin can generate JavaScript, TypeScript, or TypeScript declaration
@@ -33,8 +33,8 @@ function createEcmaScriptPlugin(init) {
         version: init.version,
         run(req) {
             var _a, _b, _c;
-            const minimumEdition = (_a = init.minimumEdition) !== null && _a !== void 0 ? _a : reflect_1.minimumEdition;
-            const maximumEdition = (_b = init.maximumEdition) !== null && _b !== void 0 ? _b : reflect_1.maximumEdition;
+            const minimumEdition = (_a = init.minimumEdition) !== null && _a !== void 0 ? _a : protobuf_2.minimumEdition;
+            const maximumEdition = (_b = init.maximumEdition) !== null && _b !== void 0 ? _b : protobuf_2.maximumEdition;
             const parameter = (0, parameter_js_1.parseParameter)(req.parameter, init.parseOption);
             const schema = (0, schema_js_1.createSchema)(req, parameter, init.name, init.version, minimumEdition, maximumEdition);
             const targetTs = schema.targets.includes("ts");

package/dist/cjs/{ecmascript/file-preamble.js → file-preamble.js}

@@ -15,7 +15,7 @@
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.makeFilePreamble = void 0;
 const wkt_1 = require("@bufbuild/protobuf/wkt");
-const source_code_info_js_1 = require("../source-code-info.js");
+const source_code_info_js_1 = require("./source-code-info.js");
 function makeFilePreamble(file, pluginName, pluginVersion, parameter, tsNoCheck) {
     const builder = [];
     const trimSuffix = (str, suffix) => str.endsWith(suffix) ? str.substring(0, str.length - suffix.length) : str;
@@ -64,6 +64,9 @@ function makeFilePreamble(file, pluginName, pluginVersion, parameter, tsNoCheck)
             break;
         }
     }
+    for (const o of (0, source_code_info_js_1.getFeatureOptionStrings)(file)) {
+        builder.push(`// option ${o};\n`);
+    }
     builder.push("/* eslint-disable */\n");
     if (tsNoCheck) {
         builder.push("// @ts-nocheck\n");

package/dist/cjs/{ecmascript/generated-file.d.ts → generated-file.d.ts}

@@ -1,4 +1,4 @@
-import type { AnyDesc, DescEnum, DescExtension, DescFile, DescMessage, DescService } from "@bufbuild/protobuf";
+import { type AnyDesc, type DescEnum, type DescExtension, type DescFile, type DescMessage, type DescService } from "@bufbuild/protobuf";
 import type { ImportSymbol } from "./import-symbol.js";
 import type { Printable } from "./printable.js";
 import type { RuntimeImports } from "./runtime-imports.js";
@@ -63,7 +63,7 @@ export interface GeneratedFile {
      * Create a printable export statement. For example:
      *
      * ```ts
-     * f.print(f.exportDecl("abstract class", "MyClass"), " {}")
+     * f.print(f.export("abstract class", "MyClass"), " {}")
      * ```
      *
      * Will generate as:
@@ -75,7 +75,7 @@ export interface GeneratedFile {
      * statement. If the plugin option `js_import_style=legacy_commonjs` is set,
      * exports will automatically be generated for CommonJS.
      */
-    exportDecl(declaration: string, name: string): Printable;
+    export(declaration: string, name: string): Printable;
     /**
      * Import a message or enumeration generated by protoc-gen-es.
      */

package/dist/cjs/{ecmascript/generated-file.js → generated-file.js}

@@ -15,7 +15,6 @@
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.createGeneratedFile = void 0;
 const protobuf_1 = require("@bufbuild/protobuf");
-const reflect_1 = require("@bufbuild/protobuf/reflect");
 const import_symbol_js_1 = require("./import-symbol.js");
 const import_path_js_1 = require("./import-path.js");
 const jsdoc_js_1 = require("./jsdoc.js");
@@ -48,7 +47,7 @@ function createGeneratedFile(name, importPath, jsImportStyle, rewriteImport, res
             }, printables);
             el.push("\n");
         },
-        exportDecl(declaration, name) {
+        export(declaration, name) {
             return {
                 kind: "es_export_stmt",
                 name,
@@ -231,25 +230,25 @@ function printableToEl(opt, printables) {
                             el.push(escapeString(p.value));
                             break;
                         case "es_proto_int64":
-                            switch (p.longType) {
-                                case reflect_1.LongType.STRING:
-                                    el.push(escapeString(p.value.toString()));
-                                    break;
-                                case reflect_1.LongType.BIGINT:
-                                    if (p.value == protobuf_1.protoInt64.zero) {
-                                        // Loose comparison will match between 0n and 0.
-                                        el.push(opt.runtime.protoInt64, ".zero");
-                                        break;
-                                    }
+                            if (p.longAsString) {
+                                el.push(escapeString(p.value.toString()));
+                            }
+                            else {
+                                if (p.value == protobuf_1.protoInt64.zero) {
+                                    // Loose comparison will match between 0n and 0.
+                                    el.push(opt.runtime.protoInt64, ".zero");
+                                }
+                                else {
                                     switch (p.type) {
-                                        case reflect_1.ScalarType.UINT64:
-                                        case reflect_1.ScalarType.FIXED64:
+                                        case protobuf_1.ScalarType.UINT64:
+                                        case protobuf_1.ScalarType.FIXED64:
                                             el.push(opt.runtime.protoInt64, ".uParse(", escapeString(p.value.toString()), ")");
                                             break;
                                         default:
                                             el.push(opt.runtime.protoInt64, ".parse(", escapeString(p.value.toString()), ")");
                                             break;
                                     }
+                                }
                             }
                             break;
                         default:

package/dist/cjs/index.d.ts

@@ -2,3 +2,10 @@ export type { Plugin } from "./plugin.js";
 export { runNodeJs } from "./run-node.js";
 export { createEcmaScriptPlugin } from "./create-es-plugin.js";
 export { getComments, getDeclarationString, getPackageComments, getSyntaxComments, } from "./source-code-info.js";
+export type { Target } from "./target.js";
+export type { Schema } from "./schema.js";
+export type { GeneratedFile, FileInfo } from "./generated-file.js";
+export type { ImportSymbol } from "./import-symbol.js";
+export { createImportSymbol } from "./import-symbol.js";
+export type { Printable } from "./printable.js";
+export { safeIdentifier } from "./safe-identifier.js";

package/dist/cjs/index.js

@@ -13,7 +13,7 @@
 // See the License for the specific language governing permissions and
 // limitations under the License.
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.getSyntaxComments = exports.getPackageComments = exports.getDeclarationString = exports.getComments = exports.createEcmaScriptPlugin = exports.runNodeJs = void 0;
+exports.safeIdentifier = exports.createImportSymbol = exports.getSyntaxComments = exports.getPackageComments = exports.getDeclarationString = exports.getComments = exports.createEcmaScriptPlugin = exports.runNodeJs = void 0;
 var run_node_js_1 = require("./run-node.js");
 Object.defineProperty(exports, "runNodeJs", { enumerable: true, get: function () { return run_node_js_1.runNodeJs; } });
 var create_es_plugin_js_1 = require("./create-es-plugin.js");
@@ -23,3 +23,7 @@ Object.defineProperty(exports, "getComments", { enumerable: true, get: function
 Object.defineProperty(exports, "getDeclarationString", { enumerable: true, get: function () { return source_code_info_js_1.getDeclarationString; } });
 Object.defineProperty(exports, "getPackageComments", { enumerable: true, get: function () { return source_code_info_js_1.getPackageComments; } });
 Object.defineProperty(exports, "getSyntaxComments", { enumerable: true, get: function () { return source_code_info_js_1.getSyntaxComments; } });
+var import_symbol_js_1 = require("./import-symbol.js");
+Object.defineProperty(exports, "createImportSymbol", { enumerable: true, get: function () { return import_symbol_js_1.createImportSymbol; } });
+var safe_identifier_js_1 = require("./safe-identifier.js");
+Object.defineProperty(exports, "safeIdentifier", { enumerable: true, get: function () { return safe_identifier_js_1.safeIdentifier; } });

package/dist/cjs/{ecmascript/jsdoc.js → jsdoc.js}

@@ -14,7 +14,8 @@
 // limitations under the License.
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.formatJsDocBlock = exports.createJsDocTextFromDesc = void 0;
-const source_code_info_js_1 = require("../source-code-info.js");
+const source_code_info_js_1 = require("./source-code-info.js");
+const reflect_1 = require("@bufbuild/protobuf/reflect");
 function createJsDocTextFromDesc(desc) {
     const comments = (0, source_code_info_js_1.getComments)(desc);
     let text = "";
@@ -50,18 +51,29 @@ function createJsDocTextFromDesc(desc) {
         case "extension":
             text += `@generated from extension: ${(0, source_code_info_js_1.getDeclarationString)(desc)};`;
             break;
+        case "message":
+        case "enum": {
+            text += `@generated from ${desc.toString()}`;
+            const featureOptions = (0, source_code_info_js_1.getFeatureOptionStrings)(desc);
+            if (featureOptions.length > 0) {
+                text += `\n@generated with ${featureOptions.length > 1 ? "options" : "option"} ${featureOptions.join(", ")}`;
+            }
+            break;
+        }
         default:
             text += `@generated from ${desc.toString()}`;
             break;
     }
-    let deprecated = desc.deprecated;
+    let deprecated;
     switch (desc.kind) {
-        case "enum":
-        case "message":
-        case "service":
-            deprecated = deprecated || desc.file.deprecated;
+        case "field":
+        case "enum_value":
+        case "rpc":
+            deprecated = desc.deprecated;
             break;
         default:
+            deprecated =
+                desc.deprecated || (0, reflect_1.parentTypes)(desc).some((d) => d.deprecated);
             break;
     }
     if (deprecated) {

package/dist/cjs/{ecmascript/parameter.js → parameter.js}

@@ -14,14 +14,14 @@
 // limitations under the License.
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.parseParameter = void 0;
-const error_js_1 = require("../error.js");
+const error_js_1 = require("./error.js");
 function parseParameter(parameter, parseExtraOption) {
     let targets = ["js", "dts"];
     let tsNocheck = false;
     let bootstrapWkt = false;
     let keepEmptyFiles = false;
     const rewriteImports = [];
-    let importExtension = ".js";
+    let importExtension = "";
     let jsImportStyle = "module";
     const rawParameters = [];
     for (const { key, value, raw } of splitParameter(parameter)) {

package/dist/{esm/ecmascript → cjs}/printable.d.ts

@@ -1,5 +1,4 @@
-import type { DescEnum, DescExtension, DescFile, DescMessage, DescService } from "@bufbuild/protobuf";
-import { ScalarType, LongType } from "@bufbuild/protobuf/reflect";
+import type { DescEnum, DescExtension, DescFile, DescMessage, DescService, ScalarType } from "@bufbuild/protobuf";
 import type { ImportSymbol } from "./import-symbol.js";
 /**
  * All types that can be passed to GeneratedFile.print()
@@ -13,7 +12,7 @@ export type ExportStatement = {
 export type LiteralProtoInt64 = {
     readonly kind: "es_proto_int64";
     type: ScalarType.INT64 | ScalarType.SINT64 | ScalarType.SFIXED64 | ScalarType.UINT64 | ScalarType.FIXED64;
-    longType: LongType;
+    longAsString: boolean;
     value: bigint | string;
 };
 export type LiteralString = {

package/dist/cjs/{ecmascript/schema.js → schema.js}

@@ -92,8 +92,16 @@ function getFilesToGenerate(request, minimumEdition, maximumEdition) {
             }
         }
     }
-    const registry = (0, reflect_1.createFileRegistry)((0, protobuf_1.create)(wkt_1.FileDescriptorSetDesc, {
-        file: request.protoFile,
+    // Our goal is to provide options with source retention to plugin authors.
+    // CodeGeneratorRequest.proto_file elides options with source retention for
+    // files to generate. For these files, we take the file from source_file_descriptors,
+    // which does include options with source retention.
+    const allProtoWithSourceOptions = request.protoFile.map((protoFile) => {
+        const sourceFile = request.sourceFileDescriptors.find((s) => s.name == protoFile.name);
+        return sourceFile !== null && sourceFile !== void 0 ? sourceFile : protoFile;
+    });
+    const registry = (0, protobuf_1.createFileRegistry)((0, protobuf_1.create)(wkt_1.FileDescriptorSetDesc, {
+        file: allProtoWithSourceOptions,
     }));
     const allFiles = [];
     const filesToGenerate = [];

package/dist/cjs/source-code-info.d.ts

@@ -1,4 +1,4 @@
-import type { DescField, DescEnumValue, DescExtension, DescComments, AnyDesc, DescFile } from "@bufbuild/protobuf";
+import { type AnyDesc, type DescComments, type DescEnumValue, type DescExtension, type DescField, type DescFile } from "@bufbuild/protobuf";
 /**
  * Get comments on the package element in the protobuf source.
  */
@@ -11,6 +11,11 @@ export declare function getSyntaxComments(desc: DescFile): DescComments;
  * Get comments on the element in the protobuf source.
  */
 export declare function getComments(desc: Exclude<AnyDesc, DescFile>): DescComments;
+/**
+ * Get feature options set on the element in the protobuf source. This returns
+ * compact (e.g. fields) or regular options (e.g. files) as an array of strings.
+ */
+export declare function getFeatureOptionStrings(desc: AnyDesc): string[];
 /**
  * Return a string that matches the definition of a field in the protobuf
  * source. Does not take custom options into account.

package/dist/cjs/source-code-info.js

@@ -13,7 +13,7 @@
 // See the License for the specific language governing permissions and
 // limitations under the License.
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.getDeclarationString = exports.getComments = exports.getSyntaxComments = exports.getPackageComments = void 0;
+exports.getDeclarationString = exports.getFeatureOptionStrings = exports.getComments = exports.getSyntaxComments = exports.getPackageComments = void 0;
 const protobuf_1 = require("@bufbuild/protobuf");
 const reflect_1 = require("@bufbuild/protobuf/reflect");
 const wkt_1 = require("@bufbuild/protobuf/wkt");
@@ -22,7 +22,7 @@ const wkt_1 = require("@bufbuild/protobuf/wkt");
  */
 function getPackageComments(desc) {
     return findComments(desc.proto.sourceCodeInfo, [
-        FieldNumber.FileDescriptorProto_Package,
+        wkt_1.FileDescriptorProtoDesc.field.package.number,
     ]);
 }
 exports.getPackageComments = getPackageComments;
@@ -31,7 +31,7 @@ exports.getPackageComments = getPackageComments;
  */
 function getSyntaxComments(desc) {
     return findComments(desc.proto.sourceCodeInfo, [
-        FieldNumber.FileDescriptorProto_Syntax,
+        wkt_1.FileDescriptorProtoDesc.field.syntax.number,
     ]);
 }
 exports.getSyntaxComments = getSyntaxComments;
@@ -46,11 +46,11 @@ function getComments(desc) {
             path = desc.parent
                 ? [
                     ...getComments(desc.parent).sourcePath,
-                    FieldNumber.DescriptorProto_EnumType,
+                    wkt_1.DescriptorProtoDesc.field.enumType.number,
                     desc.parent.proto.enumType.indexOf(desc.proto),
                 ]
                 : [
-                    FieldNumber.FileDescriptorProto_EnumType,
+                    wkt_1.FileDescriptorProtoDesc.field.enumType.number,
                     desc.file.proto.enumType.indexOf(desc.proto),
                 ];
             file = desc.file;
@@ -58,7 +58,7 @@ function getComments(desc) {
         case "oneof":
             path = [
                 ...getComments(desc.parent).sourcePath,
-                FieldNumber.DescriptorProto_OneofDecl,
+                wkt_1.DescriptorProtoDesc.field.oneofDecl.number,
                 desc.parent.proto.oneofDecl.indexOf(desc.proto),
             ];
             file = desc.parent.file;
@@ -67,11 +67,11 @@ function getComments(desc) {
             path = desc.parent
                 ? [
                     ...getComments(desc.parent).sourcePath,
-                    FieldNumber.DescriptorProto_NestedType,
+                    wkt_1.DescriptorProtoDesc.field.nestedType.number,
                     desc.parent.proto.nestedType.indexOf(desc.proto),
                 ]
                 : [
-                    FieldNumber.FileDescriptorProto_MessageType,
+                    wkt_1.FileDescriptorProtoDesc.field.messageType.number,
                     desc.file.proto.messageType.indexOf(desc.proto),
                 ];
             file = desc.file;
@@ -79,7 +79,7 @@ function getComments(desc) {
         case "enum_value":
             path = [
                 ...getComments(desc.parent).sourcePath,
-                FieldNumber.EnumDescriptorProto_Value,
+                wkt_1.EnumDescriptorProtoDesc.field.value.number,
                 desc.parent.proto.value.indexOf(desc.proto),
             ];
             file = desc.parent.file;
@@ -87,7 +87,7 @@ function getComments(desc) {
         case "field":
             path = [
                 ...getComments(desc.parent).sourcePath,
-                FieldNumber.DescriptorProto_Field,
+                wkt_1.DescriptorProtoDesc.field.field.number,
                 desc.parent.proto.field.indexOf(desc.proto),
             ];
             file = desc.parent.file;
@@ -96,18 +96,18 @@ function getComments(desc) {
             path = desc.parent
                 ? [
                     ...getComments(desc.parent).sourcePath,
-                    FieldNumber.DescriptorProto_Extension,
+                    wkt_1.DescriptorProtoDesc.field.extension.number,
                     desc.parent.proto.extension.indexOf(desc.proto),
                 ]
                 : [
-                    FieldNumber.FileDescriptorProto_Extension,
+                    wkt_1.FileDescriptorProtoDesc.field.extension.number,
                     desc.file.proto.extension.indexOf(desc.proto),
                 ];
             file = desc.file;
             break;
         case "service":
             path = [
-                FieldNumber.FileDescriptorProto_Service,
+                wkt_1.FileDescriptorProtoDesc.field.service.number,
                 desc.file.proto.service.indexOf(desc.proto),
             ];
             file = desc.file;
@@ -115,7 +115,7 @@ function getComments(desc) {
         case "rpc":
             path = [
                 ...getComments(desc.parent).sourcePath,
-                FieldNumber.ServiceDescriptorProto_Method,
+                wkt_1.ServiceDescriptorProtoDesc.field.method.number,
                 desc.parent.proto.method.indexOf(desc.proto),
             ];
             file = desc.parent.file;
@@ -124,6 +124,30 @@ function getComments(desc) {
     return findComments(file.proto.sourceCodeInfo, path);
 }
 exports.getComments = getComments;
+/**
+ * Get feature options set on the element in the protobuf source. This returns
+ * compact (e.g. fields) or regular options (e.g. files) as an array of strings.
+ */
+function getFeatureOptionStrings(desc) {
+    var _a, _b;
+    const strings = [];
+    const features = (_a = desc.proto.options) === null || _a === void 0 ? void 0 : _a.features;
+    if (features !== undefined) {
+        const r = (0, reflect_1.reflect)(wkt_1.FeatureSetDesc, features);
+        for (const f of r.fields) {
+            if (f.fieldKind != "enum" || !r.isSet(f)) {
+                continue;
+            }
+            const val = r.get(f);
+            const name = (_b = f.enum.values.find((v) => v.number == val)) === null || _b === void 0 ? void 0 : _b.name;
+            if (name !== undefined) {
+                strings.push(`features.${f.name} = ${name}`);
+            }
+        }
+    }
+    return strings;
+}
+exports.getFeatureOptionStrings = getFeatureOptionStrings;
 /**
  * Return a string that matches the definition of a field in the protobuf
  * source. Does not take custom options into account.
@@ -137,7 +161,6 @@ function getDeclarationString(desc) {
         }
         return str;
     }
-    const file = desc.kind === "extension" ? desc.file : desc.parent.file;
     const parts = [];
     function typeName(f) {
         if (f.message) {
@@ -146,18 +169,16 @@ function getDeclarationString(desc) {
         if (f.enum) {
             return f.enum.typeName;
         }
-        return reflect_1.ScalarType[f.scalar].toLowerCase();
+        return protobuf_1.ScalarType[f.scalar].toLowerCase();
     }
     switch (desc.fieldKind) {
         case "scalar":
         case "enum":
         case "message":
-            if (file.edition === wkt_1.Edition.EDITION_PROTO2 &&
-                (0, protobuf_1.isFieldSet)(wkt_1.FieldDescriptorProtoDesc, desc.proto, "label") &&
-                desc.proto.label == wkt_1.FieldDescriptorProto_Label.REQUIRED) {
+            if (fieldHasRequiredKeyword(desc)) {
                 parts.push("required");
             }
-            if (desc.optional) {
+            if (fieldHasOptionalKeyword(desc)) {
                 parts.push("optional");
             }
             parts.push(typeName(desc));
@@ -166,7 +187,7 @@ function getDeclarationString(desc) {
             parts.push("repeated", typeName(desc));
             break;
         case "map": {
-            const k = reflect_1.ScalarType[desc.mapKey].toLowerCase();
+            const k = protobuf_1.ScalarType[desc.mapKey].toLowerCase();
             const v = typeName(desc);
             parts.push(`map<${k}, ${v}>`);
             break;
@@ -176,10 +197,10 @@ function getDeclarationString(desc) {
     const options = [];
     const protoOptions = desc.proto.options;
     if (protoOptions !== undefined &&
-        (0, protobuf_1.isFieldSet)(wkt_1.FieldOptionsDesc, protoOptions, "packed")) {
+        (0, protobuf_1.isFieldSet)(protoOptions, wkt_1.FieldOptionsDesc.field.packed)) {
         options.push(`packed = ${protoOptions.packed.toString()}`);
     }
-    if ((0, protobuf_1.isFieldSet)(wkt_1.FieldDescriptorProtoDesc, desc.proto, "defaultValue")) {
+    if ((0, protobuf_1.isFieldSet)(desc.proto, wkt_1.FieldDescriptorProtoDesc.field.defaultValue)) {
         let defaultValue = desc.proto.defaultValue;
         if (desc.proto.type == wkt_1.FieldDescriptorProto_Type.BYTES ||
             desc.proto.type == wkt_1.FieldDescriptorProto_Type.STRING) {
@@ -191,19 +212,45 @@ function getDeclarationString(desc) {
         options.push(`json_name = "${desc.jsonName}"`);
     }
     if (protoOptions !== undefined &&
-        (0, protobuf_1.isFieldSet)(wkt_1.FieldOptionsDesc, protoOptions, "jstype")) {
+        (0, protobuf_1.isFieldSet)(protoOptions, wkt_1.FieldOptionsDesc.field.jstype)) {
         options.push(`jstype = ${wkt_1.FieldOptions_JSType[protoOptions.jstype]}`);
     }
     if (protoOptions !== undefined &&
-        (0, protobuf_1.isFieldSet)(wkt_1.FieldOptionsDesc, protoOptions, "deprecated")) {
+        (0, protobuf_1.isFieldSet)(protoOptions, wkt_1.FieldOptionsDesc.field.deprecated)) {
         options.push(`deprecated = true`);
     }
+    options.push(...getFeatureOptionStrings(desc));
     if (options.length > 0) {
         parts.push("[" + options.join(", ") + "]");
     }
     return parts.join(" ");
 }
 exports.getDeclarationString = getDeclarationString;
+/**
+ * Whether this field was declared with `required` in the protobuf source.
+ */
+function fieldHasRequiredKeyword(field) {
+    const edition = (field.kind == "extension" ? field.file : field.parent.file)
+        .edition;
+    return (edition == wkt_1.Edition.EDITION_PROTO2 &&
+        field.proto.label == wkt_1.FieldDescriptorProto_Label.REQUIRED);
+}
+/**
+ * Whether this field was declared with `optional` in the protobuf source.
+ * Note that message fields are always optional. It is impossible to determine
+ * whether the keyword was used.
+ */
+function fieldHasOptionalKeyword(field) {
+    const edition = (field.kind == "extension" ? field.file : field.parent.file)
+        .edition;
+    if (edition == wkt_1.Edition.EDITION_PROTO2) {
+        return (!field.oneof && field.proto.label == wkt_1.FieldDescriptorProto_Label.OPTIONAL);
+    }
+    if (edition == wkt_1.Edition.EDITION_PROTO3) {
+        return field.proto.proto3Optional;
+    }
+    return false;
+}
 /**
  * Find comments.
  */
@@ -223,10 +270,10 @@ function findComments(sourceCodeInfo, sourcePath) {
         }
         return {
             leadingDetached: location.leadingDetachedComments,
-            leading: (0, protobuf_1.isFieldSet)(wkt_1.SourceCodeInfo_LocationDesc, location, "leadingComments")
+            leading: (0, protobuf_1.isFieldSet)(location, wkt_1.SourceCodeInfo_LocationDesc.field.leadingComments)
                 ? location.leadingComments
                 : undefined,
-            trailing: (0, protobuf_1.isFieldSet)(wkt_1.SourceCodeInfo_LocationDesc, location, "trailingComments")
+            trailing: (0, protobuf_1.isFieldSet)(location, wkt_1.SourceCodeInfo_LocationDesc.field.trailingComments)
                 ? location.trailingComments
                 : undefined,
             sourcePath,
@@ -237,23 +284,3 @@ function findComments(sourceCodeInfo, sourcePath) {
         sourcePath,
     };
 }
-/**
- * The following field numbers are used to find comments in
- * google.protobuf.SourceCodeInfo.
- */
-var FieldNumber;
-(function (FieldNumber) {
-    FieldNumber[FieldNumber["FileDescriptorProto_Package"] = 2] = "FileDescriptorProto_Package";
-    FieldNumber[FieldNumber["FileDescriptorProto_MessageType"] = 4] = "FileDescriptorProto_MessageType";
-    FieldNumber[FieldNumber["FileDescriptorProto_EnumType"] = 5] = "FileDescriptorProto_EnumType";
-    FieldNumber[FieldNumber["FileDescriptorProto_Service"] = 6] = "FileDescriptorProto_Service";
-    FieldNumber[FieldNumber["FileDescriptorProto_Extension"] = 7] = "FileDescriptorProto_Extension";
-    FieldNumber[FieldNumber["FileDescriptorProto_Syntax"] = 12] = "FileDescriptorProto_Syntax";
-    FieldNumber[FieldNumber["DescriptorProto_Field"] = 2] = "DescriptorProto_Field";
-    FieldNumber[FieldNumber["DescriptorProto_NestedType"] = 3] = "DescriptorProto_NestedType";
-    FieldNumber[FieldNumber["DescriptorProto_EnumType"] = 4] = "DescriptorProto_EnumType";
-    FieldNumber[FieldNumber["DescriptorProto_Extension"] = 6] = "DescriptorProto_Extension";
-    FieldNumber[FieldNumber["DescriptorProto_OneofDecl"] = 8] = "DescriptorProto_OneofDecl";
-    FieldNumber[FieldNumber["EnumDescriptorProto_Value"] = 2] = "EnumDescriptorProto_Value";
-    FieldNumber[FieldNumber["ServiceDescriptorProto_Method"] = 2] = "ServiceDescriptorProto_Method";
-})(FieldNumber || (FieldNumber = {}));

package/dist/esm/create-es-plugin.d.ts

@@ -1,6 +1,6 @@
-import type { SupportedEdition } from "@bufbuild/protobuf";
-import type { Schema } from "./ecmascript/schema.js";
-import type { FileInfo } from "./ecmascript/generated-file.js";
+import { type SupportedEdition } from "@bufbuild/protobuf";
+import type { Schema } from "./schema.js";
+import type { FileInfo } from "./generated-file.js";
 import type { Plugin } from "./plugin.js";
 interface PluginInit {
     /**

package/dist/esm/create-es-plugin.js

@@ -12,11 +12,11 @@
 // See the License for the specific language governing permissions and
 // limitations under the License.
 import { create, protoInt64 } from "@bufbuild/protobuf";
-import { minimumEdition as minimumEditionUpstream, maximumEdition as maximumEditionUpstream, } from "@bufbuild/protobuf/reflect";
+import { minimumEdition as minimumEditionUpstream, maximumEdition as maximumEditionUpstream, } from "@bufbuild/protobuf";
 import { CodeGeneratorResponse_Feature, CodeGeneratorResponseDesc, } from "@bufbuild/protobuf/wkt";
-import { createSchema } from "./ecmascript/schema.js";
-import { transpile } from "./ecmascript/transpile.js";
-import { parseParameter } from "./ecmascript/parameter.js";
+import { createSchema } from "./schema.js";
+import { transpile } from "./transpile.js";
+import { parseParameter } from "./parameter.js";
 /**
  * Create a new code generator plugin for ECMAScript.
  * The plugin can generate JavaScript, TypeScript, or TypeScript declaration

package/dist/esm/{ecmascript/file-preamble.js → file-preamble.js}

@@ -12,7 +12,7 @@
 // See the License for the specific language governing permissions and
 // limitations under the License.
 import { Edition } from "@bufbuild/protobuf/wkt";
-import { getPackageComments, getSyntaxComments } from "../source-code-info.js";
+import { getPackageComments, getSyntaxComments, getFeatureOptionStrings, } from "./source-code-info.js";
 export function makeFilePreamble(file, pluginName, pluginVersion, parameter, tsNoCheck) {
     const builder = [];
     const trimSuffix = (str, suffix) => str.endsWith(suffix) ? str.substring(0, str.length - suffix.length) : str;
@@ -61,6 +61,9 @@ export function makeFilePreamble(file, pluginName, pluginVersion, parameter, tsN
             break;
         }
     }
+    for (const o of getFeatureOptionStrings(file)) {
+        builder.push(`// option ${o};\n`);
+    }
     builder.push("/* eslint-disable */\n");
     if (tsNoCheck) {
         builder.push("// @ts-nocheck\n");

package/dist/esm/{ecmascript/generated-file.d.ts → generated-file.d.ts}

@@ -1,4 +1,4 @@
-import type { AnyDesc, DescEnum, DescExtension, DescFile, DescMessage, DescService } from "@bufbuild/protobuf";
+import { type AnyDesc, type DescEnum, type DescExtension, type DescFile, type DescMessage, type DescService } from "@bufbuild/protobuf";
 import type { ImportSymbol } from "./import-symbol.js";
 import type { Printable } from "./printable.js";
 import type { RuntimeImports } from "./runtime-imports.js";
@@ -63,7 +63,7 @@ export interface GeneratedFile {
      * Create a printable export statement. For example:
      *
      * ```ts
-     * f.print(f.exportDecl("abstract class", "MyClass"), " {}")
+     * f.print(f.export("abstract class", "MyClass"), " {}")
      * ```
      *
      * Will generate as:
@@ -75,7 +75,7 @@ export interface GeneratedFile {
      * statement. If the plugin option `js_import_style=legacy_commonjs` is set,
      * exports will automatically be generated for CommonJS.
      */
-    exportDecl(declaration: string, name: string): Printable;
+    export(declaration: string, name: string): Printable;
     /**
      * Import a message or enumeration generated by protoc-gen-es.
      */