// @Title getUser
// @router /:name [get]
func (h *UserController) Get() {
name := h.GetString(":name")
beego.Debug("[C] Got name:", name)
defer h.ServeJSON()
if name != "" {
user := &models.Users{
Name: name,
}
users, err := models.GetUser(user, 0, 0)
if err != nil {
h.Data["json"] = map[string]string{
"message": fmt.Sprint("Failed to get with name:", name),
"error": err.Error(),
}
beego.Warn("[C] Got error:", err)
h.Ctx.Output.SetStatus(http.StatusInternalServerError)
return
}
h.Data["json"] = users
if len(users) == 0 {
beego.Debug("[C] Got nothing with name:", name)
h.Ctx.Output.SetStatus(http.StatusNotFound)
} else {
h.Ctx.Output.SetStatus(http.StatusOK)
}
}
}
// @Title listUser
// @router / [get]
func (h *UserController) GetAll() {
limit, _ := h.GetInt("limit", 0)
index, _ := h.GetInt("index", 0)
defer h.ServeJSON()
user := &models.Users{}
users, err := models.GetUser(user, limit, index)
if err != nil {
h.Data["json"] = map[string]string{
"message": fmt.Sprint("Failed to get"),
"error": err.Error(),
}
beego.Warn("[C] Got error:", err)
h.Ctx.Output.SetStatus(http.StatusInternalServerError)
return
}
h.Data["json"] = users
if len(users) == 0 {
beego.Debug("[C] Got nothing")
h.Ctx.Output.SetStatus(http.StatusNotFound)
} else {
h.Ctx.Output.SetStatus(http.StatusOK)
}
}
// @Title updateUser
// @router /:name [put]
func (h *UserController) Put() {
name := h.GetString(":name")
defer h.ServeJSON()
beego.Debug("[C] Got user name:", name)
if name != "" {
user := &models.Users{
Name: name,
}
users, err := models.GetUser(user, 0, 0)
if err != nil {
h.Data["json"] = map[string]string{
"message": fmt.Sprint("Failed to get with name:", name),
"error": err.Error(),
}
beego.Warn("[C] Got error:", err)
h.Ctx.Output.SetStatus(http.StatusInternalServerError)
return
}
if len(users) == 0 {
beego.Debug("[C] Got nothing with name:", name)
h.Ctx.Output.SetStatus(http.StatusNotFound)
return
}
sessionId := h.GetSession("id")
if sessionId != nil {
userNow := &models.Users{
Id: sessionId.(string),
}
userNows, err := models.GetUser(userNow, 0, 0)
if err != nil {
h.Data["json"] = map[string]string{
"message": fmt.Sprint("Failed to get with name:", name),
"error": err.Error(),
}
beego.Warn("[C] Got error:", err)
h.Ctx.Output.SetStatus(http.StatusInternalServerError)
return
}
if len(userNows) == 0 {
beego.Debug("[C] Invalid user id:", sessionId)
h.Ctx.Output.SetStatus(http.StatusNotFound)
return
}
}
err = json.Unmarshal(h.Ctx.Input.RequestBody, user)
if err != nil {
beego.Warn("[C] Got error:", err)
h.Data["json"] = map[string]string{
"message": "Bad request",
"error": err.Error(),
}
h.Ctx.Output.SetStatus(http.StatusBadRequest)
return
}
user.Id = users[0].Id
user.Removable = users[0].Removable // Removable should not be changed.
if user.Password != users[0].Password {
user.Password = common.EncryptPassword(user.Password)
}
beego.Debug("[C] Got user data:", user)
err = models.UpdateUser(user)
if err != nil {
h.Data["json"] = map[string]string{
"message": fmt.Sprint("Failed to update with name:", name),
"error": err.Error(),
}
beego.Warn("[C] Got error:", err)
h.Ctx.Output.SetStatus(http.StatusInternalServerError)
return
}
h.Ctx.Output.SetStatus(http.StatusAccepted)
}
}