aboutsummaryrefslogtreecommitdiffstats
path: root/golang/cmd/fatcatd/serve.go
blob: dcee4f4fc02070fac41ef4c586b6d098017cc9ee (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
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)
	}

}