Initial commit

register.go

@@ -0,0 +1,105 @@
+package mgoapi
+
+import (
+	"git.basehosts.de/gopackage/mgocrud"
+	"github.com/gin-gonic/gin"
+	mgo "gopkg.in/mgo.v2"
+)
+
+// API is wrapper for one RouterGroup and mgo DB
+type API struct {
+	DBSession            *mgo.Session
+	DBName               string
+	routerGroup          *gin.RouterGroup
+	jwtSecret            []byte
+	authenticationHeader string
+}
+
+// Context is gin.Context with extras
+type Context struct {
+	*gin.Context
+	API *API
+	DB  *mgo.Database
+}
+
+// New returns new instance of the API
+func New(session *mgo.Session, dbname string, routerGroup *gin.RouterGroup) *API {
+	return &API{
+		DBSession:   session,
+		DBName:      dbname,
+		routerGroup: routerGroup,
+	}
+}
+
+var modelRegistry = make([]mgocrud.ModelInterface, 0, 0)
+
+// RegisterModel setups gin routes for model GET, POST, PUT and DELETE
+func (api *API) RegisterModel(m mgocrud.ModelInterface) {
+
+	modelRegistry = append(modelRegistry, m)
+
+	session := api.DBSession.Copy()
+	defer session.Close()
+	db := session.DB(api.DBName)
+
+	err := mgocrud.EnsureIndex(db, m)
+	if err != nil {
+		panic(err)
+	}
+
+	cName := mgocrud.GetCollectionName(m)
+	colGroup := api.routerGroup.Group("/" + cName)
+
+	if i, ok := m.(interface {
+		Middleware(*Context)
+	}); ok {
+		colGroup.Use(func(c *gin.Context) {
+			i.Middleware(&Context{
+				Context: c,
+				API:     api,
+				DB:      db,
+			})
+		})
+	}
+
+	colGroup.GET("/:id", api.collectionGetOneHandler(m))
+	colGroup.GET("", api.collectionGetHandler(m))
+	colGroup.POST("", api.collectionPostHandler(m))
+	colGroup.PUT("/:id", api.collectionPutHandler(m))
+	colGroup.DELETE("/:id", api.collectionDeleteHandler(m))
+
+}
+
+// AddMetaRoute adds a route for retrieving meta information for all models
+// call after all registerModel calls
+func (api *API) AddMetaRoute(route string) {
+	api.routerGroup.GET(route, func(c *gin.Context) {
+		modelMeta := make(map[string]interface{})
+
+		for _, m := range modelRegistry {
+			modelMeta[mgocrud.GetCollectionName(m)] = getModelMeta(m)
+		}
+
+		c.JSON(200, map[string]interface{}{
+			"models": modelMeta,
+		})
+	})
+
+	for _, m := range modelRegistry {
+		api.routerGroup.GET("/_meta/"+mgocrud.GetCollectionName(m), api.collectionGetMetaHandler(m))
+	}
+
+}
+
+// LoginConfig configures login method
+type LoginConfig struct {
+	JWTSecret            []byte
+	AuthenticationHeader string
+}
+
+// AddLoginRoute adds a route for login
+func (api *API) AddLoginRoute(route string, m LoginModel, c LoginConfig) {
+	api.jwtSecret = c.JWTSecret
+	api.authenticationHeader = c.AuthenticationHeader
+	api.routerGroup.POST(route, api.loginPostHandler(m))
+}