From c7ca67a036e9e4cd87ff203cf3c3d1922cb79e6d Mon Sep 17 00:00:00 2001
From: Albert Y <76888457+filterpaper@users.noreply.github.com>
Date: Sun, 25 Apr 2021 09:47:43 +0800
Subject: [PATCH] Add RGB matrix suspend wake function for Planck/rev6 (#12290)

* Add RGB matrix suspend wake function for Planck/rev6

* Update suggested definition to allow user override.

Co-authored-by: Nick Brassel <nick@tzarc.org>

Co-authored-by: filterpaper <filterpaper@localhost>
Co-authored-by: Nick Brassel <nick@tzarc.org>
---
 keyboards/planck/rev6/config.h |  4 ++++
 keyboards/planck/rev6/rev6.c   | 10 ++++++++++
 2 files changed, 14 insertions(+)

diff --git a/keyboards/planck/rev6/config.h b/keyboards/planck/rev6/config.h
index 63c44001f..4bc8a509f 100644
--- a/keyboards/planck/rev6/config.h
+++ b/keyboards/planck/rev6/config.h
@@ -140,4 +140,8 @@
 #define WS2812_DMA_STREAM STM32_DMA1_STREAM2
 #define WS2812_DMA_CHANNEL 2
 
+#ifndef RGB_DISABLE_WHEN_USB_SUSPENDED
+#    define RGB_DISABLE_WHEN_USB_SUSPENDED true
+#endif
+
 #endif
diff --git a/keyboards/planck/rev6/rev6.c b/keyboards/planck/rev6/rev6.c
index 3074f8c20..4f2ff8681 100644
--- a/keyboards/planck/rev6/rev6.c
+++ b/keyboards/planck/rev6/rev6.c
@@ -39,6 +39,16 @@ led_config_t g_led_config = { {
 //     6 5 4 3
 //        0
 //     7 8 1 2
+
+void suspend_power_down_kb(void) {
+    rgb_matrix_set_suspend_state(true);
+    suspend_power_down_user();
+}
+
+void suspend_wakeup_init_kb(void) {
+    rgb_matrix_set_suspend_state(false);
+    suspend_wakeup_init_user();
+}
 #endif
 
 void matrix_init_kb(void) {