u8-mqtt 0.3.1 → 0.4.0

package/cjs/index.cjs

@@ -3,6 +3,7 @@
 Object.defineProperty(exports, '__esModule', { value: true });
 
 var node_net = require('node:net');
+var node_tls = require('node:tls');
 
 function encode_varint(n, a=[]) {
   do {
@@ -775,6 +776,191 @@ const mqtt_opts_v5 =
     encode_fns: mqtt_encode_v5,
     mqtt_writer: mqtt_writer_v5, };
 
+function parse(str, loose) {
+	if (str instanceof RegExp) return { keys:false, pattern:str };
+	var c, o, tmp, ext, keys=[], pattern='', arr = str.split('/');
+	arr[0] || arr.shift();
+
+	while (tmp = arr.shift()) {
+		c = tmp[0];
+		if (c === '*') {
+			keys.push('wild');
+			pattern += '/(.*)';
+		} else if (c === ':') {
+			o = tmp.indexOf('?', 1);
+			ext = tmp.indexOf('.', 1);
+			keys.push( tmp.substring(1, !!~o ? o : !!~ext ? ext : tmp.length) );
+			pattern += !!~o && !~ext ? '(?:/([^/]+?))?' : '/([^/]+?)';
+			if (!!~ext) pattern += (!!~o ? '?' : '') + '\\' + tmp.substring(ext);
+		} else {
+			pattern += '/' + tmp;
+		}
+	}
+
+	return {
+		keys: keys,
+		pattern: new RegExp('^' + pattern + (loose ? '(?=$|\/)' : '\/?$'), 'i')
+	};
+}
+
+/*
+class AbstractTopicRouter ::
+  async invoke(pkt, ctx) ::
+  add(topic_route, ...args) ::
+  remove(topic_route, priority) ::
+  clear(priority) ::
+  find(topic) :: // optional
+  mqtt_topic(topic_route)
+*/
+
+const with_topic_router = mqtt_topic_router =>
+  MQTTKlass => class extends MQTTKlass {
+    static _aliases() {
+      return super._aliases() +
+        ' sub_topic:subscribe_topic unsub_topic:unsubscribe_topic'}
+
+    _init_router(opt) {
+      return mqtt_topic_router(opt, this)}
+
+    get on_topic() {return this.router.add}
+
+    _sub_chain(topic, ex, topic_prefix) {
+      let res = this.subscribe([[ topic ]], ex, topic_prefix);
+      let subs = this.subs ||(this.subs = new Map());
+      subs.set((res.topic = topic), (subs.last = res));
+      return this }// fluent api -- return this and track side effects
+
+    // alias: sub_topic
+    subscribe_topic(topic_route, ...args) {
+      let router = this.router;
+      router.add(topic_route, true, args.pop() );// handler
+      let topic = router.mqtt_topic(topic_route);
+      return this._sub_chain(topic, ...args ) }// ex, topic_prefix
+
+    // alias: unsub_topic
+    unsubscribe_topic(topic_route, ...args) {
+      let router = this.router;
+      router.remove(topic_route, true);
+      let topic = router.mqtt_topic(topic_route);
+      return this.unsubscribe([[ topic ]], ...args ) } };// topic_prefix
+
+// Use [regexparam][] for url-like topic parsing
+//  [regexparam]: https://github.com/lukeed/regexparam
+
+
+const with_topic_path_router = /* #__PURE__ */
+  with_topic_router(mqtt_topic_path_router);
+
+
+const mqtt_topic = topic_route =>
+  topic_route
+    .replace(/[*].*$/, '#')
+    .replace(/:\w+\??/g, '+');
+
+const as_topic_path = topic_route =>(
+  topic_route
+    .replace(/#$/, '*') // replace MQTT # wildcard at end
+    .split(/([^\/]*[+][^\/]*)/) // split on MQTT + match tokens
+    .reduce (( sz, v, idx ) => sz +(
+      idx & 1 // even entires are body, odd are MQTT + tokens
+        ? `:$${1 + idx>>1}` // replace with `:$#` sequential ids, using ? for partial entries
+        : v ) ) );// pass through body
+
+function _ignore(pkt, params, ctx) {ctx.done = true;}
+
+function mqtt_topic_path_router() {
+  let pri_lsts = [[],[]], rm = Symbol();
+  let find = topic => _routes_iter(pri_lsts, topic);
+
+  // return duck-type compatible with AbstractTopicRouter in ./with_topic_router
+  return {find, mqtt_topic,
+    add(topic_route, ...args) {
+      let fn = args.pop();
+      let priority = args.pop();
+
+      if ('function' !== typeof fn) {
+        if (false === fn) {
+          fn = _ignore;}
+        else throw new TypeError()}
+
+      let rte = parse(as_topic_path(topic_route));
+
+      rte.key = topic_route;
+      rte.tgt = fn;
+      pri_lsts[priority ? 0 : 1].push(rte);
+      return this}
+
+  , remove(topic_route, priority) {
+      let lst = pri_lsts[priority ? 0 : 1];
+      return _route_remove([lst], topic_route)}
+
+  , clear(priority) {
+      pri_lsts[priority ? 0 : 1] = [];
+      if (null == priority) {
+        pri_lsts[1] = [];} }
+
+  , async invoke(pkt, ctx) {
+      ctx.idx = 0;
+      ctx.rm = rm;
+
+      for (let [fn, params] of find(pkt.topic)) {
+        let res = await fn(pkt, params, ctx);
+
+        if (rm === res) {
+          _route_remove(pri_lsts, fn);}
+
+        if (ctx.done) {
+          break}
+        else ctx.idx++;}
+
+      let {pkt_id, qos} = pkt;
+      if (1 === qos) {
+        await ctx.mqtt._send('puback', {pkt_id});} } } }
+
+
+function * _routes_iter(all_route_lists, topic) {
+  for (let route_list of all_route_lists) {
+    for (let route of route_list) {
+      let res = _route_match_one(topic, route);
+      if (undefined !== res) {
+        yield res;} } } }
+
+
+function _route_match_one(topic, {keys, pattern, tgt}) {
+  let match = '/' !== topic[0]
+    ? pattern.exec('/'+topic)
+    : pattern.exec(topic);
+
+  if (null === match) {
+    return}
+
+  if (false === keys) {
+    let {groups} = match;
+    if (! groups) {
+      return [tgt]}
+
+    let params = {};
+    for (let k in groups) {
+      params[k] = groups[k];}
+
+    return [tgt, params]}
+
+  if (0 === keys.length) {
+    return [tgt]}
+
+  let params = {};
+  for (let i=0; i<keys.length; i++) {
+    params[ keys[i] ] = match[1+i];}
+  return [tgt, params]}
+
+
+function _route_remove(all_route_lists, query) {
+  let match = route => route===query || route.tgt===query || route.key===query;
+  for (let lst of all_route_lists) {
+    let i = lst.findIndex(match);
+    if (0 <= i) {return !! lst.splice(i,1)} }
+  return false}
+
 /*
 export function decode_varint_loop(u8, i=0) {
   let i0 = i
@@ -873,8 +1059,6 @@ function mqtt_pkt_ctx(mqtt_level, opts, pkt_ctx) {
   }
 }
 
-Object.freeze({ao_done: true});
-
 function ao_defer_ctx(as_res = (...args) => args) {
   let y,n,_pset = (a,b) => { y=a, n=b; };
   return p =>(
@@ -958,138 +1142,13 @@ function _ping_interval(send_ping) {
     if (td) {
       tid = setInterval(send_ping, 1000 * td);
       
-
-
+        
+        
       
-      // ensure the interval allows the NodeJS event loop to exit
-      tid.unref?.();
+        // ensure the interval allows the NodeJS event loop to exit
+        tid.unref?.();
       return true} }) }
 
-function parse(str, loose) {
-	if (str instanceof RegExp) return { keys:false, pattern:str };
-	var c, o, tmp, ext, keys=[], pattern='', arr = str.split('/');
-	arr[0] || arr.shift();
-
-	while (tmp = arr.shift()) {
-		c = tmp[0];
-		if (c === '*') {
-			keys.push('wild');
-			pattern += '/(.*)';
-		} else if (c === ':') {
-			o = tmp.indexOf('?', 1);
-			ext = tmp.indexOf('.', 1);
-			keys.push( tmp.substring(1, !!~o ? o : !!~ext ? ext : tmp.length) );
-			pattern += !!~o && !~ext ? '(?:/([^/]+?))?' : '/([^/]+?)';
-			if (!!~ext) pattern += (!!~o ? '?' : '') + '\\' + tmp.substring(ext);
-		} else {
-			pattern += '/' + tmp;
-		}
-	}
-
-	return {
-		keys: keys,
-		pattern: new RegExp('^' + pattern + (loose ? '(?=$|\/)' : '\/?$'), 'i')
-	};
-}
-
-// Use [regexparam][] for url-like topic parsing
-
-function _ignore(pkt, params, ctx) {ctx.done = true;}
-
-function _mqtt_topic_router() {
-  let pri_lsts = [[],[]], rm = Symbol();
-  let find = topic => _mqtt_routes_iter(pri_lsts, topic);
-
-  return {find,
-
-    add(topic_route, ...args) {
-      let fn = args.pop();
-      let priority = args.pop();
-
-      if ('function' !== typeof fn) {
-        if (false === fn) {
-          fn = _ignore;}
-        else throw new TypeError()}
-
-      let rte = parse(
-        topic_route.replace(/[+#]$/, '*'));
-
-      rte.key = topic_route;
-      rte.tgt = fn;
-      pri_lsts[priority ? 0 : 1].push(rte);
-      return this}
-
-  , remove(topic_route, priority) {
-      let lst = pri_lsts[priority ? 0 : 1];
-      return _mqtt_route_remove([lst], topic_route)}
-
-  , clear(priority) {
-      pri_lsts[priority ? 0 : 1] = [];
-      if (null == priority) {
-        pri_lsts[1] = [];} }
-
-  , async invoke(pkt, ctx) {
-      ctx.idx = 0;
-      ctx.rm = rm;
-
-      for (let [fn, params] of find(pkt.topic)) {
-        let res = await fn(pkt, params, ctx);
-
-        if (rm === res) {
-          _mqtt_route_remove(pri_lsts, fn);}
-
-        if (ctx.done) {
-          break}
-        else ctx.idx++;}
-
-      let {pkt_id, qos} = pkt;
-      if (1 === qos) {
-        await ctx.mqtt._send('puback', {pkt_id});} } } }
-
-
-function * _mqtt_routes_iter(all_route_lists, topic) {
-  for (let route_list of all_route_lists) {
-    for (let route of route_list) {
-      let res = _mqtt_route_match_one(topic, route);
-      if (undefined !== res) {
-        yield res;} } } }
-
-
-function _mqtt_route_match_one(topic, {keys, pattern, tgt}) {
-  let match = '/' !== topic[0]
-    ? pattern.exec('/'+topic)
-    : pattern.exec(topic);
-
-  if (null === match) {
-    return}
-
-  if (false === keys) {
-    let {groups} = match;
-    if (! groups) {
-      return [tgt]}
-
-    let params = {};
-    for (let k in groups) {
-      params[k] = groups[k];}
-
-    return [tgt, params]}
-
-  if (0 === keys.length) {
-    return [tgt]}
-
-  let params = {};
-  for (let i=0; i<keys.length; i++) {
-    params[ keys[i] ] = match[1+i];}
-  return [tgt, params]}
-
-
-function _mqtt_route_remove(all_route_lists, query) {
-  let match = route => route===query || route.tgt===query || route.key===query;
-  for (let lst of all_route_lists) {
-    let i = lst.findIndex(match);
-    if (0 <= i) {return !! lst.splice(i,1)} }
-  return false}
-
 const _mqtt_cmdid_dispatch ={
   create(target) {
     return {__proto__: this, target, hashbelt: [new Map()]} }
@@ -1158,8 +1217,7 @@ const _mqtt_cmdid_dispatch ={
       let fn = target[`mqtt_${pkt.type}`]
         || target.mqtt_pkt;
 
-      if (undefined !== fn) {
-        await fn.call(target, pkt, ctx);} } })()) };
+      await fn?.call(target, pkt, ctx);} })()) };
 
 function _mqtt_dispatch(opt, target) {
   let _disp_ = _mqtt_cmdid_dispatch.create(target);
@@ -1240,53 +1298,15 @@ class MQTTBase {
 
 
   // alias: sub
-  subscribe(pkt, ex) {
-    pkt = _as_topics(pkt, ex);
+  subscribe(pkt, ex, topic_prefix) {
+    pkt = _as_topics(pkt, ex, topic_prefix);
     return this._send('subscribe', pkt, pkt)}
-  _sub_chain(topic, ex) {
-    let res = this.subscribe([[ topic ]], ex);
-    let subs = this.subs ||(this.subs = new Map());
-    subs.set((res.topic = topic), (subs.last = res));
-    return this }// fluent api -- return this and track side effects
 
   // alias: unsub
-  unsubscribe(pkt, ex) {
-    pkt = _as_topics(pkt, ex);
+  unsubscribe(pkt, ex, topic_prefix) {
+    pkt = _as_topics(pkt, ex, topic_prefix);
     return this._send('unsubscribe', pkt, pkt)}
 
-  get on_topic() {return this.router.add}
-
-  // alias: sub_topic
-  subscribe_topic(topic_route, ...args) {
-    this.router.add(topic_route, true, args.pop() );// handler
-    let topic = this.topic_for(topic_route);
-    return this._sub_chain(topic, args.pop() ) }// ex
-
-  // alias: unsub_topic
-  unsubscribe_topic(topic_route) {
-    this.router.remove(topic_route, true);
-    let topic = this.topic_for(topic_route);
-    return this.unsubscribe([[ topic ]]) }
-
-  // alias: shared_sub
-  shared_subscribe(group, topic_route, ...args) {
-    this.router.add(topic_route, true, args.pop() );// handler
-    let topic = this.topic_for(topic_route);
-    if (null != group) {
-      topic = `$share/${group}/${topic}`;}
-    return this._sub_chain(topic, args.pop() ) }// ex
-
-  // alias: shared_unsub
-  shared_unsubscribe(group, topic_route) {
-    this.router.remove(topic_route, true);
-    let topic = this.topic_for(topic_route);
-    if (null != group) {
-      topic = `$share/${group}/${topic}`;}
-    return this.unsubscribe([[ topic ]]) }
-
-  topic_for(topic_route) {
-    return topic_route.replace(/[:*].*$/, '#')}
-
 
   // alias: pub
   publish(pkt, pub_opt) {return _pub(this, pkt, pub_opt)}
@@ -1312,9 +1332,9 @@ class MQTTBase {
     if (undefined === cid) {
       this.client_id = cid = (
         
-
+          
         
-        this.new_client_id(parts)
+          this.new_client_id(parts)
         );}
 
     return cid}
@@ -1323,66 +1343,85 @@ class MQTTBase {
     return [parts[0], Math.random().toString(36).slice(2), parts[1]].join('')}
 
   
-
-
-
-
-
-
-
+    
+      
+      
+      
+        
+        
+      
 
 
   // Internal API
 
   /* async _send(type, pkt) -- provided by _conn_ and transport */
 
-  _init_router(opt) {
-    return this.router = _mqtt_topic_router()}
-
   _init_dispatch(opt) {
+    this.constructor?._once_();
+    let router = this.router =
+      this._init_router?.(opt, this);
+
     let tgt ={
       __proto__: opt.on_mqtt_type || {}
-    , router: this._init_router(opt, this)};
+    , router};
 
-    tgt.mqtt_publish ||= tgt.router.invoke;
-    return _mqtt_dispatch(this, tgt)} }
+    tgt.mqtt_publish ||= router?.invoke;
+    return _mqtt_dispatch(this, tgt)}
 
