1 files changed, 37 insertions, 4 deletions
diff --git a/quantum/template/ps2avrgb/template.c b/quantum/template/ps2avrgb/template.c
index 16fcc8469f..acc8698f56 100644
--- a/quantum/template/ps2avrgb/template.c
+++ b/quantum/template/ps2avrgb/template.c
@@ -15,10 +15,43 @@
  */
 #include "%KEYBOARD%.h"
 
-#include <avr/pgmspace.h>
-#include "action_layer.h"
-#include "i2c.h"
-#include "quantum.h"
+#ifdef RGBLIGHT_ENABLE
+
+#    include <string.h>
+#    include "i2c_master.h"
+#    include "rgblight.h"
+
+extern rgblight_config_t rgblight_config;
+
+void matrix_init_kb(void) {
+    i2c_init();
+    // call user level keymaps, if any
+    matrix_init_user();
+}
+
+// custom RGB driver
+void rgblight_set(void) {
+    if (!rgblight_config.enable) {
+        memset(led, 0, 3 * RGBLED_NUM);
+    }
+
+    i2c_transmit(0xb0, (uint8_t*)led, 3 * RGBLED_NUM, 100);
+}
+
+bool rgb_init = false;
+
+void matrix_scan_kb(void) {
+    // if LEDs were previously on before poweroff, turn them back on
+    if (rgb_init == false && rgblight_config.enable) {
+        i2c_transmit(0xb0, (uint8_t*)led, 3 * RGBLED_NUM, 100);
+        rgb_init = true;
+    }
+
+    rgblight_task();
+    matrix_scan_user();
+}
+
+#endif
 
 __attribute__ ((weak))
 void matrix_scan_user(void) {