refactor commands (cobra)

1 files changed, 110 insertions, 0 deletions

diff --git a/golang/cmd/fatcatd/serve.go b/golang/cmd/fatcatd/serve.go
new file mode 100644
index 00000000..dcee4f4f
--- /dev/null
+++ b/golang/cmd/fatcatd/serve.go
@@ -0,0 +1,110 @@
+
+package main
+
+import (
+	//"os"
+	"net/http"
+
+    "github.com/spf13/viper"
+    log "github.com/sirupsen/logrus"
+	loads "github.com/go-openapi/loads"
+	//flags "github.com/jessevdk/go-flags"
+    //"github.com/getsentry/raven-go"
+    "github.com/carbocation/interpose"
+    "github.com/carbocation/interpose/adaptors"
+    "github.com/meatballhat/negroni-logrus"
+    "github.com/go-pg/pg"
+    "github.com/spf13/cobra"
+
+	"git.archive.org/bnewbold/fatcat/golang/api/handlers"
+	"git.archive.org/bnewbold/fatcat/golang/gen/restapi"
+	"git.archive.org/bnewbold/fatcat/golang/gen/restapi/operations"
+)
+
+var serveCmd = &cobra.Command{
+    Use:    "serve [options]",
+    Short:  "Run fatcat REST API server",
+    Run:    func(cmd *cobra.Command, args[] string) {
+        main_serve()
+    },
+}
+
+func main_serve() {
+
+    log.Warn("Starting up...");
+
+    // load embedded swagger file
+    swaggerSpec, err := loads.Analyzed(restapi.SwaggerJSON, "")
+    if err != nil {
+        log.Fatalln(err)
+    }
+
+    // create new service API
+	api := operations.NewFatcatAPI(swaggerSpec)
+
+	// Set your custom logger if needed. Default one is log.Printf
+	// Expected interface func(string, ...interface{})
+    api.Logger = log.Printf
+
+	server := restapi.NewServer(api)
+	defer server.Shutdown()
+
+/*
+	parser := flags.NewParser(server, flags.Default)
+	parser.ShortDescription = "fatcat"
+	parser.LongDescription = "A scalable, versioned, API-oriented catalog of bibliographic entities and file metadata"
+
+	server.ConfigureFlags()
+	for _, optsGroup := range api.CommandLineOptionsGroups {
+		_, err := parser.AddGroup(optsGroup.ShortDescription, optsGroup.LongDescription, optsGroup.Options)
+		if err != nil {
+			log.Fatalln(err)
+		}
+	}
+
+	if _, err := parser.Parse(); err != nil {
+		code := 1
+		if fe, ok := err.(*flags.Error); ok {
+			if fe.Type == flags.ErrHelp {
+				code = 0
+			}
+		}
+		os.Exit(code)
+	}
+*/
+
+    server.Port = viper.GetInt("port")
+
+    db := pg.Connect(&pg.Options{
+        User: "bnewbold", // XXX:
+        Database: "fatcat",
+    })
+    defer db.Close()
+
+    // register all the many handlers here
+	api.GetCreatorIDHandler = handlers.NewGetCreatorIDHandler(db);
+    api.PostCreatorHandler = handlers.NewPostCreatorHandler(db);
+
+    middle := interpose.New()
+
+    // sentry and upstream
+    // XXX: middle.UseHandler(sentry.Recovery(raven.DefaultClient, false))
+
+    // logging
+    negroniMiddleware := negronilogrus.NewMiddleware()
+    middle.Use(adaptors.FromNegroni(negroniMiddleware))
+
+    // add clacks (TODO: only production)
+    middle.UseHandler(http.HandlerFunc(func(rw http.ResponseWriter, req *http.Request) {
+        rw.Header().Set("X-Clacks-Overhead:", "GNU Aaron Swartz, John Perry Barlow")
+    }))
+
+    middle.UseHandler(api.Serve(nil))
+
+    server.SetHandler(middle)
+
+	if err := server.Serve(); err != nil {
+		log.Fatalln(err)
+	}
+
+}