Roborock dock - detect water fill / change need?

I’ve been looking into this but it doesn’t seem like the Roborock integration exposes the state of the dock so that you can build automations on the level of water (e.g. when the water needs replacing)

I just created this (after first searching here) and have not yet tested it, but based on the info here this should work.

alias: Warn on Roborock empty water tank
description: Also applies to full waste water
trigger:
  - platform: state
    entity_id:
      - sensor.roborock_dock_status
    to: water_empty
  - platform: state
    entity_id:
      - sensor.roborock_dock_status
    to: waste_water_tank_full
condition: []
action:
  - service: notify.phone
    metadata: {}
    data:
      message: Refill water tank
      title: Roborock
mode: single