Better handling of channel names

porsager · porsager · commit 9bd85073cedf · 2019-12-20T23:59:56.000+01:00
diff --git a/lib/index.js b/lib/index.js
@@ -290,15 +290,14 @@ function Postgres(url, options) {
     })
   }
 
-  function listen(x, fn) {
-    if (x.match(/[^a-z0-9_-]/))
-      return Promise.reject('Only a-z A-Z 0-9 and - . _ allowed in channel names')
-
-    x in listeners
-      ? listeners[x].push(fn)
-      : (listeners[x] = [fn])
+  function listen(channel, fn) {
+    if (channel in listeners) {
+      listeners[channel].push(fn)
+      return Promise.resolve(channel)
+    }
 
-    return query({ raw: true }, getListener(), 'listen "' + x + '"').then(() => x)
+    listeners[channel] = [fn]
+    return query({ raw: true }, getListener(), 'listen ' + escape(channel)).then(() => channel)
   }
 
   function getListener() {
