package main

import (
	"bytes"
	"flag"
	"fmt"
	"io/ioutil"
	"log"
	"os"

	"github.com/cbroglie/mustache"
	mathjax "github.com/litao91/goldmark-mathjax"
	"github.com/yuin/goldmark"
	highlighting "github.com/yuin/goldmark-highlighting"
	meta "github.com/yuin/goldmark-meta"
	"github.com/yuin/goldmark/extension"
	"github.com/yuin/goldmark/parser"
)

func main() {
	var templateFilename string

	flag.StringVar(&templateFilename, "tmpl", "", "Template file to render the html inside. The content of the markdown file will replace the placeholder {{content}}.")

	flag.Parse()

	if flag.NArg() < 1 {
		fmt.Println("Please provide a markdown file for parsing.")
		os.Exit(1)
	}

	var useTemplate = false
	if templateFilename != "" {
		useTemplate = true
	}

	source, err := ioutil.ReadFile(flag.Arg(0))
	if err != nil {
		log.Fatal(err)
	}

	md := goldmark.New(
		goldmark.WithExtensions(
			extension.GFM,
			highlighting.Highlighting,
			meta.Meta,
			mathjax.MathJax,
		),
		goldmark.WithParserOptions(
			parser.WithAutoHeadingID(),
		),
	)

	htmlContent := new(bytes.Buffer)
	parserContext := parser.NewContext()
	err = md.Convert(source, htmlContent, parser.WithContext(parserContext))
	if err != nil {
		log.Fatal(err)
	}

	if useTemplate {
		tmpl, err := mustache.ParseFile(templateFilename)
		if err != nil {
			log.Fatal(err)
		}
		templateVars := meta.Get(parserContext)
		if templateVars == nil {
			templateVars = map[string]interface{}{}
		}
		templateVars["content"] = htmlContent.String()

		err = tmpl.FRender(os.Stdout, templateVars)
		if err != nil {
			log.Fatal(err)
		}
	} else {
		fmt.Print(htmlContent)
	}
}