infoFromHostname now returns type of device

kingkong123 · Apr 7, 2017 · 08c5f2d · 08c5f2d
1 parent 05c7bf6
commit 08c5f2d
Show file tree

Hide file tree

Showing 4 changed files with 11 additions and 3 deletions.
diff --git a/devices/air-purifier.js b/devices/air-purifier.js
@@ -3,10 +3,12 @@
 const Device = require('../device');
 
 class AirPurifier extends Device {
+	static get TYPE() { return 'air-purifier' };
+
 	constructor(options) {
 		super(options);
 
-		this.type = 'air-purifier';
+		this.type = AirPurifier.TYPE;
 
 		this.defineProperty('power', v => v === 'on');
 		this.defineProperty('mode');

diff --git a/devices/switch.js b/devices/switch.js
@@ -3,10 +3,12 @@
 const Device = require('../device');
 
 class Switch extends Device {
+	static get TYPE() { return 'switch' };
+
 	constructor(options) {
 		super(options);
 
-		this.type = 'switch';
+		this.type = Switch.TYPE;
 
 		this.defineProperty('power', v => v === 'on');
 

diff --git a/devices/vacuum.js b/devices/vacuum.js
@@ -13,10 +13,12 @@ function checkResult(r) {
  * doesn't use properties via get_prop but instead has a get_status.
  */
 class Vacuum extends Device {
+	static get TYPE() { return 'vacuum' };
+
 	constructor(options) {
 		super(options);
 
-		this.type = 'vacuum';
+		this.type = Vacuum.TYPE;
 
 		this.defineProperty('state', s => {
 			switch(s) {

diff --git a/index.js b/index.js
@@ -47,8 +47,10 @@ module.exports.infoFromHostname = function(hostname) {
 	const m = /(.+)_miio(\d+)/g.exec(hostname);
 	if(! m) return null;
 
+	const device = devices[m[1]];
 	return {
 		model: m[1],
+		type: (device && device.TYPE) || 'generic',
 		id: m[2]
 	};
 };