feat: add comment logging maxDepth and maxURLs inside xml output

1 files changed, 5 insertions, 3 deletions

diff --git a/xml.go b/xml.go
index 400ae8e..51c9f93 100644
--- a/xml.go
+++ b/xml.go
@@ -10,6 +10,7 @@ type XMLURLset struct {
 	XMLName xml.Name `xml:"urlset"`
 	Xmlns   string   `xml:"xmlns,attr"`
 	URLs    []URL    `xml:"url"`
+	Comment string   `xml:",comment"`
 }
 
 type URL struct {
@@ -20,7 +21,7 @@ type Loc struct {
 	Text string `xml:",chardata"`
 }
 
-func toSitemap(seen map[url.URL]int) (string, error) {
+func toSitemap(seen map[url.URL]int, maxDepth, maxURLs int) (string, error) {
 	var xmlURLs []URL
 
 	for u := range seen {
@@ -34,8 +35,9 @@ func toSitemap(seen map[url.URL]int) (string, error) {
 	}
 
 	set := XMLURLset{
-		Xmlns: "http://www.sitemaps.org/schemas/sitemap/0.9",
-		URLs:  xmlURLs,
+		Xmlns:   "http://www.sitemaps.org/schemas/sitemap/0.9",
+		URLs:    xmlURLs,
+		Comment: fmt.Sprintf("max depth: %d, max urls: %d", maxDepth, maxURLs),
 	}
 
 	output, err := xml.MarshalIndent(&set, "", "\t")