+  static _aliases() {
+    return ' pub:publish sub:subscribe unsub:unsubscribe '}
 
- {
-  let p = MQTTBase.prototype;
-  Object.assign(p,{
-    MQTTError
-  , pub: p.publish
-  , sub: p.subscribe
-  , unsub: p.unsubscribe
-  , sub_topic: p.subscribe_topic
-  , unsub_topic: p.unsubscribe_topic
-  , shared_sub: p.shared_subscribe
-  , shared_unsub: p.shared_unsubscribe} );
+  static _once_(self=this) {
+    self._once_ = _=>0;
+    self.MQTTError = MQTTError;
+    let p = self.prototype;
+    for (let alias of self._aliases().split(/\s+/)) {
+      alias = alias.split(':');
+      let fn = alias[1] && p[alias[1]];
+      if (fn) {p[alias[0]] = fn;} } } }
 
-  /*
-    p.on_mqtt_type = {
-      mqtt_auth(pkt, ctx) ::
-      mqtt_connect(pkt, ctx) ::
-      mqtt_connack(pkt, ctx) ::
-      mqtt_disconnect(pkt, ctx) ::
-
-      mqtt_publish(pkt, ctx)
-      mqtt_subscribe(pkt, ctx) ::
-      mqtt_unsubscribe(pkt, ctx) ::
-
-      mqtt_pingreq(pkt, ctx) ::
-      mqtt_pingresp(pkt, ctx) ::
-    }
-  */}
+
+/*
+on_mqtt_type = {
+  mqtt_auth(pkt, ctx) ::
+  mqtt_connect(pkt, ctx) ::
+  mqtt_connack(pkt, ctx) ::
+  mqtt_disconnect(pkt, ctx) ::
+
+  mqtt_publish(pkt, ctx)
+  mqtt_subscribe(pkt, ctx) ::
+  mqtt_unsubscribe(pkt, ctx) ::
+
+  mqtt_pingreq(pkt, ctx) ::
+  mqtt_pingresp(pkt, ctx) ::
+}
+*/
 
 
-function _as_topics(pkt, ex) {
-  if ('string' === typeof pkt) {
-    return {topics:[pkt], ... ex}}
-  if (pkt[Symbol.iterator]) {
-    return {topics:[... pkt], ... ex}}
-  return ex ? {...pkt, ...ex} : pkt}
+const _prefix_topics = (topic_prefix, iterable) =>
+  Array.from(iterable, value =>(
+    value.trim // string
+      ? _prefix_topics(topic_prefix, value)
+      : topic_prefix + value) );
+
+function _as_topics(pkt, ex, topic_prefix) {
+  if (ex?.trim) {// string
+    topic_prefix = ex;
+    ex = null;}
+
+  pkt =(
+    pkt.trim // string
+      ? {topics:[pkt], ... ex}
+    : pkt[Symbol.iterator]
+      ? {topics:[... pkt], ... ex}
+    : ex ? {...pkt, ...ex}
+    : pkt);
+
+  if (topic_prefix) {
+    // particularly useful with shared queues, e.g.
+    //   topic_prefix = '$share/some-queue-name/'
+    pkt.topics = _prefix_topics(topic_prefix, pkt.topics);}
+  return pkt}
 
 
 async function _pub(self, pkt, pub_opt) {
@@ -1454,7 +1493,7 @@ class MQTTCore extends MQTTBase {
     return this}
 
   //log_conn(evt, arg, err_arg) ::
-  //console.info @ '[[u8-mqtt log: %s]]', evt, arg, err_arg
+  //  console.info @ '[[u8-mqtt log: %s]]', evt, arg, err_arg
 
   on_live(client, is_reconnect) {
     if (is_reconnect) {
@@ -1496,30 +1535,55 @@ class MQTTCore extends MQTTBase {
     return this}
 
 
-  
-
-
-
-
-
-
-
-
-
 
+  
+    
+      
+      
+      
+        
+          
 
+    
+      
+      
+        
+          
 
+    
+      
+        
+        
+          
 
+    
+      
+      
+        
 
+      
+      
 
   
-  with_tcp(port, hostname) {
-    if (!Number.isFinite(port)) {
-      ({port, hostname} = new URL(port));}
+    with_tcp(...opt) {
+      opt = this._conn_opt(opt);
+      console.log({opt});
+      return this._use_conn (() =>
+        this.with_stream(
+          node_net.connect(opt)) ) }
+
+    with_tls(...opt) {
+      opt = this._conn_opt(opt);
+      return this._use_conn (() =>
+        this.with_stream(
+          node_tls.connect(opt)) ) }
 
-    return this._use_conn (() =>
-      this.with_stream(
-        node_net.connect(port, hostname)) ) }
+    _conn_opt([a0, a1, a2]) {
+      // (port, hostname, options) or (url, options)
+      if (Number.isFinite(a0)) {
+        return {...a2, port: a0, host: a1}}
+      a0 = new URL(a0);
+      return {...a1, port: a0.port, host: a0.hostname}}
 
 
   with_stream(read_stream, write_stream) {
@@ -1565,13 +1629,15 @@ class MQTTCore extends MQTTBase {
 
     return this} }
 
-var version = "0.3.1";
+const version = '0.4.0';
 
 const MQTTClient_v4 = /* #__PURE__ */
-  MQTTCore.mqtt_ctx(4, mqtt_opts_v5);
+  with_topic_path_router(
+    MQTTCore.mqtt_ctx(4, mqtt_opts_v5) );
 
 const MQTTClient_v5 = /* #__PURE__ */
-  MQTTCore.mqtt_ctx(5, mqtt_opts_v5);
+  with_topic_path_router(
+    MQTTCore.mqtt_ctx(5, mqtt_opts_v5) );
 
 const mqtt_v4 = opt =>
   new MQTTClient_v4(opt);
@@ -1587,13 +1653,12 @@ exports.MQTTError = MQTTError;
 exports._mqtt_cmdid_dispatch = _mqtt_cmdid_dispatch;
 exports._mqtt_conn = _mqtt_conn;
 exports._mqtt_dispatch = _mqtt_dispatch;
-exports._mqtt_route_match_one = _mqtt_route_match_one;
-exports._mqtt_route_remove = _mqtt_route_remove;
-exports._mqtt_routes_iter = _mqtt_routes_iter;
-exports._mqtt_topic_router = _mqtt_topic_router;
 exports.ao_defer_v = ao_defer_v;
-exports["default"] = mqtt_v4;
+exports.as_topic_path = as_topic_path;
+exports.default = mqtt_v4;
 exports.mqtt_v4 = mqtt_v4;
 exports.mqtt_v5 = mqtt_v5;
 exports.version = version;
+exports.with_topic_path_router = with_topic_path_router;
+exports.with_topic_router = with_topic_router;
 //# sourceMappingURL=index.cjs.map