wtf/modules/mercurial/display.go

package mercurial

import (
	"fmt"
	"strings"
	"unicode/utf8"
)

func (widget *Widget) display() {
	repoData := widget.currentData()
	if repoData == nil {
		widget.View.SetText(" Mercurial repo data is unavailable ")
		return
	}

	title := fmt.Sprintf("%s - [green]%s[white]", widget.Name(), repoData.Repository)
	widget.View.SetTitle(widget.ContextualTitle(title))

	_, _, width, _ := widget.View.GetRect()
	str := widget.settings.common.SigilStr(len(widget.Data), widget.Idx, width) + "\n"
	str = str + " [red]Branch:Bookmark[white]\n"
	str = str + fmt.Sprintf(" %s:%s\n", repoData.Branch, repoData.Bookmark)
	str = str + "\n"
	str = str + widget.formatChanges(repoData.ChangedFiles)
	str = str + "\n"
	str = str + widget.formatCommits(repoData.Commits)

	widget.View.SetText(str)
}

func (widget *Widget) formatChanges(data []string) string {
	str := ""
	str = str + " [red]Changed Files[white]\n"

	if len(data) == 1 {
		str = str + " [grey]none[white]\n"
	} else {
		for _, line := range data {
			str = str + widget.formatChange(line)
		}
	}

	return str
}

func (widget *Widget) formatChange(line string) string {
	if len(line) == 0 {
		return ""
	}

	line = strings.TrimSpace(line)
	firstChar, _ := utf8.DecodeRuneInString(line)

	// Revisit this and kill the ugly duplication
	switch firstChar {
	case 'A':
		line = strings.Replace(line, "A", "[green]A[white]", 1)
	case 'D':
		line = strings.Replace(line, "D", "[red]D[white]", 1)
	case 'M':
		line = strings.Replace(line, "M", "[yellow]M[white]", 1)
	case 'R':
		line = strings.Replace(line, "R", "[purple]R[white]", 1)
	}

	return fmt.Sprintf(" %s\n", strings.Replace(line, "\"", "", -1))
}

func (widget *Widget) formatCommits(data []string) string {
	str := ""
	str = str + " [red]Recent Commits[white]\n"

	for _, line := range data {
		str = str + widget.formatCommit(line)
	}

	return str
}

func (widget *Widget) formatCommit(line string) string {
	return fmt.Sprintf(" %s\n", strings.Replace(line, "\"", "", -1))
}