git.basschouten.com Git - openhab-addons.git/blob

   1 /**
   2  * Copyright (c) 2010-2021 Contributors to the openHAB project
   3  *
   4  * See the NOTICE file(s) distributed with this work for additional
   5  * information.
   6  *
   7  * This program and the accompanying materials are made available under the
   8  * terms of the Eclipse Public License 2.0 which is available at
   9  * http://www.eclipse.org/legal/epl-2.0
  10  *
  11  * SPDX-License-Identifier: EPL-2.0
  12  */
  13 package org.openhab.binding.miio.internal.handler;
  14
  15 import static org.openhab.binding.miio.internal.MiIoBindingConstants.*;
  16
  17 import java.io.ByteArrayInputStream;
  18 import java.io.ByteArrayOutputStream;
  19 import java.io.File;
  20 import java.io.IOException;
  21 import java.time.Instant;
  22 import java.time.LocalDateTime;
  23 import java.time.ZoneId;
  24 import java.time.ZonedDateTime;
  25 import java.time.format.DateTimeFormatter;
  26 import java.util.Collections;
  27 import java.util.Map.Entry;
  28 import java.util.Set;
  29 import java.util.concurrent.ConcurrentHashMap;
  30 import java.util.concurrent.TimeUnit;
  31 import java.util.stream.Collectors;
  32 import java.util.stream.Stream;
  33
  34 import javax.imageio.ImageIO;
  35
  36 import org.eclipse.jdt.annotation.NonNullByDefault;
  37 import org.eclipse.jdt.annotation.Nullable;
  38 import org.openhab.binding.miio.internal.MiIoBindingConfiguration;
  39 import org.openhab.binding.miio.internal.MiIoCommand;
  40 import org.openhab.binding.miio.internal.MiIoSendCommand;
  41 import org.openhab.binding.miio.internal.basic.MiIoDatabaseWatchService;
  42 import org.openhab.binding.miio.internal.cloud.CloudConnector;
  43 import org.openhab.binding.miio.internal.cloud.CloudUtil;
  44 import org.openhab.binding.miio.internal.cloud.MiCloudException;
  45 import org.openhab.binding.miio.internal.robot.ConsumablesType;
  46 import org.openhab.binding.miio.internal.robot.FanModeType;
  47 import org.openhab.binding.miio.internal.robot.RRMapDraw;
  48 import org.openhab.binding.miio.internal.robot.RRMapDrawOptions;
  49 import org.openhab.binding.miio.internal.robot.RobotCababilities;
  50 import org.openhab.binding.miio.internal.robot.StatusDTO;
  51 import org.openhab.binding.miio.internal.robot.StatusType;
  52 import org.openhab.binding.miio.internal.robot.VacuumErrorType;
  53 import org.openhab.binding.miio.internal.transport.MiIoAsyncCommunication;
  54 import org.openhab.core.cache.ExpiringCache;
  55 import org.openhab.core.i18n.LocaleProvider;
  56 import org.openhab.core.i18n.TranslationProvider;
  57 import org.openhab.core.library.types.DateTimeType;
  58 import org.openhab.core.library.types.DecimalType;
  59 import org.openhab.core.library.types.OnOffType;
  60 import org.openhab.core.library.types.QuantityType;
  61 import org.openhab.core.library.types.RawType;
  62 import org.openhab.core.library.types.StringType;
  63 import org.openhab.core.library.unit.SIUnits;
  64 import org.openhab.core.library.unit.Units;
  65 import org.openhab.core.thing.Channel;
  66 import org.openhab.core.thing.ChannelUID;
  67 import org.openhab.core.thing.Thing;
  68 import org.openhab.core.thing.ThingStatusDetail;
  69 import org.openhab.core.thing.binding.builder.ChannelBuilder;
  70 import org.openhab.core.thing.binding.builder.ThingBuilder;
  71 import org.openhab.core.thing.type.ChannelType;
  72 import org.openhab.core.thing.type.ChannelTypeRegistry;
  73 import org.openhab.core.types.Command;
  74 import org.openhab.core.types.RefreshType;
  75 import org.openhab.core.types.State;
  76 import org.openhab.core.types.UnDefType;
  77 import org.slf4j.Logger;
  78 import org.slf4j.LoggerFactory;
  79
  80 import com.google.gson.Gson;
  81 import com.google.gson.GsonBuilder;
  82 import com.google.gson.JsonArray;
  83 import com.google.gson.JsonObject;
  84
  85 /**
  86  * The {@link MiIoVacuumHandler} is responsible for handling commands, which are
  87  * sent to one of the channels.
  88  *
  89  * @author Marcel Verpaalen - Initial contribution
  90  */
  91 @NonNullByDefault
  92 public class MiIoVacuumHandler extends MiIoAbstractHandler {
  93     private final Logger logger = LoggerFactory.getLogger(MiIoVacuumHandler.class);
  94     private static final DateTimeFormatter DATEFORMATTER = DateTimeFormatter.ofPattern("yyyyMMdd-HHmmss");
  95     private static final Gson GSON = new GsonBuilder().serializeNulls().create();
  96     private final ChannelUID mapChannelUid;
  97
  98     private static final Set<RobotCababilities> FEATURES_CHANNELS = Collections.unmodifiableSet(Stream
  99             .of(RobotCababilities.SEGMENT_STATUS, RobotCababilities.MAP_STATUS, RobotCababilities.LED_STATUS,
 100                     RobotCababilities.CARPET_MODE, RobotCababilities.FW_FEATURES, RobotCababilities.ROOM_MAPPING,
 101                     RobotCababilities.MULTI_MAP_LIST, RobotCababilities.CUSTOMIZE_CLEAN_MODE)
 102             .collect(Collectors.toSet()));
 103
 104     private ExpiringCache<String> status;
 105     private ExpiringCache<String> consumables;
 106     private ExpiringCache<String> dnd;
 107     private ExpiringCache<String> history;
 108     private int stateId;
 109     private ExpiringCache<String> map;
 110     private String lastHistoryId = "";
 111     private String lastMap = "";
 112     private boolean hasChannelStructure;
 113     private ConcurrentHashMap<RobotCababilities, Boolean> deviceCapabilities = new ConcurrentHashMap<>();
 114     private ChannelTypeRegistry channelTypeRegistry;
 115     private RRMapDrawOptions mapDrawOptions = new RRMapDrawOptions();
 116
 117     public MiIoVacuumHandler(Thing thing, MiIoDatabaseWatchService miIoDatabaseWatchService,
 118             CloudConnector cloudConnector, ChannelTypeRegistry channelTypeRegistry, TranslationProvider i18nProvider,
 119             LocaleProvider localeProvider) {
 120         super(thing, miIoDatabaseWatchService, cloudConnector, i18nProvider, localeProvider);
 121         this.channelTypeRegistry = channelTypeRegistry;
 122         mapChannelUid = new ChannelUID(thing.getUID(), CHANNEL_VACUUM_MAP);
 123         status = new ExpiringCache<>(CACHE_EXPIRY, () -> {
 124             try {
 125                 int ret = sendCommand(MiIoCommand.GET_STATUS);
 126                 if (ret != 0) {
 127                     return "id:" + ret;
 128                 }
 129             } catch (Exception e) {
 130                 logger.debug("Error during status refresh: {}", e.getMessage(), e);
 131             }
 132             return null;
 133         });
 134         consumables = new ExpiringCache<>(CACHE_EXPIRY, () -> {
 135             try {
 136                 int ret = sendCommand(MiIoCommand.CONSUMABLES_GET);
 137                 if (ret != 0) {
 138                     return "id:" + ret;
 139                 }
 140             } catch (Exception e) {
 141                 logger.debug("Error during consumables refresh: {}", e.getMessage(), e);
 142             }
 143             return null;
 144         });
 145         dnd = new ExpiringCache<>(CACHE_EXPIRY, () -> {
 146             try {
 147                 int ret = sendCommand(MiIoCommand.DND_GET);
 148                 if (ret != 0) {
 149                     return "id:" + ret;
 150                 }
 151             } catch (Exception e) {
 152                 logger.debug("Error during dnd refresh: {}", e.getMessage(), e);
 153             }
 154             return null;
 155         });
 156         history = new ExpiringCache<>(CACHE_EXPIRY, () -> {
 157             try {
 158                 int ret = sendCommand(MiIoCommand.CLEAN_SUMMARY_GET);
 159                 if (ret != 0) {
 160                     return "id:" + ret;
 161                 }
 162             } catch (Exception e) {
 163                 logger.debug("Error during cleaning data refresh: {}", e.getMessage(), e);
 164             }
 165             return null;
 166         });
 167         map = new ExpiringCache<String>(CACHE_EXPIRY, () -> {
 168             try {
 169                 int ret = sendCommand(MiIoCommand.GET_MAP);
 170                 if (ret != 0) {
 171                     return "id:" + ret;
 172                 }
 173             } catch (Exception e) {
 174                 logger.debug("Error during dnd refresh: {}", e.getMessage(), e);
 175             }
 176             return null;
 177         });
 178     }
 179
 180     @Override
 181     public void handleCommand(ChannelUID channelUID, Command command) {
 182         if (getConnection() == null) {
 183             logger.debug("Vacuum {} not online. Command {} ignored", getThing().getUID(), command.toString());
 184             return;
 185         }
 186         if (command == RefreshType.REFRESH) {
 187             logger.debug("Refreshing {}", channelUID);
 188             updateData();
 189             lastMap = "";
 190             if (channelUID.getId().equals(CHANNEL_VACUUM_MAP)) {
 191                 sendCommand(MiIoCommand.GET_MAP);
 192             }
 193             return;
 194         }
 195         if (handleCommandsChannels(channelUID, command)) {
 196             forceStatusUpdate();
 197             return;
 198         }
 199         if (channelUID.getId().equals(CHANNEL_VACUUM)) {
 200             if (command instanceof OnOffType) {
 201                 if (command.equals(OnOffType.ON)) {
 202                     sendCommand(MiIoCommand.START_VACUUM);
 203                     forceStatusUpdate();
 204                     return;
 205                 } else {
 206                     sendCommand(MiIoCommand.STOP_VACUUM);
 207                     miIoScheduler.schedule(() -> {
 208                         sendCommand(MiIoCommand.CHARGE);
 209                         forceStatusUpdate();
 210                     }, 2000, TimeUnit.MILLISECONDS);
 211                     return;
 212                 }
 213             }
 214         }
 215         if (channelUID.getId().equals(CHANNEL_CONTROL)) {
 216             if (command.toString().equals("vacuum")) {
 217                 sendCommand(MiIoCommand.START_VACUUM);
 218             } else if (command.toString().equals("spot")) {
 219                 sendCommand(MiIoCommand.START_SPOT);
 220             } else if (command.toString().equals("pause")) {
 221                 sendCommand(MiIoCommand.PAUSE);
 222             } else if (command.toString().equals("dock")) {
 223                 sendCommand(MiIoCommand.STOP_VACUUM);
 224                 miIoScheduler.schedule(() -> {
 225                     sendCommand(MiIoCommand.CHARGE);
 226                     forceStatusUpdate();
 227                 }, 2000, TimeUnit.MILLISECONDS);
 228                 return;
 229             } else {
 230                 logger.info("Command {} not recognised", command.toString());
 231             }
 232             forceStatusUpdate();
 233             return;
 234         }
 235         if (channelUID.getId().equals(CHANNEL_FAN_POWER)) {
 236             sendCommand(MiIoCommand.SET_MODE, "[" + command.toString() + "]");
 237             forceStatusUpdate();
 238             return;
 239         }
 240         if (channelUID.getId().equals(RobotCababilities.WATERBOX_MODE.getChannel())) {
 241             sendCommand(MiIoCommand.SET_WATERBOX_MODE, "[" + command.toString() + "]");
 242             forceStatusUpdate();
 243             return;
 244         }
 245         if (channelUID.getId().equals(RobotCababilities.SEGMENT_CLEAN.getChannel()) && !command.toString().isEmpty()
 246                 && !command.toString().contentEquals("-")) {
 247             sendCommand(MiIoCommand.START_SEGMENT, "[" + command.toString() + "]");
 248             updateState(RobotCababilities.SEGMENT_CLEAN.getChannel(), new StringType("-"));
 249             forceStatusUpdate();
 250             return;
 251         }
 252         if (channelUID.getId().equals(CHANNEL_FAN_CONTROL)) {
 253             if (Integer.valueOf(command.toString()) > 0) {
 254                 sendCommand(MiIoCommand.SET_MODE, "[" + command.toString() + "]");
 255             }
 256             forceStatusUpdate();
 257             return;
 258         }
 259         if (channelUID.getId().equals(CHANNEL_CONSUMABLE_RESET)) {
 260             sendCommand(MiIoCommand.CONSUMABLES_RESET, "[" + command.toString() + "]");
 261             updateState(CHANNEL_CONSUMABLE_RESET, new StringType("none"));
 262         }
 263     }
 264
 265     private void forceStatusUpdate() {
 266         status.invalidateValue();
 267         miIoScheduler.schedule(() -> {
 268             status.getValue();
 269         }, 3000, TimeUnit.MILLISECONDS);
 270     }
 271
 272     private void safeUpdateState(String channelID, @Nullable Integer state) {
 273         if (state != null) {
 274             updateState(channelID, new DecimalType(state));
 275         } else {
 276             logger.debug("Channel {} not update. value not available.", channelID);
 277         }
 278     }
 279
 280     private boolean updateVacuumStatus(JsonObject statusData) {
 281         StatusDTO statusInfo = GSON.fromJson(statusData, StatusDTO.class);
 282         safeUpdateState(CHANNEL_BATTERY, statusInfo.getBattery());
 283         if (statusInfo.getCleanArea() != null) {
 284             updateState(CHANNEL_CLEAN_AREA,
 285                     new QuantityType<>(statusInfo.getCleanArea() / 1000000.0, SIUnits.SQUARE_METRE));
 286         }
 287         if (statusInfo.getCleanTime() != null) {
 288             updateState(CHANNEL_CLEAN_TIME,
 289                     new QuantityType<>(TimeUnit.SECONDS.toMinutes(statusInfo.getCleanTime()), Units.MINUTE));
 290         }
 291         safeUpdateState(CHANNEL_DND_ENABLED, statusInfo.getDndEnabled());
 292
 293         if (statusInfo.getErrorCode() != null) {
 294             updateState(CHANNEL_ERROR_CODE,
 295                     new StringType(VacuumErrorType.getType(statusInfo.getErrorCode()).getDescription()));
 296             safeUpdateState(CHANNEL_ERROR_ID, statusInfo.getErrorCode());
 297         }
 298
 299         if (statusInfo.getFanPower() != null) {
 300             updateState(CHANNEL_FAN_POWER, new DecimalType(statusInfo.getFanPower()));
 301             updateState(CHANNEL_FAN_CONTROL, new DecimalType(FanModeType.getType(statusInfo.getFanPower()).getId()));
 302         }
 303         safeUpdateState(CHANNEL_IN_CLEANING, statusInfo.getInCleaning());
 304         safeUpdateState(CHANNEL_MAP_PRESENT, statusInfo.getMapPresent());
 305         if (statusInfo.getState() != null) {
 306             stateId = statusInfo.getState();
 307             StatusType state = StatusType.getType(statusInfo.getState());
 308             updateState(CHANNEL_STATE, new StringType(state.getDescription()));
 309             updateState(CHANNEL_STATE_ID, new DecimalType(statusInfo.getState()));
 310
 311             State vacuum = OnOffType.OFF;
 312             String control;
 313             switch (state) {
 314                 case ZONE:
 315                 case ROOM:
 316                 case CLEANING:
 317                 case RETURNING:
 318                     control = "vacuum";
 319                     vacuum = OnOffType.ON;
 320                     break;
 321                 case CHARGING:
 322                 case CHARGING_ERROR:
 323                 case DOCKING:
 324                 case FULL:
 325                     control = "dock";
 326                     break;
 327                 case SLEEPING:
 328                 case PAUSED:
 329                 case IDLE:
 330                     control = "pause";
 331                     break;
 332                 case SPOTCLEAN:
 333                     control = "spot";
 334                     vacuum = OnOffType.ON;
 335                     break;
 336                 default:
 337                     control = "undef";
 338                     break;
 339             }
 340             if ("undef".equals(control)) {
 341                 updateState(CHANNEL_CONTROL, UnDefType.UNDEF);
 342             } else {
 343                 updateState(CHANNEL_CONTROL, new StringType(control));
 344             }
 345             updateState(CHANNEL_VACUUM, vacuum);
 346         }
 347         if (deviceCapabilities.containsKey(RobotCababilities.WATERBOX_MODE)) {
 348             safeUpdateState(RobotCababilities.WATERBOX_MODE.getChannel(), statusInfo.getWaterBoxMode());
 349         }
 350         if (deviceCapabilities.containsKey(RobotCababilities.WATERBOX_STATUS)) {
 351             safeUpdateState(RobotCababilities.WATERBOX_STATUS.getChannel(), statusInfo.getWaterBoxStatus());
 352         }
 353         if (deviceCapabilities.containsKey(RobotCababilities.WATERBOX_CARRIAGE)) {
 354             safeUpdateState(RobotCababilities.WATERBOX_CARRIAGE.getChannel(), statusInfo.getWaterBoxCarriageStatus());
 355         }
 356         if (deviceCapabilities.containsKey(RobotCababilities.LOCKSTATUS)) {
 357             safeUpdateState(RobotCababilities.LOCKSTATUS.getChannel(), statusInfo.getLockStatus());
 358         }
 359         if (deviceCapabilities.containsKey(RobotCababilities.MOP_FORBIDDEN)) {
 360             safeUpdateState(RobotCababilities.MOP_FORBIDDEN.getChannel(), statusInfo.getMopForbiddenEnable());
 361         }
 362         if (deviceCapabilities.containsKey(RobotCababilities.LOCATING)) {
 363             safeUpdateState(RobotCababilities.LOCATING.getChannel(), statusInfo.getIsLocating());
 364         }
 365         return true;
 366     }
 367
 368     private boolean updateConsumables(JsonObject consumablesData) {
 369         int mainBrush = consumablesData.get("main_brush_work_time").getAsInt();
 370         int sideBrush = consumablesData.get("side_brush_work_time").getAsInt();
 371         int filter = consumablesData.get("filter_work_time").getAsInt();
 372         int sensor = consumablesData.get("sensor_dirty_time").getAsInt();
 373         updateState(CHANNEL_CONSUMABLE_MAIN_TIME,
 374                 new QuantityType<>(ConsumablesType.remainingHours(mainBrush, ConsumablesType.MAIN_BRUSH), Units.HOUR));
 375         updateState(CHANNEL_CONSUMABLE_MAIN_PERC,
 376                 new DecimalType(ConsumablesType.remainingPercent(mainBrush, ConsumablesType.MAIN_BRUSH)));
 377         updateState(CHANNEL_CONSUMABLE_SIDE_TIME,
 378                 new QuantityType<>(ConsumablesType.remainingHours(sideBrush, ConsumablesType.SIDE_BRUSH), Units.HOUR));
 379         updateState(CHANNEL_CONSUMABLE_SIDE_PERC,
 380                 new DecimalType(ConsumablesType.remainingPercent(sideBrush, ConsumablesType.SIDE_BRUSH)));
 381         updateState(CHANNEL_CONSUMABLE_FILTER_TIME,
 382                 new QuantityType<>(ConsumablesType.remainingHours(filter, ConsumablesType.FILTER), Units.HOUR));
 383         updateState(CHANNEL_CONSUMABLE_FILTER_PERC,
 384                 new DecimalType(ConsumablesType.remainingPercent(filter, ConsumablesType.FILTER)));
 385         updateState(CHANNEL_CONSUMABLE_SENSOR_TIME,
 386                 new QuantityType<>(ConsumablesType.remainingHours(sensor, ConsumablesType.SENSOR), Units.HOUR));
 387         updateState(CHANNEL_CONSUMABLE_SENSOR_PERC,
 388                 new DecimalType(ConsumablesType.remainingPercent(sensor, ConsumablesType.SENSOR)));
 389         return true;
 390     }
 391
 392     private boolean updateDnD(JsonObject dndData) {
 393         logger.trace("Do not disturb data: {}", dndData.toString());
 394         updateState(CHANNEL_DND_FUNCTION, new DecimalType(dndData.get("enabled").getAsBigDecimal()));
 395         updateState(CHANNEL_DND_START, new StringType(String.format("%02d:%02d", dndData.get("start_hour").getAsInt(),
 396                 dndData.get("start_minute").getAsInt())));
 397         updateState(CHANNEL_DND_END, new StringType(
 398                 String.format("%02d:%02d", dndData.get("end_hour").getAsInt(), dndData.get("end_minute").getAsInt())));
 399         return true;
 400     }
 401
 402     private boolean updateHistory(JsonArray historyData) {
 403         logger.trace("Cleaning history data: {}", historyData.toString());
 404         updateState(CHANNEL_HISTORY_TOTALTIME,
 405                 new QuantityType<>(TimeUnit.SECONDS.toMinutes(historyData.get(0).getAsLong()), Units.MINUTE));
 406         updateState(CHANNEL_HISTORY_TOTALAREA,
 407                 new QuantityType<>(historyData.get(1).getAsDouble() / 1000000D, SIUnits.SQUARE_METRE));
 408         updateState(CHANNEL_HISTORY_COUNT, new DecimalType(historyData.get(2).toString()));
 409         if (historyData.get(3).getAsJsonArray().size() > 0) {
 410             String lastClean = historyData.get(3).getAsJsonArray().get(0).getAsString();
 411             if (!lastClean.equals(lastHistoryId)) {
 412                 lastHistoryId = lastClean;
 413                 sendCommand(MiIoCommand.CLEAN_RECORD_GET, "[" + lastClean + "]");
 414             }
 415         }
 416         return true;
 417     }
 418
 419     private void updateHistoryRecord(JsonArray historyData) {
 420         ZonedDateTime startTime = ZonedDateTime.ofInstant(Instant.ofEpochSecond(historyData.get(0).getAsLong()),
 421                 ZoneId.systemDefault());
 422         ZonedDateTime endTime = ZonedDateTime.ofInstant(Instant.ofEpochSecond(historyData.get(1).getAsLong()),
 423                 ZoneId.systemDefault());
 424         long duration = TimeUnit.SECONDS.toMinutes(historyData.get(2).getAsLong());
 425         double area = historyData.get(3).getAsDouble() / 1000000D;
 426         int error = historyData.get(4).getAsInt();
 427         int finished = historyData.get(5).getAsInt();
 428         JsonObject historyRecord = new JsonObject();
 429         historyRecord.addProperty("start", startTime.toString());
 430         historyRecord.addProperty("end", endTime.toString());
 431         historyRecord.addProperty("duration", duration);
 432         historyRecord.addProperty("area", area);
 433         historyRecord.addProperty("error", error);
 434         historyRecord.addProperty("finished", finished);
 435         updateState(CHANNEL_HISTORY_START_TIME, new DateTimeType(startTime));
 436         updateState(CHANNEL_HISTORY_END_TIME, new DateTimeType(endTime));
 437         updateState(CHANNEL_HISTORY_DURATION, new QuantityType<>(duration, Units.MINUTE));
 438         updateState(CHANNEL_HISTORY_AREA, new QuantityType<>(area, SIUnits.SQUARE_METRE));
 439         updateState(CHANNEL_HISTORY_ERROR, new DecimalType(error));
 440         updateState(CHANNEL_HISTORY_FINISH, new DecimalType(finished));
 441         updateState(CHANNEL_HISTORY_RECORD, new StringType(historyRecord.toString()));
 442     }
 443
 444     @Override
 445     protected boolean skipUpdate() {
 446         if (!hasConnection()) {
 447             logger.debug("Skipping periodic update for '{}'. No Connection", getThing().getUID().toString());
 448             return true;
 449         }
 450         if (ThingStatusDetail.CONFIGURATION_ERROR.equals(getThing().getStatusInfo().getStatusDetail())) {
 451             logger.debug("Skipping periodic update for '{}' UID '{}'. Thing Status", getThing().getUID().toString(),
 452                     getThing().getStatusInfo().getStatusDetail());
 453             refreshNetwork();
 454             return true;
 455         }
 456         final MiIoAsyncCommunication mc = miioCom;
 457         if (mc != null && mc.getQueueLength() > MAX_QUEUE) {
 458             logger.debug("Skipping periodic update for '{}'. {} elements in queue.", getThing().getUID().toString(),
 459                     mc.getQueueLength());
 460             return true;
 461         }
 462         return false;
 463     }
 464
 465     @Override
 466     protected synchronized void updateData() {
 467         if (!hasConnection() || skipUpdate()) {
 468             return;
 469         }
 470         logger.debug("Periodic update for '{}' ({})", getThing().getUID().toString(), getThing().getThingTypeUID());
 471         try {
 472             dnd.getValue();
 473             history.getValue();
 474             status.getValue();
 475             refreshNetwork();
 476             consumables.getValue();
 477             if (lastMap.isEmpty() || stateId != 8) {
 478                 if (isLinked(mapChannelUid)) {
 479                     map.getValue();
 480                 }
 481             }
 482             for (RobotCababilities cmd : FEATURES_CHANNELS) {
 483                 if (isLinked(cmd.getChannel())) {
 484                     sendCommand(cmd.getCommand());
 485                 }
 486             }
 487         } catch (Exception e) {
 488             logger.debug("Error while updating '{}': '{}", getThing().getUID().toString(), e.getLocalizedMessage());
 489         }
 490     }
 491
 492     @Override
 493     public void initialize() {
 494         super.initialize();
 495         hasChannelStructure = false;
 496         this.mapDrawOptions = RRMapDrawOptions
 497                 .getOptionsFromFile(BINDING_USERDATA_PATH + File.separator + "mapConfig.json", logger);
 498         updateState(RobotCababilities.SEGMENT_CLEAN.getChannel(), new StringType("-"));
 499     }
 500
 501     @Override
 502     protected boolean initializeData() {
 503         updateState(CHANNEL_CONSUMABLE_RESET, new StringType("none"));
 504         return super.initializeData();
 505     }
 506
 507     @Override
 508     public void onMessageReceived(MiIoSendCommand response) {
 509         super.onMessageReceived(response);
 510         if (response.isError()) {
 511             return;
 512         }
 513         switch (response.getCommand()) {
 514             case GET_STATUS:
 515                 if (response.getResult().isJsonArray()) {
 516                     JsonObject statusResponse = response.getResult().getAsJsonArray().get(0).getAsJsonObject();
 517                     if (!hasChannelStructure) {
 518                         setCapabilities(statusResponse);
 519                         createCapabilityChannels();
 520                     }
 521                     updateVacuumStatus(statusResponse);
 522                 }
 523                 break;
 524             case CONSUMABLES_GET:
 525                 if (response.getResult().isJsonArray()) {
 526                     updateConsumables(response.getResult().getAsJsonArray().get(0).getAsJsonObject());
 527                 }
 528                 break;
 529             case DND_GET:
 530                 if (response.getResult().isJsonArray()) {
 531                     updateDnD(response.getResult().getAsJsonArray().get(0).getAsJsonObject());
 532                 }
 533                 break;
 534             case CLEAN_SUMMARY_GET:
 535                 if (response.getResult().isJsonArray()) {
 536                     updateHistory(response.getResult().getAsJsonArray());
 537                 }
 538                 break;
 539             case CLEAN_RECORD_GET:
 540                 if (response.getResult().isJsonArray() && response.getResult().getAsJsonArray().size() > 0
 541                         && response.getResult().getAsJsonArray().get(0).isJsonArray()) {
 542                     updateHistoryRecord(response.getResult().getAsJsonArray().get(0).getAsJsonArray());
 543                 } else {
 544                     logger.debug("Could not extract cleaning history record from: {}", response);
 545                 }
 546                 break;
 547             case GET_MAP:
 548                 if (response.getResult().isJsonArray()) {
 549                     String mapresponse = response.getResult().getAsJsonArray().get(0).getAsString();
 550                     if (!mapresponse.contentEquals("retry") && !mapresponse.contentEquals(lastMap)) {
 551                         lastMap = mapresponse;
 552                         miIoScheduler.submit(() -> updateState(CHANNEL_VACUUM_MAP, getMap(mapresponse)));
 553                     }
 554                 }
 555                 break;
 556             case GET_MAP_STATUS:
 557             case GET_SEGMENT_STATUS:
 558             case GET_LED_STATUS:
 559                 updateNumericChannel(response);
 560                 break;
 561             case GET_CARPET_MODE:
 562             case GET_FW_FEATURES:
 563             case GET_CUSTOMIZED_CLEAN_MODE:
 564             case GET_MULTI_MAP_LIST:
 565             case GET_ROOM_MAPPING:
 566                 for (RobotCababilities cmd : FEATURES_CHANNELS) {
 567                     if (response.getCommand().getCommand().contentEquals(cmd.getCommand())) {
 568                         updateState(cmd.getChannel(), new StringType(response.getResult().toString()));
 569                         break;
 570                     }
 571                 }
 572                 break;
 573             default:
 574                 break;
 575         }
 576     }
 577
 578     private void updateNumericChannel(MiIoSendCommand response) {
 579         RobotCababilities capabilityChannel = null;
 580         for (RobotCababilities cmd : FEATURES_CHANNELS) {
 581             if (response.getCommand().getCommand().contentEquals(cmd.getCommand())) {
 582                 capabilityChannel = cmd;
 583                 break;
 584             }
 585         }
 586         if (capabilityChannel != null) {
 587             if (response.getResult().isJsonArray() && response.getResult().getAsJsonArray().get(0).isJsonPrimitive()) {
 588                 try {
 589                     Integer stat = response.getResult().getAsJsonArray().get(0).getAsInt();
 590                     updateState(capabilityChannel.getChannel(), new DecimalType(stat));
 591                     return;
 592                 } catch (ClassCastException | IllegalStateException e) {
 593                     logger.debug("Could not update numeric channel {} with '{}': {}", capabilityChannel.getChannel(),
 594                             response.getResult(), e.getMessage());
 595                 }
 596             } else {
 597                 logger.debug("Could not update numeric channel {} with '{}': Not in expected format",
 598                         capabilityChannel.getChannel(), response.getResult());
 599             }
 600             updateState(capabilityChannel.getChannel(), UnDefType.UNDEF);
 601         }
 602     }
 603
 604     private void setCapabilities(JsonObject statusResponse) {
 605         for (RobotCababilities capability : RobotCababilities.values()) {
 606             if (statusResponse.has(capability.getStatusFieldName())) {
 607                 deviceCapabilities.putIfAbsent(capability, false);
 608                 logger.debug("Setting additional vacuum {}", capability);
 609             }
 610         }
 611     }
 612
 613     private void createCapabilityChannels() {
 614         ThingBuilder thingBuilder = editThing();
 615         int cnt = 0;
 616
 617         for (Entry<RobotCababilities, Boolean> robotCapability : deviceCapabilities.entrySet()) {
 618             RobotCababilities capability = robotCapability.getKey();
 619             Boolean channelCreated = robotCapability.getValue();
 620             if (!channelCreated) {
 621                 if (thing.getChannels().stream()
 622                         .anyMatch(ch -> ch.getUID().getId().equalsIgnoreCase(capability.getChannel()))) {
 623                     logger.debug("Channel already available...skip creation of channel '{}'.", capability.getChannel());
 624                     deviceCapabilities.replace(capability, true);
 625                     continue;
 626                 }
 627                 logger.debug("Creating dynamic channel for capability {}", capability);
 628                 ChannelType channelType = channelTypeRegistry.getChannelType(capability.getChannelType());
 629                 if (channelType != null) {
 630                     logger.debug("Found channelType '{}' for capability {}", channelType, capability.name());
 631                     ChannelUID channelUID = new ChannelUID(getThing().getUID(), capability.getChannel());
 632                     Channel channel = ChannelBuilder.create(channelUID, channelType.getItemType())
 633                             .withType(capability.getChannelType()).withLabel(channelType.getLabel()).build();
 634                     thingBuilder.withChannel(channel);
 635                     cnt++;
 636                 } else {
 637                     logger.debug("ChannelType {} not found (Unexpected). Available types:",
 638                             capability.getChannelType());
 639                     for (ChannelType ct : channelTypeRegistry.getChannelTypes()) {
 640                         logger.debug("Available channelType: '{}' '{}' '{}'", ct.getUID(), ct.toString(),
 641                                 ct.getConfigDescriptionURI());
 642                     }
 643                 }
 644             }
 645         }
 646         if (cnt > 0) {
 647             updateThing(thingBuilder.build());
 648         }
 649         hasChannelStructure = true;
 650     }
 651
 652     private State getMap(String map) {
 653         final MiIoBindingConfiguration configuration = this.configuration;
 654         if (configuration != null && cloudConnector.isConnected()) {
 655             try {
 656                 final @Nullable RawType mapDl = cloudConnector.getMap(map, configuration.cloudServer);
 657                 if (mapDl != null) {
 658                     byte[] mapData = mapDl.getBytes();
 659                     RRMapDraw rrMap = RRMapDraw.loadImage(new ByteArrayInputStream(mapData));
 660                     rrMap.setDrawOptions(mapDrawOptions);
 661                     ByteArrayOutputStream baos = new ByteArrayOutputStream();
 662                     if (logger.isDebugEnabled()) {
 663                         final String mapPath = BINDING_USERDATA_PATH + File.separator + map
 664                                 + LocalDateTime.now().format(DATEFORMATTER) + ".rrmap";
 665                         CloudUtil.writeBytesToFileNio(mapData, mapPath);
 666                         logger.debug("Mapdata saved to {}", mapPath);
 667                     }
 668                     ImageIO.write(rrMap.getImage(), "jpg", baos);
 669                     byte[] byteArray = baos.toByteArray();
 670                     if (byteArray != null && byteArray.length > 0) {
 671                         return new RawType(byteArray, "image/jpeg");
 672                     } else {
 673                         logger.debug("Mapdata empty removing image");
 674                         return UnDefType.UNDEF;
 675                     }
 676                 }
 677             } catch (MiCloudException e) {
 678                 logger.debug("Error getting data from Xiaomi cloud. Mapdata could not be updated: {}", e.getMessage());
 679             } catch (IOException e) {
 680                 logger.debug("Mapdata could not be updated: {}", e.getMessage());
 681             }
 682         } else {
 683             logger.debug("Not connected to Xiaomi cloud. Cannot retreive new map: {}", map);
 684         }
 685         return UnDefType.UNDEF;
 686     }
 687 }