add command line flags

Author: muhammadn

cmd/flags.go

@@ -0,0 +1,29 @@
+package cmd
+
+import (
+        "github.com/spf13/cobra"
+	"muhammadn/avahi-proxy/pkg/proxy"
+)
+
+var runCmd = &cobra.Command{
+    Use:   "run",
+    Aliases: []string{"r"},
+    Short:  "Start running the proxy",
+    RunE: func(cmd *cobra.Command, args []string) error {
+        baseDomain, _               := cmd.Flags().GetString("basedomain")
+	port, _                     := cmd.Flags().GetString("port")
+
+	baseDomain = baseDomain + "."
+
+	proxy.RunProxy(baseDomain, port)
+        return nil
+    },
+}
+
+func init() {
+        rootCmd.AddCommand(runCmd)
+	runCmd.Flags().StringP("baseDomain", "", "", "Your local LAN domain name, example: home.lan")
+	runCmd.Flags().StringP("port", "p", "5354", "Port number to run avahi-proxy (default: 5354)")
+
+        runCmd.MarkFlagRequired("baseDomain")
+}

cmd/root.go

@@ -0,0 +1,19 @@
+package cmd
+
+import (
+    "github.com/spf13/cobra"
+)
+
+var (
+	rootCmd = &cobra.Command{
+		Use:   "avahi-proxy",
+		Short: "Map Multicast DNS .local hostnames to your LAN domain name",
+		Long: `Automatically map Multicast DNS .local hostnames to your LAN domain name (example, ".home.lan" or ".internal" and use your favourite DNS server to query avahi-proxy to get machine hostnames on .local multicast network. 
+
+		Let your Windows and Android phones resolve your Multicast devices like iPhone/Mac and iPad, or maybe you want to expose .local hostnames over IPSec with your LAN domain name`,
+	}
+)
+
+func Execute() error {
+	return rootCmd.Execute()
+}

main.go

@@ -0,0 +1,9 @@
+package main
+
+import (
+        "muhammadn/avahi-proxy/cmd"
+)
+
+func main() {
+        cmd.Execute()
+}

cmd/avahi-proxy/main.go renamed to pkg/proxy/main.go

@@ -1,10 +1,10 @@
-package main
+package proxy
 
 import (
 	"fmt"
 	"log"
-	"strconv"
 	"strings"
+	"strconv"
 
 	"github.com/godbus/dbus/v5"
 	"github.com/holoplot/go-avahi"
@@ -73,14 +73,14 @@ func parseQuery(m *dns.Msg) {
 	}
 }
 
-func main() {
+func RunProxy(baseDomain string, port string) {
 	// attach request handler func
-	dns.HandleFunc("home.lan.", handleDnsRequest)
+	dns.HandleFunc(baseDomain, handleDnsRequest)
 
 	// start server
-	port := 5354
-	server := &dns.Server{Addr: ":" + strconv.Itoa(port), Net: "udp"}
-	log.Printf("Starting at %d\n", port)
+	server := &dns.Server{Addr: ":" + port, Net: "udp"}
+	nport, _ := strconv.Atoi(port)
+	log.Printf("Starting at %d\n", nport)
 	err := server.ListenAndServe()
 	defer server.Shutdown()
 	if err != nil {