From 141e28221608c60631a7c0aa0adc8177e1e09c4d Mon Sep 17 00:00:00 2001
From: HelgeKeck <helgekeck@hotmail.com>
Date: Sun, 31 Mar 2024 08:46:25 +0200
Subject: [PATCH] rmmu extension

---
 scripts/ratos-common.sh  | 9 +++++++++
 scripts/ratos-install.sh | 1 +
 scripts/ratos-update.sh  | 1 +
 3 files changed, 11 insertions(+)

diff --git a/scripts/ratos-common.sh b/scripts/ratos-common.sh
index f3459b4e..b0a92e5b 100755
--- a/scripts/ratos-common.sh
+++ b/scripts/ratos-common.sh
@@ -112,6 +112,15 @@ register_z_offset_probe()
     _register_klippy_extension $EXT_NAME "$EXT_PATH" $EXT_FILE "false"
 }
 
+register_rmmu()
+{
+    EXT_NAME="rmmu_extension"
+    EXT_PATH=$(realpath "$SCRIPT_DIR"/../klippy)
+    EXT_FILE="rmmu.py"
+	# Don't error if extension is already registered
+    _register_klippy_extension $EXT_NAME "$EXT_PATH" $EXT_FILE "false"
+}
+
 install_hooks()
 {
     report_status "Installing git hooks"
diff --git a/scripts/ratos-install.sh b/scripts/ratos-install.sh
index 06dc79bd..4bcc0886 100755
--- a/scripts/ratos-install.sh
+++ b/scripts/ratos-install.sh
@@ -49,4 +49,5 @@ ensure_sudo_command_whitelisting sudo
 register_gcode_shell_command
 register_ratos_homing
 register_z_offset_probe
+register_rmmu
 register_resonance_generator
\ No newline at end of file
diff --git a/scripts/ratos-update.sh b/scripts/ratos-update.sh
index e5e5ad73..45c7fc5c 100755
--- a/scripts/ratos-update.sh
+++ b/scripts/ratos-update.sh
@@ -57,6 +57,7 @@ install_beacon
 install_hooks
 ensure_node_18
 register_z_offset_probe
+register_rmmu
 register_resonance_generator
 symlink_extensions
 restart_configurator
\ No newline at end of file