app/attribute/service.go

79 lines
1.9 KiB
Go

package attribute
import (
"github.com/gin-gonic/gin"
"kumoly.io/kumoly/app/auth"
"kumoly.io/kumoly/app/server"
"kumoly.io/kumoly/app/store"
)
type Service struct{}
func (srv Service) GetName() string { return "attribute.Service" }
func (srv Service) GetDependencies() []string { return []string{} }
func (srv Service) IsService() bool { return false }
func (srv Service) Main() error { return nil }
func (srv Service) Del() {}
func (srv Service) Health() error { return nil }
func (srv Service) Init() error {
Init()
return nil
}
func (srv Service) Load() error {
api := server.API.Group("attr", auth.ACAdmin())
api.GET("", func(c *gin.Context) {
key := c.Query("key")
if key == "" {
var attrs []Attribute
store.DB.Find(&attrs)
server.OK(c, attrs)
} else {
var attr Attribute
store.DB.Find(&attr, "key = ?", key)
server.OK(c, attr)
}
})
api.PUT("", func(c *gin.Context) {
var data struct {
Key string `json:"key" binding:"required"`
Value string `json:"value" binding:"required"`
}
if err := c.ShouldBindJSON(&data); err != nil {
panic(err)
}
if err := store.DB.Model(&Attribute{}).Where("key = ?", data.Key).
Update("value", data.Value).Error; err != nil {
panic(err)
}
server.OK(c, "ok")
})
api.POST("restore", func(c *gin.Context) {
var data struct {
Key string `json:"key" binding:"required"`
}
if err := c.ShouldBindJSON(&data); err != nil {
panic(err)
}
result := store.DB.Exec(`update "attributes" set value = "default" where "key" = ?`, data.Key)
if result.Error != nil {
panic(result.Error)
}
server.OK(c, result.RowsAffected)
})
api.POST("restore_all", func(c *gin.Context) {
result := store.DB.Exec(`update "attributes" set value = "default"`)
if result.Error != nil {
panic(result.Error)
}
server.OK(c, result.RowsAffected)
})
return nil
}