// SPDX-License-Identifier: GPL-2.0 OR BSD-3-Clause
/*
 * Copyright (C) 2024 Intel Corporation
 */
#include <linux/leds.h>
#include <net/mac80211.h>

#include "fw/api/led.h"
#include "mld.h"
#include "led.h"
#include "hcmd.h"

static void iwl_mld_send_led_fw_cmd(struct iwl_mld *mld, bool on)
{
	struct iwl_led_cmd led_cmd = {
		.status = cpu_to_le32(on),
	};
	int err;

	if (WARN_ON(!mld->fw_status.running))
		return;

	err = iwl_mld_send_cmd_with_flags_pdu(mld, WIDE_ID(LONG_GROUP,
							   LEDS_CMD),
					      CMD_ASYNC, &led_cmd);

	if (err)
		IWL_WARN(mld, "LED command failed: %d\n", err);
}

static void iwl_led_brightness_set(struct led_classdev *led_cdev,
				   enum led_brightness brightness)
{
	struct iwl_mld *mld = container_of(led_cdev, struct iwl_mld, led);

	if (!mld->fw_status.running)
		return;

	iwl_mld_send_led_fw_cmd(mld, brightness > 0);
}

int iwl_mld_leds_init(struct iwl_mld *mld)
{
	int mode = iwlwifi_mod_params.led_mode;
	int ret;

	switch (mode) {
	case IWL_LED_BLINK:
		IWL_ERR(mld, "Blink led mode not supported, used default\n");
		fallthrough;
	case IWL_LED_DEFAULT:
	case IWL_LED_RF_STATE:
		mode = IWL_LED_RF_STATE;
		break;
	case IWL_LED_DISABLE:
		IWL_INFO(mld, "Led disabled\n");
		return 0;
	default:
		return -EINVAL;
	}

	mld->led.name = kasprintf(GFP_KERNEL, "%s-led",
				  wiphy_name(mld->hw->wiphy));
	if (!mld->led.name)
		return -ENOMEM;

	mld->led.brightness_set = iwl_led_brightness_set;
	mld->led.max_brightness = 1;

	if (mode == IWL_LED_RF_STATE)
		mld->led.default_trigger =
			ieee80211_get_radio_led_name(mld->hw);

	ret = led_classdev_register(mld->trans->dev, &mld->led);
	if (ret) {
		kfree(mld->led.name);
		mld->led.name = NULL;
		IWL_INFO(mld, "Failed to enable led\n");
	}

	return ret;
}

void iwl_mld_led_config_fw(struct iwl_mld *mld)
{
	if (!mld->led.name)
		return;

	iwl_mld_send_led_fw_cmd(mld, mld->led.brightness > 0);
}

void iwl_mld_leds_exit(struct iwl_mld *mld)
{
	if (!mld->led.name)
		return;

	led_classdev_unregister(&mld->led);
	kfree(mld->led.name);
	mld->led.name = NULL;
}