1 files changed, 30 insertions, 16 deletions

diff --git a/keyboards/dk60/dk60.c b/keyboards/dk60/dk60.c
index 8b9dc547d..6d72ff6bd 100644
--- a/keyboards/dk60/dk60.c
+++ b/keyboards/dk60/dk60.c
@@ -1,3 +1,20 @@
+/*
+Copyright 2017 Damien Broqua <contact@darkou.fr>
+
+This program is free software: you can redistribute it and/or modify
+it under the terms of the GNU General Public License as published by
+the Free Software Foundation, either version 2 of the License, or
+(at your option) any later version.
+
+This program is distributed in the hope that it will be useful,
+but WITHOUT ANY WARRANTY; without even the implied warranty of
+MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+GNU General Public License for more details.
+
+You should have received a copy of the GNU General Public License
+along with this program.  If not, see <http://www.gnu.org/licenses/>.
+*/
+
 #include "dk60.h"
 
 extern inline void dk60_caps_led_on(void);
@@ -9,12 +26,10 @@ extern inline void dk60_esc_led_off(void);
 extern inline void dk60_led_all_on(void);
 extern inline void dk60_led_all_off(void);
 
-
-void dk60_blink_all_leds(void)
-{
+void dk60_blink_all_leds(void) {
     dk60_led_all_off();
     dk60_led_all_on();
-    _delay_ms(500);
+    wait_ms(500);
     dk60_led_all_off();
 }
 
@@ -26,19 +41,18 @@ void matrix_init_kb(void) {
 }
 
 void led_init_ports(void) {
-    // * Set our LED pins as output
-    DDRE |= (1<<6);
-    DDRF |= (1<<0);
+    setPinOutput(E6);
+    setPinOutput(F0);
 }
 
-void led_set_kb(uint8_t usb_led) {
-        if (usb_led & (1<<USB_LED_CAPS_LOCK)) {
-                // Turn capslock on
-                dk60_caps_led_on();
-        } else {
-                // Turn capslock off
-                dk60_caps_led_off();
-        }
+bool led_update_kb(led_t led_state) {
+    if (led_update_user(led_state)) {
+        if (led_state.caps_lock) {
+            dk60_caps_led_on();
+        } else {
+            dk60_caps_led_off();
+        }
+    }
 
-        led_set_user(usb_led);
+    return true;
 }