From f3e1945faf343c51c28e13c5f0415401231e2414 Mon Sep 17 00:00:00 2001
From: dualtone-tony <tony.gardner@dualtone.co.uk>
Date: Fri, 26 Apr 2024 17:02:21 +0100
Subject: [PATCH] Add reregisterForMessages method and fix deleting from verbs
 once unregistered

---
 lib/drachtio-agent.js | 17 ++---------------
 lib/srf.js            |  4 ++++
 2 files changed, 6 insertions(+), 15 deletions(-)

diff --git a/lib/drachtio-agent.js b/lib/drachtio-agent.js
index 374d3b5..a58fed7 100644
--- a/lib/drachtio-agent.js
+++ b/lib/drachtio-agent.js
@@ -470,25 +470,12 @@ class DrachtioAgent extends Emitter {
 
     this.mapServer.forEach((obj, socket) => {
       if (obj.authenticated) {
-        this.unrouteVerbs(socket) ;
+        this.wp.send(socket, 'remove_route|' + verb) ;
+        this.verbs.delete(verb) ;
       }
     });
   }
 
-  unrouteVerbs(socket) {
-    this.verbs.forEach((obj, verb) => {
-      if (obj.sent === false) {
-        return ;
-      }
-
-      obj = {
-        sent: false,
-        acknowledged: false,
-        rid: this.wp.send(socket, 'remove_route|' + verb)
-      } ;
-    });
-  }
-
   disconnect(socket) {
     const sock = socket || this._getDefaultSocket();
     debugSocket(`disconnect: removing socket ${sockPort(sock)}`);
diff --git a/lib/srf.js b/lib/srf.js
index 4107bc9..d8a2dd8 100644
--- a/lib/srf.js
+++ b/lib/srf.js
@@ -1229,6 +1229,10 @@ class Srf extends Emitter {
     this._app.client.removeRoute(sipVerb) ;
   }
 
+  reregisterForMessages(sipVerb) {
+    this._app.client.route(sipVerb) ;
+  }
+
   _b2bRequestWithinDialog(dlg, req, res, proxyRequestHeaders, proxyResponseHeaders, callback) {
     callback = callback || noop ;
     let headers = {} ;