feat(gengo): also put headers in internal package

Author: raskyld

cmd/gen.go

@@ -79,7 +79,7 @@ The code generated for your main package will be written in zz_generated.tektask
 
 			genFunc = genFuncPtr
 
-			genInternalPtr, err := gengo.NewGoInternal(ctx.Logger, outputPkgName)
+			genInternalPtr, err := gengo.NewGoInternal(ctx.Logger, outputPkgName, headerFile, year)
 			if err != nil {
 				return err
 			}

internal/gengo/gen_func.go

@@ -68,11 +68,11 @@ type FuncArgs struct {
 	TemplatesArgs PerTemplateArgs
 }
 
-func NewGoFunc(logger *slog.Logger, headerfile, year string) (*TaskGoFuncGenerator, error) {
+func NewGoFunc(logger *slog.Logger, headerFile, year string) (*TaskGoFuncGenerator, error) {
 	g := &TaskGoFuncGenerator{
 		Logger:     logger.With("generator", "goFunc"),
 		Template:   &template.Template{},
-		HeaderFile: headerfile,
+		HeaderFile: headerFile,
 		Year:       year,
 	}
 

internal/gengo/gen_internal.go

@@ -22,22 +22,25 @@ import (
 	"log/slog"
 	"sigs.k8s.io/controller-tools/pkg/genall"
 	"sigs.k8s.io/controller-tools/pkg/markers"
+	"strings"
 	"text/template"
 )
 
 type TaskGoInternalGenerator struct {
-	Logger *slog.Logger
-
-	Template *template.Template
-
+	Logger      *slog.Logger
+	Template    *template.Template
 	PackageName string
+	HeaderFile  string
+	Year        string
 }
 
-func NewGoInternal(logger *slog.Logger, pkgName string) (*TaskGoInternalGenerator, error) {
+func NewGoInternal(logger *slog.Logger, pkgName, headerFile, year string) (*TaskGoInternalGenerator, error) {
 	g := &TaskGoInternalGenerator{
 		Logger:      logger.With("generator", "goInternal"),
 		Template:    &template.Template{},
 		PackageName: pkgName,
+		HeaderFile:  headerFile,
+		Year:        year,
 	}
 
 	g.RegisterTemplate(GoHeaderName, GoHeaderTpl).
@@ -63,10 +66,20 @@ func (*TaskGoInternalGenerator) RegisterMarkers(into *markers.Registry) error {
 
 func (g *TaskGoInternalGenerator) Generate(ctx *genall.GenerationContext) error {
 	var headerBytes bytes.Buffer
+	var headerText string
+
+	if g.HeaderFile != "" {
+		buf, err := ctx.ReadFile(g.HeaderFile)
+		if err != nil {
+			return err
+		}
+
+		headerText = strings.ReplaceAll(string(buf), " YEAR", " "+g.Year)
+	}
 
 	headerArgs := GoHeaderArgs{
 		PkgName: g.PackageName,
-		Header:  "// generated code",
+		Header:  headerText,
 		ImportPaths: []string{
 			"errors",
 			"fmt",

internal/gengo/go_header.go

@@ -20,7 +20,8 @@ const GoHeaderName = "go.header"
 
 const GoHeaderTpl = `{{- with .Header -}}
 {{.}}
-{{- end}}
+{{end -}}
+
 package {{.PkgName}}
 
 // This code is generated by Tektasker, DO NOT EDIT