/*
 * Copyright (c) 2024 Nordic Semiconductor ASA
 *
 * SPDX-License-Identifier: BSD-3-Clause
 */

/**
 * @brief File containing QSPI Bus Layer specific function definitions of the
 * Wi-Fi driver.
 */

#include "bal_structs.h"
#include "spi.h"
#include "pal.h"


static int nrf_wifi_bus_spi_irq_handler(void *data)
{
	struct nrf_wifi_bus_spi_dev_ctx *dev_ctx = NULL;
	struct nrf_wifi_bus_spi_priv *spi_priv = NULL;
	int ret = 0;

	dev_ctx = (struct nrf_wifi_bus_spi_dev_ctx *)data;
	spi_priv = dev_ctx->spi_priv;

	ret = spi_priv->intr_callbk_fn(dev_ctx->bal_dev_ctx);

	return ret;
}


static void *nrf_wifi_bus_spi_dev_add(void *bus_priv,
				       void *bal_dev_ctx)
{
	enum nrf_wifi_status status = NRF_WIFI_STATUS_FAIL;
	struct nrf_wifi_bus_spi_priv *spi_priv = NULL;
	struct nrf_wifi_bus_spi_dev_ctx *spi_dev_ctx = NULL;
	struct nrf_wifi_osal_host_map host_map;

	spi_priv = bus_priv;

	spi_dev_ctx = nrf_wifi_osal_mem_zalloc(sizeof(*spi_dev_ctx));

	if (!spi_dev_ctx) {
		nrf_wifi_osal_log_err("%s: Unable to allocate spi_dev_ctx", __func__);
		goto out;
	}

	spi_dev_ctx->spi_priv = spi_priv;
	spi_dev_ctx->bal_dev_ctx = bal_dev_ctx;

	spi_dev_ctx->os_spi_dev_ctx = nrf_wifi_osal_bus_spi_dev_add(spi_priv->os_spi_priv,
								    spi_dev_ctx);

	if (!spi_dev_ctx->os_spi_dev_ctx) {
		nrf_wifi_osal_log_err("%s: nrf_wifi_osal_bus_spi_dev_add failed", __func__);

		nrf_wifi_osal_mem_free(spi_dev_ctx);

		spi_dev_ctx = NULL;

		goto out;
	}

	nrf_wifi_osal_bus_spi_dev_host_map_get(spi_dev_ctx->os_spi_dev_ctx,
					       &host_map);

	spi_dev_ctx->host_addr_base = host_map.addr;

	spi_dev_ctx->addr_pktram_base = spi_dev_ctx->host_addr_base +
		spi_priv->cfg_params.addr_pktram_base;

out:
	return spi_dev_ctx;
}


static void nrf_wifi_bus_spi_dev_rem(void *bus_dev_ctx)
{
	struct nrf_wifi_bus_spi_dev_ctx *spi_dev_ctx = NULL;

	spi_dev_ctx = bus_dev_ctx;

	nrf_wifi_osal_mem_free(spi_dev_ctx);
}


static enum nrf_wifi_status nrf_wifi_bus_spi_dev_init(void *bus_dev_ctx)
{
	enum nrf_wifi_status status = NRF_WIFI_STATUS_FAIL;
	struct nrf_wifi_bus_spi_dev_ctx *spi_dev_ctx = NULL;

	spi_dev_ctx = bus_dev_ctx;


	status = nrf_wifi_osal_bus_spi_dev_intr_reg(spi_dev_ctx->os_spi_dev_ctx, spi_dev_ctx,
					&nrf_wifi_bus_spi_irq_handler);

	if (status != NRF_WIFI_STATUS_SUCCESS) {
		nrf_wifi_osal_log_err(
				      "%s: Unable to register interrupt to the OS",
				      __func__);
		goto out;
	}

	status = nrf_wifi_osal_bus_spi_dev_init(spi_dev_ctx->os_spi_dev_ctx);

	if (status != NRF_WIFI_STATUS_SUCCESS) {
		nrf_wifi_osal_log_err("%s: nrf_wifi_osal_spi_dev_init failed", __func__);

		nrf_wifi_osal_bus_spi_dev_intr_unreg(spi_dev_ctx->os_spi_dev_ctx);
		goto out;
	}
out:
	return status;
}


