add recovery handler generator.

Author: cheikhshift

cmd/samb-medic/main.go

@@ -0,0 +1,33 @@
+package main
+
+import (
+	"flag"
+	"fmt"
+	"io/ioutil"
+	"os"
+)
+
+func main() {
+
+	name := flag.String("name", "FooBar", "Name of recovery function to add.")
+	project := flag.String("project", "./", "Path to project")
+
+	flag.Parse()
+
+	os.Chdir(*project)
+
+	// set path to where the handler source will be saved
+	savePath := "./pkg/api/" + *name + "_recover.go"
+
+	recoverySource := fmt.Sprintf(recoveryHandlerTemplate, *name, *name)
+
+	fmt.Println("Saving file to : ", savePath)
+	err := ioutil.WriteFile(savePath, []byte(recoverySource), 0700)
+
+	if err != nil {
+		panic(err)
+	}
+
+	fmt.Println("Add this to your server's recover directive : do ", *name + ";" )
+
+}

cmd/samb-medic/template.go

@@ -0,0 +1,16 @@
+package main
+
+var recoveryHandlerTemplate = `package api
+
+import (
+	"net/http"
+)
+
+// %s handles recovery of requests.
+func %s(w http.ResponseWriter, r *http.Request, m string) {
+
+
+	w.Write([]byte("Please implement me."))
+
+}
+`