From b19b5a829bae837dad75fea9fbaa9c5a55c8a811 Mon Sep 17 00:00:00 2001
From: Alex Yatskov <alex@foosoft.net>
Date: Sun, 9 Aug 2015 14:18:09 +0900
Subject: [PATCH] Making it better

---
 vimdoc.go  | 41 +++++++++++++++++++++++++++++++----------
 vimdown.go | 42 +++++++++++++++---------------------------
 2 files changed, 46 insertions(+), 37 deletions(-)

diff --git a/vimdoc.go b/vimdoc.go
index 9688581..69e93ec 100644
--- a/vimdoc.go
+++ b/vimdoc.go
@@ -26,6 +26,7 @@ import (
 	"bytes"
 	"fmt"
 	"log"
+	"path"
 	"regexp"
 	"strings"
 
@@ -48,17 +49,32 @@ type header struct {
 }
 
 type vimDoc struct {
-	rootHead *header
-	lastHead *header
+	filename string
 	title    string
+	desc     string
 	cols     int
 	tabs     int
 	tocPos   int
 	lists    []*list
+	rootHead *header
+	lastHead *header
 }
 
-func VimDocRenderer(cols, tabs int) blackfriday.Renderer {
-	return &vimDoc{title: "test", cols: cols, tabs: tabs, tocPos: -1}
+func VimDocRenderer(filename, desc string, cols, tabs int) blackfriday.Renderer {
+	filename = path.Base(filename)
+	title := filename
+
+	if index := strings.LastIndex(filename, "."); index > -1 {
+		title = strings.ToLower(filename[:index])
+	}
+
+	return &vimDoc{
+		filename: filename,
+		title:    title,
+		desc:     desc,
+		cols:     cols,
+		tabs:     tabs,
+		tocPos:   -1}
 }
 
 func (v *vimDoc) pushl() {
@@ -98,11 +114,12 @@ func (v *vimDoc) buildTag(header []byte) []byte {
 
 func (v *vimDoc) writeStraddle(out *bytes.Buffer, left, right []byte, trim int) {
 	padding := v.cols - (len(left) + len(right)) + trim
+	if padding <= 0 {
+		padding = 1
+	}
 
 	out.Write(left)
-	if padding > 0 {
-		out.WriteString(strings.Repeat(" ", padding))
-	}
+	out.WriteString(strings.Repeat(" ", padding))
 	out.Write(right)
 	out.WriteString("\n")
 }
@@ -347,9 +364,13 @@ func (v *vimDoc) NormalText(out *bytes.Buffer, text []byte) {
 }
 
 // Header and footer
-func (*vimDoc) DocumentHeader(out *bytes.Buffer) {
-	// unimplemented
-	log.Println("DocumentHeader is a stub")
+func (v *vimDoc) DocumentHeader(out *bytes.Buffer) {
+	if len(v.desc) > 0 {
+		v.writeStraddle(out, []byte(v.filename), []byte(v.desc), 0)
+	} else {
+		out.WriteString(v.filename)
+		out.WriteString("\n")
+	}
 }
 
 func (v *vimDoc) DocumentFooter(out *bytes.Buffer) {
diff --git a/vimdown.go b/vimdown.go
index 1d793ed..9765da3 100644
--- a/vimdown.go
+++ b/vimdown.go
@@ -33,7 +33,7 @@ import (
 )
 
 func usage() {
-	fmt.Fprintf(os.Stderr, "Usage: %s [options] [input] [output]\n", os.Args[0])
+	fmt.Fprintf(os.Stderr, "Usage: %s [options] input output\n", os.Args[0])
 	fmt.Fprintf(os.Stderr, "http://foosoft.net/projects/vimdown/\n\n")
 	fmt.Fprintf(os.Stderr, "Parameters:\n")
 	flag.PrintDefaults()
@@ -42,44 +42,32 @@ func usage() {
 func main() {
 	cols := flag.Int("cols", 80, "number of columns to use for alignment and rules")
 	tabs := flag.Int("tabs", 4, "size of the tab to use specified as number of spaces")
+	desc := flag.String("desc", "", "short description of extension")
 	flag.Usage = usage
 	flag.Parse()
 
 	args := flag.Args()
-
-	var input []byte
-	var err error
-	switch len(args) {
-	case 0:
-		if input, err = ioutil.ReadAll(os.Stdin); err != nil {
-			log.Fatal("error reading from stdin")
-		}
-	case 1, 2:
-		if input, err = ioutil.ReadFile(args[0]); err != nil {
-			log.Fatalf("error reading from %s", args[0])
-		}
-	default:
+	if len(args) < 2 {
 		flag.Usage()
 		os.Exit(-1)
 	}
 
-	renderer := VimDocRenderer(*cols, *tabs)
+	input, err := ioutil.ReadFile(args[0])
+	if err != nil {
+		log.Fatalf("error: unable to read from file %s", args[0])
+	}
+
+	renderer := VimDocRenderer(args[1], *desc, *cols, *tabs)
 	extensions := blackfriday.EXTENSION_FENCED_CODE | blackfriday.EXTENSION_NO_INTRA_EMPHASIS | blackfriday.EXTENSION_SPACE_HEADERS
 	output := blackfriday.Markdown(input, renderer, extensions)
 
-	var file *os.File
-	switch len(args) {
-	case 2:
-		if file, err = os.Create(args[1]); err != nil {
-			fmt.Fprintf(os.Stderr, "Error creating %s: %v", args[1], err)
-			os.Exit(-1)
-		}
-		defer file.Close()
-	default:
-		file = os.Stdout
+	file, err := os.Create(args[1])
+	if err != nil {
+		log.Fatalf("error: unable to write to file %s", args[1])
 	}
+	defer file.Close()
 
-	if _, err = file.Write(output); err != nil {
-		log.Fatal("error writing output")
+	if _, err := file.Write(output); err != nil {
+		log.Fatal("error: unable to write output")
 	}
 }