main.go

package main

import (
	"os"
	"fmt"
	"flag"
	"strings"
	"io/ioutil"
	"github.com/fairlyblank/md2min"
)

var mode string
var modeList []string

func pFail() {
	if err := recover(); err != nil {
		fmt.Fprintf(os.Stderr, "%v\n", err)
	}
}

func usage() {
	fmt.Fprintf(os.Stderr, "Usage: md2min [-nav=h2] name.md\n  name.md: markdown file name\n")
	flag.PrintDefaults()
}

func init() {
	modeList = []string{"none", "h1", "h2", "h3", "h4", "h5", "h6"}
	flag.StringVar(&mode, "nav", "none", `navigate level ["none", "h1", "h2", "h3", "h4", "h5", "h6"]`)
}

func main() {
	defer pFail()

	flag.Parse()

	i := 0
	for ; i<len(modeList); i++ {
		if modeList[i] == mode {
			break
		}
	}
	
	if flag.NArg() < 1 || i >= len(modeList) {
		usage()
		return
	}

	name := flag.Args()[0]
	file, err := os.Open(name)
	if err != nil {
		panic(err)
	}
	defer file.Close()

	input, err := ioutil.ReadAll(file)
	if err != nil {
		panic(err)
	}

	md := md2min.New(mode)
	
	newname := strings.TrimRight(name, ".md") + ".html"
	outfile, err := os.Create(newname)
	if err != nil {
		panic(err)
	}
	defer outfile.Close()

	err = md.Parse(input, outfile)
	if err != nil {
		panic(err)
	}
}