mirrors/Hypfer-Valetudo
2
0
Fork 0
mirror of https://github.com/Hypfer/Valetudo.git synced 2025-10-26 11:27:27 +00:00
Code Issues Packages Projects Releases Wiki Activity

refactor(vendor.midea): Move specific capabilities into actual implementation class

Browse Source
This commit is contained in:
Sören Beye 2025-09-13 10:05:18 +02:00
parent 7c0fc3a094
commit c0d1640ae3
2 changed files with 78 additions and 64 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

74
backend/lib/robots/midea/MideaJ15ProUltraValetudoRobot.js
View File

@ -1,8 +1,82 @@
const capabilities = require("./capabilities");
const fs = require("node:fs");
const Logger = require("../../Logger");
const MideaQuirkFactory = require("./MideaQuirkFactory");
const MideaValetudoRobot = require("./MideaValetudoRobot");
const MissingResourceValetudoEvent = require("../../valetudo_events/events/MissingResourceValetudoEvent");
const QuirksCapability = require("../../core/capabilities/QuirksCapability");
const {IMAGE_FILE_FORMAT} = require("../../utils/const");
class MideaJ15ProUltraValetudoRobot extends MideaValetudoRobot {
constructor(options) {
super(options);
const quirkFactory = new MideaQuirkFactory({
robot: this
});
[
capabilities.MideaMopExtensionControlCapability,
capabilities.MideaCameraLightControlCapability,
capabilities.MideaObstacleAvoidanceControlCapability,
capabilities.MideaMopDockMopWashTemperatureControlCapability,
capabilities.MideaCarpetSensorModeControlCapability,
capabilities.MideaPetObstacleAvoidanceControlCapability,
capabilities.MideaMopTwistControlCapability,
capabilities.MideaMopExtensionFurnitureLegHandlingControlCapability,
].forEach(capability => {
this.registerCapability(new capability({robot: this}));
});
this.registerCapability(new capabilities.MideaObstacleImagesCapability({
robot: this,
fileFormat: IMAGE_FILE_FORMAT.JPG,
dimensions: {
width: 640,
height: 480
}
}));
this.registerCapability(new QuirksCapability({
robot: this,
quirks: [
quirkFactory.getQuirk(MideaQuirkFactory.KNOWN_QUIRKS.HAIR_CUTTING),
quirkFactory.getQuirk(MideaQuirkFactory.KNOWN_QUIRKS.HAIR_CUTTING_ONE_TIME_TURBO),
quirkFactory.getQuirk(MideaQuirkFactory.KNOWN_QUIRKS.AI_OBSTACLE_CLASSIFICATION),
quirkFactory.getQuirk(MideaQuirkFactory.KNOWN_QUIRKS.QUIET_AUTO_EMPTY),
quirkFactory.getQuirk(MideaQuirkFactory.KNOWN_QUIRKS.CLIFF_SENSORS),
quirkFactory.getQuirk(MideaQuirkFactory.KNOWN_QUIRKS.CARPET_FIRST),
quirkFactory.getQuirk(MideaQuirkFactory.KNOWN_QUIRKS.DEEP_CARPET_CLEANING),
quirkFactory.getQuirk(MideaQuirkFactory.KNOWN_QUIRKS.INCREASED_CARPET_AVOIDANCE),
quirkFactory.getQuirk(MideaQuirkFactory.KNOWN_QUIRKS.STAIN_CLEANING),
]
}));
if (this.config.get("embedded") === true) {
if (!fs.existsSync("/userdata/ai_models/cod-detect-large.bin")) {
this.valetudoEventStore.raise(new MissingResourceValetudoEvent({
id: "midea_ai_model",
message: "The large obstacle detection AI model is missing."
}));
}
}
}
startup() {
super.startup();
if (this.config.get("embedded") === true) {
let aiModelVersion = "unknown/none";
try {
aiModelVersion = fs.readFileSync("/userdata/ai_models/version.txt").toString();
} catch (e) {
/* intentional */
}
Logger.info(`Obstacle detection AI model version: ${aiModelVersion}`);
}
}
getManufacturer() {
return "Eureka";
}

68
backend/lib/robots/midea/MideaValetudoRobot.js
View File

@ -2,24 +2,17 @@ const forge = require("node-forge");
const fs = require("fs");
const BEightParser = require("../../msmart/BEightParser");
const capabilities = require("./capabilities");
const dtos = require("../../msmart/dtos");
const DummyCloudCertManager = require("../../utils/DummyCloudCertManager");
const entities = require("../../entities");
const Logger = require("../../Logger");
const MideaMapParser = require("./MideaMapParser");
const MSmartConst = require("../../msmart/MSmartConst");
const MSmartDummycloud = require("../../msmart/MSmartDummycloud");
const MSmartPacket = require("../../msmart/MSmartPacket");
const ValetudoRobot = require("../../core/ValetudoRobot");
const MideaMapParser = require("./MideaMapParser");
const MideaQuirkFactory = require("./MideaQuirkFactory");
const capabilities = require("./capabilities");
const QuirksCapability = require("../../core/capabilities/QuirksCapability");
const {IMAGE_FILE_FORMAT} = require("../../utils/const");
const entities = require("../../entities");
const stateAttrs = entities.state.attributes;
const MissingResourceValetudoEvent = require("../../valetudo_events/events/MissingResourceValetudoEvent");
const ValetudoRobotError = require("../../entities/core/ValetudoRobotError");
const ValetudoSelectionPreset = require("../../entities/core/ValetudoSelectionPreset");
@ -90,20 +83,8 @@ class MideaValetudoRobot extends ValetudoRobot {
robot: this,
networkInterface: "wlan0"
}));
if (!fs.existsSync("/userdata/ai_models/cod-detect-large.bin")) {
this.valetudoEventStore.raise(new MissingResourceValetudoEvent({
id: "midea_ai_model",
message: "The large obstacle detection AI model is missing."
}));
}
}
const quirkFactory = new MideaQuirkFactory({
robot: this
});
this.registerCapability(new capabilities.MideaFanSpeedControlCapability({
robot: this,
presets: Object.keys(MideaValetudoRobot.FAN_SPEEDS).map(k => {
@ -141,44 +122,12 @@ class MideaValetudoRobot extends ValetudoRobot {
capabilities.MideaAutoEmptyDockManualTriggerCapability,
capabilities.MideaMopDockCleanManualTriggerCapability,
capabilities.MideaMopDockDryManualTriggerCapability,
capabilities.MideaMopExtensionControlCapability,
capabilities.MideaCameraLightControlCapability,
capabilities.MideaObstacleAvoidanceControlCapability,
capabilities.MideaAutoEmptyDockAutoEmptyIntervalControlCapability,
capabilities.MideaMopDockMopWashTemperatureControlCapability,
capabilities.MideaCarpetModeControlCapability,
capabilities.MideaCarpetSensorModeControlCapability,
capabilities.MideaPetObstacleAvoidanceControlCapability,
capabilities.MideaMopTwistControlCapability,
capabilities.MideaMopExtensionFurnitureLegHandlingControlCapability,
].forEach(capability => {
this.registerCapability(new capability({robot: this}));
});
this.registerCapability(new capabilities.MideaObstacleImagesCapability({
robot: this,
fileFormat: IMAGE_FILE_FORMAT.JPG,
dimensions: {
width: 640,
height: 480
}
}));
this.registerCapability(new QuirksCapability({
robot: this,
quirks: [
quirkFactory.getQuirk(MideaQuirkFactory.KNOWN_QUIRKS.HAIR_CUTTING),
quirkFactory.getQuirk(MideaQuirkFactory.KNOWN_QUIRKS.HAIR_CUTTING_ONE_TIME_TURBO),
quirkFactory.getQuirk(MideaQuirkFactory.KNOWN_QUIRKS.AI_OBSTACLE_CLASSIFICATION),
quirkFactory.getQuirk(MideaQuirkFactory.KNOWN_QUIRKS.QUIET_AUTO_EMPTY),
quirkFactory.getQuirk(MideaQuirkFactory.KNOWN_QUIRKS.CLIFF_SENSORS),
quirkFactory.getQuirk(MideaQuirkFactory.KNOWN_QUIRKS.CARPET_FIRST),
quirkFactory.getQuirk(MideaQuirkFactory.KNOWN_QUIRKS.DEEP_CARPET_CLEANING),
quirkFactory.getQuirk(MideaQuirkFactory.KNOWN_QUIRKS.INCREASED_CARPET_AVOIDANCE),
quirkFactory.getQuirk(MideaQuirkFactory.KNOWN_QUIRKS.STAIN_CLEANING),
]
}));
this.state.upsertFirstMatchingAttribute(new entities.state.attributes.DockStatusStateAttribute({
value: entities.state.attributes.DockStatusStateAttribute.VALUE.IDLE
}));
@ -206,16 +155,7 @@ class MideaValetudoRobot extends ValetudoRobot {
super.startup();
if (this.config.get("embedded") === true) {
const firmwareVersion = this.getFirmwareVersion() ?? "unknown";
let aiModelVersion = "unknown/none";
try {
aiModelVersion = fs.readFileSync("/userdata/ai_models/version.txt").toString();
} catch (e) {
/* intentional */
}
Logger.info("Firmware Version: " + firmwareVersion);
Logger.info(`Obstacle detection AI model version: ${aiModelVersion}`);
Logger.info(`Firmware Version: ${this.getFirmwareVersion() ?? "unknown"}`);
}
}