From b092c9cb4c2e9fc8ad70eb893c7571d2165c2bf2 Mon Sep 17 00:00:00 2001
From: Arpad Ryszka <arpio@squareroundforest.org>
Date: Tue, 26 Aug 2025 21:02:10 +0200
Subject: [PATCH] add header to the generated output

---
 generate/generate.go      | 7 +++++++
 generate/generate_test.go | 6 +++---
 2 files changed, 10 insertions(+), 3 deletions(-)

diff --git a/generate/generate.go b/generate/generate.go
index f48c8fc..4dea088 100644
--- a/generate/generate.go
+++ b/generate/generate.go
@@ -21,6 +21,12 @@ import (
 	"strings"
 )
 
+const header = `/*
+Generated by https://code.squareroundforest.org/arpio/docreflect
+*/
+
+`
+
 type options struct {
 	wd      string
 	goroot  string
@@ -600,6 +606,7 @@ func format(w io.Writer, pname string, docs map[string]string) error {
 		_, err = fmt.Fprintln(w, a...)
 	}
 
+	println(header)
 	printf("package %s\n", pname)
 	println("import \"code.squareroundforest.org/arpio/docreflect\"")
 	println("func init() {")
diff --git a/generate/generate_test.go b/generate/generate_test.go
index 2650038..32a76ba 100644
--- a/generate/generate_test.go
+++ b/generate/generate_test.go
@@ -404,12 +404,12 @@ func TestFormat(t *testing.T) {
 	}
 
 	o := b.String()
-	if o != `package testpackage
+	if !strings.Contains(o, `package testpackage
 import "code.squareroundforest.org/arpio/docreflect"
 func init() {
 docreflect.Register("baz", "qux")
 docreflect.Register("foo", "bar")
-}` {
-		t.Fatal()
+}`) {
+		t.Fatal(o)
 	}
 }