From eb1b2cb7c412984de6ab8dafac42e0f0d2ff8888 Mon Sep 17 00:00:00 2001 From: Toby Padilla Date: Tue, 10 Aug 2021 17:21:46 -0500 Subject: [PATCH] Use templatize function --- tui/bubbles/repo/bubble.go | 21 ++++++++++++--------- 1 file changed, 12 insertions(+), 9 deletions(-) diff --git a/tui/bubbles/repo/bubble.go b/tui/bubbles/repo/bubble.go index 417ffc9b4b44dea31f59c7c84cd3a09ddb0ff8b1..45dba827efa893fa0903a4bf6a90f6a7a28adc8b 100644 --- a/tui/bubbles/repo/bubble.go +++ b/tui/bubbles/repo/bubble.go @@ -55,16 +55,10 @@ func (b *Bubble) setupCmd() tea.Msg { } md := r.Readme if b.templateObject != nil { - t, err := template.New("readme").Parse(md) + md, err = b.templatize(md) if err != nil { return ErrMsg{err} } - buf := &bytes.Buffer{} - err = t.Execute(buf, b.templateObject) - if err != nil { - return ErrMsg{err} - } - md = buf.String() } md, err = b.glamourize(md) if err != nil { @@ -75,8 +69,17 @@ func (b *Bubble) setupCmd() tea.Msg { return nil } -func (b *Bubble) templatize(t string) (string, error) { - return "", nil +func (b *Bubble) templatize(mdt string) (string, error) { + t, err := template.New("readme").Parse(mdt) + if err != nil { + return "", err + } + buf := &bytes.Buffer{} + err = t.Execute(buf, b.templateObject) + if err != nil { + return "", err + } + return buf.String(), nil } func (b *Bubble) glamourize(md string) (string, error) {