peter/websub-hub
1
0
Fork 0
Code Issues 1 Pull Requests Releases Activity

Add unsubscription

Browse Source
This commit is contained in:
Peter Stuifzand 2018-01-30 21:48:35 +01:00
parent 62b4e4012d
commit 7448b0f878
1 changed files with 26 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

27
cmd/hubserver/main.go
View File

@ -79,6 +79,30 @@ func (handler *subscriptionHandler) handlePublish(w http.ResponseWriter, r *http
return nil
}
func (handler *subscriptionHandler) handleUnsubscription(w http.ResponseWriter, r *http.Request) error {
log.Println(r.Form.Encode())
callback := r.Form.Get("hub.callback")
topic := r.Form.Get("hub.topic")
if subs, e := handler.Subscribers[topic]; e {
for i, sub := range subs {
if sub.Callback != callback {
continue
}
ourChallenge := randStringBytes(12)
if validateURL(sub.Callback, ourChallenge) {
subs = append(subs[:i], subs[i+1:]...)
break
}
}
w.WriteHeader(200)
fmt.Fprintf(w, "Unsubscribed\n")
} else {
http.Error(w, "Hub does not handle subscription for topic", 400)
}
return nil
}
func (handler *subscriptionHandler) handleSubscription(w http.ResponseWriter, r *http.Request) error {
log.Println(r.Form.Encode())
callback := r.Form.Get("hub.callback")
@ -181,7 +205,8 @@ func (handler *subscriptionHandler) ServeHTTP(w http.ResponseWriter, r *http.Req
if mode == "subscribe" {
handler.handleSubscription(w, r)
return
} else if mode == "unsubcribe" {
} else if mode == "unsubscribe" {
handler.handleUnsubscription(w, r)
return
} else if mode == "publish" {
handler.handlePublish(w, r)