193 lines
6.0 KiB
Dart
193 lines
6.0 KiB
Dart
|
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"}}');
|
||
|
}
|
||
|
}
|