From 9e59f510a138f8b3e17ef470ed7925511060d3e7 Mon Sep 17 00:00:00 2001 From: Gregory Ballantine Date: Sat, 14 Jul 2018 16:03:54 -0400 Subject: [PATCH] Added some global flags to specify LDAP host and port to connect to --- cmd/root.go | 13 +++++++++++++ cmd/search.go | 2 +- 2 files changed, 14 insertions(+), 1 deletion(-) diff --git a/cmd/root.go b/cmd/root.go index 0048359..1476f32 100644 --- a/cmd/root.go +++ b/cmd/root.go @@ -7,6 +7,18 @@ import ( "github.com/spf13/cobra" ) +var ( + flagHost string + flagPort int +) + +func init() { + // define global CLI flags + rootCmd.PersistentFlags().StringVarP(&flagHost, "host", "H", "ldap.example.com", "LDAP host to perform operations on") + rootCmd.PersistentFlags().IntVarP(&flagPort, "port", "p", 389, "TCP port that the LDAP host is listening on") +} + +// define root command var rootCmd = &cobra.Command{ Use: "muldap", Short: "muldap is Metaunix.net's LDAP management tool", @@ -16,6 +28,7 @@ var rootCmd = &cobra.Command{ }, } +// start CLI app func Execute() { if err := rootCmd.Execute(); err != nil { fmt.Println(err) diff --git a/cmd/search.go b/cmd/search.go index 3e350b4..5e594bd 100644 --- a/cmd/search.go +++ b/cmd/search.go @@ -31,7 +31,7 @@ var searchCmd = &cobra.Command{ Long: `Perform an LDAP search operation on an LDAP directory`, Run: func(cmd *cobra.Command, args []string) { // create new LDAP connection - l, err := ldap.Dial("tcp", fmt.Sprintf("%s:%d", "ldap.example.com", 389)) + l, err := ldap.Dial("tcp", fmt.Sprintf("%s:%d", flagHost, flagPort)) if err != nil { log.Fatal(err) }