Htmlize

Author: mattn

ccat.go

@@ -42,6 +42,14 @@ func (p PlainTextPrinter) Print(r io.Reader, w io.Writer) error {
 	return err
 }
 
+type HtmlPrinter struct {
+	ColorPalettes ColorPalettes
+}
+
+func (c HtmlPrinter) Print(r io.Reader, w io.Writer) error {
+	return HtmlPrint(r, w, c.ColorPalettes)
+}
+
 func CCat(fname string, p CCatPrinter, w io.Writer) error {
 	var r io.Reader
 

html.go

@@ -0,0 +1,106 @@
+package main
+
+import (
+	"bytes"
+	"fmt"
+	"sort"
+	"strings"
+)
+
+type HtmlCodes map[string]string
+
+func (c HtmlCodes) String() string {
+	var cc []string
+	for k, _ := range c {
+		if k == "" {
+			continue
+		}
+
+		cc = append(cc, k)
+	}
+	sort.Strings(cc)
+
+	var s []string
+	for _, ss := range cc {
+		s = append(s, Htmlize(ss, ss))
+	}
+
+	return strings.Join(s, ", ")
+}
+
+var htmlCodes = HtmlCodes{
+	"":          "",
+	"reset":     `</span>`,
+	"bold":      `<span class="bold">`,
+	"faint":     `<span class="faint">`,
+	"standout":  `<span class="standout">`,
+	"underline": `<span class="underline">`,
+	"blink":     `<span class="blink">`,
+	"overline":  `<span class="overline">`,
+}
+
+func init() {
+	darkHtmls := []string{
+		"black",
+		"darkred",
+		"darkgreen",
+		"brown",
+		"darkblue",
+		"purple",
+		"teal",
+		"lightgray",
+	}
+
+	lightHtmls := []string{
+		"darkgray",
+		"red",
+		"green",
+		"yellow",
+		"blue",
+		"fuchsia",
+		"turquoise",
+		"white",
+	}
+
+	for i, x := 0, 30; i < len(darkHtmls); i, x = i+1, x+1 {
+		htmlCodes[darkHtmls[i]] = fmt.Sprintf(`<span class="%s">`, darkHtmls[i])
+		htmlCodes[lightHtmls[i]] = fmt.Sprintf(`<span class="%s">`, lightHtmls[i])
+	}
+
+	htmlCodes["darkteal"] = htmlCodes["turquoise"]
+	htmlCodes["darkyellow"] = htmlCodes["brown"]
+	htmlCodes["fuscia"] = htmlCodes["fuchsia"]
+	htmlCodes["white"] = htmlCodes["bold"]
+}
+
+func Htmlize(attr, text string) string {
+	if attr == "" {
+		return text
+	}
+
+	result := new(bytes.Buffer)
+
+	if strings.HasPrefix(attr, "+") && strings.HasSuffix(attr, "+") {
+		result.WriteString(htmlCodes["blink"])
+		attr = strings.TrimPrefix(attr, "+")
+		attr = strings.TrimSuffix(attr, "+")
+	}
+
+	if strings.HasPrefix(attr, "*") && strings.HasSuffix(attr, "*") {
+		result.WriteString(htmlCodes["bold"])
+		attr = strings.TrimPrefix(attr, "*")
+		attr = strings.TrimSuffix(attr, "*")
+	}
+
+	if strings.HasPrefix(attr, "_") && strings.HasSuffix(attr, "_") {
+		result.WriteString(htmlCodes["underline"])
+		attr = strings.TrimPrefix(attr, "_")
+		attr = strings.TrimSuffix(attr, "_")
+	}
+
+	result.WriteString(htmlCodes[attr])
+	result.WriteString(text)
+	result.WriteString(htmlCodes["reset"])
+
+	return result.String()
+}

main.go

@@ -16,6 +16,7 @@ type ccatCmd struct {
 	BG          string
 	Color       string
 	ColorCodes  mapValue
+	HTML        bool
 	ShowPalette bool
 	ShowVersion bool
 }
@@ -61,7 +62,9 @@ Value of color can be %s
 	}
 
 	var printer CCatPrinter
