match other executables

tvanriper · Mar 24, 2022 · 90ee94c · 90ee94c
1 parent dc75e9a
commit 90ee94c
Showing 1 changed file with 11 additions and 3 deletions.
diff --git a/cmd/apscan/main.go b/cmd/apscan/main.go
@@ -1,15 +1,23 @@
 package main
 
 import (
+	"flag"
 	"fmt"
 
 	"github.com/tvanriper/go-wireless"
 )
 
 func main() {
-	iface, ok := wireless.DefaultInterface()
-	if !ok {
-		panic("no wifi cards on the system")
+	var iface string
+	flag.StringVar(&iface, "i", "", "interface to use")
+	flag.Parse()
+
+	if iface == "" {
+		var ok bool
+		iface, ok = wireless.DefaultInterface()
+		if !ok {
+			panic("no wifi cards on the system")
+		}
 	}
 	fmt.Printf("Using interface: %s\n", iface)