Merge pull request #145 from ogen-go/feat/nested-enum

feat: add support nested enum as query

Author: tdakkota

example/message.proto

@@ -25,6 +25,7 @@ message Item {
 message GetItemsRequest {
   int32 limit = 1 [(google.api.field_behavior) = REQUIRED];
   int32 offset = 2 [(google.api.field_behavior) = REQUIRED];
+  NestedEnum.ItemType nested_type = 3 [(google.api.field_behavior) = REQUIRED];
 }
 
 message GetItemsResponse {

example/openapi.yaml

@@ -13,6 +13,11 @@ paths:
           schema:
             type: integer
             format: int32
+        - name: nestedType
+          in: query
+          required: true
+          schema:
+            $ref: '#/components/schemas/NestedEnum.ItemType'
         - name: offset
           in: query
           required: true

internal/gen/generator.go

@@ -32,9 +32,7 @@ func NewGenerator(files []*protogen.File, opts ...GeneratorOption) (*Generator,
 		}
 
 		for _, e := range f.Enums {
-			if err := g.mkEnum(e); err != nil {
-				return nil, err
-			}
+			g.mkEnum(e)
 		}
 
 		for _, s := range f.Services {
@@ -346,6 +344,10 @@ func (g *Generator) mkQueryParameters(op *ogen.Operation, fields map[string]*pro
 					delete(seen, msg)
 					continue
 				}
+			case protoreflect.EnumKind:
+				descName := descriptorName(fd.Enum())
+				s := mkEnumOgenSchema(fd.Enum())
+				g.spec.AddSchema(descName, s)
 			case protoreflect.GroupKind:
 				return errors.Errorf("unsupported kind: %s", kind)
 			}

internal/gen/schema.go

@@ -14,15 +14,11 @@ import (
 	"github.com/ogen-go/ogen"
 )
 
-func (g *Generator) mkEnum(e *protogen.Enum) error {
-	s := &ogen.Schema{
-		Type: "string",
-		Enum: enum(e.Desc),
-	}
+func (g *Generator) mkEnum(e *protogen.Enum) {
+	s := mkEnumOgenSchema(e.Desc)
 
 	name := descriptorName(e.Desc)
 	g.spec.AddSchema(name, s)
-	return nil
 }
 
 func enum(ed protoreflect.EnumDescriptor) []json.RawMessage {
@@ -41,6 +37,15 @@ func enum(ed protoreflect.EnumDescriptor) []json.RawMessage {
 	return enum
 }
 
+func mkEnumOgenSchema(ed protoreflect.EnumDescriptor) *ogen.Schema {
+	s := &ogen.Schema{
+		Type: "string",
+		Enum: enum(ed),
+	}
+
+	return s
+}
+
 func (g *Generator) mkSchema(msg *protogen.Message) error {
 	s := ogen.NewSchema().SetType("object")
 
@@ -59,9 +64,7 @@ func (g *Generator) mkSchema(msg *protogen.Message) error {
 			}
 		}
 		if field.Enum != nil {
-			if err := g.mkEnum(field.Enum); err != nil {
-				return err
-			}
+			g.mkEnum(field.Enum)
 		}
 	}
 
@@ -72,9 +75,7 @@ func (g *Generator) mkSchema(msg *protogen.Message) error {
 	}
 
 	for _, e := range msg.Enums {
-		if err := g.mkEnum(e); err != nil {
-			return err
-		}
+		g.mkEnum(e)
 	}
 
 	name := descriptorName(msg.Desc)