Merge pull request #12 from shiftky/add_exec_sub_cmd

Add 'exec' sub-command to tmsh-cli

cmd/tmsh/cmd/root.go

@@ -2,6 +2,7 @@ package cmd
 
 import (
 	"fmt"
+	"os"
 
 	"github.com/spf13/cobra"
 	"github.com/spf13/viper"
@@ -16,9 +17,31 @@ var RootCmd = &cobra.Command{
 
 var versionCmd = &cobra.Command{
 	Use:   "version",
-	Short: "Print the version number of tmsh command.",
+	Short: "Print the version number of tmsh-cli command.",
 	Run: func(cmd *cobra.Command, args []string) {
-		fmt.Println("tmsh v0.1.0")
+		fmt.Println("tmsh v0.2.0")
+	},
+}
+
+var execCmd = &cobra.Command{
+	Use:   "exec [tmsh command]",
+	Short: "Execute any command of TMSH",
+	Run: func(cmd *cobra.Command, args []string) {
+		if len(args) < 1 {
+			cmd.Usage()
+			os.Exit(2)
+		}
+
+		bigip := NewSession()
+		defer bigip.Close()
+
+		ret, err := bigip.ExecuteCommand(args[0])
+		if err != nil {
+			fmt.Fprintln(os.Stderr, err)
+			os.Exit(1)
+		}
+
+		fmt.Println(ret)
 	},
 }
 
@@ -38,4 +61,5 @@ func init() {
 	viper.BindPFlag("TMSH_PORT", RootCmdFlags.Lookup("port"))
 
 	RootCmd.AddCommand(versionCmd)
+	RootCmd.AddCommand(execCmd)
 }