summaryrefslogtreecommitdiffstats
path: root/modules/highlight/highlight.go
blob: 39b5d6d1532575ef15e03887f6ab72c8836b3ef6 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
// Copyright 2015 The Gogs Authors. All rights reserved.
// Use of this source code is governed by a MIT-style
// license that can be found in the LICENSE file.

package highlight

import (
	"path"
	"strings"

	"code.gitea.io/gitea/modules/setting"
)

var (
	// File name should ignore highlight.
	ignoreFileNames = map[string]bool{
		"license": true,
		"copying": true,
	}

	// File names that are representing highlight classes.
	highlightFileNames = map[string]bool{
		"dockerfile": true,
		"makefile":   true,
	}

	// Extensions that are same as highlight classes.
	highlightExts = map[string]bool{
		".arm":    true,
		".as":     true,
		".sh":     true,
		".cs":     true,
		".cpp":    true,
		".c":      true,
		".css":    true,
		".cmake":  true,
		".bat":    true,
		".dart":   true,
		".patch":  true,
		".elixir": true,
		".erlang": true,
		".go":     true,
		".html":   true,
		".xml":    true,
		".hs":     true,
		".ini":    true,
		".json":   true,
		".java":   true,
		".js":     true,
		".less":   true,
		".lua":    true,
		".php":    true,
		".py":     true,
		".rb":     true,
		".scss":   true,
		".sql":    true,
		".scala":  true,
		".swift":  true,
		".ts":     true,
		".vb":     true,
	}

	// Extensions that are not same as highlight classes.
	highlightMapping = map[string]string{}
)

// NewContext loads highlight map
func NewContext() {
	keys := setting.Cfg.Section("highlight.mapping").Keys()
	for i := range keys {
		highlightMapping[keys[i].Name()] = keys[i].Value()
	}
}

// FileNameToHighlightClass returns the best match for highlight class name
// based on the rule of highlight.js.
func FileNameToHighlightClass(fname string) string {
	fname = strings.ToLower(fname)
	if ignoreFileNames[fname] {
		return "nohighlight"
	}

	if highlightFileNames[fname] {
		return fname
	}

	ext := path.Ext(fname)
	if highlightExts[ext] {
		return ext[1:]
	}

	name, ok := highlightMapping[ext]
	if ok {
		return name
	}

	return ""
}