From a057138880bb016b3dac971b6381d75f29870181 Mon Sep 17 00:00:00 2001
From: Xavier Michelon <xavier.michelon@proton.ch>
Date: Thu, 7 Dec 2023 10:48:33 +0100
Subject: [PATCH] feat(GODT-3121): KB suggestion test tool now support
 multi-line input.

---
 utils/kb-suggester/kb-suggester.go | 17 ++++++++++-------
 1 file changed, 10 insertions(+), 7 deletions(-)

diff --git a/utils/kb-suggester/kb-suggester.go b/utils/kb-suggester/kb-suggester.go
index eb37ad31..9ae878ac 100644
--- a/utils/kb-suggester/kb-suggester.go
+++ b/utils/kb-suggester/kb-suggester.go
@@ -18,8 +18,8 @@
 package main
 
 import (
-	"bufio"
 	"fmt"
+	"io"
 	"os"
 
 	"github.com/ProtonMail/proton-bridge/v3/internal/kb"
@@ -33,14 +33,17 @@ func checkErrors(err error) {
 }
 
 func main() {
-	var input string
-	fmt.Print("Type your input: ")
-
-	buffer := bufio.NewReader(os.Stdin)
-	input, err := buffer.ReadString('\n')
+	fi, err := os.Stdin.Stat()
 	checkErrors(err)
 
-	suggestions, err := kb.GetSuggestions(input)
+	if (fi.Mode() & os.ModeNamedPipe) == 0 {
+		fmt.Println("Type your input, Ctrl+D to finish: ")
+	}
+
+	bytes, err := io.ReadAll(os.Stdin)
+	checkErrors(err)
+
+	suggestions, err := kb.GetSuggestions(string(bytes))
 	checkErrors(err)
 
 	if len(suggestions) == 0 {