package main

import (
	"bufio"
	"fmt"
	"io/ioutil"
	"os"
	"path/filepath"
	"regexp"
	"strings"
)

// Reads all .ronn files & and converts them to string literals
// triggered by "go generate" comment
// Literals are inserted into a map using an init function, this means
// that there are no compilation errors if 'go generate' hasn't been run, just
// blank man files.
func main() {
	fmt.Fprintf(os.Stderr, "Converting man pages into code...\n")
	mandir := "../docs/man"
	fs, err := ioutil.ReadDir(mandir)
	if err != nil {
		fmt.Fprintf(os.Stderr, "Failed to open man dir: %v\n", err)
		os.Exit(2)
	}
	out, err := os.Create("../commands/mancontent_gen.go")
	if err != nil {
		fmt.Fprintf(os.Stderr, "Failed to create go file: %v\n", err)
		os.Exit(2)
	}
	out.WriteString("package commands\n\nfunc init() {\n")
	out.WriteString("// THIS FILE IS GENERATED, DO NOT EDIT\n")
	out.WriteString("// Use 'go generate ./commands' to update\n")
	fileregex := regexp.MustCompile(`git-lfs(?:-([A-Za-z\-]+))?.\d.ronn`)
	headerregex := regexp.MustCompile(`^###?\s+([A-Za-z0-9 ]+)`)
	// only pick up caps in links to avoid matching optional args
	linkregex := regexp.MustCompile(`\[([A-Z\- ]+)\]`)
	count := 0
	for _, f := range fs {
		if match := fileregex.FindStringSubmatch(f.Name()); match != nil {
			fmt.Fprintf(os.Stderr, "%v\n", f.Name())
			cmd := match[1]
			if len(cmd) == 0 {
				// This is git-lfs.1.ronn
				cmd = "git-lfs"
			}
			out.WriteString("ManPages[\"" + cmd + "\"] = `")
			contentf, err := os.Open(filepath.Join(mandir, f.Name()))
			if err != nil {
				fmt.Fprintf(os.Stderr, "Failed to open %v: %v\n", f.Name(), err)
				os.Exit(2)
			}
			// Process the ronn to make it nicer as help text
			scanner := bufio.NewScanner(contentf)
			firstHeaderDone := false
			skipNextLineIfBlank := false
			lastLineWasBullet := false
		scanloop:
			for scanner.Scan() {
				line := scanner.Text()
				trimmedline := strings.TrimSpace(line)
				if skipNextLineIfBlank && len(trimmedline) == 0 {
					skipNextLineIfBlank = false
					lastLineWasBullet = false
					continue
				}

				// Special case headers
				if hmatch := headerregex.FindStringSubmatch(line); hmatch != nil {
					header := strings.ToLower(hmatch[1])
					switch header {
					case "synopsis":
						// Ignore this, just go direct to command

					case "description":
						// Just skip the header & newline
						skipNextLineIfBlank = true
					case "options":
						out.WriteString("Options:" + "\n")
					case "see also":
						// don't include any content after this
						break scanloop
					default:
						out.WriteString(strings.ToUpper(header[:1]) + header[1:] + "\n")
						out.WriteString(strings.Repeat("-", len(header)) + "\n")
					}
					firstHeaderDone = true
					lastLineWasBullet = false
					continue
				} else if lmatches := linkregex.FindAllStringSubmatch(line, -1); lmatches != nil {
					for _, lmatch := range lmatches {
						linktext := strings.ToLower(lmatch[1])
						line = strings.Replace(line, lmatch[0], `"`+strings.ToUpper(linktext[:1])+linktext[1:]+`"`, 1)
					}
				}
				// Skip content until after first header
				if !firstHeaderDone {
					continue
				}
				// OK, content here

				// Remove backticks since it won't format & that's delimiting the string
				line = strings.Replace(line, "`", "", -1)
				// indent bullets
				if strings.HasPrefix(line, "*") {
					lastLineWasBullet = true
				} else if lastLineWasBullet && !strings.HasPrefix(line, " ") {
					// indent paragraphs under bullets if not already done
					line = "  " + line
				}

				out.WriteString(line + "\n")
			}
			out.WriteString("`\n")
			contentf.Close()
			count++
		}
	}
	out.WriteString("}\n")
	fmt.Fprintf(os.Stderr, "Successfully processed %d man pages.\n", count)

}