From e19752ac34cfa42eb813cbe4eee20f2adf1e0fc0 Mon Sep 17 00:00:00 2001 From: Miclle Zheng Date: Sun, 30 Jun 2024 22:48:27 +0800 Subject: [PATCH] Only one of 'file path' and 'unix character device' parameters can be selected. --- main.go | 17 ++++++++--------- 1 file changed, 8 insertions(+), 9 deletions(-) diff --git a/main.go b/main.go index dd71c33..3fc1cd5 100644 --- a/main.go +++ b/main.go @@ -4,7 +4,6 @@ import ( "bufio" "flag" "fmt" - "io/ioutil" "log" "os" @@ -34,13 +33,13 @@ func main() { if len(file) > 0 { align.Init(file) - } - - stat, _ := os.Stdin.Stat() - if (stat.Mode() & os.ModeCharDevice) == 0 { - reader := bufio.NewReader(os.Stdin) - align.Init(reader) - writeToConsole = true + } else { + stat, _ := os.Stdin.Stat() + if (stat.Mode() & os.ModeCharDevice) == 0 { + reader := bufio.NewReader(os.Stdin) + align.Init(reader) + writeToConsole = true + } } b, err := align.Do() @@ -53,5 +52,5 @@ func main() { return } - ioutil.WriteFile(file, b, 0) + os.WriteFile(file, b, 0) }