1 package main
2
3 import (
4 "io/ioutil"
5 "os"
6 "strings"
7 "text/template"
8
9 "github.com/aymerick/douceur/css"
10 "github.com/aymerick/douceur/parser"
11 "gopkg.in/alecthomas/kingpin.v3-unstable"
12
13 "github.com/alecthomas/chroma/v2"
14 )
15
16 const (
17 outputTemplate = `package styles
18
19 import (
20 "github.com/alecthomas/chroma/v2"
21 )
22
23 // {{.Name}} style.
24 var {{.Name}} = Register(chroma.MustNewStyle("{{.Name|Lower}}", chroma.StyleEntries{
25 {{- range .Rules}}
26 {{- if .Prelude|TokenType}}
27 chroma.{{.Prelude|TokenType}}: "{{.Declarations|TranslateDecls}}",
28 {{- end}}
29 {{- end}}
30 }))
31 `
32 )
33
34 var (
35 typeByClass = map[string]chroma.TokenType{
36 ".hll": chroma.Background,
37 }
38
39 cssNamedColours = map[string]string{
40 "black": "#000000", "silver": "#c0c0c0", "gray": "#808080", "white": "#ffffff",
41 "maroon": "#800000", "red": "#ff0000", "purple": "#800080", "fuchsia": "#ff00ff",
42 "green": "#008000", "lime": "#00ff00", "olive": "#808000", "yellow": "#ffff00",
43 "navy": "#000080", "blue": "#0000ff", "teal": "#008080", "aqua": "#00ffff",
44 "orange": "#ffa500", "aliceblue": "#f0f8ff", "antiquewhite": "#faebd7", "aquamarine": "#7fffd4",
45 "azure": "#f0ffff", "beige": "#f5f5dc", "bisque": "#ffe4c4", "blanchedalmond": "#ffebcd",
46 "blueviolet": "#8a2be2", "brown": "#a52a2a", "burlywood": "#deb887", "cadetblue": "#5f9ea0",
47 "chartreuse": "#7fff00", "chocolate": "#d2691e", "coral": "#ff7f50", "cornflowerblue": "#6495ed",
48 "cornsilk": "#fff8dc", "crimson": "#dc143c", "cyan": "#00ffff", "darkblue": "#00008b",
49 "darkcyan": "#008b8b", "darkgoldenrod": "#b8860b", "darkgray": "#a9a9a9", "darkgreen": "#006400",
50 "darkgrey": "#a9a9a9", "darkkhaki": "#bdb76b", "darkmagenta": "#8b008b", "darkolivegreen": "#556b2f",
51 "darkorange": "#ff8c00", "darkorchid": "#9932cc", "darkred": "#8b0000", "darksalmon": "#e9967a",
52 "darkseagreen": "#8fbc8f", "darkslateblue": "#483d8b", "darkslategray": "#2f4f4f", "darkslategrey": "#2f4f4f",
53 "darkturquoise": "#00ced1", "darkviolet": "#9400d3", "deeppink": "#ff1493", "deepskyblue": "#00bfff",
54 "dimgray": "#696969", "dimgrey": "#696969", "dodgerblue": "#1e90ff", "firebrick": "#b22222",
55 "floralwhite": "#fffaf0", "forestgreen": "#228b22", "gainsboro": "#dcdcdc", "ghostwhite": "#f8f8ff",
56 "gold": "#ffd700", "goldenrod": "#daa520", "greenyellow": "#adff2f", "grey": "#808080",
57 "honeydew": "#f0fff0", "hotpink": "#ff69b4", "indianred": "#cd5c5c", "indigo": "#4b0082",
58 "ivory": "#fffff0", "khaki": "#f0e68c", "lavender": "#e6e6fa", "lavenderblush": "#fff0f5",
59 "lawngreen": "#7cfc00", "lemonchiffon": "#fffacd", "lightblue": "#add8e6", "lightcoral": "#f08080",
60 "lightcyan": "#e0ffff", "lightgoldenrodyellow": "#fafad2", "lightgray": "#d3d3d3", "lightgreen": "#90ee90",
61 "lightgrey": "#d3d3d3", "lightpink": "#ffb6c1", "lightsalmon": "#ffa07a", "lightseagreen": "#20b2aa",
62 "lightskyblue": "#87cefa", "lightslategray": "#778899", "lightslategrey": "#778899", "lightsteelblue": "#b0c4de",
63 "lightyellow": "#ffffe0", "limegreen": "#32cd32", "linen": "#faf0e6", "magenta": "#ff00ff",
64 "mediumaquamarine": "#66cdaa", "mediumblue": "#0000cd", "mediumorchid": "#ba55d3", "mediumpurple": "#9370db",
65 "mediumseagreen": "#3cb371", "mediumslateblue": "#7b68ee", "mediumspringgreen": "#00fa9a", "mediumturquoise": "#48d1cc",
66 "mediumvioletred": "#c71585", "midnightblue": "#191970", "mintcream": "#f5fffa", "mistyrose": "#ffe4e1",
67 "moccasin": "#ffe4b5", "navajowhite": "#ffdead", "oldlace": "#fdf5e6", "olivedrab": "#6b8e23",
68 "orangered": "#ff4500", "orchid": "#da70d6", "palegoldenrod": "#eee8aa", "palegreen": "#98fb98",
69 "paleturquoise": "#afeeee", "palevioletred": "#db7093", "papayawhip": "#ffefd5", "peachpuff": "#ffdab9",
70 "peru": "#cd853f", "pink": "#ffc0cb", "plum": "#dda0dd", "powderblue": "#b0e0e6",
71 "rosybrown": "#bc8f8f", "royalblue": "#4169e1", "saddlebrown": "#8b4513", "salmon": "#fa8072",
72 "sandybrown": "#f4a460", "seagreen": "#2e8b57", "seashell": "#fff5ee", "sienna": "#a0522d",
73 "skyblue": "#87ceeb", "slateblue": "#6a5acd", "slategray": "#708090", "slategrey": "#708090",
74 "snow": "#fffafa", "springgreen": "#00ff7f", "steelblue": "#4682b4", "tan": "#d2b48c",
75 "thistle": "#d8bfd8", "tomato": "#ff6347", "turquoise": "#40e0d0", "violet": "#ee82ee",
76 "wheat": "#f5deb3", "whitesmoke": "#f5f5f5", "yellowgreen": "#9acd32", "rebeccapurple": "#663399",
77 }
78
79 nameArg = kingpin.Arg("name", "Name of output style.").Required().String()
80 fileArg = kingpin.Arg("stylesheets", ".css file to import").Required().ExistingFile()
81 )
82
83 func init() {
84 for tt, str := range chroma.StandardTypes {
85 typeByClass["."+str] = tt
86 }
87 }
88
89 func translateDecls(decls []*css.Declaration) string {
90 out := []string{}
91 for _, decl := range decls {
92 switch decl.Property {
93 case "color":
94 clr := decl.Value
95 if c, ok := cssNamedColours[clr]; ok {
96 clr = c
97 }
98 out = append(out, clr)
99 case "background-color":
100 out = append(out, "bg:"+decl.Value)
101 case "font-style":
102 if strings.Contains(decl.Value, "italic") {
103 out = append(out, "italic")
104 }
105 case "font-weight":
106 if strings.Contains(decl.Value, "bold") {
107 out = append(out, "bold")
108 }
109 case "text-decoration":
110 if strings.Contains(decl.Value, "underline") {
111 out = append(out, "underline")
112 }
113 }
114 }
115 return strings.Join(out, " ")
116 }
117
118 func main() {
119 kingpin.Parse()
120 source, err := ioutil.ReadFile(*fileArg)
121 kingpin.FatalIfError(err, "")
122 css, err := parser.Parse(string(source))
123 kingpin.FatalIfError(err, "")
124
125 context := map[string]interface{}{
126 "Name": *nameArg,
127 "Rules": css.Rules,
128 }
129 tmpl := template.Must(template.New("style").Funcs(template.FuncMap{
130 "Lower": strings.ToLower,
131 "TranslateDecls": translateDecls,
132 "TokenType": func(s string) chroma.TokenType { return typeByClass[s] },
133 }).Parse(outputTemplate))
134 err = tmpl.Execute(os.Stdout, context)
135 kingpin.FatalIfError(err, "")
136 }
137
View as plain text