Skip to content
Snippets Groups Projects
Commit fc04b598 authored by Jake Taylor's avatar Jake Taylor :lips:
Browse files

Merge branch 'hotfix/endpoint-auth' into 'release'

add auth check to nb endpoints

See merge request elixxir/notifications-bot!13
parents 47b993a9 f9ae81bd
Branches
Tags
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