static void nrf_wifi_bus_spi_dev_deinit(void *bus_dev_ctx)
{
	struct nrf_wifi_bus_spi_dev_ctx *spi_dev_ctx = NULL;

	spi_dev_ctx = bus_dev_ctx;

	nrf_wifi_osal_bus_spi_dev_intr_unreg(
					      spi_dev_ctx->os_spi_dev_ctx);

	nrf_wifi_osal_bus_spi_dev_deinit(
					  spi_dev_ctx->os_spi_dev_ctx);
}


static void *nrf_wifi_bus_spi_init(void *params,
				   enum nrf_wifi_status (*intr_callbk_fn)(void *bal_dev_ctx))
{
	struct nrf_wifi_bus_spi_priv *spi_priv = NULL;

	spi_priv = nrf_wifi_osal_mem_zalloc(sizeof(*spi_priv));

	if (!spi_priv) {
		nrf_wifi_osal_log_err("%s: Unable to allocate memory for spi_priv",
				      __func__);
		goto out;
	}

	nrf_wifi_osal_mem_cpy(&spi_priv->cfg_params,
			      params,
			      sizeof(spi_priv->cfg_params));

	spi_priv->intr_callbk_fn = intr_callbk_fn;

	spi_priv->os_spi_priv = nrf_wifi_osal_bus_spi_init();
	if (!spi_priv->os_spi_priv) {
		nrf_wifi_osal_log_err("%s: Unable to register QSPI driver",
				      __func__);

		nrf_wifi_osal_mem_free(spi_priv);

		spi_priv = NULL;

		goto out;
	}
out:
	return spi_priv;
}


static void nrf_wifi_bus_spi_deinit(void *bus_priv)
{
	struct nrf_wifi_bus_spi_priv *spi_priv = NULL;

	spi_priv = bus_priv;

	nrf_wifi_osal_bus_spi_deinit(spi_priv->os_spi_priv);

	nrf_wifi_osal_mem_free(spi_priv);
}


static unsigned int nrf_wifi_bus_spi_read_word(void *dev_ctx,
					       unsigned long addr_offset)
{
	struct nrf_wifi_bus_spi_dev_ctx *spi_dev_ctx = NULL;
	unsigned int val = 0;

	spi_dev_ctx = (struct nrf_wifi_bus_spi_dev_ctx *)dev_ctx;

	val = nrf_wifi_osal_spi_read_reg32(spi_dev_ctx->os_spi_dev_ctx,
					   spi_dev_ctx->host_addr_base + addr_offset);
	return val;
}


static void nrf_wifi_bus_spi_write_word(void *dev_ctx,
					unsigned long addr_offset,
					unsigned int val)
{
	struct nrf_wifi_bus_spi_dev_ctx *spi_dev_ctx = NULL;

	spi_dev_ctx = (struct nrf_wifi_bus_spi_dev_ctx *)dev_ctx;

	nrf_wifi_osal_spi_write_reg32(spi_dev_ctx->os_spi_dev_ctx,
				      spi_dev_ctx->host_addr_base + addr_offset,
				      val);
}


static void nrf_wifi_bus_spi_read_block(void *dev_ctx,
					void *dest_addr,
					unsigned long src_addr_offset,
					size_t len)
{
	struct nrf_wifi_bus_spi_dev_ctx *spi_dev_ctx = NULL;

	spi_dev_ctx = (struct nrf_wifi_bus_spi_dev_ctx *)dev_ctx;

	nrf_wifi_osal_spi_cpy_from(spi_dev_ctx->os_spi_dev_ctx,
				   dest_addr,
				   spi_dev_ctx->host_addr_base + src_addr_offset,
				   len);
}


static void nrf_wifi_bus_spi_write_block(void *dev_ctx,
					 unsigned long dest_addr_offset,
					 const void *src_addr,
					 size_t len)
{
	struct nrf_wifi_bus_spi_dev_ctx *spi_dev_ctx = NULL;

	spi_dev_ctx = (struct nrf_wifi_bus_spi_dev_ctx *)dev_ctx;

	nrf_wifi_osal_spi_cpy_to(spi_dev_ctx->os_spi_dev_ctx,
				 spi_dev_ctx->host_addr_base + dest_addr_offset,
				 src_addr,
				 len);
}


