Skip to content
Snippets Groups Projects
Commit f9ae81bd authored by Jonah Husson's avatar Jonah Husson
Browse files

add auth check to nb endpoints

parent 47b993a9
No related branches found
No related tags found
No related merge requests found
......@@ -193,6 +193,9 @@ func pollForNotifications(nb *Impl) (strings []string, e error) {
// RegisterForNotifications is called by the client, and adds a user registration to our database
func (nb *Impl) RegisterForNotifications(clientToken []byte, auth *connect.Auth) error {
if !auth.IsAuthenticated {
return errors.New("Cannot register for notifications: client is not authenticated")
}
// Implement this
u := &storage.User{
Id: auth.Sender.GetId(),
......@@ -207,6 +210,9 @@ func (nb *Impl) RegisterForNotifications(clientToken []byte, auth *connect.Auth)
// UnregisterForNotifications is called by the client, and removes a user registration from our database
func (nb *Impl) UnregisterForNotifications(auth *connect.Auth) error {
if !auth.IsAuthenticated {
return errors.New("Cannot unregister for notifications: client is not authenticated")
}
err := nb.Storage.DeleteUser(auth.Sender.GetId())
if err != nil {
return errors.Wrap(err, "Failed to unregister user with notifications")
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment