mirror of https://github.com/sunface/rust-course
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
80 lines
1.9 KiB
80 lines
1.9 KiB
package internal
|
|
|
|
import (
|
|
"net/http"
|
|
"sort"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
"github.com/imdotdev/im.dev/server/internal/user"
|
|
"github.com/imdotdev/im.dev/server/pkg/common"
|
|
"github.com/imdotdev/im.dev/server/pkg/db"
|
|
"github.com/imdotdev/im.dev/server/pkg/e"
|
|
"github.com/imdotdev/im.dev/server/pkg/models"
|
|
)
|
|
|
|
func GetNavbars(c *gin.Context) {
|
|
navbars := make(models.Navbars, 0)
|
|
|
|
rows, err := db.Conn.Query("SELECT id,label,value,weight FROM navbar")
|
|
if err != nil {
|
|
c.JSON(http.StatusInternalServerError, common.RespError(e.Internal))
|
|
return
|
|
}
|
|
|
|
for rows.Next() {
|
|
nv := &models.Navbar{}
|
|
rows.Scan(&nv.ID, &nv.Label, &nv.Value, &nv.Weight)
|
|
navbars = append(navbars, nv)
|
|
}
|
|
|
|
sort.Sort(navbars)
|
|
|
|
c.JSON(http.StatusOK, common.RespSuccess(navbars))
|
|
}
|
|
|
|
func SubmitNavbar(c *gin.Context) {
|
|
nav := &models.Navbar{}
|
|
c.Bind(&nav)
|
|
|
|
u := user.CurrentUser(c)
|
|
if !u.Role.IsAdmin() {
|
|
c.JSON(http.StatusForbidden, common.RespError(e.NoPermission))
|
|
return
|
|
}
|
|
|
|
var err error
|
|
if nav.ID == 0 {
|
|
_, err = db.Conn.Exec("INSERT INTO navbar (label,value,weight) VALUES (?,?,?)",
|
|
nav.Label, nav.Value, nav.Weight)
|
|
} else {
|
|
_, err = db.Conn.Exec("UPDATE navbar SET label=?,value=?,weight=? WHERE id=?", nav.Label, nav.Value, nav.Weight, nav.ID)
|
|
}
|
|
|
|
if err != nil {
|
|
logger.Warn("submit navbar error", "error", err)
|
|
c.JSON(http.StatusInternalServerError, common.RespError(e.Internal))
|
|
return
|
|
}
|
|
|
|
c.JSON(http.StatusOK, common.RespSuccess(nil))
|
|
}
|
|
|
|
func DeleteNavbar(c *gin.Context) {
|
|
u := user.CurrentUser(c)
|
|
if !u.Role.IsAdmin() {
|
|
c.JSON(http.StatusForbidden, common.RespError(e.NoPermission))
|
|
return
|
|
}
|
|
|
|
id := c.Param("id")
|
|
_, err := db.Conn.Exec("DELETE FROM navbar WHERE id=?", id)
|
|
if err != nil {
|
|
logger.Warn("delete navbar error", "error", err)
|
|
c.JSON(http.StatusInternalServerError, common.RespError(e.Internal))
|
|
return
|
|
}
|
|
|
|
c.JSON(http.StatusOK, common.RespSuccess(nil))
|
|
|
|
}
|