openbilibili/vendor/gopkg.in/AlecAivazis/survey.v1/core/template.go

package core

import (
	"bytes"
	"sync"
	"text/template"

	"github.com/mgutz/ansi"
)

var DisableColor = false

var (
	// HelpInputRune is the rune which the user should enter to trigger
	// more detailed question help
	HelpInputRune = '?'

	// ErrorIcon will be be shown before an error
	ErrorIcon = "X"

	// HelpIcon will be shown before more detailed question help
	HelpIcon = "?"
	// QuestionIcon will be shown before a question Message
	QuestionIcon = "?"

	// MarkedOptionIcon will be prepended before a selected multiselect option
	MarkedOptionIcon = "[x]"
	// UnmarkedOptionIcon will be prepended before an unselected multiselect option
	UnmarkedOptionIcon = "[ ]"

	// SelectFocusIcon is prepended to an option to signify the user is
	// currently focusing that option
	SelectFocusIcon = ">"
)

/*
  SetFancyIcons changes the err, help, marked, and focus input icons to their
  fancier forms. These forms may not be compatible with most terminals.
  This function will not touch the QuestionIcon as its fancy and non fancy form
  are the same.
*/
func SetFancyIcons() {
	ErrorIcon = "✘"
	HelpIcon = "ⓘ"
	// QuestionIcon fancy and non-fancy form are the same

	MarkedOptionIcon = "◉"
	UnmarkedOptionIcon = "◯"

	SelectFocusIcon = "❯"
}

var TemplateFuncs = map[string]interface{}{
	// Templates with Color formatting. See Documentation: https://github.com/mgutz/ansi#style-format
	"color": func(color string) string {
		if DisableColor {
			return ""
		}
		return ansi.ColorCode(color)
	},
	"HelpInputRune": func() string {
		return string(HelpInputRune)
	},
	"ErrorIcon": func() string {
		return ErrorIcon
	},
	"HelpIcon": func() string {
		return HelpIcon
	},
	"QuestionIcon": func() string {
		return QuestionIcon
	},
	"MarkedOptionIcon": func() string {
		return MarkedOptionIcon
	},
	"UnmarkedOptionIcon": func() string {
		return UnmarkedOptionIcon
	},
	"SelectFocusIcon": func() string {
		return SelectFocusIcon
	},
}

var (
	memoizedGetTemplate = map[string]*template.Template{}

	memoMutex = &sync.RWMutex{}
)

func getTemplate(tmpl string) (*template.Template, error) {
	memoMutex.RLock()
	if t, ok := memoizedGetTemplate[tmpl]; ok {
		memoMutex.RUnlock()
		return t, nil
	}
	memoMutex.RUnlock()

	t, err := template.New("prompt").Funcs(TemplateFuncs).Parse(tmpl)
	if err != nil {
		return nil, err
	}

	memoMutex.Lock()
	memoizedGetTemplate[tmpl] = t
	memoMutex.Unlock()
	return t, nil
}

func RunTemplate(tmpl string, data interface{}) (string, error) {
	t, err := getTemplate(tmpl)
	if err != nil {
		return "", err
	}
	buf := bytes.NewBufferString("")
	err = t.Execute(buf, data)
	if err != nil {
		return "", err
	}
	return buf.String(), err
}