mirror of
https://github.com/Koenkk/zigbee2mqtt.git
synced 2024-11-17 02:48:31 -07:00
304 lines
12 KiB
JavaScript
304 lines
12 KiB
JavaScript
const ZigbeeHerdsman = require('zigbee-herdsman');
|
|
const logger = require('./util/logger');
|
|
const settings = require('./util/settings');
|
|
const data = require('./util/data');
|
|
const assert = require('assert');
|
|
const utils = require('./util/utils');
|
|
const events = require('events');
|
|
const objectAssignDeep = require('object-assign-deep');
|
|
const zigbeeHerdsmanConverters = require('zigbee-herdsman-converters');
|
|
const stringify = require('json-stable-stringify');
|
|
|
|
const endpointNames = utils.getEndpointNames();
|
|
const keyEndpointByNumber = new RegExp(`.*/([0-9]*)$`);
|
|
|
|
class Zigbee extends events.EventEmitter {
|
|
constructor() {
|
|
super();
|
|
this.acceptJoiningDeviceHandler = this.acceptJoiningDeviceHandler.bind(this);
|
|
}
|
|
|
|
async start() {
|
|
logger.info(`Starting zigbee-herdsman...`);
|
|
const herdsmanSettings = {
|
|
network: {
|
|
panID: settings.get().advanced.pan_id,
|
|
extendedPanID: settings.get().advanced.ext_pan_id,
|
|
channelList: [settings.get().advanced.channel],
|
|
networkKey: settings.get().advanced.network_key,
|
|
},
|
|
databasePath: data.joinPath('database.db'),
|
|
databaseBackupPath: data.joinPath('database.db.backup'),
|
|
backupPath: data.joinPath('coordinator_backup.json'),
|
|
serialPort: {
|
|
baudRate: settings.get().advanced.baudrate,
|
|
rtscts: settings.get().advanced.rtscts,
|
|
path: settings.get().serial.port,
|
|
adapter: settings.get().serial.adapter,
|
|
},
|
|
adapter: {
|
|
concurrent: settings.get().advanced.adapter_concurrent,
|
|
},
|
|
};
|
|
|
|
const herdsmanSettingsLog = objectAssignDeep.noMutate(herdsmanSettings);
|
|
herdsmanSettingsLog.network.networkKey = 'HIDDEN';
|
|
logger.debug(`Using zigbee-herdsman with settings: '${stringify(herdsmanSettingsLog)}'`);
|
|
|
|
if (herdsmanSettings.network.networkKey === 'GENERATE') {
|
|
const newKey = Array.from({length: 16}, () => Math.floor(Math.random() * 255));
|
|
settings.set(['advanced', 'network_key'], newKey);
|
|
herdsmanSettings.network.networkKey = newKey;
|
|
}
|
|
|
|
try {
|
|
herdsmanSettings.acceptJoiningDeviceHandler = this.acceptJoiningDeviceHandler;
|
|
this.herdsman = new ZigbeeHerdsman.Controller(herdsmanSettings);
|
|
await this.herdsman.start();
|
|
} catch (error) {
|
|
logger.error(`Error while starting zigbee-herdsman`);
|
|
throw error;
|
|
}
|
|
|
|
this.herdsman.on('adapterDisconnected', () => this.emit('adapterDisconnected'));
|
|
this.herdsman.on('deviceAnnounce', (data) => this.emit('event', 'deviceAnnounce', data));
|
|
this.herdsman.on('deviceInterview', (data) => this.emit('event', 'deviceInterview', data));
|
|
this.herdsman.on('deviceJoined', (data) => this.emit('event', 'deviceJoined', data));
|
|
this.herdsman.on('deviceLeave', (data) => this.emit('event', 'deviceLeave', data));
|
|
this.herdsman.on('message', (data) => this.emit('event', 'message', data));
|
|
|
|
logger.info('zigbee-herdsman started');
|
|
logger.info(`Coordinator firmware version: '${stringify(await this.getCoordinatorVersion())}'`);
|
|
logger.debug(`Zigbee network parameters: ${stringify(await this.herdsman.getNetworkParameters())}`);
|
|
|
|
for (const device of this.getClients()) {
|
|
// If a passlist is used, all other device will be removed from the network.
|
|
const passlist = settings.get().passlist.concat(settings.get().whitelist);
|
|
const blocklist = settings.get().blocklist.concat(settings.get().ban);
|
|
if (passlist.length > 0) {
|
|
if (!passlist.includes(device.ieeeAddr)) {
|
|
logger.warn(`Device which is not on passlist connected (${device.ieeeAddr}), removing...`);
|
|
device.removeFromNetwork();
|
|
}
|
|
} else if (blocklist.includes(device.ieeeAddr)) {
|
|
logger.warn(`Device on blocklist is connected (${device.ieeeAddr}), removing...`);
|
|
device.removeFromNetwork();
|
|
}
|
|
}
|
|
|
|
// Check if we have to turn off the led
|
|
if (settings.get().serial.disable_led) {
|
|
await this.herdsman.setLED(false);
|
|
}
|
|
|
|
// Check if we have to set a transmit power
|
|
if (settings.get().experimental.hasOwnProperty('transmit_power')) {
|
|
const transmitPower = settings.get().experimental.transmit_power;
|
|
await this.herdsman.setTransmitPower(transmitPower);
|
|
logger.info(`Set transmit power to '${transmitPower}'`);
|
|
}
|
|
}
|
|
|
|
async getCoordinatorVersion() {
|
|
return this.herdsman.getCoordinatorVersion();
|
|
}
|
|
|
|
async getNetworkParameters() {
|
|
return this.herdsman.getNetworkParameters();
|
|
}
|
|
|
|
async reset(type) {
|
|
await this.herdsman.reset(type);
|
|
}
|
|
|
|
async stop() {
|
|
await this.herdsman.stop();
|
|
logger.info('zigbee-herdsman stopped');
|
|
}
|
|
|
|
async permitJoin(permit) {
|
|
permit ?
|
|
logger.info('Zigbee: allowing new devices to join.') :
|
|
logger.info('Zigbee: disabling joining new devices.');
|
|
|
|
await this.herdsman.permitJoin(permit);
|
|
}
|
|
|
|
async getPermitJoin() {
|
|
return this.herdsman.getPermitJoin();
|
|
}
|
|
|
|
getClients() {
|
|
return this.herdsman.getDevices().filter((device) => device.type !== 'Coordinator');
|
|
}
|
|
|
|
getDevices() {
|
|
return this.herdsman.getDevices();
|
|
}
|
|
|
|
getDeviceByIeeeAddr(ieeeAddr) {
|
|
return this.herdsman.getDeviceByIeeeAddr(ieeeAddr);
|
|
}
|
|
|
|
getDeviceByNetworkAddress(networkAddress) {
|
|
return this.herdsman.getDeviceByNetworkAddress(networkAddress);
|
|
}
|
|
|
|
getDevicesByType(type) {
|
|
return this.herdsman.getDevicesByType(type);
|
|
}
|
|
|
|
/**
|
|
* @param {string} key
|
|
* @return {object} {
|
|
* type: device | coordinator
|
|
* device|group: zigbee-herdsman entity
|
|
* endpoint: selected endpoint (only if type === device)
|
|
* settings: from configuration.yaml
|
|
* name: name of the entity
|
|
* definition: zigbee-herdsman-converters definition (only if type === device)
|
|
* }
|
|
*/
|
|
resolveEntity(key) {
|
|
assert(
|
|
typeof key === 'string' || typeof key === 'number' ||
|
|
key.constructor.name === 'Device' || key.constructor.name === 'Group',
|
|
`Wrong type '${typeof key}'`,
|
|
);
|
|
|
|
const deviceOptions = settings.get().device_options;
|
|
if (typeof key === 'string' || typeof key === 'number') {
|
|
if (typeof key === 'number') {
|
|
key = key.toString();
|
|
}
|
|
|
|
if (typeof key === 'string' && key.toLowerCase() === 'coordinator') {
|
|
const coordinator = this.getDevicesByType('Coordinator')[0];
|
|
return {
|
|
type: 'device',
|
|
device: coordinator,
|
|
endpoint: coordinator.getEndpoint(1),
|
|
settings: {friendlyName: 'Coordinator'},
|
|
name: 'Coordinator',
|
|
};
|
|
}
|
|
|
|
let endpointKey = endpointNames.find((p) => key.endsWith(`/${p}`));
|
|
const endpointByNumber = key.match(keyEndpointByNumber);
|
|
if (!endpointKey && endpointByNumber) {
|
|
endpointKey = Number(endpointByNumber[1]);
|
|
}
|
|
if (endpointKey) {
|
|
key = key.replace(`/${endpointKey}`, '');
|
|
}
|
|
|
|
const entity = settings.getEntity(key);
|
|
if (!entity) {
|
|
return null;
|
|
} else if (entity.type === 'device') {
|
|
const device = this.getDeviceByIeeeAddr(entity.ID);
|
|
if (!device) {
|
|
return null;
|
|
}
|
|
|
|
const definition = zigbeeHerdsmanConverters.findByDevice(device);
|
|
const endpoints = definition && definition.endpoint ? definition.endpoint(device) : null;
|
|
let endpoint;
|
|
if (endpointKey) {
|
|
if (endpointByNumber) {
|
|
endpoint = device.getEndpoint(endpointKey);
|
|
} else {
|
|
assert(definition != null, `Endpoint name '${endpointKey}' is given but device is unsupported`);
|
|
assert(endpoints != null, `Endpoint name '${endpointKey}' is given but no endpoints defined`);
|
|
const endpointID = endpoints[endpointKey];
|
|
assert(endpointID, `Endpoint name '${endpointKey}' is given but device has no such endpoint`);
|
|
endpoint = device.getEndpoint(endpointID);
|
|
}
|
|
} else if (endpoints && endpoints['default']) {
|
|
endpoint = device.getEndpoint(endpoints['default']);
|
|
} else {
|
|
endpoint = device.endpoints[0];
|
|
}
|
|
|
|
return {
|
|
type: 'device', device, endpoint, settings: {...deviceOptions, ...entity},
|
|
name: entity.friendlyName, definition,
|
|
};
|
|
} else {
|
|
let group = this.getGroupByID(entity.ID);
|
|
if (!group) group = this.createGroup(entity.ID);
|
|
return {type: 'group', group, settings: {...deviceOptions, ...entity}, name: entity.friendlyName};
|
|
}
|
|
} else if (key.constructor.name === 'Device') {
|
|
const setting = settings.getEntity(key.ieeeAddr);
|
|
return {
|
|
type: 'device',
|
|
device: key,
|
|
endpoint: key.endpoints[0],
|
|
settings: {...deviceOptions, ...(setting || {})},
|
|
name: setting ? setting.friendlyName : (key.type === 'Coordinator' ? 'Coordinator' : key.ieeeAddr),
|
|
definition: zigbeeHerdsmanConverters.findByDevice(key),
|
|
};
|
|
} else { // Group
|
|
const setting = settings.getEntity(key.groupID);
|
|
return {
|
|
type: 'group',
|
|
group: key,
|
|
settings: {...deviceOptions, ...(setting || {})},
|
|
name: setting ? setting.friendlyName : key.groupID,
|
|
};
|
|
}
|
|
}
|
|
|
|
getGroupByID(ID) {
|
|
return this.herdsman.getGroupByID(ID);
|
|
}
|
|
|
|
getGroups() {
|
|
return this.herdsman.getGroups();
|
|
}
|
|
|
|
createGroup(groupID) {
|
|
return this.herdsman.createGroup(groupID);
|
|
}
|
|
|
|
acceptJoiningDeviceHandler(ieeeAddr) {
|
|
// If passlist is set, all devices not on passlist will be rejected to join the network
|
|
const passlist = settings.get().passlist.concat(settings.get().whitelist);
|
|
const blocklist = settings.get().blocklist.concat(settings.get().ban);
|
|
if (passlist.length > 0) {
|
|
if (passlist.includes(ieeeAddr)) {
|
|
logger.info(`Accepting joining device which is on passlist '${ieeeAddr}'`);
|
|
return true;
|
|
} else {
|
|
logger.info(`Rejecting joining not in passlist device '${ieeeAddr}'`);
|
|
return false;
|
|
}
|
|
} else if (blocklist.length > 0) {
|
|
if (blocklist.includes(ieeeAddr)) {
|
|
logger.info(`Rejecting joining device which is on blocklist '${ieeeAddr}'`);
|
|
return false;
|
|
} else {
|
|
logger.info(`Accepting joining not in blocklist device '${ieeeAddr}'`);
|
|
return true;
|
|
}
|
|
} else {
|
|
return true;
|
|
}
|
|
}
|
|
|
|
async touchlinkFactoryResetFirst() {
|
|
return this.herdsman.touchlinkFactoryResetFirst();
|
|
}
|
|
|
|
async touchlinkFactoryReset(ieeeAddr, channel) {
|
|
return this.herdsman.touchlinkFactoryReset(ieeeAddr, channel);
|
|
}
|
|
|
|
async touchlinkScan() {
|
|
return this.herdsman.touchlinkScan();
|
|
}
|
|
}
|
|
|
|
module.exports = Zigbee;
|