trafficlunar/api
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

feat: add authorization to websocket

Browse source
This commit is contained in:
trafficlunar 2024-11-29 19:45:14 +00:00
parent e7e06cf646
commit 51557d616e
2 changed files with 12 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
.env.example
View file

@ -11,4 +11,7 @@ UPTIME_KUMA_ENABLED=false
# URL for Uptime Kuma # URL for Uptime Kuma
UPTIME_KUMA_URL="http://localhost:3001/metrics" UPTIME_KUMA_URL="http://localhost:3001/metrics"
# API key for Uptime Kuma # API key for Uptime Kuma
UPTIME_KUMA_API_KEY="API_KEY_GOES_HERE" UPTIME_KUMA_API_KEY="API_KEY_GOES_HERE"
# The password for the computer WebSocket
WEBSOCKET_PASSWORD="PASSWORD_GOES_HERE"

9
internal/handler/computer.go
View file

@ -6,6 +6,7 @@ import (
"encoding/json" "encoding/json"
"log/slog" "log/slog"
"net/http" "net/http"
"os"
"github.com/gorilla/websocket" "github.com/gorilla/websocket"
) )
@ -18,6 +19,12 @@ var upgrader = websocket.Upgrader{
} }
func HandleComputerWebSocket(w http.ResponseWriter, r *http.Request) { func HandleComputerWebSocket(w http.ResponseWriter, r *http.Request) {
if r.Header.Get("Authorization") != os.Getenv("WEBSOCKET_PASSWORD") {
w.WriteHeader(http.StatusUnauthorized)
w.Write([]byte("Unauthorized"))
return
}
conn, err := upgrader.Upgrade(w, r, nil) conn, err := upgrader.Upgrade(w, r, nil)
if err != nil { if err != nil {
slog.Error("Error when upgrading websocket connection", slog.Any("error", err)) slog.Error("Error when upgrading websocket connection", slog.Any("error", err))
@ -25,7 +32,7 @@ func HandleComputerWebSocket(w http.ResponseWriter, r *http.Request) {
} }
defer conn.Close() defer conn.Close()
slog.Info("Websocket connection established!") slog.Info("WebSocket connection established!")
service.ComputerData.Online = true service.ComputerData.Online = true
// Read messages // Read messages