mmktomato / go-nascii Goto Github PK

--- main.orig.go	2019-02-18 07:28:38 +0800
+++ main.go	2020-08-26 15:36:54 +0800
@@ -3,6 +3,7 @@ package main
 import (
 	"fmt"
 	"os"
+	"bufio"
 	"regexp"
 	"strconv"
 	"strings"
@@ -22,16 +23,24 @@ func main() {
 	var opt option
 	args, err := flags.ParseArgs(&opt, os.Args)
 	if err != nil {
-		panic(err)
+        os.Exit(0);
+		//panic(err)
 	}
-
+    var instr string
+    for _, arg := range args[1:] {
+        instr = arg
+    }
+
+    if len(instr) == 0{
+        input := bufio.NewScanner(os.Stdin)
+        input.Scan()
+        instr = input.Text()
+    }
 	isReverse := 0 < len(opt.IsReverse)
-	for _, arg := range args[1:] {
-		if isReverse {
-			asciiToUtf8(arg)
-		} else {
-			utf8ToAscii(arg)
-		}
+	if isReverse {
+		asciiToUtf8(instr)
+	} else {
+		utf8ToAscii(instr)
 	}
 }