forked from niklasfasching/go-org
-
Notifications
You must be signed in to change notification settings - Fork 0
/
main.go
115 lines (108 loc) · 2.79 KB
/
main.go
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
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
package main
import (
"bytes"
"fmt"
"io/ioutil"
"log"
"os"
"strings"
"github.com/alecthomas/chroma"
"github.com/alecthomas/chroma/formatters/html"
"github.com/alecthomas/chroma/lexers"
"github.com/alecthomas/chroma/styles"
"github.com/niklasfasching/go-org/blorg"
"github.com/niklasfasching/go-org/org"
)
func main() {
log := log.New(os.Stderr, "", 0)
if len(os.Args) < 2 {
log.Println("USAGE: org COMMAND [ARGS]")
log.Println("- org render FILE OUTPUT_FORMAT")
log.Println(" OUTPUT_FORMAT: org, html, html-chroma")
log.Println("- org blorg init")
log.Println("- org blorg build")
log.Println("- org blorg serve")
os.Exit(1)
}
switch cmd := strings.ToLower(os.Args[1]); cmd {
case "render":
if len(os.Args) < 4 {
log.Fatal("USAGE: org render FILE OUTPUT_FORMAT")
}
out, err := render(os.Args[2], os.Args[3])
if err != nil {
log.Fatalf("Error: %v", err)
}
fmt.Fprint(os.Stdout, out)
case "blorg":
if err := runBlorg(os.Args[2]); err != nil {
log.Fatalf("Error: %v", err)
}
default:
log.Fatalf("Unsupported command: %s", cmd)
}
}
func runBlorg(cmd string) error {
switch strings.ToLower(cmd) {
case "init":
if _, err := os.Stat(blorg.DefaultConfigFile); !os.IsNotExist(err) {
return err
}
if err := ioutil.WriteFile(blorg.DefaultConfigFile, []byte(blorg.DefaultConfig), os.ModePerm); err != nil {
return err
}
log.Printf("blorg init finished: Wrote ./%s", blorg.DefaultConfigFile)
return nil
case "build":
config, err := blorg.ReadConfig(blorg.DefaultConfigFile)
if err != nil {
return err
}
if err := config.Render(); err != nil {
return err
}
log.Println("blorg build finished")
return nil
case "serve":
config, err := blorg.ReadConfig(blorg.DefaultConfigFile)
if err != nil {
return err
}
return config.Serve()
default:
return fmt.Errorf("Supported commands: init build serve")
}
}
func render(path, format string) (string, error) {
bs, err := ioutil.ReadFile(path)
if err != nil {
return "", err
}
d := org.New().Parse(bytes.NewReader(bs), path)
switch strings.ToLower(format) {
case "org":
return d.Write(org.NewOrgWriter())
case "html":
return d.Write(org.NewHTMLWriter())
case "html-chroma":
writer := org.NewHTMLWriter()
writer.HighlightCodeBlock = highlightCodeBlock
return d.Write(writer)
default:
return "", fmt.Errorf("unsupported output format: %s", format)
}
}
func highlightCodeBlock(source, lang string, inline bool) string {
var w strings.Builder
l := lexers.Get(lang)
if l == nil {
l = lexers.Fallback
}
l = chroma.Coalesce(l)
it, _ := l.Tokenise(nil, source)
_ = html.New().Format(&w, styles.Get("friendly"), it)
if inline {
return `<div class="highlight-inline">` + "\n" + w.String() + "\n" + `</div>`
}
return `<div class="highlight">` + "\n" + w.String() + "\n" + `</div>`
}