glamour-temp-MUTN/baselement.go

package gold

import (
	"fmt"
	"io"
	"strconv"

	"github.com/logrusorgru/aurora"
	bf "gopkg.in/russross/blackfriday.v2"
)

type BaseElement struct {
	Token  string
	Prefix string
	Suffix string
	Style  StyleType
}

func (e *BaseElement) renderText(w io.Writer, rules *ElementStyle, s string) error {
	out := aurora.Reset(s)
	if rules.Color != "" {
		i, err := strconv.Atoi(rules.Color)
		if err == nil && i >= 0 && i <= 255 {
			out = out.Index(uint8(i))
		}
	}
	if rules.BackgroundColor != "" {
		i, err := strconv.Atoi(rules.BackgroundColor)
		if err == nil && i >= 0 && i <= 255 {
			out = out.Index(uint8(i))
		}
	}

	if rules.Underline {
		out = out.Underline()
	}
	if rules.Bold {
		out = out.Bold()
	}
	if rules.Italic {
		out = out.Italic()
	}
	if rules.CrossedOut {
		out = out.CrossedOut()
	}
	if rules.Overlined {
		out = out.Overlined()
	}
	if rules.Inverse {
		out = out.Reverse()
	}
	if rules.Blink {
		out = out.Blink()
	}

	_, err := w.Write([]byte(aurora.Sprintf("%s", out)))
	return err
}

func (e *BaseElement) Render(w io.Writer, node *bf.Node, tr *TermRenderer) error {
	if e.Prefix != "" {
		fmt.Fprintf(w, "%s", e.Prefix)
	}
	defer func() {
		if e.Suffix != "" {
			fmt.Fprintf(w, "%s", e.Suffix)
		}
	}()

	rules := tr.style[e.Style]
	if rules == nil {
		fmt.Fprintf(w, "%s", e.Token)
		return nil
	}

	if rules.Prefix != "" {
		e.renderText(w, rules, rules.Prefix)
	}
	defer func() {
		if rules.Suffix != "" {
			e.renderText(w, rules, rules.Suffix)
		}
	}()

	return e.renderText(w, rules, e.Token)
}