package main import ( "io/ioutil" "os" "strings" "text/template" "github.com/aymerick/douceur/css" "github.com/aymerick/douceur/parser" "gopkg.in/alecthomas/kingpin.v3-unstable" "github.com/alecthomas/chroma/v2" ) const ( outputTemplate = `package styles import ( "github.com/alecthomas/chroma/v2" ) // {{.Name}} style. var {{.Name}} = Register(chroma.MustNewStyle("{{.Name|Lower}}", chroma.StyleEntries{ {{- range .Rules}} {{- if .Prelude|TokenType}} chroma.{{.Prelude|TokenType}}: "{{.Declarations|TranslateDecls}}", {{- end}} {{- end}} })) ` ) var ( typeByClass = map[string]chroma.TokenType{ ".hll": chroma.Background, } cssNamedColours = map[string]string{ "black": "#000000", "silver": "#c0c0c0", "gray": "#808080", "white": "#ffffff", "maroon": "#800000", "red": "#ff0000", "purple": "#800080", "fuchsia": "#ff00ff", "green": "#008000", "lime": "#00ff00", "olive": "#808000", "yellow": "#ffff00", "navy": "#000080", "blue": "#0000ff", "teal": "#008080", "aqua": "#00ffff", "orange": "#ffa500", "aliceblue": "#f0f8ff", "antiquewhite": "#faebd7", "aquamarine": "#7fffd4", "azure": "#f0ffff", "beige": "#f5f5dc", "bisque": "#ffe4c4", "blanchedalmond": "#ffebcd", "blueviolet": "#8a2be2", "brown": "#a52a2a", "burlywood": "#deb887", "cadetblue": "#5f9ea0", "chartreuse": "#7fff00", "chocolate": "#d2691e", "coral": "#ff7f50", "cornflowerblue": "#6495ed", "cornsilk": "#fff8dc", "crimson": "#dc143c", "cyan": "#00ffff", "darkblue": "#00008b", "darkcyan": "#008b8b", "darkgoldenrod": "#b8860b", "darkgray": "#a9a9a9", "darkgreen": "#006400", "darkgrey": "#a9a9a9", "darkkhaki": "#bdb76b", "darkmagenta": "#8b008b", "darkolivegreen": "#556b2f", "darkorange": "#ff8c00", "darkorchid": "#9932cc", "darkred": "#8b0000", "darksalmon": "#e9967a", "darkseagreen": "#8fbc8f", "darkslateblue": "#483d8b", "darkslategray": "#2f4f4f", "darkslategrey": "#2f4f4f", "darkturquoise": "#00ced1", "darkviolet": "#9400d3", "deeppink": "#ff1493", "deepskyblue": "#00bfff", "dimgray": "#696969", "dimgrey": "#696969", "dodgerblue": "#1e90ff", "firebrick": "#b22222", "floralwhite": "#fffaf0", "forestgreen": "#228b22", "gainsboro": "#dcdcdc", "ghostwhite": "#f8f8ff", "gold": "#ffd700", "goldenrod": "#daa520", "greenyellow": "#adff2f", "grey": "#808080", "honeydew": "#f0fff0", "hotpink": "#ff69b4", "indianred": "#cd5c5c", "indigo": "#4b0082", "ivory": "#fffff0", "khaki": "#f0e68c", "lavender": "#e6e6fa", "lavenderblush": "#fff0f5", "lawngreen": "#7cfc00", "lemonchiffon": "#fffacd", "lightblue": "#add8e6", "lightcoral": "#f08080", "lightcyan": "#e0ffff", "lightgoldenrodyellow": "#fafad2", "lightgray": "#d3d3d3", "lightgreen": "#90ee90", "lightgrey": "#d3d3d3", "lightpink": "#ffb6c1", "lightsalmon": "#ffa07a", "lightseagreen": "#20b2aa", "lightskyblue": "#87cefa", "lightslategray": "#778899", "lightslategrey": "#778899", "lightsteelblue": "#b0c4de", "lightyellow": "#ffffe0", "limegreen": "#32cd32", "linen": "#faf0e6", "magenta": "#ff00ff", "mediumaquamarine": "#66cdaa", "mediumblue": "#0000cd", "mediumorchid": "#ba55d3", "mediumpurple": "#9370db", "mediumseagreen": "#3cb371", "mediumslateblue": "#7b68ee", "mediumspringgreen": "#00fa9a", "mediumturquoise": "#48d1cc", "mediumvioletred": "#c71585", "midnightblue": "#191970", "mintcream": "#f5fffa", "mistyrose": "#ffe4e1", "moccasin": "#ffe4b5", "navajowhite": "#ffdead", "oldlace": "#fdf5e6", "olivedrab": "#6b8e23", "orangered": "#ff4500", "orchid": "#da70d6", "palegoldenrod": "#eee8aa", "palegreen": "#98fb98", "paleturquoise": "#afeeee", "palevioletred": "#db7093", "papayawhip": "#ffefd5", "peachpuff": "#ffdab9", "peru": "#cd853f", "pink": "#ffc0cb", "plum": "#dda0dd", "powderblue": "#b0e0e6", "rosybrown": "#bc8f8f", "royalblue": "#4169e1", "saddlebrown": "#8b4513", "salmon": "#fa8072", "sandybrown": "#f4a460", "seagreen": "#2e8b57", "seashell": "#fff5ee", "sienna": "#a0522d", "skyblue": "#87ceeb", "slateblue": "#6a5acd", "slategray": "#708090", "slategrey": "#708090", "snow": "#fffafa", "springgreen": "#00ff7f", "steelblue": "#4682b4", "tan": "#d2b48c", "thistle": "#d8bfd8", "tomato": "#ff6347", "turquoise": "#40e0d0", "violet": "#ee82ee", "wheat": "#f5deb3", "whitesmoke": "#f5f5f5", "yellowgreen": "#9acd32", "rebeccapurple": "#663399", } nameArg = kingpin.Arg("name", "Name of output style.").Required().String() fileArg = kingpin.Arg("stylesheets", ".css file to import").Required().ExistingFile() ) func init() { for tt, str := range chroma.StandardTypes { typeByClass["."+str] = tt } } func translateDecls(decls []*css.Declaration) string { out := []string{} for _, decl := range decls { switch decl.Property { case "color": clr := decl.Value if c, ok := cssNamedColours[clr]; ok { clr = c } out = append(out, clr) case "background-color": out = append(out, "bg:"+decl.Value) case "font-style": if strings.Contains(decl.Value, "italic") { out = append(out, "italic") } case "font-weight": if strings.Contains(decl.Value, "bold") { out = append(out, "bold") } case "text-decoration": if strings.Contains(decl.Value, "underline") { out = append(out, "underline") } } } return strings.Join(out, " ") } func main() { kingpin.Parse() source, err := ioutil.ReadFile(*fileArg) kingpin.FatalIfError(err, "") css, err := parser.Parse(string(source)) kingpin.FatalIfError(err, "") context := map[string]interface{}{ "Name": *nameArg, "Rules": css.Rules, } tmpl := template.Must(template.New("style").Funcs(template.FuncMap{ "Lower": strings.ToLower, "TranslateDecls": translateDecls, "TokenType": func(s string) chroma.TokenType { return typeByClass[s] }, }).Parse(outputTemplate)) err = tmpl.Execute(os.Stdout, context) kingpin.FatalIfError(err, "") }