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.
130 lines
4.1 KiB
130 lines
4.1 KiB
package controllers
|
|
|
|
import (
|
|
"database/sql"
|
|
"fmt"
|
|
"log"
|
|
"net/http"
|
|
"strconv"
|
|
|
|
"git.aiterp.net/lucifer/lucifer/internal/respond"
|
|
"git.aiterp.net/lucifer/lucifer/models"
|
|
"github.com/gorilla/mux"
|
|
)
|
|
|
|
// The GroupController is a controller for all user inports.
|
|
type GroupController struct {
|
|
groups models.GroupRepository
|
|
users models.UserRepository
|
|
lights models.LightRepository
|
|
}
|
|
|
|
// getGroups (`GET /:id`): Get user by id
|
|
func (c *GroupController) getGroups(w http.ResponseWriter, r *http.Request) {
|
|
user := models.UserFromContext(r.Context())
|
|
|
|
groups, err := c.groups.ListByUser(r.Context(), *user)
|
|
if err != nil {
|
|
log.Printf("Getting groups for user %s (%d) failed: %s", user.Name, user.ID, err)
|
|
respond.Error(w, http.StatusInternalServerError, "internal_error", "Failed to get groups.")
|
|
return
|
|
}
|
|
|
|
respond.Data(w, groups)
|
|
}
|
|
|
|
// getGroup (`GET /:id`): Get user by id
|
|
func (c *GroupController) getGroup(w http.ResponseWriter, r *http.Request) {
|
|
session := models.SessionFromContext(r.Context())
|
|
|
|
idStr := mux.Vars(r)["id"]
|
|
id, err := strconv.ParseInt(idStr, 10, 32)
|
|
if err != nil {
|
|
respond.Error(w, http.StatusBadRequest, "invalid_id", "The id"+idStr+"is not valid.")
|
|
return
|
|
}
|
|
|
|
group, err := c.groups.FindByID(r.Context(), int(id))
|
|
if err != nil || !group.Permission(session.UserID).Read {
|
|
respond.Error(w, http.StatusNotFound, "not_found", "The group cannot be found or you are not authorized to view it.")
|
|
return
|
|
}
|
|
|
|
respond.Data(w, group)
|
|
}
|
|
|
|
// getGroupLights (`GET /:id/light/`): Get user by id
|
|
func (c *GroupController) getGroupLights(w http.ResponseWriter, r *http.Request) {
|
|
session := models.SessionFromContext(r.Context())
|
|
|
|
idStr := mux.Vars(r)["id"]
|
|
id, err := strconv.ParseInt(idStr, 10, 32)
|
|
if err != nil {
|
|
respond.Error(w, http.StatusBadRequest, "invalid_id", "The id "+idStr+" is not valid.")
|
|
return
|
|
}
|
|
|
|
group, err := c.groups.FindByID(r.Context(), int(id))
|
|
if err != nil || !group.Permission(session.UserID).Read {
|
|
respond.Error(w, http.StatusNotFound, "group_not_found", "The group cannot be found or you are not authorized to view it.")
|
|
return
|
|
}
|
|
|
|
lights, err := c.lights.ListByGroup(r.Context(), group)
|
|
if err != nil && err != sql.ErrNoRows {
|
|
log.Printf("Getting lights for group %s (%d) failed: %s", group.Name, group.ID, err)
|
|
respond.Error(w, http.StatusInternalServerError, "internal_error", "Failed to get groups.")
|
|
return
|
|
}
|
|
|
|
respond.Data(w, lights)
|
|
}
|
|
|
|
// getGroupLight (`GET /:group_id/light/:id`): Get user by id
|
|
func (c *GroupController) getGroupLight(w http.ResponseWriter, r *http.Request) {
|
|
session := models.SessionFromContext(r.Context())
|
|
|
|
groupIDStr := mux.Vars(r)["group_id"]
|
|
groupID, err := strconv.ParseInt(groupIDStr, 10, 32)
|
|
if err != nil {
|
|
respond.Error(w, http.StatusBadRequest, "invalid_id", "The group id "+groupIDStr+" is not valid.")
|
|
return
|
|
}
|
|
|
|
idStr := mux.Vars(r)["id"]
|
|
id, err := strconv.ParseInt(idStr, 10, 32)
|
|
if err != nil {
|
|
respond.Error(w, http.StatusBadRequest, "invalid_id", "The light id "+idStr+" is not valid.")
|
|
return
|
|
}
|
|
|
|
group, err := c.groups.FindByID(r.Context(), int(groupID))
|
|
if err != nil || !group.Permission(session.UserID).Read {
|
|
respond.Error(w, http.StatusNotFound, "group_not_found", "The group cannot be found or you are not authorized to view it.")
|
|
return
|
|
}
|
|
|
|
light, err := c.lights.FindByID(r.Context(), int(id))
|
|
if err != nil || light.GroupID != group.ID {
|
|
fmt.Println(light, id)
|
|
respond.Error(w, http.StatusNotFound, "light_not_found", "The light cannot be found in this group.")
|
|
return
|
|
}
|
|
|
|
respond.Data(w, light)
|
|
}
|
|
|
|
// Mount mounts the controller
|
|
func (c *GroupController) Mount(router *mux.Router, prefix string) {
|
|
sub := router.PathPrefix(prefix).Subrouter()
|
|
|
|
sub.HandleFunc("/", c.getGroups).Methods("GET")
|
|
sub.HandleFunc("/{id}", c.getGroup).Methods("GET")
|
|
sub.HandleFunc("/{id}/light/", c.getGroupLights).Methods("GET")
|
|
sub.HandleFunc("/{group_id}/light/{id}", c.getGroupLight).Methods("GET")
|
|
}
|
|
|
|
// NewGroupController creates a new GroupController.
|
|
func NewGroupController(groups models.GroupRepository, users models.UserRepository, lights models.LightRepository) *GroupController {
|
|
return &GroupController{groups: groups, users: users, lights: lights}
|
|
}
|