/*      $OpenBSD: gpioleds.c,v 1.4 2022/04/06 18:59:28 naddy Exp $      */
/*
 * Copyright (c) 2021 Klemens Nanni <kn@openbsd.org>
 *
 * Permission to use, copy, modify, and distribute this software for any
 * purpose with or without fee is hereby granted, provided that the above
 * copyright notice and this permission notice appear in all copies.
 *
 * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
 * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
 * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
 * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
 * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
 * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
 * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
 */

#include <sys/param.h>
#include <sys/systm.h>
#include <sys/device.h>
#include <sys/gpio.h>
#include <sys/malloc.h>

#include <machine/bus.h>
#include <machine/fdt.h>

#include <dev/gpio/gpiovar.h>
#include <dev/ofw/ofw_gpio.h>
#include <dev/ofw/ofw_pinctrl.h>
#include <dev/ofw/openfirm.h>
#include <dev/ofw/fdt.h>

struct gpioleds_softc {
        struct device    sc_dev;
        int              sc_node;
};

int     gpioleds_match(struct device *, void *, void *);
void    gpioleds_attach(struct device *, struct device *, void *);

const struct cfattach gpioleds_ca = {
        sizeof (struct gpioleds_softc), gpioleds_match, gpioleds_attach
};

struct cfdriver gpioleds_cd = {
        NULL, "gpioleds", DV_DULL
};

int
gpioleds_match(struct device *parent, void *match, void *aux)
{
        const struct fdt_attach_args    *faa = aux;

        return OF_is_compatible(faa->fa_node, "gpio-leds");
}

void
gpioleds_attach(struct device *parent, struct device *self, void *aux)
{
        struct fdt_attach_args  *faa = aux;
        uint32_t                *led_pin;
        char                    *function, *default_state;
        char                    *function_prop = "function";
        int                      function_len, default_state_len, gpios_len;
        int                      node, leds = 0;

        pinctrl_byname(faa->fa_node, "default");

        for (node = OF_child(faa->fa_node); node; node = OF_peer(node)) {
                function_len = OF_getproplen(node, function_prop);
                if (function_len <= 0) {
                        function_prop = "label";
                        function_len = OF_getproplen(node, function_prop);
                        if (function_len <= 0)
                                continue;
                }
                default_state_len = OF_getproplen(node, "default-state");
                if (default_state_len <= 0)
                        continue;
                gpios_len = OF_getproplen(node, "gpios");
                if (gpios_len <= 0)
                        continue;

                function = malloc(function_len, M_TEMP, M_WAITOK);
                OF_getprop(node, function_prop, function, function_len);
                default_state = malloc(default_state_len, M_TEMP, M_WAITOK);
                OF_getprop(node, "default-state", default_state, default_state_len);
                led_pin = malloc(gpios_len, M_TEMP, M_WAITOK);
                OF_getpropintarray(node, "gpios", led_pin, gpios_len);
                gpio_controller_config_pin(led_pin, GPIO_CONFIG_OUTPUT);
                if (strcmp(default_state, "on") == 0)
                        gpio_controller_set_pin(led_pin, 1);
                else if (strcmp(default_state, "off") == 0)
                        gpio_controller_set_pin(led_pin, 0);

                printf("%s \"%s\"", leds++ ? "," : ":", function);
                free(function, M_TEMP, function_len);
                free(default_state, M_TEMP, default_state_len);
                free(led_pin, M_TEMP, gpios_len);
        }

        if (leds == 0)
                printf(": no LEDs");
        printf("\n");
}