From 2df64026ba93b3971ed7bf557d37b85b7c339480 Mon Sep 17 00:00:00 2001
From: Alexander Thaller <alexander.thaller@atraveo.de>
Date: Mon, 9 Feb 2015 23:44:14 +0100
Subject: [PATCH] added pre and post run hooks.

---
 command.go | 18 +++++++++++++++++-
 1 file changed, 17 insertions(+), 1 deletion(-)

diff --git a/command.go b/command.go
index 2b85d43..ede3243 100644
--- a/command.go
+++ b/command.go
@@ -59,6 +59,12 @@ type Command struct {
 	// Run runs the command.
 	// The args are the arguments after the command name.
 	Run func(cmd *Command, args []string)
+	// PreRun runs the command after the flags are parsed and before run.
+	// The args are the arguments after the command name.
+	PreRun func(cmd *Command, args []string)
+	// PostRun runs the command after run.
+	// The args are the arguments after the command name.
+	PostRun func(cmd *Command, args []string)
 	// Commands is the list of commands supported by this program.
 	commands []*Command
 	// Parent Command for this command
@@ -448,7 +454,17 @@ func (c *Command) execute(a []string) (err error) {
 
 	c.preRun()
 	argWoFlags := c.Flags().Args()
+
+	if c.PreRun != nil {
+		c.PreRun(c, argWoFlags)
+	}
+
 	c.Run(c, argWoFlags)
+
+	if c.PostRun != nil {
+		c.PostRun(c, argWoFlags)
+	}
+
 	return nil
 }
 
@@ -942,7 +958,7 @@ func (c *Command) mergePersistentFlags() {
 		c.PersistentFlags().VisitAll(addtolocal)
 	}
 	rmerge = func(x *Command) {
-		if ! x.HasParent() {
+		if !x.HasParent() {
 			flag.CommandLine.VisitAll(func(f *flag.Flag) {
 				if x.PersistentFlags().Lookup(f.Name) == nil {
 					x.PersistentFlags().AddFlag(f)