keychron_qmk_firmware/keyboards/atom47/rev2/rev2.c

#include "rev2.h"
#include "led.h"

void matrix_init_kb(void) {
	// put your keyboard start-up code here
	// runs once when the firmware starts up
	matrix_init_user();
	led_init_ports();
};

void matrix_scan_kb(void) {
	// put your looping keyboard code here
	// runs every cycle (a lot)
	matrix_scan_user();
};

void led_init_ports(void) {
    // * Set our LED pins as output
    DDRB &= ~(1<<5);

	//Set output high, so the capslock led is off
	PORTB |= (1 << 5);
}

void led_set_kb(uint8_t usb_led) {
	if (usb_led & (1<<USB_LED_CAPS_LOCK)) {
        // Turn capslock on
        PORTF |= (1<<5);
    } else {
        // Turn capslock off
        PORTF &= ~(1<<5);
    }
	led_set_user(usb_led);
}