UNPKG

node-red-iot-mqtt-api

Version:

return right result to node-red

1,027 lines (972 loc) 74.1 kB
/** * Copyright JS Foundation and other contributors, http://js.foundation * * Licensed under the Apache License, Version 2.0 (the "License"); * you may not use this file except in compliance with the License. * You may obtain a copy of the License at * * http://www.apache.org/licenses/LICENSE-2.0 * * Unless required by applicable law or agreed to in writing, software * distributed under the License is distributed on an "AS IS" BASIS, * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. * See the License for the specific language governing permissions and * limitations under the License. **/ module.exports = function (RED) { "use strict"; var mqtt = require("mqtt"); var isUtf8 = require('is-utf8'); var HttpsProxyAgent = require('https-proxy-agent'); var url = require('url'); const crypto = require('crypto'); const knownMediaTypes = { "text/css": "string", "text/html": "string", "text/plain": "string", "text/html": "string", "application/json": "json", "application/octet-stream": "buffer", "application/pdf": "buffer", "application/x-gtar": "buffer", "application/x-gzip": "buffer", "application/x-tar": "buffer", "application/xml": "string", "application/zip": "buffer", "audio/aac": "buffer", "audio/ac3": "buffer", "audio/basic": "buffer", "audio/mp4": "buffer", "audio/ogg": "buffer", "image/bmp": "buffer", "image/gif": "buffer", "image/jpeg": "buffer", "image/tiff": "buffer", "image/png": "buffer", } const generateUUID = () => { if (typeof crypto === 'object') { if (typeof crypto.randomUUID === 'function') { return crypto.randomUUID() } if (typeof crypto.getRandomValues === 'function' && typeof Uint8Array === 'function') { const callback = (c) => { const num = Number(c) return (num ^ (crypto.getRandomValues(new Uint8Array(1))[0] & (15 >> (num / 4)))).toString(16) } return '10000000-1000-4000-8000-100000000000'.replace(/[018]/g, callback) } } let timestamp = new Date().getTime() let performanceNow = (typeof performance !== 'undefined' && performance.now && performance.now() * 1000) || 0 return 'xxxxxxxx-xxxx-4xxx-yxxx-xxxxxxxxxxxx'.replace(/[xy]/g, (c) => { let random = Math.random() * 16 if (timestamp > 0) { random = (timestamp + random) % 16 | 0 timestamp = Math.floor(timestamp / 16) } else { random = (performanceNow + random) % 16 | 0 performanceNow = Math.floor(performanceNow / 16) } return (c === 'x' ? random : (random & 0x3) | 0x8).toString(16) }) } //#region "Supporting functions" function matchTopic(ts, t) { if (ts == "#") { return true; } /* The following allows shared subscriptions (as in MQTT v5) http://docs.oasis-open.org/mqtt/mqtt/v5.0/cs02/mqtt-v5.0-cs02.html#_Toc514345522 4.8.2 describes shares like: $share/{ShareName}/{filter} $share is a literal string that marks the Topic Filter as being a Shared Subscription Topic Filter. {ShareName} is a character string that does not include "/", "+" or "#" {filter} The remainder of the string has the same syntax and semantics as a Topic Filter in a non-shared subscription. Refer to section 4.7. */ else if (ts.startsWith("$share")) { ts = ts.replace(/^\$share\/[^#+/]+\/(.*)/g, "$1"); } var re = new RegExp("^" + ts.replace(/([\[\]\?\(\)\\\\$\^\*\.|])/g, "\\$1").replace(/\+/g, "[^/]+").replace(/\/#$/, "(\/.*)?") + "$"); return re.test(t); } /** * Helper function for setting integer property values in the MQTT V5 properties object * @param {object} src Source object containing properties * @param {object} dst Destination object to set/add properties * @param {string} propName The property name to set in the Destination object * @param {integer} [minVal] The minimum value. If the src value is less than minVal, it will NOT be set in the destination * @param {integer} [maxVal] The maximum value. If the src value is greater than maxVal, it will NOT be set in the destination * @param {integer} [def] An optional default to set in the destination object if prop is NOT present in the soruce object */ function setIntProp(src, dst, propName, minVal, maxVal, def) { if (hasProperty(src, propName)) { var v = parseInt(src[propName]); if (isNaN(v)) return; if (minVal != null) { if (v < minVal) return; } if (maxVal != null) { if (v > maxVal) return; } dst[propName] = v; } else { if (def != undefined) dst[propName] = def; } } /** * Test a topic string is valid for subscription * @param {string} topic * @returns `true` if it is a valid topic */ function isValidSubscriptionTopic(topic) { return /^(#$|(\+|[^+#]*)(\/(\+|[^+#]*))*(\/(\+|#|[^+#]*))?$)/.test(topic); } /** * Test a topic string is valid for publishing * @param {string} topic * @returns `true` if it is a valid topic */ function isValidPublishTopic(topic) { return !/[\+#\b\f\n\r\t\v\0]/.test(topic); } /** * Helper function for setting string property values in the MQTT V5 properties object * @param {object} src Source object containing properties * @param {object} dst Destination object to set/add properties * @param {string} propName The property name to set in the Destination object * @param {string} [def] An optional default to set in the destination object if prop is NOT present in the soruce object */ function setStrProp(src, dst, propName, def) { if (src[propName] && typeof src[propName] == "string") { dst[propName] = src[propName]; } else { if (def != undefined) dst[propName] = def; } } /** * Helper function for setting boolean property values in the MQTT V5 properties object * @param {object} src Source object containing properties * @param {object} dst Destination object to set/add properties * @param {string} propName The property name to set in the Destination object * @param {boolean} [def] An optional default to set in the destination object if prop is NOT present in the soruce object */ function setBoolProp(src, dst, propName, def) { if (src[propName] != null) { if (src[propName] === "true" || src[propName] === true) { dst[propName] = true; } else if (src[propName] === "false" || src[propName] === false) { dst[propName] = false; } } else { if (def != undefined) dst[propName] = def; } } /** * Helper function for copying the MQTT v5 srcUserProperties object (parameter1) to the properties object (parameter2). * Any property in srcUserProperties that is NOT a key/string pair will be silently discarded. * NOTE: if no sutable properties are present, the userProperties object will NOT be added to the properties object * @param {object} srcUserProperties An object with key/value string pairs * @param {object} properties A properties object in which userProperties will be copied to */ function setUserProperties(srcUserProperties, properties) { if (srcUserProperties && typeof srcUserProperties == "object") { let _clone = {}; let count = 0; let keys = Object.keys(srcUserProperties); if (!keys || !keys.length) return null; keys.forEach(key => { let val = srcUserProperties[key]; if (typeof val == "string") { count++; _clone[key] = val; } }); if (count) properties.userProperties = _clone; } } /** * Helper function for copying the MQTT v5 buffer type properties * NOTE: if src[propName] is not a buffer, dst[propName] will NOT be assigned a value (unless def is set) * @param {object} src Source object containing properties * @param {object} dst Destination object to set/add properties * @param {string} propName The property name to set in the Destination object * @param {boolean} [def] An optional default to set in the destination object if prop is NOT present in the Source object */ function setBufferProp(src, dst, propName, def) { if (!dst) return; if (src && dst) { var buf = src[propName]; if (buf && typeof Buffer.isBuffer(buf)) { dst[propName] = Buffer.from(buf); } } else { if (def != undefined) dst[propName] = def; } } /** * Helper function for applying changes to an objects properties ONLY when the src object actually has the property. * This avoids setting a `dst` property null/undefined when the `src` object doesnt have the named property. * @param {object} src Source object containing properties * @param {object} dst Destination object to set property * @param {string} propName The property name to set in the Destination object * @param {boolean} force force the dst property to be updated/created even if src property is empty */ function setIfHasProperty(src, dst, propName, force) { if (src && dst && propName) { const ok = force || hasProperty(src, propName); if (ok) { dst[propName] = src[propName]; } } } /** * Helper function to test an object has a property * @param {object} obj Object to test * @param {string} propName Name of property to find * @returns true if object has property `propName` */ function hasProperty(obj, propName) { //JavaScript does not protect the property name hasOwnProperty //Object.prototype.hasOwnProperty.call is the recommended/safer test return Object.prototype.hasOwnProperty.call(obj, propName); } /** * Handle the payload / packet recieved in MQTT In and MQTT Sub nodes */ function subscriptionHandler(node, datatype, topic, payload, packet) { const v5 = node.brokerConn.options && node.brokerConn.options.protocolVersion == 5; var msg = { topic: topic, payload: null, qos: packet.qos, retain: packet.retain }; if (v5 && packet.properties) { setStrProp(packet.properties, msg, "responseTopic"); setBufferProp(packet.properties, msg, "correlationData"); setStrProp(packet.properties, msg, "contentType"); setIntProp(packet.properties, msg, "messageExpiryInterval", 0); setBoolProp(packet.properties, msg, "payloadFormatIndicator"); setStrProp(packet.properties, msg, "reasonString"); setUserProperties(packet.properties.userProperties, msg); } const v5isUtf8 = v5 ? msg.payloadFormatIndicator === true : null; const v5HasMediaType = v5 ? !!msg.contentType : null; const v5MediaTypeLC = v5 ? (msg.contentType + "").toLowerCase() : null; if (datatype === "buffer") { // payload = payload; } else if (datatype === "base64") { payload = payload.toString('base64'); } else if (datatype === "utf8") { payload = payload.toString('utf8'); } else if (datatype === "json") { if (v5isUtf8 || isUtf8(payload)) { try { payload = JSON.parse(payload.toString()); } catch (e) { node.error(RED._("mqtt.errors.invalid-json-parse"), { payload: payload, topic: topic, qos: packet.qos, retain: packet.retain }); return; } } else { node.error((RED._("mqtt.errors.invalid-json-string")), { payload: payload, topic: topic, qos: packet.qos, retain: packet.retain }); return; } } else { //"auto" (legacy) or "auto-detect" (new default) if (v5isUtf8 || v5HasMediaType) { const outputType = knownMediaTypes[v5MediaTypeLC] switch (outputType) { case "string": payload = payload.toString(); break; case "buffer": //no change break; case "json": try { //since v5 type states this should be JSON, parse it & error out if NOT JSON payload = payload.toString() const obj = JSON.parse(payload); if (datatype === "auto-detect") { payload = obj; //as mode is "auto-detect", return the parsed JSON } } catch (e) { node.error(RED._("mqtt.errors.invalid-json-parse"), { payload: payload, topic: topic, qos: packet.qos, retain: packet.retain }); return; } break; default: if (v5isUtf8 || isUtf8(payload)) { payload = payload.toString(); //auto String if (datatype === "auto-detect") { try { payload = JSON.parse(payload); //auto to parsed object (attempt) } catch (e) { /* mute error - it simply isnt JSON, just leave payload as a string */ } } } break; } } else if (isUtf8(payload)) { payload = payload.toString(); //auto String if (datatype === "auto-detect") { try { payload = JSON.parse(payload); } catch (e) { /* mute error - it simply isnt JSON, just leave payload as a string */ } } } //else { //leave as buffer //} } msg.payload = payload; if ((node.brokerConn.broker === "localhost") || (node.brokerConn.broker === "127.0.0.1")) { msg._topic = topic; } node.send(msg); } /** * Send an mqtt message to broker * @param {MQTTOutNode} node the owner node * @param {object} msg The msg to prepare for publishing * @param {function} done callback when done */ function doPublish(node, msg, done) { try { done = typeof done == "function" ? done : function noop() { }; let v5 = node.brokerConn.options && node.brokerConn.options.protocolVersion == 5; const bsp = (node.brokerConn && node.brokerConn.serverProperties) || {}; //Sanitise the `msg` object properties ready for publishing if (msg.qos) { msg.qos = parseInt(msg.qos); if ((msg.qos !== 0) && (msg.qos !== 1) && (msg.qos !== 2)) { msg.qos = null; } } if (node.topicCategory) { msg.topicCategory = node.topicCategory; } /* If node properties exists, override/set that to property in msg */ if (node.topic) { msg.topic = node.topic; } msg.qos = Number(node.qos || msg.qos || 0); msg.retain = node.retain || msg.retain || false; msg.retain = ((msg.retain === true) || (msg.retain === "true")) || false; if (node.topicCategory === "eventUp") { msg.eventName = node.eventName; } else if (node.topicCategory === "serviceInvoke") { msg.serviceName = node.serviceName; } if (v5) { if (node.userProperties) { msg.userProperties = node.userProperties; } if (node.responseTopic) { msg.responseTopic = node.responseTopic; } if (node.correlationData) { msg.correlationData = node.correlationData; } if (node.contentType) { msg.contentType = node.contentType; } if (node.messageExpiryInterval) { msg.messageExpiryInterval = node.messageExpiryInterval; } } if (hasProperty(msg, "payload")) { // send the message node.brokerConn.publish(msg, function (err) { if (err && err.warn) { node.warn(err); return; } done(err); }); } else { done(); } } catch (error) { done(error); } } function updateStatus(node, allNodes) { let setStatus = setStatusDisconnected if (node.connecting) { setStatus = setStatusConnecting } else if (node.connected) { setStatus = setStatusConnected } setStatus(node, allNodes) } function setStatusDisconnected(node, allNodes) { if (allNodes) { for (var id in node.users) { if (hasProperty(node.users, id)) { node.users[id].status({ fill: "red", shape: "ring", text: "node-red:common.status.disconnected" }); } } } else { node.status({ fill: "red", shape: "ring", text: "node-red:common.status.disconnected" }); } } function setStatusConnecting(node, allNodes) { if (allNodes) { for (var id in node.users) { if (hasProperty(node.users, id)) { node.users[id].status({ fill: "yellow", shape: "ring", text: "node-red:common.status.connecting" }); } } } else { node.status({ fill: "yellow", shape: "ring", text: "node-red:common.status.connecting" }); } } function setStatusConnected(node, allNodes) { if (allNodes) { for (var id in node.users) { if (hasProperty(node.users, id)) { node.users[id].status({ fill: "green", shape: "dot", text: "node-red:common.status.connected" }); } } } else { node.status({ fill: "green", shape: "dot", text: "node-red:common.status.connected" }); } } function handleConnectAction(node, msg, done) { let actionData = typeof msg.broker === 'object' ? msg.broker : null; if (node.brokerConn.canConnect()) { // Not currently connected/connecting - trigger the connect if (actionData) { node.brokerConn.setOptions(actionData); } node.brokerConn.connect(function () { done(); }); } else { // Already Connected/Connecting if (!actionData) { // All is good - already connected and no broker override provided done() } else if (actionData.force) { // The force flag tells us to cycle the connection. node.brokerConn.disconnect(function () { node.brokerConn.setOptions(actionData); node.brokerConn.connect(function () { done(); }); }) } else { // Without force flag, we will refuse to cycle an active connection done(new Error(RED._('mqtt.errors.invalid-action-alreadyconnected'))); } } } function handleDisconnectAction(node, done) { node.brokerConn.disconnect(function () { done(); }); } //#endregion "Supporting functions" //#region "Broker node" function MQTTBrokerNode(n) { RED.nodes.createNode(this, n); const node = this; node.users = {}; // Config node state node.brokerurl = ""; node.connected = false; node.connecting = false; node.closing = false; node.options = {}; node.queue = []; node.subscriptions = {}; node.clientListeners = [] node.deviceName = n.deviceName; node.productKey = n.productKey; /** @type {mqtt.MqttClient}*/ this.client; node.setOptions = function (opts, init) { if (!opts || typeof opts !== "object") { return; //nothing to change, simply return } //apply property changes (only if the property exists in the opts object) setIfHasProperty(opts, node, "url", init); setIfHasProperty(opts, node, "broker", init); setIfHasProperty(opts, node, "port", init); setIfHasProperty({ ...opts, clientid: `${n.productKey}_${n.deviceName}_${n.model}` }, node, "clientid", init); setIfHasProperty(opts, node, "model", init); setIfHasProperty(opts, node, "autoConnect", init); setIfHasProperty(opts, node, "usetls", init); setIfHasProperty(opts, node, "verifyservercert", init); setIfHasProperty(opts, node, "compatmode", init); setIfHasProperty(opts, node, "protocolVersion", init); setIfHasProperty(opts, node, "keepalive", init); setIfHasProperty(opts, node, "cleansession", init); setIfHasProperty(opts, node, "topicAliasMaximum", init); setIfHasProperty(opts, node, "maximumPacketSize", init); setIfHasProperty(opts, node, "receiveMaximum", init); //https://docs.oasis-open.org/mqtt/mqtt/v5.0/os/mqtt-v5.0-os.html#_Toc3901116 if (hasProperty(opts, "userProperties")) { node.userProperties = opts.userProperties; } else if (hasProperty(opts, "userProps")) { node.userProperties = opts.userProps; } if (hasProperty(opts, "sessionExpiry")) { node.sessionExpiryInterval = opts.sessionExpiry; } else if (hasProperty(opts, "sessionExpiryInterval")) { node.sessionExpiryInterval = opts.sessionExpiryInterval } function createLWT(topic, payload, qos, retain, v5opts, v5SubPropName) { let message = undefined; if (topic) { message = { topic: topic, payload: payload || "", qos: Number(qos || 0), retain: retain == "true" || retain === true, } if (v5opts) { let v5Properties = message; if (v5SubPropName) { v5Properties = message[v5SubPropName] = {}; } //re-align local prop name to mqttjs std if (hasProperty(v5opts, "respTopic")) { v5opts.responseTopic = v5opts.respTopic; } if (hasProperty(v5opts, "correl")) { v5opts.correlationData = v5opts.correl; } if (hasProperty(v5opts, "expiry")) { v5opts.messageExpiryInterval = v5opts.expiry; } if (hasProperty(v5opts, "delay")) { v5opts.willDelayInterval = v5opts.delay; } if (hasProperty(v5opts, "userProps")) { v5opts.userProperties = v5opts.userProps; } //setup v5 properties if (typeof v5opts.userProperties == "string" && /^ *{/.test(v5opts.userProperties)) { try { setUserProperties(JSON.parse(v5opts.userProps), v5Properties); } catch (err) { } } else if (typeof v5opts.userProperties == "object") { setUserProperties(v5opts.userProperties, v5Properties); } setStrProp(v5opts, v5Properties, "contentType"); setStrProp(v5opts, v5Properties, "responseTopic"); setBufferProp(v5opts, v5Properties, "correlationData"); setIntProp(v5opts, v5Properties, "messageExpiryInterval"); setIntProp(v5opts, v5Properties, "willDelayInterval"); } } return message; } if (init) { if (hasProperty(opts, "birthTopic")) { node.birthMessage = createLWT(opts.birthTopic, opts.birthPayload, opts.birthQos, opts.birthRetain, opts.birthMsg, ""); }; if (hasProperty(opts, "closeTopic")) { node.closeMessage = createLWT(opts.closeTopic, opts.closePayload, opts.closeQos, opts.closeRetain, opts.closeMsg, ""); }; if (hasProperty(opts, "willTopic")) { //will v5 properties must be set in the "properties" sub object node.options.will = createLWT(opts.willTopic, opts.willPayload, opts.willQos, opts.willRetain, opts.willMsg, "properties"); }; } else { //update options if (hasProperty(opts, "birth")) { if (typeof opts.birth !== "object") { opts.birth = {}; } node.birthMessage = createLWT(opts.birth.topic, opts.birth.payload, opts.birth.qos, opts.birth.retain, opts.birth.properties, ""); } if (hasProperty(opts, "close")) { if (typeof opts.close !== "object") { opts.close = {}; } node.closeMessage = createLWT(opts.close.topic, opts.close.payload, opts.close.qos, opts.close.retain, opts.close.properties, ""); } if (hasProperty(opts, "will")) { if (typeof opts.will !== "object") { opts.will = {}; } //will v5 properties must be set in the "properties" sub object node.options.will = createLWT(opts.will.topic, opts.will.payload, opts.will.qos, opts.will.retain, opts.will.properties, "properties"); } } // mqttClientId: productKey_deviceName_model // node.options.clientId = `${n.productKey}_${n.deviceName}_${n.model}`; node.username = n.deviceName // mqttPassword: md5(产品密钥,mqttClientId) node.password = crypto.createHash('md5').update(`${n.secrete}${node.clientid}`).digest('hex'); // if (node.credentials) { // node.username = node.credentials.user; // node.password = node.credentials.password; // } if (!init & hasProperty(opts, "username")) { node.username = opts.username; }; if (!init & hasProperty(opts, "password")) { node.password = opts.password; }; // If the config node is missing certain options (it was probably deployed prior to an update to the node code), // select/generate sensible options for the new fields if (typeof node.usetls === 'undefined') { node.usetls = false; } if (typeof node.verifyservercert === 'undefined') { node.verifyservercert = false; } if (typeof node.keepalive === 'undefined') { node.keepalive = 60; } else if (typeof node.keepalive === 'string') { node.keepalive = Number(node.keepalive); } if (typeof node.cleansession === 'undefined') { node.cleansession = true; } //use url or build a url from usetls://broker:port if (node.url && node.brokerurl !== node.url) { node.brokerurl = node.url; } else { // if the broker is ws:// or wss:// or tcp:// if (node.broker.indexOf("://") > -1) { node.brokerurl = node.broker; // Only for ws or wss, check if proxy env var for additional configuration if (node.brokerurl.indexOf("wss://") > -1 || node.brokerurl.indexOf("ws://") > -1) { // check if proxy is set in env let prox, noprox, noproxy; if (process.env.http_proxy) { prox = process.env.http_proxy; } if (process.env.HTTP_PROXY) { prox = process.env.HTTP_PROXY; } if (process.env.no_proxy) { noprox = process.env.no_proxy.split(","); } if (process.env.NO_PROXY) { noprox = process.env.NO_PROXY.split(","); } if (noprox) { for (var i = 0; i < noprox.length; i += 1) { if (node.brokerurl.indexOf(noprox[i].trim()) !== -1) { noproxy = true; } } } if (prox && !noproxy) { var parsedUrl = url.parse(node.brokerurl); var proxyOpts = url.parse(prox); // true for wss proxyOpts.secureEndpoint = parsedUrl.protocol ? parsedUrl.protocol === 'wss:' : true; // Set Agent for wsOption in MQTT var agent = new HttpsProxyAgent(proxyOpts); node.options.wsOptions = { agent: agent }; } } } else { // construct the std mqtt:// url if (node.usetls) { node.brokerurl = "mqtts://"; } else { node.brokerurl = "mqtt://"; } if (node.broker !== "") { //Check for an IPv6 address if (/(?:^|(?<=\s))(([0-9a-fA-F]{1,4}:){7,7}[0-9a-fA-F]{1,4}|([0-9a-fA-F]{1,4}:){1,7}:|([0-9a-fA-F]{1,4}:){1,6}:[0-9a-fA-F]{1,4}|([0-9a-fA-F]{1,4}:){1,5}(:[0-9a-fA-F]{1,4}){1,2}|([0-9a-fA-F]{1,4}:){1,4}(:[0-9a-fA-F]{1,4}){1,3}|([0-9a-fA-F]{1,4}:){1,3}(:[0-9a-fA-F]{1,4}){1,4}|([0-9a-fA-F]{1,4}:){1,2}(:[0-9a-fA-F]{1,4}){1,5}|[0-9a-fA-F]{1,4}:((:[0-9a-fA-F]{1,4}){1,6})|:((:[0-9a-fA-F]{1,4}){1,7}|:)|fe80:(:[0-9a-fA-F]{0,4}){0,4}%[0-9a-zA-Z]{1,}|::(ffff(:0{1,4}){0,1}:){0,1}((25[0-5]|(2[0-4]|1{0,1}[0-9]){0,1}[0-9])\.){3,3}(25[0-5]|(2[0-4]|1{0,1}[0-9]){0,1}[0-9])|([0-9a-fA-F]{1,4}:){1,4}:((25[0-5]|(2[0-4]|1{0,1}[0-9]){0,1}[0-9])\.){3,3}(25[0-5]|(2[0-4]|1{0,1}[0-9]){0,1}[0-9]))(?=\s|$)/.test(node.broker)) { node.brokerurl = node.brokerurl + "[" + node.broker + "]:"; } else { node.brokerurl = node.brokerurl + node.broker + ":"; } // port now defaults to 1883 if unset. if (!node.port) { node.brokerurl = node.brokerurl + "1883"; } else { node.brokerurl = node.brokerurl + node.port; } } else { node.brokerurl = node.brokerurl + "localhost:1883"; } } } // Ensure cleansession set if clientid not supplied if (!node.cleansession && !node.model) { node.cleansession = true; node.warn(RED._("mqtt.errors.nonclean-missingclientid")); } // Build options for passing to the MQTT.js API node.options.username = node.username; node.options.password = node.password; node.options.keepalive = node.keepalive; node.options.clean = node.cleansession; node.options.clientId = node.clientid || 'nodered_' + RED.util.generateId(); node.options.reconnectPeriod = RED.settings.mqttReconnectTime || 5000; delete node.options.protocolId; //V4+ default delete node.options.protocolVersion; //V4 default delete node.options.properties;//V5 only if (node.compatmode == "true" || node.compatmode === true || node.protocolVersion == 3) { node.options.protocolId = 'MQIsdp';//V3 compat only node.options.protocolVersion = 3; } else if (node.protocolVersion == 5) { delete node.options.protocolId; node.options.protocolVersion = 5; node.options.properties = {}; node.options.properties.requestResponseInformation = true; node.options.properties.requestProblemInformation = true; if (node.userProperties && /^ *{/.test(node.userProperties)) { try { setUserProperties(JSON.parse(node.userProperties), node.options.properties); } catch (err) { } } if (node.sessionExpiryInterval && node.sessionExpiryInterval !== "0") { setIntProp(node, node.options.properties, "sessionExpiryInterval"); } } if (node.usetls && n.tls) { var tlsNode = RED.nodes.getNode(n.tls); if (tlsNode) { tlsNode.addTLSOptions(node.options); } } // If there's no rejectUnauthorized already, then this could be an // old config where this option was provided on the broker node and // not the tls node if (typeof node.options.rejectUnauthorized === 'undefined') { node.options.rejectUnauthorized = (node.verifyservercert == "true" || node.verifyservercert === true); } } node.v5 = () => node.options && node.options.protocolVersion == 5 node.subscriptionIdentifiersAvailable = () => node.v5() && node.serverProperties && node.serverProperties.subscriptionIdentifiersAvailable n.autoConnect = n.autoConnect === "false" || n.autoConnect === false ? false : true; node.setOptions(n, true); // Define functions called by MQTT in and out nodes node.register = function (mqttNode) { node.users[mqttNode.id] = mqttNode; if (Object.keys(node.users).length === 1) { if (node.autoConnect) { node.connect(); //update nodes status setTimeout(function () { updateStatus(node, true) }, 1) } } }; node.deregister = function (mqttNode, done, autoDisconnect) { delete node.users[mqttNode.id]; if (autoDisconnect && !node.closing && node.connected && Object.keys(node.users).length === 0) { node.disconnect(done); } else { done(); } }; node.canConnect = function () { return !node.connected && !node.connecting; } node.connect = function (callback) { if (node.canConnect()) { node.closing = false; node.connecting = true; setStatusConnecting(node, true); try { node.serverProperties = {}; if (node.client) { //belt and braces to avoid left over clients node.client.end(true); node._clientRemoveListeners(); } node.client = mqtt.connect(node.brokerurl, node.options); node.client.setMaxListeners(0); let callbackDone = false; //prevent re-connects causing node._clientOn('connect' firing callback multiple times // Register successful connect or reconnect handler node._clientOn('connect', function (connack) { node.closing = false; node.connecting = false; node.connected = true; if (!callbackDone && typeof callback == "function") { callback(); } callbackDone = true; node.topicAliases = {}; node.log(RED._("mqtt.state.connected", { broker: (node.clientid ? node.clientid + "@" : "") + node.brokerurl })); if (node.options.protocolVersion == 5 && connack && hasProperty(connack, "properties")) { if (typeof connack.properties == "object") { //clean & assign all props sent from server. setIntProp(connack.properties, node.serverProperties, "topicAliasMaximum", 0); setIntProp(connack.properties, node.serverProperties, "receiveMaximum", 0); setIntProp(connack.properties, node.serverProperties, "sessionExpiryInterval", 0, 0xFFFFFFFF); setIntProp(connack.properties, node.serverProperties, "maximumQoS", 0, 2); setBoolProp(connack.properties, node.serverProperties, "retainAvailable", true); setBoolProp(connack.properties, node.serverProperties, "wildcardSubscriptionAvailable", true); setBoolProp(connack.properties, node.serverProperties, "subscriptionIdentifiersAvailable", true); setBoolProp(connack.properties, node.serverProperties, "sharedSubscriptionAvailable"); setIntProp(connack.properties, node.serverProperties, "maximumPacketSize", 0); setIntProp(connack.properties, node.serverProperties, "serverKeepAlive"); setStrProp(connack.properties, node.serverProperties, "responseInformation"); setStrProp(connack.properties, node.serverProperties, "serverReference"); setStrProp(connack.properties, node.serverProperties, "assignedClientIdentifier"); setStrProp(connack.properties, node.serverProperties, "reasonString"); setUserProperties(connack.properties, node.serverProperties); } } setStatusConnected(node, true); // Remove any existing listeners before resubscribing to avoid duplicates in the event of a re-connection node._clientRemoveListeners('message'); // Re-subscribe to stored topics for (var s in node.subscriptions) { if (node.subscriptions.hasOwnProperty(s)) { let topic = s; let qos = 0; let _options = {}; for (var r in node.subscriptions[s]) { if (node.subscriptions[s].hasOwnProperty(r)) { qos = Math.max(qos, node.subscriptions[s][r].qos); _options = node.subscriptions[s][r].options; node._clientOn('message', node.subscriptions[s][r].handler); } } _options.qos = _options.qos || qos; node.client.subscribe(topic, _options); } } // Send any birth message if (node.birthMessage) { setTimeout(() => { node.publish(node.birthMessage); }, 1); } }); node._clientOn("reconnect", function () { setStatusConnecting(node, true); }); //Broker Disconnect - V5 event node._clientOn("disconnect", function (packet) { //Emitted after receiving disconnect packet from broker. MQTT 5.0 feature. const rc = (packet && packet.properties && packet.reasonCode) || packet.reasonCode; const rs = packet && packet.properties && packet.properties.reasonString || ""; const details = { broker: (node.clientid ? node.clientid + "@" : "") + node.brokerurl, reasonCode: rc, reasonString: rs } node.connected = false; node.log(RED._("mqtt.state.broker-disconnected", details)); setStatusDisconnected(node, true); }); // Register disconnect handlers node._clientOn('close', function () { if (node.connected) { node.connected = false; node.log(RED._("mqtt.state.disconnected", { broker: (node.clientid ? node.clientid + "@" : "") + node.brokerurl })); setStatusDisconnected(node, true); } else if (node.connecting) { node.log(RED._("mqtt.state.connect-failed", { broker: (node.clientid ? node.clientid + "@" : "") + node.brokerurl })); } }); // Register connect error handler // The client's own reconnect logic will take care of errors node._clientOn('error', function (error) { node.log(RED._("mqtt.state.error-connect", { broker: (node.clientid ? node.clientid + "@" : "") + node.brokerurl, error: error.message })); }); } catch (err) { console.log(err); } } }; node.disconnect = function (callback) { const _callback = function () { if (node.connected || node.connecting) { setStatusDisconnected(node, true); } if (node.client) { node._clientRemoveListeners(); } node.connecting = false; node.connected = false; callback && typeof callback == "function" && callback(); }; if (!node.client) { return _callback(); } if (node.closing) { return _callback(); } let waitEnd = (client, ms) => { return new Promise((resolve, reject) => { node.closing = true; if (!client) { resolve(); } else { const t = setTimeout(() => { //clean end() has exceeded WAIT_END, lets force end! client && client.end(true); reject(); }, ms); client.end(() => { clearTimeout(t); resolve() }); } }); }; if (node.connected && node.closeMessage) { node.publish(node.closeMessage, function (err) { waitEnd(node.client, 2000).then(() => { _callback(); }).catch((e) => { _callback(); }) }); } else { waitEnd(node.client, 2000).then(() => { _callback(); }).catch((e) => { _callback(); }) } } node.subscriptionIds = {}; node.subid = 1; node.subscribe = function (topic, options, callback, ref) { ref = ref || 0; var qos; if (typeof options == "object") { qos = options.qos; } else { qos = options; options = {}; } options.qos = qos; if (!node.subscriptionIds[topic]) { node.subscriptionIds[topic] = node.subid++; } options.properties = options.properties || {}; options.properties.subscriptionIdentifier = node.subscriptionIds[topic]; node.subscriptions[topic] = node.subscriptions[topic] || {}; var sub = { topic: topic, qos: qos, options: options, handler: function (mtopic, mpayload, mpacket) { if (mpacket.properties && options.properties && mpacket.properties.subscriptionIdentifier && options.properties.subscriptionIdentifier && (mpacket.properties.subscriptionIdentifier !== options.properties.subscriptionIdentifier)) { //do nothing as subscriptionIdentifier does not match } else if (matchTopic(topic, mtopic)) { callback(mtopic, mpayload, mpacket); } }, ref: ref }; node.subscriptions[topic][ref] = sub; if (node.connected) { const subIdsAvailable = node.subscriptionIdentifiersAvailable() node._clientOn('message', sub.handler); // if the broker doesn't support subscription identifiers (e.g. AWS core), then don't send them if (options.properties && options.properties.subscriptionIdentifier && subIdsAvailable !== true) { delete options.properties.subscriptionIdentifier } node.client.subscribe(topic, options); } }; node.unsubscribe = function (topic, ref, removed) { ref = ref || 0; var sub = node.subscriptions[topic]; if (sub) { if (sub[ref]) { if (node.client) { node._clientRemoveListeners('message', sub[ref].handler); } delete sub[ref]; } //TODO: Review. The `if(removed)` was commented out to always delete and remove subscriptions. // if we dont then property changes dont get applied and old subs still trigger //if (removed) { if (Object.keys(sub).length === 0) { delete node.subscriptions[topic]; delete node.subscriptionIds[topic]; if (node.connected) { node.client.unsubscribe(topic); } } //} } }; node.topicAliases = {}; node.publish = function (msg, done) { if (node.connected) { if (msg.payload === null || msg.payload === undefined) { msg.payload = ""; } else if (!Buffer.isBuffer(msg.payload)) { if (typeof msg.payload === "object") { msg.payload = JSON.stringify(msg.payload); } else if (typeof msg.payload !== "string") { msg.payload = "" + msg.payload; } } var options = {