/
permissions.go
executable file
·93 lines (79 loc) · 2.44 KB
/
permissions.go
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
package model
import (
"reflect"
"github.com/Datosystem/go_api_core/app"
"github.com/Datosystem/go_api_core/message"
"github.com/gin-gonic/gin"
"gorm.io/gorm/schema"
)
type PermissionFunc func(c *gin.Context) message.Message
type ModelWithPermissionsPrefix interface {
PermissionsPrefix() string
}
type ModelWithPermissionsGet interface {
PermissionsGet(c *gin.Context) message.Message
}
type ModelWithPermissionsPost interface {
PermissionsPost(c *gin.Context) message.Message
}
type ModelWithPermissionsPatch interface {
PermissionsPatch(c *gin.Context) message.Message
}
type ModelWithPermissionsDelete interface {
PermissionsDelete(c *gin.Context) message.Message
}
func PermissionsPrefix(model interface{}) string {
var prefix string
if prefixModel, ok := model.(ModelWithPermissionsPrefix); ok {
prefix = prefixModel.PermissionsPrefix()
} else if tableModel, ok := model.(TableModel); ok {
prefix = tableModel.TableName()
} else {
typ := reflect.TypeOf(model)
if typ.Kind() == reflect.Ptr {
typ = typ.Elem()
}
// Note: this uses the default naming strategy at the moment to keep things simple
return schema.NamingStrategy{
NoLowerCase: true,
SingularTable: true,
}.TableName(typ.Name())
}
return prefix
}
func PermissionsGet(model interface{}) PermissionFunc {
if modelPerm, ok := model.(ModelWithPermissionsGet); ok {
return modelPerm.PermissionsGet
} else {
return func(c *gin.Context) message.Message {
return c.MustGet("s").(*app.Session).CheckOne(c, PermissionsPrefix(model)+"_GET")
}
}
}
func PermissionsPost(model interface{}) PermissionFunc {
if modelPerm, ok := model.(ModelWithPermissionsPost); ok {
return modelPerm.PermissionsPost
} else {
return func(c *gin.Context) message.Message {
return c.MustGet("s").(*app.Session).CheckOne(c, PermissionsPrefix(model)+"_POST")
}
}
}
func PermissionsPatch(model interface{}) PermissionFunc {
if modelPerm, ok := model.(ModelWithPermissionsPatch); ok {
return modelPerm.PermissionsPatch
} else {
return func(c *gin.Context) message.Message {
return c.MustGet("s").(*app.Session).CheckOne(c, PermissionsPrefix(model)+"_PATCH")
}
}
}
func PermissionsDelete(model interface{}) PermissionFunc {
if modelPerm, ok := model.(ModelWithPermissionsDelete); ok {
return modelPerm.PermissionsDelete
} else {
return func(c *gin.Context) message.Message {
return c.MustGet("s").(*app.Session).CheckOne(c, PermissionsPrefix(model)+"_DELETE")
}
}
}