hugo/helpers/docshelper.go

55 lines
1.1 KiB
Go

package helpers
import (
"path/filepath"
"sort"
"strings"
"github.com/alecthomas/chroma/lexers"
"github.com/gohugoio/hugo/docshelper"
)
// This is is just some helpers used to create some JSON used in the Hugo docs.
func init() {
docsProvider := func() docshelper.DocProvider {
var chromaLexers []interface{}
sort.Sort(lexers.Registry.Lexers)
for _, l := range lexers.Registry.Lexers {
config := l.Config()
var filenames []string
filenames = append(filenames, config.Filenames...)
filenames = append(filenames, config.AliasFilenames...)
aliases := config.Aliases
for _, filename := range filenames {
alias := strings.TrimSpace(strings.TrimPrefix(filepath.Ext(filename), "."))
if alias != "" {
aliases = append(aliases, alias)
}
}
aliases = UniqueStringsSorted(aliases)
lexerEntry := struct {
Name string
Aliases []string
}{
config.Name,
aliases,
}
chromaLexers = append(chromaLexers, lexerEntry)
}
return docshelper.DocProvider{"chroma": map[string]interface{}{"lexers": chromaLexers}}
}
docshelper.AddDocProviderFunc(docsProvider)
}