forked from Silverfish/proton-bridge
feat(GODT-3121): KB suggestion test tool now support multi-line input.
This commit is contained in:
@ -18,8 +18,8 @@
|
|||||||
package main
|
package main
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"bufio"
|
|
||||||
"fmt"
|
"fmt"
|
||||||
|
"io"
|
||||||
"os"
|
"os"
|
||||||
|
|
||||||
"github.com/ProtonMail/proton-bridge/v3/internal/kb"
|
"github.com/ProtonMail/proton-bridge/v3/internal/kb"
|
||||||
@ -33,14 +33,17 @@ func checkErrors(err error) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func main() {
|
func main() {
|
||||||
var input string
|
fi, err := os.Stdin.Stat()
|
||||||
fmt.Print("Type your input: ")
|
|
||||||
|
|
||||||
buffer := bufio.NewReader(os.Stdin)
|
|
||||||
input, err := buffer.ReadString('\n')
|
|
||||||
checkErrors(err)
|
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)
|
checkErrors(err)
|
||||||
|
|
||||||
if len(suggestions) == 0 {
|
if len(suggestions) == 0 {
|
||||||
|
|||||||
Reference in New Issue
Block a user