-
Notifications
You must be signed in to change notification settings - Fork 0
/
scaleController.go
102 lines (83 loc) · 2.38 KB
/
scaleController.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
94
95
96
97
98
99
100
101
102
package controllers
import (
"github.com/gin-gonic/gin"
"github.com/jefgodesky/rnrapi/helpers"
"github.com/jefgodesky/rnrapi/initializers"
"github.com/jefgodesky/rnrapi/models"
"github.com/jefgodesky/rnrapi/parsers"
"github.com/jefgodesky/rnrapi/serializers"
"gorm.io/gorm/clause"
)
func ScaleCreate(c *gin.Context) {
scale := parsers.BodyToScale(c)
if result := initializers.DB.Create(&scale); result.Error != nil {
c.JSON(500, gin.H{"error": "Failed to create scale"})
return
}
c.JSON(200, serializers.SerializeScale(*scale))
}
func ScaleIndex(c *gin.Context) {
var scales []models.Scale
user := helpers.GetUserFromContext(c, false)
query := initializers.DB.
Preload(clause.Associations).
Model(&models.Scale{})
if user != nil {
query.Where("public = ? OR author_id = ?", true, user.ID)
} else {
query.Where("public = ?", true)
}
var total int64
query.Count(&total)
query.Scopes(helpers.Paginate(c)).Find(&scales)
c.JSON(200, gin.H{
"total": total,
"page": c.GetInt("page"),
"page_size": c.GetInt("page_size"),
"scales": serializers.SerializeScales(scales),
})
}
func ScaleRetrieve(c *gin.Context) {
scale := helpers.GetScaleFromSlug(c)
user := helpers.GetUserFromContext(c, false)
allowed := scale.Public || scale.Author.ID == user.ID
if !allowed {
c.JSON(403, gin.H{"error": "Forbidden"})
return
}
c.JSON(200, serializers.SerializeScale(*scale))
}
func ScaleUpdate(c *gin.Context) {
scale := helpers.ScaleAuthorOnly(c)
if scale == nil {
return
}
newScale := parsers.BodyToScale(c)
scale.Name = newScale.Name
scale.Slug = newScale.Slug
scale.Description = newScale.Description
scale.Levels = newScale.Levels
scale.Public = newScale.Public
scale.AuthorID = newScale.AuthorID
scale.Author = newScale.Author
if err := initializers.DB.Where("scale_id = ?", scale.ID).Delete(&models.Level{}).Error; err != nil {
c.JSON(500, gin.H{"error": "Failed to update scale"})
return
}
if err := initializers.DB.Save(scale).Error; err != nil {
c.JSON(500, gin.H{"Error": "Failed to update scale"})
return
}
c.JSON(200, serializers.SerializeScale(*scale))
}
func ScaleDestroy(c *gin.Context) {
scale := helpers.ScaleAuthorOnly(c)
if scale == nil {
return
}
if err := initializers.DB.Delete(&scale).Error; err != nil {
c.JSON(500, gin.H{"Error": "Failed to destroy scale"})
return
}
c.JSON(200, serializers.SerializeScale(*scale))
}