Initial commit

app/method.go

@@ -0,0 +1,63 @@
+package app
+
+import (
+	"database/sql"
+	"io/ioutil"
+
+	"github.com/gin-gonic/gin"
+	"github.com/osuripple/api/common"
+)
+
+// Method wraps an API method to a HandlerFunc.
+func Method(f func(md common.MethodData) common.Response, db *sql.DB, privilegesNeeded ...int) gin.HandlerFunc {
+	return func(c *gin.Context) {
+		data, err := ioutil.ReadAll(c.Request.Body)
+		if err != nil {
+			c.Error(err)
+		}
+		c.Request.Body.Close()
+
+		token := ""
+		switch {
+		case c.Request.Header.Get("X-Ripple-Token") != "":
+			token = c.Request.Header.Get("X-Ripple-Token")
+		case c.Query("token") != "":
+			token = c.Query("token")
+		case c.Query("k") != "":
+			token = c.Query("k")
+		}
+
+		md := common.MethodData{
+			DB:          db,
+			RequestData: data,
+			C:           c,
+		}
+		if token != "" {
+			tokenReal, exists := GetTokenFull(token, db)
+			if exists {
+				md.User = tokenReal
+			}
+		}
+
+		missingPrivileges := 0
+		for _, privilege := range privilegesNeeded {
+			if int(md.User.Privileges)&privilege == 0 {
+				missingPrivileges |= privilege
+			}
+		}
+		if missingPrivileges != 0 {
+			c.IndentedJSON(401, common.Response{
+				Code:    401,
+				Message: "You don't have the privilege(s): " + common.Privileges(missingPrivileges).String() + ".",
+			})
+			return
+		}
+
+		resp := f(md)
+		if resp.Code == 0 {
+			c.IndentedJSON(500, resp)
+		} else {
+			c.IndentedJSON(200, resp)
+		}
+	}
+}