[feat] Insertion of Gopher specific content and ommition of web-only content

main.go

@@ -10,15 +10,18 @@ import(
 )
 
 type State struct {
-	out io.Writer	
+	out io.Writer
 	maxLineLength int
+	insertionMap []string
+	insertionIndex int
 	isInCodeBlock bool
 	wasPreviousLink bool
+	isWebOnly bool
 }
 
 func main() {
 	if len(os.Args) < 3 {
-		fmt.Printf("Usage: %s input.gmi output_gophermap [max line length (default 69)]\n", os.Args[0])
+		fmt.Printf("Usage: %s input.gmi output_gophermap [max line length (default 70)] [preprocessor insertion map file]\n", os.Args[0])
 		return
 	}
 
@@ -52,7 +55,17 @@ func main() {
 		outputWriter = os.Stdout
 	}
 
-	s := State{outputWriter, 69, false, false}
+	preprocessorMap := []string{}
+	if len(os.Args) >= 5 {
+		data, err := os.ReadFile(os.Args[4])
+		if err != nil {
+			fmt.Fprintln(os.Stderr, err)
+			os.Exit(1)
+		}
+		preprocessorMap = strings.Split(string(data), "<!--break-->")
+	}
+
+	s := State{outputWriter, 70, preprocessorMap, 0, false, false, false}
 	if len(os.Args) >= 4 {
 		maxLineLength, err := strconv.Atoi(os.Args[3])
 		if err != nil {
@@ -72,6 +85,17 @@ var gopherTypeCharParams = GopherTypeCharParams{
 }
 
 func (s *State) handleElement(e p.Element) {
+	if strings.HasPrefix(e.Text(), "<!--web-only-->") {
+		s.isWebOnly = !s.isWebOnly
+		return
+	}
+	if s.isWebOnly { return }
+	if strings.HasPrefix(e.Text(), "<!--insert-gopher-->") {
+		s.out.Write([]byte(s.insertionMap[s.insertionIndex]))
+		s.insertionIndex += 1
+		return
+	}
+
 	switch (e.Type()) {
 		case p.ElementTypes.Text():
 			if e.Text() == "" { return }
@@ -110,5 +134,7 @@ func (s *State) handleElement(e p.Element) {
 			s.isInCodeBlock = !s.isInCodeBlock
 			s.wasPreviousLink = false
 	}
-	s.out.Write([]byte("\n"))
+	if e.Text() != "" {
+		s.out.Write([]byte("\n"))
+	}
 }