Valetudo/backend/lib/robots/roborock/capabilities/RoborockMapSegmentEditCapability.js

const MapSegmentEditCapability = require("../../../core/capabilities/MapSegmentEditCapability");
const RoborockMapParser = require("../RoborockMapParser");

/**
 * @extends MapSegmentEditCapability<import("../RoborockValetudoRobot")>
 */
class RoborockMapSegmentEditCapability extends MapSegmentEditCapability {
    /**
     * @param {import("../../../entities/core/ValetudoMapSegment")} segmentA
     * @param {import("../../../entities/core/ValetudoMapSegment")} segmentB
     * @returns {Promise<void>}
     */
    async joinSegments(segmentA, segmentB) {
        await this.robot.sendCommand("merge_segment", [parseInt(segmentA.id), parseInt(segmentB.id)], {timeout: 5000});

        this.robot.pollMap();
    }

    /**
     * @param {import("../../../entities/core/ValetudoMapSegment")} segment
     * @param {object} pA
     * @param {number} pA.x
     * @param {number} pA.y
     * @param {object} pB
     * @param {number} pB.x
     * @param {number} pB.y
     * @returns {Promise<void>}
     */
    async splitSegment(segment, pA, pB) {
        const flippedSplitLine = [
            parseInt(segment.id),
            Math.floor(pA.x * 10),
            Math.floor(RoborockMapParser.DIMENSION_MM - pA.y * 10),
            Math.floor(pB.x * 10),
            Math.floor(RoborockMapParser.DIMENSION_MM - pB.y * 10)
        ];

        await this.robot.sendCommand("split_segment", flippedSplitLine, {timeout: 5000});

        this.robot.pollMap();
    }
}

module.exports = RoborockMapSegmentEditCapability;