snapshot of https://github.com/ent/ent.git (master, 2026-09-15; blobs > 131072B omitted)

examples/jsonencode/ent/entc.go

@@ -0,0 +1,81 @@
+// Copyright 2019-present Facebook Inc. All rights reserved.
+// This source code is licensed under the Apache 2.0 license found
+// in the LICENSE file in the root directory of this source tree.
+
+//go:build ignore
+// +build ignore
+
+package main
+
+import (
+	"log"
+
+	"entgo.io/ent/entc"
+	"entgo.io/ent/entc/gen"
+	"entgo.io/ent/schema/edge"
+)
+
+func main() {
+	opts := []entc.Option{
+		entc.Extensions(
+			&EncodeExtension{},
+		),
+	}
+	err := entc.Generate("./schema", &gen.Config{
+		Header: `
+			// Copyright 2019-present Facebook Inc. All rights reserved.
+			// This source code is licensed under the Apache 2.0 license found
+			// in the LICENSE file in the root directory of this source tree.
+
+			// Code generated by ent, DO NOT EDIT.
+		`,
+	}, opts...)
+	if err != nil {
+		log.Fatalf("running ent codegen: %v", err)
+	}
+}
+
+// EncodeExtension is an implementation of entc.Extension that adds a MarshalJSON
+// method to each generated type <T> and inlines the Edges field to the top level JSON.
+type EncodeExtension struct {
+	entc.DefaultExtension
+}
+
+// Templates of the extension.
+func (e *EncodeExtension) Templates() []*gen.Template {
+	return []*gen.Template{
+		gen.MustParse(gen.NewTemplate("model/additional/jsonencode").
+			Parse(`
+{{ if $.Edges }}
+	// MarshalJSON implements the json.Marshaler interface.
+	func ({{ $.Receiver }} *{{ $.Name }}) MarshalJSON() ([]byte, error) {
+		type Alias {{ $.Name }}
+		return json.Marshal(&struct {
+			*Alias
+			{{ $.Name }}Edges
+		}{
+			Alias: (*Alias)({{ $.Receiver }}),
+			{{ $.Name }}Edges: {{ $.Receiver }}.Edges,
+		})
+	}
+{{ end }}
+`)),
+	}
+}
+
+// Hooks of the extension.
+func (e *EncodeExtension) Hooks() []gen.Hook {
+	return []gen.Hook{
+		func(next gen.Generator) gen.Generator {
+			return gen.GenerateFunc(func(g *gen.Graph) error {
+				tag := edge.Annotation{StructTag: `json:"-"`}
+				for _, n := range g.Nodes {
+					n.Annotations.Set(tag.Name(), tag)
+				}
+				return next.Generate(g)
+			})
+		},
+	}
+}
+
+var _ entc.Extension = (*EncodeExtension)(nil)