/*-
 * SPDX-License-Identifier: BSD-2-Clause
 *
 * Copyright 2017 Emmanuel Vadot <manu@freebsd.org>
 * Copyright (c) 2024 The FreeBSD Foundation
 *
 * Portions of this software were developed by Mitchell Horne
 * <mhorne@FreeBSD.org> under sponsorship from the FreeBSD Foundation.
 */

#include <sys/param.h>
#include <sys/kernel.h>
#include <sys/bus.h>
#include <sys/module.h>
#include <sys/queue.h>
#include <sys/taskqueue.h>

#include <machine/bus.h>

#include <dev/mmc/bridge.h>
#include <dev/mmc/mmc_fdt_helpers.h>

#include <dev/mmc/host/dwmmc_var.h>

#include <dev/ofw/ofw_bus_subr.h>

#include "opt_mmccam.h"

enum dwmmc_type {
	DWMMC_GENERIC = 1,
	DWMMC_JH7110
};

static struct ofw_compat_data compat_data[] = {
	{"snps,dw-mshc",	DWMMC_GENERIC},
	{"starfive,jh7110-mmc",	DWMMC_JH7110},
	{NULL,			0}
};

static int dwmmc_starfive_update_ios(struct dwmmc_softc *sc,
    struct mmc_ios *ios)
{
	int err;

	if (ios->clock != 0 && ios->clock != sc->bus_hz) {
		err = clk_set_freq(sc->ciu, ios->clock, CLK_SET_ROUND_DOWN);
		if (err != 0) {
			printf("%s, Failed to set freq for ciu clock\n",
			    __func__);
			return (err);
		}
		sc->bus_hz = ios->clock;
	}

	return (0);
}

static int
starfive_dwmmc_probe(device_t dev)
{
	phandle_t node;
	int type;

	if (!ofw_bus_status_okay(dev))
		return (ENXIO);

	type = ofw_bus_search_compatible(dev, compat_data)->ocd_data;
	if (type == 0)
		return (ENXIO);

	/*
	 * If we matched the generic compat string, check the top-level board
	 * compatible, to ensure we should actually use the starfive driver.
	 */
	if (type == DWMMC_GENERIC) {
		node = OF_finddevice("/");
		if (!ofw_bus_node_is_compatible(node, "starfive,jh7110"))
			return (ENXIO);
	}

	device_set_desc(dev, "Synopsys DesignWare Mobile Storage "
	    "Host Controller (StarFive)");

	return (BUS_PROBE_VENDOR);
}

static int
starfive_dwmmc_attach(device_t dev)
{
	struct dwmmc_softc *sc;

	sc = device_get_softc(dev);
	sc->update_ios = &dwmmc_starfive_update_ios;

	return (dwmmc_attach(dev));
}

static device_method_t starfive_dwmmc_methods[] = {
	/* bus interface */
	DEVMETHOD(device_probe,		starfive_dwmmc_probe),
	DEVMETHOD(device_attach,	starfive_dwmmc_attach),
	DEVMETHOD(device_detach,	dwmmc_detach),

	DEVMETHOD_END
};

DEFINE_CLASS_1(starfive_dwmmc, starfive_dwmmc_driver, starfive_dwmmc_methods,
    sizeof(struct dwmmc_softc), dwmmc_driver);

DRIVER_MODULE(starfive_dwmmc, simplebus, starfive_dwmmc_driver, 0, 0);

#ifndef MMCCAM
MMC_DECLARE_BRIDGE(starfive_dwmmc);
#endif