before publish and after publish commands

hugo.go

@@ -101,7 +101,23 @@ func (h Hugo) ServeHTTP(w http.ResponseWriter, r *http.Request) (int, error) {
 			}
 
 			if r.Header.Get("Regenerate") == "true" {
+				if err = h.Config.BeforePublish(
+					r,
+					&h.FileManager.Configs[0],
+					h.FileManager.Configs[0].User,
+				); err != nil {
+					return http.StatusInternalServerError, err
+				}
+
 				RunHugo(h.Config, false)
+
+				if err = h.Config.AfterPublish(
+					r,
+					&h.FileManager.Configs[0],
+					h.FileManager.Configs[0].User,
+				); err != nil {
+					return http.StatusInternalServerError, err
+				}
 			}
 
 			if r.Header.Get("Schedule") != "" {

setup.go

@@ -69,6 +69,8 @@ type Config struct {
 	Args          []string // Hugo arguments
 	BaseURL       string   // BaseURL of admin interface
 	WebDavURL     string
+	BeforePublish config.CommandFunc
+	AfterPublish  config.CommandFunc
 }
 
 // Parse parses the configuration set by the user so it can be
@@ -127,6 +129,14 @@ func parse(c *caddy.Controller, root string) (*Config, *filemanager.FileManager,
 				}
 
 				cfg.Args = append(cfg.Args, "--"+values[0]+"="+value)
+			case "before_publish":
+				if cfg.BeforePublish, err = config.CommandRunner(c); err != nil {
+					return cfg, &filemanager.FileManager{}, err
+				}
+			case "after_publish":
+				if cfg.AfterPublish, err = config.CommandRunner(c); err != nil {
+					return cfg, &filemanager.FileManager{}, err
+				}
 			default:
 				line := "\n\t" + c.Val()