From 7e3518c314463906df85e760045eedfe7b42fc34 Mon Sep 17 00:00:00 2001
From: Luke Emmet <luke.emmet@gmail.com>
Date: Thu, 13 Aug 2020 12:38:23 +0100
Subject: [PATCH] working version of sequential numbering in batches added
 version to build

---
 html2gmi.go | 150 ++++++++++++++++++++++++++++------------------------
 1 file changed, 82 insertions(+), 68 deletions(-)

diff --git a/html2gmi.go b/html2gmi.go
index fc0bbee..be0cdf4 100644
--- a/html2gmi.go
+++ b/html2gmi.go
@@ -1,93 +1,107 @@
 package main
 
 import (
-    flag "github.com/spf13/pflag"
-    "fmt"
-    "io/ioutil"
-    "os"
-    "github.com/LukeEmmet/html2gemini"
-    "bufio"
-    "io"
-    "errors"
+	"bufio"
+	"errors"
+	"fmt"
+	"github.com/LukeEmmet/html2gemini"
+	flag "github.com/spf13/pflag"
+	"io"
+	"io/ioutil"
+	"os"
 )
 
+var version = "0.2.1"
+
 var (
-    output = flag.StringP("output", "o", "", "Output path. Otherwise uses stdout\n")
-    input = flag.StringP("input", "i", "", "Input path. Otherwise uses stdin\n")
+	output            = flag.StringP("output", "o", "", "Output path. Otherwise uses stdout\n")
+	input             = flag.StringP("input", "i", "", "Input path. Otherwise uses stdin\n")
+	citationStart     = flag.IntP("citationStart", "c", 1, "Start citations from this index")
+	linkEmitFrequency = flag.IntP("linkEmitFrequency", "l", 2, "Emit gathered links through the document after this number of paragraphs")
+	verFlag           = flag.BoolP("version", "v", false, "Find out what version of html2gmi you're running")
 )
 
 func check(e error) {
-    if e != nil {
-        panic(e)
-        os.Exit(1)
-    }
+	if e != nil {
+		panic(e)
+	}
 }
 
 func saveFile(contents []byte, path string) {
-    d1 := []byte(contents)
-    err := ioutil.WriteFile(path, d1, 0644)
-    check(err)
+	d1 := contents
+	err := ioutil.WriteFile(path, d1, 0644)
+	check(err)
 }
 
-func readStdin () (string) {
-    // based on https://flaviocopes.com/go-shell-pipes/
-    reader := bufio.NewReader(os.Stdin)     //default size is 4096 apparently
-    var output []rune
+func readStdin() string {
+	// based on https://flaviocopes.com/go-shell-pipes/
+	reader := bufio.NewReader(os.Stdin) //default size is 4096 apparently
+	var output []rune
 
-    for {
-        input, _, err := reader.ReadRune()
-        if err != nil && err == io.EOF {
-            break
-        }
-        output = append(output, input)
-    }   
+	for {
+		input, _, err := reader.ReadRune()
+		if err != nil && err == io.EOF {
+			break
+		}
+		output = append(output, input)
+	}
 
-    return string(output)
+	return string(output)
 }
 
 func getInput() (string, error) {
-        var inputHtml string
-    
-        info, err := os.Stdin.Stat()
-        check(err)
-    
-        if info.Mode()&os.ModeNamedPipe != 0 {
-            // we have a pipe input
-            inputHtml = readStdin()
-            
-        } else if (*input != "") {
-              //get the input file from the command line
-              dat, err := ioutil.ReadFile(*input)
-              check(err)
-              inputHtml = string(dat)
+	var inputHtml string
 
-        }  else  {
-            //we shouldnt get here 
-            return "", errors.New("Invalid option for input - use -i <path> or pipe to stdin")
-        } 
-        
-        return inputHtml, nil
+	info, err := os.Stdin.Stat()
+	check(err)
+
+	if info.Mode()&os.ModeNamedPipe != 0 {
+		// we have a pipe input
+		inputHtml = readStdin()
+
+	} else if *input != "" {
+		//get the input file from the command line
+		dat, err := ioutil.ReadFile(*input)
+		check(err)
+		inputHtml = string(dat)
+
+	} else {
+		//we shouldn't get here
+		return "", errors.New("invalid option for input - use -i <path> or pipe to stdin")
+	}
+
+	return inputHtml, nil
 }
 
 func main() {
-        var inputHtml string
-    
-        flag.Parse()
+	var inputHtml string
 
-        //get the input from commandline or stdin
-        inputHtml, err:= getInput()
-        check(err)
-        
-        //convert html to gmi
-        text, err := html2text.FromString(inputHtml, html2text.Options{PrettyTables: true, GeminiCitationStyleLinks: true})
-        check(err)
-        
-        //process the output
-        if (*output == "") {
-            fmt.Print(text)
-        } else {
-            //save to the specified output
-            gmiBytes := []byte(text)            
-            saveFile(gmiBytes, *output)
-        }
+	flag.Parse()
+
+	if *verFlag {
+		fmt.Println("html2gmi " + version)
+		return
+	}
+
+	//get the input from commandline or stdin
+	inputHtml, err := getInput()
+	check(err)
+
+	//convert html to gmi
+	options := html2gemini.NewOptions()
+	options.PrettyTables = true
+	options.CitationStart = *citationStart
+	options.LinkEmitFrequency = *linkEmitFrequency
+
+	text, err := html2gemini.FromString(inputHtml, *options)
+	check(err)
+
+	//process the output
+	if *output == "" {
+		fmt.Print(text)
+	} else {
+		//save to the specified output
+		gmiBytes := []byte(text)
+		saveFile(gmiBytes, *output)
+	}
 }