buzzert/Kordophone
Private
Public Access
1
0
Fork 0
Code Packages Activity

server: enforce auth for updates

Browse Source
This commit is contained in:
James Magahern
2024-04-04 22:59:59 -07:00
parent c666083e4b
commit 63876104aa
1 changed files with 10 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
web/server.go
View File

@@ -330,6 +330,15 @@ func (m *MockHTTPServer) handleMarkConversation(w http.ResponseWriter, r *http.R
w.WriteHeader(http.StatusOK) w.WriteHeader(http.StatusOK)
} }
func (m *MockHTTPServer) handleUpdates(w http.ResponseWriter, r *http.Request) {
if !m.requireAuthentication(w, r) {
return
}
s := websocket.Server{Handler: websocket.Handler(m.handleUpdatesWebsocket)}
s.ServeHTTP(w, r)
}
func (m *MockHTTPServer) handleUpdatesWebsocket(c *websocket.Conn) { func (m *MockHTTPServer) handleUpdatesWebsocket(c *websocket.Conn) {
// Fetch updates continuously // Fetch updates continuously
defer c.Close() defer c.Close()
@@ -371,12 +380,7 @@ func NewMockHTTPServer(config MockHTTPServerConfiguration) *MockHTTPServer {
this.mux.Handle("/pollUpdates", http.HandlerFunc(this.handlePollUpdates)) this.mux.Handle("/pollUpdates", http.HandlerFunc(this.handlePollUpdates))
this.mux.Handle("/sendMessage", http.HandlerFunc(this.handleSendMessage)) this.mux.Handle("/sendMessage", http.HandlerFunc(this.handleSendMessage))
this.mux.Handle("/markConversation", http.HandlerFunc(this.handleMarkConversation)) this.mux.Handle("/markConversation", http.HandlerFunc(this.handleMarkConversation))
this.mux.Handle("/updates", http.HandlerFunc(this.handleUpdates))
// /updates websocket
this.mux.Handle("/updates", http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
s := websocket.Server{Handler: websocket.Handler(this.handleUpdatesWebsocket)}
s.ServeHTTP(w, r)
}))
this.mux.Handle("/", http.HandlerFunc(this.handleNotFound)) this.mux.Handle("/", http.HandlerFunc(this.handleNotFound))