md2vim/vimdoc.go

/*
 * Copyright (c) 2015 Alex Yatskov <alex@foosoft.net>
 * Author: Alex Yatskov <alex@foosoft.net>
 *
 * Permission is hereby granted, free of charge, to any person obtaining a copy of
 * this software and associated documentation files (the "Software"), to deal in
 * the Software without restriction, including without limitation the rights to
 * use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of
 * the Software, and to permit persons to whom the Software is furnished to do so,
 * subject to the following conditions:
 *
 * The above copyright notice and this permission notice shall be included in all
 * copies or substantial portions of the Software.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS
 * FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR
 * COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER
 * IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN
 * CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
 */

package main

import (
	"bytes"
	"fmt"
	"log"
	"strings"

	"github.com/russross/blackfriday"
)

const (
	LIST_STYLE_ORDERED = iota + 1
	LIST_STYLE_UNORDERED
)

func indent(text []byte) []byte {
	return nil
}

type listCtx struct {
	style, index int
}

type vimDoc struct {
	lists []*listCtx
}

func (v *vimDoc) pushList(style int) {
	v.lists = append(v.lists, &listCtx{style, 1})
}

func (v *vimDoc) popList() {
	if len(v.lists) == 0 {
		log.Fatal("invalid list operation")
	}

	v.lists = v.lists[:len(v.lists)-1]
}

func (v *vimDoc) getList() *listCtx {
	if len(v.lists) == 0 {
		log.Fatal("invalid list operation")
	}

	return v.lists[len(v.lists)-1]
}

func VimDocRenderer() blackfriday.Renderer {
	return &vimDoc{}
}

// Block-level callbacks
func (*vimDoc) BlockCode(out *bytes.Buffer, text []byte, lang string) {
	out.WriteString(">\n")
	out.Write(text)
	out.WriteString("<\n")
}

func (*vimDoc) BlockQuote(out *bytes.Buffer, text []byte) {
	out.WriteString(">\n")
	out.Write(text)
	out.WriteString("<\n")
}

func (*vimDoc) BlockHtml(out *bytes.Buffer, text []byte) {
	out.WriteString(">\n")
	out.Write(text)
	out.WriteString("<\n")
}

func (*vimDoc) Header(out *bytes.Buffer, text func() bool, level int, id string) {
	if text() {
		out.WriteString(" ~\n")
	}
}

func (*vimDoc) HRule(out *bytes.Buffer) {
	out.WriteString(strings.Repeat("=", 80))
}

func (v *vimDoc) List(out *bytes.Buffer, text func() bool, flags int) {
	style := LIST_STYLE_UNORDERED
	if flags&blackfriday.LIST_TYPE_ORDERED == blackfriday.LIST_TYPE_ORDERED {
		style = LIST_STYLE_ORDERED
	}
	v.pushList(style)

	text()

	v.popList()
}

func (v *vimDoc) ListItem(out *bytes.Buffer, text []byte, flags int) {
	list := v.getList()
	out.WriteString(fmt.Sprintf("\n%d.", list.index))
	out.Write(text)
	list.index++
}

func (*vimDoc) Paragraph(out *bytes.Buffer, text func() bool) {
	marker := out.Len()
	out.WriteString("\n")
	if !text() {
		out.Truncate(marker)
		return
	}
	out.WriteString("\n")
}

func (*vimDoc) Table(out *bytes.Buffer, header []byte, body []byte, columnData []int) {
}

func (*vimDoc) TableRow(out *bytes.Buffer, text []byte) {
}

func (*vimDoc) TableHeaderCell(out *bytes.Buffer, text []byte, flags int) {
}

func (*vimDoc) TableCell(out *bytes.Buffer, text []byte, flags int) {
}

func (*vimDoc) Footnotes(out *bytes.Buffer, text func() bool) {
	text()
}

func (*vimDoc) FootnoteItem(out *bytes.Buffer, name, text []byte, flags int) {
	out.Write(text)
}

func (*vimDoc) TitleBlock(out *bytes.Buffer, text []byte) {
	out.Write(text)
}

// Span-level callbacks
func (*vimDoc) AutoLink(out *bytes.Buffer, link []byte, kind int) {
	out.Write(link)
}

func (*vimDoc) CodeSpan(out *bytes.Buffer, text []byte) {
	out.Write(text)
}

func (*vimDoc) DoubleEmphasis(out *bytes.Buffer, text []byte) {
	out.Write(text)
}

func (*vimDoc) Emphasis(out *bytes.Buffer, text []byte) {
	out.Write(text)
}

func (*vimDoc) Image(out *bytes.Buffer, link []byte, title []byte, alt []byte) {
	// not implemented
}

func (*vimDoc) LineBreak(out *bytes.Buffer) {
	out.WriteString("\n")
}

func (*vimDoc) Link(out *bytes.Buffer, link []byte, title []byte, content []byte) {
	out.WriteString(fmt.Sprintf("%s (%s)", content, link))
}

func (*vimDoc) RawHtmlTag(out *bytes.Buffer, tag []byte) {
	out.Write(tag)
}

func (*vimDoc) TripleEmphasis(out *bytes.Buffer, text []byte) {
	out.Write(text)
}

func (*vimDoc) StrikeThrough(out *bytes.Buffer, text []byte) {
	out.Write(text)
}

func (*vimDoc) FootnoteRef(out *bytes.Buffer, ref []byte, id int) {
	// not implemented
}

// Low-level callbacks
func (*vimDoc) Entity(out *bytes.Buffer, entity []byte) {
	out.Write(entity)
}

func (*vimDoc) NormalText(out *bytes.Buffer, text []byte) {
	out.Write(text)
}

// Header and footer
func (*vimDoc) DocumentHeader(out *bytes.Buffer) {
	// not implemented
}

func (*vimDoc) DocumentFooter(out *bytes.Buffer) {
	// not implemented
}

func (*vimDoc) GetFlags() int {
	return 0
}