func main() {
// parse command-line options
var page, xml, xml2 bool
var css, head string
flag.BoolVar(&page, "page", false, "generate a standalone HTML page")
flag.BoolVar(&xml, "xml", false, "generate XML2RFC v3 output")
flag.BoolVar(&xml2, "xml2", false, "generate XML2RFC v2 output")
flag.StringVar(&css, "css", "", "link to a CSS stylesheet (implies -page)")
flag.StringVar(&head, "head", "", "link to HTML to be included in head (implies -page)")
flag.StringVar(&mmark.CitationsID, "bib-id", mmark.CitationsID, "ID bibliography URL")
flag.StringVar(&mmark.CitationsRFC, "bib-rfc", mmark.CitationsRFC, "RFC bibliography URL")
flag.Usage = func() {
fmt.Fprintf(os.Stderr, "Mmark Markdown Processor"+
"\nAvailable at http://github.com/miekg/mmark\n\n"+
"Copyright © 2014 Miek Gieben <*****@*****.**>\n"+
"Copyright © 2011 Russ Ross <*****@*****.**>\n"+
"Distributed under the Simplified BSD License\n\n"+
"Usage:\n"+
" %s [options] [inputfile [outputfile]]\n\n"+
"Options:\n",
os.Args[0])
flag.PrintDefaults()
}
flag.Parse()
// enforce implied options
if css != "" {
page = true
}
if head != "" {
page = true
}
// read the input
var input []byte
var err error
args := flag.Args()
switch len(args) {
case 0:
if input, err = ioutil.ReadAll(os.Stdin); err != nil {
log.Fatalf("error reading from Stdin:", err)
}
case 1, 2:
if input, err = ioutil.ReadFile(args[0]); err != nil {
log.Fatalf("error reading from", args[0], ":", err)
}
default:
flag.Usage()
return
}
// set up options
extensions := 0
extensions |= mmark.EXTENSION_TABLES
extensions |= mmark.EXTENSION_FENCED_CODE
extensions |= mmark.EXTENSION_AUTOLINK
extensions |= mmark.EXTENSION_SPACE_HEADERS
extensions |= mmark.EXTENSION_CITATION
extensions |= mmark.EXTENSION_TITLEBLOCK_TOML
extensions |= mmark.EXTENSION_HEADER_IDS
extensions |= mmark.EXTENSION_AUTO_HEADER_IDS
extensions |= mmark.EXTENSION_UNIQUE_HEADER_IDS
extensions |= mmark.EXTENSION_FOOTNOTES
extensions |= mmark.EXTENSION_SHORT_REF
extensions |= mmark.EXTENSION_NO_EMPTY_LINE_BEFORE_BLOCK
extensions |= mmark.EXTENSION_INCLUDE
extensions |= mmark.EXTENSION_PARTS
var renderer mmark.Renderer
xmlFlags := 0
switch {
case xml:
if page {
xmlFlags = mmark.XML_STANDALONE
}
renderer = mmark.XmlRenderer(xmlFlags)
case xml2:
if page {
xmlFlags = mmark.XML2_STANDALONE
}
renderer = mmark.Xml2Renderer(xmlFlags)
default:
// render the data into HTML
htmlFlags := 0
if page {
htmlFlags |= mmark.HTML_COMPLETE_PAGE
}
renderer = mmark.HtmlRenderer(htmlFlags, css, head)
}
// parse and render
output := mmark.Parse(input, renderer, extensions).Bytes()
// output the result
out := os.Stdout
if len(args) == 2 {
if out, err = os.Create(args[1]); err != nil {
log.Fatalf("error creating %s: %v", args[1], err)
}
defer out.Close()
}
if _, err = out.Write(output); err != nil {
log.Fatalf("error writing output:", err)
}
}
func main() {
// parse command-line options
var page, xml, xml2, commonmark bool
var css, head, cpuprofile string
var repeat int
flag.BoolVar(&page, "page", false, "generate a standalone HTML page")
flag.BoolVar(&xml, "xml", false, "generate XML2RFC v3 output")
flag.BoolVar(&xml2, "xml2", false, "generate XML2RFC v2 output")
flag.BoolVar(&commonmark, "commonmark", false, "input is commonmark")
flag.StringVar(&css, "css", "", "link to a CSS stylesheet (implies -page)")
flag.StringVar(&head, "head", "", "link to HTML to be included in head (implies -page)")
flag.StringVar(&cpuprofile, "cpuprofile", "", "write cpu profile to a file")
flag.IntVar(&repeat, "repeat", 1, "process the input multiple times (for benchmarking)")
flag.Usage = func() {
fmt.Fprintf(os.Stderr, "Mmark Markdown Processor v"+mmark.VERSION+
"\nAvailable at http://github.com/miekg/mmark\n\n"+
"Copyright © 2014 Miek Gieben <*****@*****.**>\n"+
"Copyright © 2011 Russ Ross <*****@*****.**>\n"+
"Distributed under the Simplified BSD License\n"+
"See website for details\n\n"+
"Usage:\n"+
" %s [options] [inputfile [outputfile]]\n\n"+
"Options:\n",
os.Args[0])
flag.PrintDefaults()
}
flag.Parse()
// enforce implied options
if css != "" {
page = true
}
if head != "" {
page = true
}
// turn on profiling?
if cpuprofile != "" {
f, err := os.Create(cpuprofile)
if err != nil {
fmt.Fprintln(os.Stderr, err)
}
pprof.StartCPUProfile(f)
defer pprof.StopCPUProfile()
}
// read the input
var input []byte
var err error
args := flag.Args()
switch len(args) {
case 0:
if input, err = ioutil.ReadAll(os.Stdin); err != nil {
fmt.Fprintln(os.Stderr, "error reading from Stdin:", err)
os.Exit(-1)
}
case 1, 2:
if input, err = ioutil.ReadFile(args[0]); err != nil {
fmt.Fprintln(os.Stderr, "error reading from", args[0], ":", err)
os.Exit(-1)
}
default:
flag.Usage()
os.Exit(-1)
}
// set up options
extensions := 0
extensions |= mmark.EXTENSION_TABLES
extensions |= mmark.EXTENSION_FENCED_CODE
extensions |= mmark.EXTENSION_AUTOLINK
extensions |= mmark.EXTENSION_SPACE_HEADERS
extensions |= mmark.EXTENSION_CITATION
extensions |= mmark.EXTENSION_TITLEBLOCK_TOML
extensions |= mmark.EXTENSION_HEADER_IDS
extensions |= mmark.EXTENSION_AUTO_HEADER_IDS
extensions |= mmark.EXTENSION_UNIQUE_HEADER_IDS
extensions |= mmark.EXTENSION_FOOTNOTES
extensions |= mmark.EXTENSION_SHORT_REF
extensions |= mmark.EXTENSION_NO_EMPTY_LINE_BEFORE_BLOCK
extensions |= mmark.EXTENSION_INCLUDE
if commonmark {
extensions &= ^mmark.EXTENSION_AUTO_HEADER_IDS
extensions &= ^mmark.EXTENSION_AUTOLINK
extensions |= mmark.EXTENSION_LAX_HTML_BLOCKS
}
var renderer mmark.Renderer
xmlFlags := 0
switch {
case xml:
if page {
xmlFlags = mmark.XML_STANDALONE
}
renderer = mmark.XmlRenderer(xmlFlags)
case xml2:
if page {
xmlFlags = mmark.XML2_STANDALONE
}
renderer = mmark.Xml2Renderer(xmlFlags)
default:
// render the data into HTML
htmlFlags := 0
if page {
htmlFlags |= mmark.HTML_COMPLETE_PAGE
}
renderer = mmark.HtmlRenderer(htmlFlags, css, head)
}
// parse and render
var output []byte
for i := 0; i < repeat; i++ {
// TODO(miek): io.Copy
output = mmark.Parse(input, renderer, extensions).Bytes()
}
// output the result
var out *os.File
if len(args) == 2 {
if out, err = os.Create(args[1]); err != nil {
fmt.Fprintf(os.Stderr, "error creating %s: %v", args[1], err)
os.Exit(-1)
}
defer out.Close()
} else {
out = os.Stdout
}
if _, err = out.Write(output); err != nil {
fmt.Fprintln(os.Stderr, "error writing output:", err)
os.Exit(-1)
}
}