Fixed bug in os.Args iteration.

README.md

@@ -0,0 +1,31 @@
+### The "UI" package
+
+This package provides --verbose and --quiet flag features to your cli.
+It recognize both short and long flags. The package contains three functions:
+- **PrintOut** to print regular messages unless --quiet flag is set.
+- **PrintVerbose** to print messages only if --verbose flag is set, but --quiet
+flag is not.
+- **SetupOutputMode** to reconfigure output mode in the runtime.
+
+-------------
+
+### Usage
+1. Download the package by **go get** command:
+```shell
+go get code.foilhatguy.casa/pub/ui
+```
+2. Import package to your golang code:
+```go
+import (
+	"code.foilhatguy.casa/pub/ui"
+)
+```
+There is no need for additional configuration. The package have its own **init** 
+function to detect --verbose and --quiet flags.
+
+3. Use **PrintOut** and **PrintVerbose** functions to print
+messages to the standard output:
+```go
+    ui.PrintOut("Some regular message.")
+    ui.PrintVerbose("Some message printed only if --verbose flag is set.")
+```

ui.go

@@ -1,33 +1,38 @@
-// This package provides --quiet and --verbose flags for cli and corresponding output print functions PrintOut and PrintVerbose.
-// Another line
 package ui
 
 import (
 	"fmt"
 	"io"
 	"os"
-
+    "strings"
-	"github.com/spf13/pflag"
 )
 
 var stdOutput io.Writer
 var verboseOutput io.Writer
 
 func init() {
-	outputFlagSet := pflag.NewFlagSet("output", pflag.ContinueOnError)
+    isQuiet := detectFlag("q", "--quiet")
-	outputFlagSet.ParseErrorsWhitelist.UnknownFlags = true
+    isVerbose := detectFlag("V", "--verbose")
-	outputFlagSet.SetOutput(nil)
 
-	isVerbose := outputFlagSet.BoolP("verbose", "V", false, "If set adds additional details to output.")
+	SetupOutputMode(isQuiet, isVerbose)
-	isQuiet := outputFlagSet.BoolP("quiet", "q", false, "If set makes no output at all.")
-
-	err := outputFlagSet.Parse(os.Args)
-	if err != nil {
-		fmt.Println(err)
-		os.Exit(1)
 }
 
-	SetupOutputMode(*isQuiet, *isVerbose)
+func detectFlag(shortFlag string, fullFlag string) bool {
+    for _, arg := range os.Args {
+        if containFlag(arg, shortFlag) || arg == fullFlag {
+            return true
+        } 
+    }
+    return false
+}
+
+func containFlag(arg string, shortFlag string) bool {
+    if len(arg)>1 {
+        if arg[0] == "-"[0] && arg[1] != "-"[0] {
+            return strings.Contains(arg, shortFlag)
+        }
+    }
+    return false
 }
 
 func SetupOutputMode(isQuiet bool, isVerbose bool) {