lucifer/controllers/light-controller.go

package controllers

import (
	"database/sql"
	"net/http"
	"strconv"

	"git.aiterp.net/lucifer/lucifer/internal/httperr"

	"git.aiterp.net/lucifer/lucifer/internal/respond"
	"git.aiterp.net/lucifer/lucifer/models"
	"github.com/gorilla/mux"
	"golang.org/x/sync/errgroup"
)

// The LightController is a controller for /api/light/.
type LightController struct {
	groups models.GroupRepository
	users  models.UserRepository
	lights models.LightRepository
}

// getLights (`GET /:id`): Get user by id
func (c *LightController) getLights(w http.ResponseWriter, r *http.Request) {
	user := models.UserFromContext(r.Context())

	groups, err := c.groups.ListByUser(r.Context(), *user)
	if err != nil {
		httperr.Respond(w, err)
		return
	}

	allLights := make([]models.Light, 0, len(groups)*8)

	eg, egCtx := errgroup.WithContext(r.Context())
	for i := range groups {
		group := groups[i]

		eg.Go(func() error {
			lights, err := c.lights.ListByGroup(egCtx, group)
			if err != nil {
				return err
			}

			allLights = append(allLights, lights...)

			return nil
		})
	}

	if err := eg.Wait(); err != nil {
		httperr.Respond(w, err)
		return
	}

	respond.Data(w, allLights)
}

func (c *LightController) getLight(w http.ResponseWriter, r *http.Request) {
	user := models.UserFromContext(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 light id "+idStr+" is not valid.")
		return
	}

	light, err := c.lights.FindByID(r.Context(), int(id))
	if err == sql.ErrNoRows {
		httperr.Respond(w, httperr.NotFound("Light"))
		return
	} else if err != nil {
		httperr.Respond(w, err)
		return
	}

	group, err := c.groups.FindByID(r.Context(), light.GroupID)
	if err == sql.ErrNoRows {
		httperr.Respond(w, httperr.NotFound("Group"))
		return
	} else if err != nil {
		httperr.Respond(w, err)
		return
	}

	if !group.Permission(user.ID).Read {
		respond.Error(w, http.StatusForbidden, "permission_denied", "You do not have permission to see this light.")
		return
	}

	respond.Data(w, light)
}

// Mount mounts the controller
func (c *LightController) Mount(router *mux.Router, prefix string) {
	sub := router.PathPrefix(prefix).Subrouter()

	sub.HandleFunc("/", c.getLights).Methods("GET")
	sub.HandleFunc("/{id}", c.getLight).Methods("GET")
}

// NewLightController creates a new LightController.
func NewLightController(groups models.GroupRepository, users models.UserRepository, lights models.LightRepository) *LightController {
	return &LightController{groups: groups, users: users, lights: lights}
}