[#10][#16][#4][#20] A lot about connection and getting data.

lib/data_model.dart

@@ -0,0 +1,193 @@
+part of 'main.dart';
+
+class HassioDataModel {
+  String _hassioAPIEndpoint;
+  String _hassioPassword;
+  IOWebSocketChannel _hassioChannel;
+  int _currentMssageId = 0;
+  int _statesMessageId = 0;
+  int _servicesMessageId = 0;
+  Map _entitiesData = {};
+  Map _servicesData = {};
+  Map _uiStructure = {};
+  Completer _fetchCompleter;
+  Completer _statesCompleter;
+  Completer _servicesCompleter;
+
+  Map get entities => _entitiesData;
+  Map get services => _servicesData;
+  Map get uiStructure => _uiStructure;
+
+  HassioDataModel(String url, String password) {
+    _hassioAPIEndpoint = url;
+    _hassioPassword = password;
+  }
+
+  Future fetch() {
+    _fetchCompleter = new Completer();
+    _reConnectSocket().then((r) {
+      _getData();
+    }).catchError((e){
+      _fetchCompleter.completeError(e);
+    });
+    return _fetchCompleter.future;
+  }
+
+  Future _reConnectSocket() {
+    var _connectionCompleter = new Completer();
+    if ((_hassioChannel == null) || (_hassioChannel.closeCode != null)) {
+      debugPrint("Socket connecting...");
+      _hassioChannel = IOWebSocketChannel.connect(_hassioAPIEndpoint);
+      _hassioChannel.stream.handleError((e) {
+        debugPrint("Socket error: ${e.toString()}");
+      });
+      _hassioChannel.stream.listen((message) =>
+          _handleMessage(_connectionCompleter, message));
+    } else {
+      _connectionCompleter.complete();
+    }
+    return _connectionCompleter.future;
+  }
+
+  _getData() {
+    _getStates().then((result) {
+      _getServices().then((result) {
+        _fetchCompleter.complete();
+      }).catchError((e) {
+        _fetchCompleter.completeError(e);
+      });
+    }).catchError((e) {
+      _fetchCompleter.completeError(e);
+    });
+
+
+  }
+
+  _handleMessage(Completer connectionCompleter, String message) {
+    debugPrint("<[Receive]Message from Home Assistant:");
+    var data = json.decode(message);
+    debugPrint("   type: ${data['type']}");
+    if (data["type"] == "auth_required") {
+      debugPrint("   sending auth!");
+      _sendMessageRaw('{"type": "auth","api_password": "$_hassioPassword"}');
+    } else if (data["type"] == "auth_ok") {
+      debugPrint("   auth done");
+      debugPrint("Connection done");
+      connectionCompleter.complete();
+    } else if (data["type"] == "auth_invalid") {
+      connectionCompleter.completeError({message: "Auth error: ${data["message"]}"});
+    } else if (data["type"] == "result") {
+      if (data["success"] == true) {
+        if (data["id"] == _statesMessageId) {
+          _parseEntities(data["result"]);
+          _statesCompleter.complete();
+        } else if (data["id"] == _servicesMessageId) {
+          _parseServices(data["result"]);
+          _servicesCompleter.complete();
+        } else if (data["id"] == _currentMssageId) {
+          debugPrint("Request id:$_currentMssageId was successful");
+        } else {
+          _handleErrorMessage({"message" : "Wrong message ID"});
+        }
+      } else {
+        _handleErrorMessage(data["error"]);
+      }
+    }
+  }
+
+  _handleErrorMessage(Object error) {
+    debugPrint("Error: ${error.toString()}");
+    if (!_statesCompleter.isCompleted) _statesCompleter.completeError(error);
+    if (!_servicesCompleter.isCompleted) _servicesCompleter.completeError(error);
+  }
+
+  Future _getStates() {
+    _statesCompleter = new Completer();
+    _incrementMessageId();
+    _statesMessageId = _currentMssageId;
+    _sendMessageRaw('{"id": $_currentMssageId, "type": "get_states"}');
+
+    return _statesCompleter.future;
+  }
+
+  Future _getServices() {
+    _servicesCompleter = new Completer();
+    _incrementMessageId();
+    _servicesMessageId = _currentMssageId;
+    _sendMessageRaw('{"id": $_currentMssageId, "type": "get_services"}');
+
+    return _servicesCompleter.future;
+  }
+
+  _incrementMessageId() {
+    _currentMssageId += 1;
+  }
+
+  _sendMessageRaw(message) {
+    _reConnectSocket().then((r) {
+      debugPrint(">[Send]Sending to Home Assistant:");
+      debugPrint("   $message");
+      _hassioChannel.sink.add(message);
+    }).catchError((e){
+      debugPrint("Unable to connect for sending =(");
+    });
+
+
+  }
+
+  void _parseServices(Map data) {
+    Map result = {};
+    debugPrint("Parsing ${data.length} Home Assistant service domains");
+    data.forEach((domain, services){
+      result[domain] = Map.from(services);
+      services.forEach((serviceName, serviceData){
+        if (_entitiesData["$domain.$serviceName"] != null) {
+          result[domain].remove(serviceName);
+        }
+      });
+    });
+    _servicesData = result;
+  }
+
+  void _parseEntities(List data) async {
+    Map switchServices = {
+      "turn_on": {},
+      "turn_off": {},
+      "toggle": {}
+    };
+    debugPrint("Parsing ${data.length} Home Assistant entities");
+    data.forEach((entity) {
+      var composedEntity = Map.from(entity);
+      String entityDomain = entity["entity_id"].split(".")[0];
+      String entityId = entity["entity_id"];
+
+      composedEntity["display_name"] = "${entity["attributes"]!=null ? entity["attributes"]["friendly_name"] ?? entity["attributes"]["name"] : "_"}";
+      composedEntity["domain"] = entityDomain;
+
+      if ((entityDomain == "automation") || (entityDomain == "light") || (entityDomain == "switch") || (entityDomain == "script")) {
+        composedEntity["services"] = Map.from(switchServices);
+      }
+
+      _entitiesData[entityId] = Map.from(composedEntity);
+    });
+    var defaultView = _entitiesData["group.default_view"];
+    debugPrint("Gethering default view");
+    if (defaultView!= null) {
+      defaultView["attributes"]["entity_id"].forEach((entityId) {
+        if (_entitiesData[entityId]["domain"] != "group") {
+          _uiStructure[entityId] = _entitiesData[entityId];
+        } else {
+          _entitiesData[entityId]["attributes"]["entity_id"].forEach((groupedEntityId) {
+            _uiStructure[groupedEntityId] = _entitiesData[groupedEntityId];
+          });
+        }
+      });
+    }
+
+  }
+
+  callService(String domain, String service, String entity_id) {
+    _incrementMessageId();
+    _sendMessageRaw('{"id": $_currentMssageId, "type": "call_service", "domain": "$domain", "service": "$service", "service_data": {"entity_id": "$entity_id"}}');
+  }
+}