Add quiet output

msharris · Mar 15, 2024 · 3d3df71 · 3d3df71
1 parent ecfbb27
commit 3d3df71
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 8 deletions.
diff --git a/app/app.go b/app/app.go
@@ -14,6 +14,7 @@ type Options struct {
 	Locations []string
 	Sort      Field
 	Reverse   bool
+	Quiet     bool
 }
 
 type Field string
@@ -89,7 +90,7 @@ func Run(options Options) error {
 		slices.Reverse(stations)
 	}
 
-	show(stations)
+	show(stations, options.Quiet)
 
 	return nil
 }
@@ -146,15 +147,21 @@ func sort(stations []Station, f Field) {
 	}
 }
 
-func show(stations []Station) {
+func show(stations []Station, quiet bool) {
 	w := tabwriter.NewWriter(os.Stdout, 0, 0, 2, ' ', 0)
-	fmt.Fprintln(w, fmt.Sprintf("%v\t%v\t%v\t%v\t%v\t", "Id", "Location", "Index", "Time", "Status"))
-	for _, s := range stations {
-		status := "-"
-		if !s.Available {
-			status = "n/a"
+	if quiet {
+		for _, s := range stations {
+			fmt.Fprintln(w, fmt.Sprintf("%v\t%.1f\t", s.Name, s.UVIndex))
+		}
+	} else {
+		fmt.Fprintln(w, fmt.Sprintf("%v\t%v\t%v\t%v\t%v\t", "Id", "Location", "Index", "Time", "Status"))
+		for _, s := range stations {
+			status := "-"
+			if !s.Available {
+				status = "n/a"
+			}
+			fmt.Fprintln(w, fmt.Sprintf("%v\t%v\t%.1f\t%v\t%v\t", s.Id, s.Name, s.UVIndex, s.Time.Format("15:04"), status))
 		}
-		fmt.Fprintln(w, fmt.Sprintf("%v\t%v\t%.1f\t%v\t%v\t", s.Id, s.Name, s.UVIndex, s.Time.Format("15:04"), status))
 	}
 	w.Flush()
 }
diff --git a/cmd/root.go b/cmd/root.go
@@ -32,4 +32,5 @@ func init() {
 	rootCmd.Flags().StringSliceVarP(&flags.Locations, "locations", "l", nil, "comma-separated list of locations to display (accepts id and name)")
 	rootCmd.Flags().VarP(&flags.Sort, "sort", "s", "field to sort observations by")
 	rootCmd.Flags().BoolVarP(&flags.Reverse, "reverse", "r", false, "print observations in reverse order")
+	rootCmd.Flags().BoolVarP(&flags.Quiet, "quiet", "q", false, "reduce the output")
 }