Gisle Aune
5 years ago
10 changed files with 292 additions and 36 deletions
-
3cmd/lucifer-server/main.go
-
220controllers/bridge-controller.go
-
2controllers/group-controller.go
-
2controllers/light-controller.go
-
2controllers/user-controller.go
-
4database/sqlite/bridge-repository.go
-
3internal/httperr/error.go
-
11light/driver.go
-
2light/hue/driver.go
-
75light/service.go
@ -0,0 +1,220 @@ |
|||
package controllers |
|||
|
|||
import ( |
|||
"context" |
|||
"database/sql" |
|||
"encoding/json" |
|||
"errors" |
|||
"net/http" |
|||
"strconv" |
|||
|
|||
"git.aiterp.net/lucifer/lucifer/internal/httperr" |
|||
"git.aiterp.net/lucifer/lucifer/internal/respond" |
|||
"git.aiterp.net/lucifer/lucifer/light" |
|||
"git.aiterp.net/lucifer/lucifer/models" |
|||
"github.com/gorilla/mux" |
|||
) |
|||
|
|||
// The BridgeController is a controller for all bridge things.
|
|||
type BridgeController struct { |
|||
service *light.Service |
|||
groups models.GroupRepository |
|||
} |
|||
|
|||
// getBridges (`GET /`): Get all bridges
|
|||
func (c *BridgeController) getBridges(w http.ResponseWriter, r *http.Request) { |
|||
bridges, err := c.service.Bridges(r.Context()) |
|||
if err != nil { |
|||
httperr.Respond(w, err) |
|||
return |
|||
} |
|||
|
|||
respond.Data(w, bridges) |
|||
} |
|||
|
|||
// getBridge (`GET /:bridge_id`): Get bridge by ID
|
|||
func (c *BridgeController) getBridge(w http.ResponseWriter, r *http.Request) { |
|||
idStr := mux.Vars(r)["bridge_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 |
|||
} |
|||
|
|||
bridge, err := c.service.Bridge(r.Context(), int(id)) |
|||
if err == sql.ErrNoRows { |
|||
respond.Error(w, 404, "bridge_not_found", "The bridge cannot be found.") |
|||
return |
|||
} else if err != nil { |
|||
httperr.Respond(w, err) |
|||
return |
|||
} |
|||
|
|||
respond.Data(w, bridge) |
|||
} |
|||
|
|||
// postBridge (`POST /`): Post bridge
|
|||
func (c *BridgeController) postBridge(w http.ResponseWriter, r *http.Request) { |
|||
postData := struct { |
|||
Name string `json:"name"` |
|||
Driver string `json:"driver"` |
|||
Addr string `json:"addr"` |
|||
}{} |
|||
if err := json.NewDecoder(r.Body).Decode(&postData); err != nil || postData.Name == "" || postData.Driver == "" || postData.Addr == "" { |
|||
respond.Error(w, http.StatusBadRequest, "invalid_json", "Input is not valid JSON.") |
|||
return |
|||
} |
|||
|
|||
bridge, err := c.service.DirectConnect(r.Context(), postData.Driver, postData.Addr, postData.Name) |
|||
if err != nil { |
|||
httperr.Respond(w, err) |
|||
return |
|||
} |
|||
|
|||
respond.Data(w, bridge) |
|||
} |
|||
|
|||
// updateBridge (`PUT/PATCH /:bridge_id`): Update bridge by ID
|
|||
func (c *BridgeController) updateBridge(w http.ResponseWriter, r *http.Request) { |
|||
idStr := mux.Vars(r)["bridge_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 |
|||
} |
|||
|
|||
putData := struct { |
|||
Name string |
|||
}{} |
|||
if err := json.NewDecoder(r.Body).Decode(&putData); err != nil { |
|||
respond.Error(w, http.StatusBadRequest, "invalid_json", "Input is not valid JSON.") |
|||
return |
|||
} |
|||
if putData.Name == "" { |
|||
respond.Error(w, http.StatusBadRequest, "invalid_name", "The name cannot be blank.") |
|||
return |
|||
} |
|||
|
|||
bridge, err := c.service.Bridge(r.Context(), int(id)) |
|||
if err == sql.ErrNoRows { |
|||
respond.Error(w, 404, "bridge_not_found", "The bridge cannot be found.") |
|||
return |
|||
} else if err != nil { |
|||
httperr.Respond(w, err) |
|||
return |
|||
} |
|||
|
|||
bridge.Name = putData.Name |
|||
err = c.service.UpdateBridge(r.Context(), bridge) |
|||
if err != nil { |
|||
httperr.Respond(w, err) |
|||
return |
|||
} |
|||
|
|||
respond.Data(w, bridge) |
|||
} |
|||
|
|||
// postBridgeDiscover (`POST /:bridge_id/discover`): Delete bridge by ID
|
|||
func (c *BridgeController) postBridgeDiscover(w http.ResponseWriter, r *http.Request) { |
|||
idStr := mux.Vars(r)["bridge_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 |
|||
} |
|||
|
|||
bridge, err := c.service.Bridge(r.Context(), int(id)) |
|||
if err == sql.ErrNoRows { |
|||
respond.Error(w, 404, "bridge_not_found", "The bridge cannot be found.") |
|||
return |
|||
} else if err != nil { |
|||
httperr.Respond(w, err) |
|||
return |
|||
} |
|||
|
|||
err = c.service.DiscoverLights(r.Context(), bridge) |
|||
if err != nil { |
|||
httperr.Respond(w, err) |
|||
return |
|||
} |
|||
|
|||
respond.Data(w, bridge) |
|||
} |
|||
|
|||
// deleteBridge (`DELETE /:bridge_id`): Delete bridge by ID
|
|||
func (c *BridgeController) deleteBridge(w http.ResponseWriter, r *http.Request) { |
|||
idStr := mux.Vars(r)["bridge_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 |
|||
} |
|||
|
|||
bridge, err := c.service.Bridge(r.Context(), int(id)) |
|||
if err == sql.ErrNoRows { |
|||
respond.Error(w, 404, "bridge_not_found", "The bridge cannot be found.") |
|||
return |
|||
} else if err != nil { |
|||
httperr.Respond(w, err) |
|||
return |
|||
} |
|||
|
|||
err = c.service.DeleteBridge(r.Context(), bridge) |
|||
if err != nil { |
|||
httperr.Respond(w, err) |
|||
return |
|||
} |
|||
|
|||
respond.Data(w, bridge) |
|||
} |
|||
|
|||
// Mount mounts the controller
|
|||
func (c *BridgeController) Mount(router *mux.Router, prefix string) { |
|||
sub := router.PathPrefix(prefix).Subrouter() |
|||
|
|||
sub.Use(c.adminMiddleware()) |
|||
|
|||
sub.HandleFunc("/", c.getBridges).Methods("GET") |
|||
sub.HandleFunc("/", c.postBridge).Methods("POST") |
|||
sub.HandleFunc("/{bridge_id}", c.getBridge).Methods("GET") |
|||
sub.HandleFunc("/{bridge_id}", c.updateBridge).Methods("PUT", "PATCH") |
|||
sub.HandleFunc("/{bridge_id}", c.deleteBridge).Methods("DELETE") |
|||
sub.HandleFunc("/{bridge_id}/discover", c.postBridgeDiscover).Methods("POST") |
|||
} |
|||
|
|||
func (c *BridgeController) adminMiddleware() mux.MiddlewareFunc { |
|||
return func(next http.Handler) http.Handler { |
|||
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) { |
|||
err := c.isAdmin(r.Context()) |
|||
if err != nil { |
|||
httperr.Respond(w, err) |
|||
return |
|||
} |
|||
|
|||
next.ServeHTTP(w, r) |
|||
}) |
|||
} |
|||
} |
|||
|
|||
func (c *BridgeController) isAdmin(ctx context.Context) error { |
|||
user := models.UserFromContext(ctx) |
|||
|
|||
lonelyLights, err := c.groups.FindByID(ctx, 0) |
|||
if err != nil { |
|||
return errors.New("Lonely Lights group could not be found") |
|||
} |
|||
|
|||
if !lonelyLights.Permission(user.ID).Write { |
|||
return httperr.ErrAccessDenied |
|||
} |
|||
|
|||
return nil |
|||
} |
|||
|
|||
// NewBridgeController makes a new bridge controller
|
|||
func NewBridgeController(service *light.Service, groups models.GroupRepository) *BridgeController { |
|||
return &BridgeController{ |
|||
service: service, |
|||
groups: groups, |
|||
} |
|||
} |
Write
Preview
Loading…
Cancel
Save
Reference in new issue