tpl/diagrams: Move Goat to its own file

2 files changed, 68 insertions, 50 deletions

diff --git a/tpl/diagrams/diagrams.go b/tpl/diagrams/diagrams.go
index d5d19b08b..dfa29a978 100644
--- a/tpl/diagrams/diagrams.go
+++ b/tpl/diagrams/diagrams.go
@@ -14,14 +14,7 @@
 package diagrams
 
 import (
-	"bytes"
 	"html/template"
-	"io"
-	"strings"
-
-	"github.com/bep/goat"
-	"github.com/gohugoio/hugo/deps"
-	"github.com/spf13/cast"
 )
 
 type SVGDiagram interface {
@@ -38,46 +31,3 @@ type SVGDiagram interface {
 	// Height returns the height of the SVG.
 	Height() int
 }
-
-type goatDiagram struct {
-	d goat.SVG
-}
-
-func (d goatDiagram) Inner() template.HTML {
-	return template.HTML(d.d.Body)
-}
-
-func (d goatDiagram) Wrapped() template.HTML {
-	return template.HTML(d.d.String())
-}
-
-func (d goatDiagram) Width() int {
-	return d.d.Width
-}
-
-func (d goatDiagram) Height() int {
-	return d.d.Height
-}
-
-// Namespace provides template functions for the diagrams namespace.
-type Namespace struct {
-	d *deps.Deps
-}
-
-// Goat creates a new SVG diagram from input v.
-func (d *Namespace) Goat(v any) SVGDiagram {
-	var r io.Reader
-
-	switch vv := v.(type) {
-	case io.Reader:
-		r = vv
-	case []byte:
-		r = bytes.NewReader(vv)
-	default:
-		r = strings.NewReader(cast.ToString(v))
-	}
-
-	return goatDiagram{
-		d: goat.BuildSVG(r),
-	}
-}
diff --git a/tpl/diagrams/goat.go b/tpl/diagrams/goat.go
new file mode 100644
index 000000000..f3d4f4bfb
--- /dev/null
+++ b/tpl/diagrams/goat.go
@@ -0,0 +1,68 @@
+// Copyright 2022 The Hugo Authors. All rights reserved.
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+// http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+package diagrams
+
+import (
+	"bytes"
+	"html/template"
+	"io"
+	"strings"
+
+	"github.com/bep/goat"
+	"github.com/gohugoio/hugo/deps"
+	"github.com/spf13/cast"
+)
+
+type goatDiagram struct {
+	d goat.SVG
+}
+
+func (d goatDiagram) Inner() template.HTML {
+	return template.HTML(d.d.Body)
+}
+
+func (d goatDiagram) Wrapped() template.HTML {
+	return template.HTML(d.d.String())
+}
+
+func (d goatDiagram) Width() int {
+	return d.d.Width
+}
+
+func (d goatDiagram) Height() int {
+	return d.d.Height
+}
+
+// Namespace provides template functions for the diagrams namespace.
+type Namespace struct {
+	d *deps.Deps
+}
+
+// Goat creates a new SVG diagram from input v.
+func (d *Namespace) Goat(v any) SVGDiagram {
+	var r io.Reader
+
+	switch vv := v.(type) {
+	case io.Reader:
+		r = vv
+	case []byte:
+		r = bytes.NewReader(vv)
+	default:
+		r = strings.NewReader(cast.ToString(v))
+	}
+
+	return goatDiagram{
+		d: goat.BuildSVG(r),
+	}
+}