oh-my-posh/src/segment_time.go
Jan De Dobbeleer 79fa990205 feat: env vars in templates
resolves #743
2021-05-27 22:06:33 +02:00

56 lines
1.2 KiB
Go

package main
import "time"
type tempus struct {
props *properties
env environmentInfo
templateText string
CurrentDate time.Time
}
const (
// TimeFormat uses the reference time Mon Jan 2 15:04:05 MST 2006 to show the pattern with which to format the current time
TimeFormat Property = "time_format"
)
func (t *tempus) enabled() bool {
// if no date set, use now(unit testing)
if t.CurrentDate.IsZero() {
t.CurrentDate = time.Now()
}
segmentTemplate := t.props.getString(SegmentTemplate, "")
if segmentTemplate != "" {
template := &textTemplate{
Template: segmentTemplate,
Context: t,
Env: t.env,
}
var err error
t.templateText, err = template.render()
if err != nil {
t.templateText = err.Error()
}
return len(t.templateText) > 0
}
return true
}
func (t *tempus) string() string {
return t.getFormattedText()
}
func (t *tempus) init(props *properties, env environmentInfo) {
t.props = props
t.env = env
}
func (t *tempus) getFormattedText() string {
if len(t.templateText) > 0 {
return t.templateText
}
// keep old behaviour if no template
timeFormatProperty := t.props.getString(TimeFormat, "15:04:05")
return t.CurrentDate.Format(timeFormatProperty)
}