-	if c.Color == "always" {
+	if c.HTML {
+		printer = HtmlPrinter{colorPalettes}
+	} else if c.Color == "always" {
 		printer = ColorPrinter{colorPalettes}
 	} else if c.Color == "never" {
 		printer = PlainTextPrinter{}
@@ -91,6 +94,7 @@ func main() {
 		Long: "Colorize FILE(s), or standard input, to standard output.",
 		Example: `$ ccat FILE1 FILE2 ...
   $ ccat --bg=dark FILE1 FILE2 ... # dark background
+  $ ccat --html # output html
   $ ccat -G String="_darkblue_" -G Plaintext="darkred" FILE # set color codes
   $ ccat --palette # show palette
   $ ccat # read from standard input
@@ -116,6 +120,7 @@ Examples:
 	rootCmd.PersistentFlags().StringVarP(&ccatCmd.BG, "bg", "", "light", `set to "light" or "dark" depending on the terminal's background`)
 	rootCmd.PersistentFlags().StringVarP(&ccatCmd.Color, "color", "C", "auto", `colorize the output; value can be "never", "always" or "auto"`)
 	rootCmd.PersistentFlags().VarP(&ccatCmd.ColorCodes, "color-code", "G", `set color codes`)
+	rootCmd.PersistentFlags().BoolVarP(&ccatCmd.HTML, "html", "", false, `output html`)
 	rootCmd.PersistentFlags().BoolVarP(&ccatCmd.ShowPalette, "palette", "", false, `show color palettes`)
 	rootCmd.PersistentFlags().BoolVarP(&ccatCmd.ShowVersion, "version", "v", false, `show version`)
 

printer.go

@@ -3,6 +3,7 @@ package main
 import (
 	"fmt"
 	"io"
+	"sort"
 	"strings"
 
 	"github.com/jingweno/ccat/Godeps/_workspace/src/github.com/sourcegraph/syntaxhighlight"
@@ -143,3 +144,38 @@ func (p Printer) Print(w io.Writer, kind syntaxhighlight.Kind, tokText string) e
 
 	return err
 }
+
+func HtmlPrint(r io.Reader, w io.Writer, palettes ColorPalettes) error {
+	keys := []string{}
+	for k := range htmlCodes {
+		keys = append(keys, k)
+	}
+	sort.Strings(keys)
+	w.Write([]byte("<style>\n"))
+	for _, s := range keys {
+		if s == "" {
+			continue
+		}
+		w.Write([]byte(fmt.Sprintf(".%s { color: %s; }\n", s, s)))
+	}
+	w.Write([]byte("</style>\n"))
+	w.Write([]byte("<pre>\n"))
+	err := syntaxhighlight.Print(syntaxhighlight.NewScannerReader(r), w, HtmlCodePrinter{palettes})
+	w.Write([]byte("\n</pre>\n"))
+	return err
+}
+
+type HtmlCodePrinter struct {
+	ColorPalettes ColorPalettes
+}
+
+func (p HtmlCodePrinter) Print(w io.Writer, kind syntaxhighlight.Kind, tokText string) error {
+	c := p.ColorPalettes.Get(kind)
+	if len(c) > 0 {
+		tokText = Htmlize(c, tokText)
+	}
+
+	_, err := io.WriteString(w, tokText)
+
+	return err
+}

printer_test.go

@@ -19,3 +19,51 @@ func TestCPrint(t *testing.T) {
 		t.Errorf("output is wrong: %s", s)
 	}
 }
+
+func TestHtmlPrint(t *testing.T) {
+	r := bytes.NewBufferString("hello")
+	var w bytes.Buffer
+
+	err := HtmlPrint(r, &w, LightColorPalettes)
+	if err != nil {
+		t.Errorf("error should be nil, but it's %s", err)
+	}
+
+	expect := `<style>
+.black { color: black; }
+.blink { color: blink; }
+.blue { color: blue; }
+.bold { color: bold; }
+.brown { color: brown; }
+.darkblue { color: darkblue; }
+.darkgray { color: darkgray; }
+.darkgreen { color: darkgreen; }
+.darkred { color: darkred; }
+.darkteal { color: darkteal; }
+.darkyellow { color: darkyellow; }
+.faint { color: faint; }
+.fuchsia { color: fuchsia; }
+.fuscia { color: fuscia; }
+.green { color: green; }
+.lightgray { color: lightgray; }
+.overline { color: overline; }
+.purple { color: purple; }
+.red { color: red; }
+.reset { color: reset; }
+.standout { color: standout; }
+.teal { color: teal; }
+.turquoise { color: turquoise; }
+.underline { color: underline; }
+.white { color: white; }
+.yellow { color: yellow; }
+</style>
+<pre>
+<span class="darkblue">hello</span>
+</pre>
+`
+
+	s := w.String()
+	if s != expect {
+		t.Errorf("output is wrong: %s", s)
+	}
+}