static unsigned long nrf_wifi_bus_spi_dma_map(void *dev_ctx,
					      unsigned long virt_addr,
					      size_t len,
					      enum nrf_wifi_osal_dma_dir dma_dir)
{
	struct nrf_wifi_bus_spi_dev_ctx *spi_dev_ctx = NULL;
	unsigned long phy_addr = 0;

	spi_dev_ctx = (struct nrf_wifi_bus_spi_dev_ctx *)dev_ctx;

	phy_addr = spi_dev_ctx->host_addr_base + (virt_addr - spi_dev_ctx->addr_pktram_base);


	return phy_addr;
}


static unsigned long nrf_wifi_bus_spi_dma_unmap(void *dev_ctx,
						unsigned long phy_addr,
						size_t len,
						enum nrf_wifi_osal_dma_dir dma_dir)
{
	struct nrf_wifi_bus_spi_dev_ctx *spi_dev_ctx = NULL;
	unsigned long virt_addr = 0;

	spi_dev_ctx = (struct nrf_wifi_bus_spi_dev_ctx *)dev_ctx;

	virt_addr = spi_dev_ctx->addr_pktram_base + (phy_addr - spi_dev_ctx->host_addr_base);

	return virt_addr;
}


#ifdef NRF_WIFI_LOW_POWER
static void nrf_wifi_bus_spi_ps_sleep(void *dev_ctx)
{
	struct nrf_wifi_bus_spi_dev_ctx *spi_dev_ctx = NULL;

	spi_dev_ctx = (struct nrf_wifi_bus_spi_dev_ctx *)dev_ctx;

	nrf_wifi_osal_bus_qspi_ps_sleep(spi_dev_ctx->os_spi_dev_ctx);
}


static void nrf_wifi_bus_spi_ps_wake(void *dev_ctx)
{
	struct nrf_wifi_bus_spi_dev_ctx *spi_dev_ctx = NULL;

	spi_dev_ctx = (struct nrf_wifi_bus_spi_dev_ctx *)dev_ctx;

	nrf_wifi_osal_bus_qspi_ps_wake(spi_dev_ctx->os_spi_dev_ctx);
}


static int nrf_wifi_bus_spi_ps_status(void *dev_ctx)
{
	struct nrf_wifi_bus_spi_dev_ctx *spi_dev_ctx = NULL;

	spi_dev_ctx = (struct nrf_wifi_bus_spi_dev_ctx *)dev_ctx;

	return nrf_wifi_osal_bus_qspi_ps_status(spi_dev_ctx->os_spi_dev_ctx);
}
#endif /* NRF_WIFI_LOW_POWER */


static struct nrf_wifi_bal_ops nrf_wifi_bus_spi_ops = {
	.init = &nrf_wifi_bus_spi_init,
	.deinit = &nrf_wifi_bus_spi_deinit,
	.dev_add = &nrf_wifi_bus_spi_dev_add,
	.dev_rem = &nrf_wifi_bus_spi_dev_rem,
	.dev_init = &nrf_wifi_bus_spi_dev_init,
	.dev_deinit = &nrf_wifi_bus_spi_dev_deinit,
	.read_word = &nrf_wifi_bus_spi_read_word,
	.write_word = &nrf_wifi_bus_spi_write_word,
	.read_block = &nrf_wifi_bus_spi_read_block,
	.write_block = &nrf_wifi_bus_spi_write_block,
	.dma_map = &nrf_wifi_bus_spi_dma_map,
	.dma_unmap = &nrf_wifi_bus_spi_dma_unmap,
#ifdef NRF_WIFI_LOW_POWER
	.rpu_ps_sleep = &nrf_wifi_bus_spi_ps_sleep,
	.rpu_ps_wake = &nrf_wifi_bus_spi_ps_wake,
	.rpu_ps_status = &nrf_wifi_bus_spi_ps_status,
#endif /* NRF_WIFI_LOW_POWER */
};


struct nrf_wifi_bal_ops *get_bus_ops(void)
{
	return &nrf_wifi_bus_spi_ops;
}