// SPDX-License-Identifier: GPL-2.0
/*
 * Copyright 2023 Google LLC.
 */

#include <linux/device.h>
#include <linux/spmi.h>

static void devm_spmi_controller_release(struct device *parent, void *res)
{
        spmi_controller_put(*(struct spmi_controller **)res);
}

struct spmi_controller *devm_spmi_controller_alloc(struct device *parent, size_t size)
{
        struct spmi_controller **ptr, *ctrl;

        ptr = devres_alloc(devm_spmi_controller_release, sizeof(*ptr), GFP_KERNEL);
        if (!ptr)
                return ERR_PTR(-ENOMEM);

        ctrl = spmi_controller_alloc(parent, size);
        if (IS_ERR(ctrl)) {
                devres_free(ptr);
                return ctrl;
        }

        *ptr = ctrl;
        devres_add(parent, ptr);

        return ctrl;
}
EXPORT_SYMBOL_GPL(devm_spmi_controller_alloc);

static void devm_spmi_controller_remove(struct device *parent, void *res)
{
        spmi_controller_remove(*(struct spmi_controller **)res);
}

int devm_spmi_controller_add(struct device *parent, struct spmi_controller *ctrl)
{
        struct spmi_controller **ptr;
        int ret;

        ptr = devres_alloc(devm_spmi_controller_remove, sizeof(*ptr), GFP_KERNEL);
        if (!ptr)
                return -ENOMEM;

        ret = spmi_controller_add(ctrl);
        if (ret) {
                devres_free(ptr);
                return ret;
        }

        *ptr = ctrl;
        devres_add(parent, ptr);

        return 0;

}
EXPORT_SYMBOL_GPL(devm_spmi_controller_add);

MODULE_LICENSE("GPL");
MODULE_DESCRIPTION("SPMI devres helpers");