From 3ba9f8dd04968adddfbe85e54373619996bfdf1f Mon Sep 17 00:00:00 2001 From: Josh Bleecher Snyder Date: Mon, 27 Mar 2023 15:33:42 -0700 Subject: [PATCH] util/codegen: add -copyright to control presence of copyright headers Fixes #7702 Signed-off-by: Josh Bleecher Snyder --- util/codegen/codegen.go | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/util/codegen/codegen.go b/util/codegen/codegen.go index 364a9b69e..b1e012fad 100644 --- a/util/codegen/codegen.go +++ b/util/codegen/codegen.go @@ -6,6 +6,7 @@ package codegen import ( "bytes" + "flag" "fmt" "go/ast" "go/token" @@ -20,6 +21,8 @@ import ( "tailscale.com/util/mak" ) +var flagCopyright = flag.Bool("copyright", true, "add Tailscale copyright to generated file headers") + // LoadTypes returns all named types in pkgName, keyed by their type name. func LoadTypes(buildTags string, pkgName string) (*packages.Package, map[string]*types.Named, error) { cfg := &packages.Config{ @@ -104,7 +107,9 @@ func (it *ImportTracker) Write(w io.Writer) { } func writeHeader(w io.Writer, tool, pkg string) { - fmt.Fprint(w, copyrightHeader) + if *flagCopyright { + fmt.Fprint(w, copyrightHeader) + } fmt.Fprintf(w, genAndPackageHeader, tool, pkg) }