// SPDX-License-Identifier: GPL-2.0 OR BSD-3-Clause
/* Copyright(c) 2025  Realtek Corporation
 */

#include "main.h"
#include "debug.h"
#include "led.h"

static int rtw_led_set(struct led_classdev *led,
		       enum led_brightness brightness)
{
	struct rtw_dev *rtwdev = container_of(led, struct rtw_dev, led_cdev);

	mutex_lock(&rtwdev->mutex);

	rtwdev->chip->ops->led_set(led, brightness);

	mutex_unlock(&rtwdev->mutex);

	return 0;
}

void rtw_led_init(struct rtw_dev *rtwdev)
{
	static const struct ieee80211_tpt_blink rtw_tpt_blink[] = {
		{ .throughput = 0 * 1024, .blink_time = 334 },
		{ .throughput = 1 * 1024, .blink_time = 260 },
		{ .throughput = 5 * 1024, .blink_time = 220 },
		{ .throughput = 10 * 1024, .blink_time = 190 },
		{ .throughput = 20 * 1024, .blink_time = 170 },
		{ .throughput = 50 * 1024, .blink_time = 150 },
		{ .throughput = 70 * 1024, .blink_time = 130 },
		{ .throughput = 100 * 1024, .blink_time = 110 },
		{ .throughput = 200 * 1024, .blink_time = 80 },
		{ .throughput = 300 * 1024, .blink_time = 50 },
	};
	struct led_classdev *led = &rtwdev->led_cdev;
	int err;

	if (!rtwdev->chip->ops->led_set)
		return;

	led->brightness_set_blocking = rtw_led_set;

	snprintf(rtwdev->led_name, sizeof(rtwdev->led_name),
		 "rtw88-%s", dev_name(rtwdev->dev));

	led->name = rtwdev->led_name;
	led->max_brightness = LED_ON;
	led->default_trigger =
		ieee80211_create_tpt_led_trigger(rtwdev->hw,
						 IEEE80211_TPT_LEDTRIG_FL_RADIO,
						 rtw_tpt_blink,
						 ARRAY_SIZE(rtw_tpt_blink));

	err = led_classdev_register(rtwdev->dev, led);
	if (err) {
		rtw_warn(rtwdev, "Failed to register the LED, error %d\n", err);
		return;
	}

	rtwdev->led_registered = true;
}

void rtw_led_deinit(struct rtw_dev *rtwdev)
{
	struct led_classdev *led = &rtwdev->led_cdev;

	if (!rtwdev->led_registered)
		return;

	rtwdev->chip->ops->led_set(led, LED_OFF);
	led_classdev_unregister(led);
}