From 4842b4ad52166d39ae2595e14e2941c716d5f467 Mon Sep 17 00:00:00 2001
From: b1ek <me@blek.codes>
Date: Mon, 3 Jun 2024 09:04:47 +1000
Subject: [PATCH] use a different stdin read method

---
 main.go | 15 +++++++--------
 1 file changed, 7 insertions(+), 8 deletions(-)

diff --git a/main.go b/main.go
index 7962ed1..01584eb 100644
--- a/main.go
+++ b/main.go
@@ -1,8 +1,8 @@
 package main
 
 import (
-	"bufio"
 	"fmt"
+	"io"
 	"os"
 	"strings"
 
@@ -40,14 +40,13 @@ func main() {
 		}
 		file = f
 	} else {
-		scanner := bufio.NewScanner(os.Stdin)
-		for scanner.Scan() {
-			file = scanner.Bytes()
-		}
+		data, err := io.ReadAll(os.Stdin)
 
-		if scanner.Err() != nil {
-			panic(scanner.Err())
+		if err != nil {
+			fmt.Printf("Coudln't read from stdin: %s", err.Error())
+			os.Exit(1)
 		}
+		file = data
 	}
 
 	code := string(file)
@@ -70,7 +69,7 @@ func main() {
 		os.Exit(2)
 	}
 
-	gen := shebang + "\n\n" + gencode(found) + "\n" + strings.Join(codelines[1:], "\n")
+	gen := shebang + "\n\n" + gencode(found) + "\n\n" + strings.Join(codelines[1:], "\n")
 
 	if args.Outfile == "" {
 		fmt.Printf("%s", gen)