Add basic language display

This needs to be cleaned up a bit still, the stats need to move into the
`#repo` element.

quit.go

@@ -6,10 +6,12 @@ import (
	"html/template"
	"io"
	"log"
	"math"
	"net/http"
	"os"
	"path"
	"path/filepath"
	"sort"
	"strings"
	"time"
@@ -52,11 +54,12 @@ func main() {
}
type FancyRepo struct {
	repo         *git.Repository
	commit       *FancyCommit
	commitCount  int
	branches     []*git.Branch
	contributors []*git.Signature
	repo          *git.Repository
	commit        *FancyCommit
	commitCount   int
	branches      []*git.Branch
	contributors  []*git.Signature
	languageStats *LanguageStats
}
func NewFancyRepo(repo *git.Repository) *FancyRepo {
@@ -166,6 +169,134 @@ func (fr *FancyRepo) Tags() ([]string, error) {
	return fr.repo.Tags.List()
}
var languageSuffixes = map[string]string{
	".c":            "C",
	".h":            "C",
	".cpp":          "C++",
	"CMakeList.txt": "CMake",
	".css":          "CSS",
	".go":           "Go",
	".html":         "HTML",
	".js":           "JavaScript",
	".java":         "Java",
	"Makefile":      "Makefile",
	".py":           "Python",
	".rb":           "Ruby",
	".rs":           "Rust",
	".sh":           "Shell",
	".bash":         "Shell",
}
var languageColors = map[string]string{
	"C":   "#555555",
	"C++": "#f34b7d",
	//"CMake": "",
	"CSS":        "#563d7c",
	"Go":         "#375eab",
	"HTML":       "#e34c26",
	"JavaScript": "#f1e05a",
	"Java":       "#b07219",
	"Makefile":   "#427819",
	"Python":     "#3572A5",
	"Ruby":       "#701516",
	"Rust":       "#dea584",
	"Shell":      "#89e051",
}
func fileToLanguage(filename string) string {
	filename = strings.ToLower(filename)
	for suffix, lang := range languageSuffixes {
		if strings.HasSuffix(filename, suffix) {
			return lang
		}
	}
	return ""
}
type LanguageStats struct {
	percentages map[string]float64
	Languages   []*LanguageInfo
}
type LanguageInfo struct {
	Language   string
	Percentage float64
}
func (li *LanguageInfo) String() string {
	return fmt.Sprintf("%s (%.2f%%)", li.Language, li.Percentage)
}
type languageByPercent []*LanguageInfo
func (l languageByPercent) Len() int           { return len(l) }
func (l languageByPercent) Less(i, j int) bool { return l[i].Percentage < l[j].Percentage }
func (l languageByPercent) Swap(i, j int)      { l[i], l[j] = l[j], l[i] }
func NewLanguageStats(percentages map[string]float64) *LanguageStats {
	for lang, percentage := range percentages {
		if percentage < 0.001 {
			delete(percentages, lang)
			continue
		}
		percentage = math.Trunc(percentage*10000) / 100
		percentages[lang] = percentage
	}
	languages := make([]*LanguageInfo, 0, len(percentages))
	for lang, percentage := range percentages {
		languages = append(languages, &LanguageInfo{Language: lang, Percentage: percentage})
	}
	sort.Sort(sort.Reverse(languageByPercent(languages)))
	return &LanguageStats{
		percentages: percentages,
		Languages:   languages,
	}
}
func (li *LanguageStats) String() string {
	return fmt.Sprintf("%v", li.percentages)
}
func (fr *FancyRepo) LanguageStats() (*LanguageStats, error) {
	if fr.languageStats == nil {
		commit, err := fr.Commit()
		if err != nil {
			return nil, err
		}
		files, err := commit.Files(true)
		if err != nil {
			return nil, err
		}
		total := 0
		languages := make(map[string]int)
		for _, f := range files {
			if f.entry.Type != git.ObjectBlob {
				continue
			}
			lang := fileToLanguage(f.Name())
			if lang == "" {
				continue
			}
			total += 1
			languages[lang] += 1
		}
		percentages := make(map[string]float64)
		for lang, count := range languages {
			percentages[lang] = float64(count) / float64(total)
		}
		fr.languageStats = NewLanguageStats(percentages)
	}
	return fr.languageStats, nil
}
type FancyCommit struct {
	commit     *git.Commit
	files      []*FancyFile
@@ -227,16 +358,20 @@ func (fc *FancyCommit) Description() string {
	return fc.commit.Message()
}
func (fc *FancyCommit) Files() ([]*FancyFile, error) {
func (fc *FancyCommit) Files(recursive bool) ([]*FancyFile, error) {
	if fc.files == nil {
		tree, err := fc.commit.Tree()
		if err != nil {
			return nil, err
		}
		recurse := 1
		if recursive {
			recurse = 0
		}
		err = tree.Walk(func(s string, entry *git.TreeEntry) int {
			fc.files = append(fc.files, NewFancyFile(fc.commit.Owner(), entry))
			return 1
			return recurse
		})
		if err != nil {
			return nil, err
@@ -331,7 +466,16 @@ func (fc *FancyFile) Contents() (template.HTML, error) {
	return template.HTML(buf.String()), nil
}
var repoTmpl = template.Must(template.New("").Parse(`<!doctype html>
var repoFuncs = template.FuncMap{
	"languageToColor": func(l string) string {
		c, ok := languageColors[l]
		if !ok {
			return "black"
		}
		return c
	},
}
var repoTmpl = template.Must(template.New("").Funcs(repoFuncs).Parse(`<!doctype html>
<html>
	<head>
		<meta charset="utf-8" />
@@ -347,6 +491,23 @@ var repoTmpl = template.Must(template.New("").Parse(`<!doctype html>
			<a id="contributors" class="repo-stat" href="#"><span class="icon">👥</span><span class="count">{{ len .Repo.Contributors }}</span> contributors</a>
		</section>
		<div id="languages">
			<ol class="language-list">
			{{ range $language := .Repo.LanguageStats.Languages }}
				<li>
					<span class="language">{{ $language.Language }}</span>
					<span class="percentage">{{ $language.Percentage }}%</span>
				</li>
			{{ end }}
			</ol>
			<div id="language-bar">
			{{ range $language := .Repo.LanguageStats.Languages }}
				<span class="language" title="{{ $language }}" style="width: {{ $language.Percentage }}%; background-color: {{ languageToColor $language.Language }}"><!--{{ $language.Language }}--></span>
			{{ end }}
			</div>
		</div>
		<section id="commit" class="commit-info">
			<div class="commit-author">{{ .Repo.Commit.Author }}</div>
			<div class="commit-summary" title="{{ .Repo.Commit.Description }}">{{ .Repo.Commit.Summary }}</div>
@@ -355,7 +516,7 @@ var repoTmpl = template.Must(template.New("").Parse(`<!doctype html>
		</section>
		<table id="files">
			{{ range $file := .Repo.Commit.Files }}
			{{ range $file := .Repo.Commit.Files false }}
			<tr class="file">
				<td class="file-name file-type-{{ $file.Type }}">{{ $file.Name }}</td>
			</tr>
@@ -416,6 +577,17 @@ body {
	padding-right: 0.2em;
}
.language-list {
  display: none;
}
#language-bar {
  display: flex;
  width: 70vw;
  height: 0.5em;
  margin: -1em 0 1em;
}
#commit {
  display: flex;
  justify-content: space-between;