67 lines
1.4 KiB
Go
67 lines
1.4 KiB
Go
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 {
|
|
resp.Code = 500
|
|
}
|
|
if _, exists := c.GetQuery("pls200"); exists {
|
|
c.IndentedJSON(200, resp)
|
|
} else {
|
|
c.IndentedJSON(resp.Code, resp)
|
|
}
|
|
}
|
|
}
|