mirror of
https://github.com/gohugoio/hugo.git
synced 2024-11-07 20:30:36 -05:00
tpl/diagrams: Move Goat to its own file
This commit is contained in:
parent
002cd52807
commit
e754d5cb3e
2 changed files with 68 additions and 50 deletions
|
@ -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),
|
||||
}
|
||||
}
|
||||
|
|
68
tpl/diagrams/goat.go
Normal file
68
tpl/diagrams/goat.go
Normal file
|
@ -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),
|
||||
}
|
||||
}
|
Loading…
Reference in a new issue