/*
 * Copyright 2021 Red Hat Inc.
 *
 * Permission is hereby granted, free of charge, to any person obtaining a
 * copy of this software and associated documentation files (the "Software"),
 * to deal in the Software without restriction, including without limitation
 * the rights to use, copy, modify, merge, publish, distribute, sublicense,
 * and/or sell copies of the Software, and to permit persons to whom the
 * Software is furnished to do so, subject to the following conditions:
 *
 * The above copyright notice and this permission notice shall be included in
 * all copies or substantial portions of the Software.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.  IN NO EVENT SHALL
 * THE COPYRIGHT HOLDER(S) OR AUTHOR(S) BE LIABLE FOR ANY CLAIM, DAMAGES OR
 * OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE,
 * ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR
 * OTHER DEALINGS IN THE SOFTWARE.
 */
#define nvkm_uconn(p) container_of((p), struct nvkm_conn, object)
#include "conn.h"
#include "outp.h"

#include <core/client.h>
#include <core/event.h>
#include <subdev/gpio.h>
#include <subdev/i2c.h>

#include <nvif/if0011.h>

static int
nvkm_uconn_uevent_gsp(struct nvkm_object *object, u64 token, u32 bits)
{
        union nvif_conn_event_args args;

        args.v0.version = 0;
        args.v0.types = 0;
        if (bits & NVKM_DPYID_PLUG)
                args.v0.types |= NVIF_CONN_EVENT_V0_PLUG;
        if (bits & NVKM_DPYID_UNPLUG)
                args.v0.types |= NVIF_CONN_EVENT_V0_UNPLUG;
        if (bits & NVKM_DPYID_IRQ)
                args.v0.types |= NVIF_CONN_EVENT_V0_IRQ;

        return object->client->event(token, &args, sizeof(args.v0));
}

static int
nvkm_uconn_uevent_aux(struct nvkm_object *object, u64 token, u32 bits)
{
        union nvif_conn_event_args args;

        args.v0.version = 0;
        args.v0.types = 0;
        if (bits & NVKM_I2C_PLUG)
                args.v0.types |= NVIF_CONN_EVENT_V0_PLUG;
        if (bits & NVKM_I2C_UNPLUG)
                args.v0.types |= NVIF_CONN_EVENT_V0_UNPLUG;
        if (bits & NVKM_I2C_IRQ)
                args.v0.types |= NVIF_CONN_EVENT_V0_IRQ;

        return object->client->event(token, &args, sizeof(args.v0));
}

static int
nvkm_uconn_uevent_gpio(struct nvkm_object *object, u64 token, u32 bits)
{
        union nvif_conn_event_args args;

        args.v0.version = 0;
        args.v0.types = 0;
        if (bits & NVKM_GPIO_HI)
                args.v0.types |= NVIF_CONN_EVENT_V0_PLUG;
        if (bits & NVKM_GPIO_LO)
                args.v0.types |= NVIF_CONN_EVENT_V0_UNPLUG;

        return object->client->event(token, &args, sizeof(args.v0));
}

static bool
nvkm_connector_is_dp_dms(u8 type)
{
        switch (type) {
        case DCB_CONNECTOR_DMS59_DP0:
        case DCB_CONNECTOR_DMS59_DP1:
                return true;
        default:
                return false;
        }
}

static int
nvkm_uconn_uevent(struct nvkm_object *object, void *argv, u32 argc, struct nvkm_uevent *uevent)
{
        struct nvkm_conn *conn = nvkm_uconn(object);
        struct nvkm_disp *disp = conn->disp;
        struct nvkm_device *device = disp->engine.subdev.device;
        struct nvkm_outp *outp;
        union nvif_conn_event_args *args = argv;
        u64 bits = 0;

        if (!uevent) {
                if (!disp->rm.client.gsp && conn->info.hpd == DCB_GPIO_UNUSED)
                        return -ENOSYS;
                return 0;
        }

        if (argc != sizeof(args->v0) || args->v0.version != 0)
                return -ENOSYS;

        list_for_each_entry(outp, &conn->disp->outps, head) {
                if (outp->info.connector == conn->index)
                        break;
        }

        if (&outp->head == &conn->disp->outps)
                return -EINVAL;

        if (disp->rm.client.gsp) {
                if (args->v0.types & NVIF_CONN_EVENT_V0_PLUG  ) bits |= NVKM_DPYID_PLUG;
                if (args->v0.types & NVIF_CONN_EVENT_V0_UNPLUG) bits |= NVKM_DPYID_UNPLUG;
                if (args->v0.types & NVIF_CONN_EVENT_V0_IRQ   ) bits |= NVKM_DPYID_IRQ;

                return nvkm_uevent_add(uevent, &disp->rm.event, outp->index, bits,
                                       nvkm_uconn_uevent_gsp);
        }

        if (outp->dp.aux && !outp->info.location) {
                if (args->v0.types & NVIF_CONN_EVENT_V0_PLUG  ) bits |= NVKM_I2C_PLUG;
                if (args->v0.types & NVIF_CONN_EVENT_V0_UNPLUG) bits |= NVKM_I2C_UNPLUG;
                if (args->v0.types & NVIF_CONN_EVENT_V0_IRQ   ) bits |= NVKM_I2C_IRQ;

                return nvkm_uevent_add(uevent, &device->i2c->event, outp->dp.aux->id, bits,
                                       nvkm_uconn_uevent_aux);
        }

        if (args->v0.types & NVIF_CONN_EVENT_V0_PLUG  ) bits |= NVKM_GPIO_HI;
        if (args->v0.types & NVIF_CONN_EVENT_V0_UNPLUG) bits |= NVKM_GPIO_LO;
        if (args->v0.types & NVIF_CONN_EVENT_V0_IRQ) {
                /* TODO: support DP IRQ on ANX9805 and remove this hack. */
                if (!outp->info.location && !nvkm_connector_is_dp_dms(conn->info.type))
                        return -EINVAL;
        }

        return nvkm_uevent_add(uevent, &device->gpio->event, conn->info.hpd, bits,
                               nvkm_uconn_uevent_gpio);
}

static void *
nvkm_uconn_dtor(struct nvkm_object *object)
{
        struct nvkm_conn *conn = nvkm_uconn(object);
        struct nvkm_disp *disp = conn->disp;

        spin_lock(&disp->client.lock);
        conn->object.func = NULL;
        spin_unlock(&disp->client.lock);
        return NULL;
}

static const struct nvkm_object_func
nvkm_uconn = {
        .dtor = nvkm_uconn_dtor,
        .uevent = nvkm_uconn_uevent,
};

int
nvkm_uconn_new(const struct nvkm_oclass *oclass, void *argv, u32 argc, struct nvkm_object **pobject)
{
        struct nvkm_disp *disp = nvkm_udisp(oclass->parent);
        struct nvkm_conn *cont, *conn = NULL;
        union nvif_conn_args *args = argv;
        int ret;

        if (argc != sizeof(args->v0) || args->v0.version != 0)
                return -ENOSYS;

        list_for_each_entry(cont, &disp->conns, head) {
                if (cont->index == args->v0.id) {
                        conn = cont;
                        break;
                }
        }

        if (!conn)
                return -EINVAL;

        ret = -EBUSY;
        spin_lock(&disp->client.lock);
        if (!conn->object.func) {
                switch (conn->info.type) {
                /* VGA */
                case DCB_CONNECTOR_DVI_A        :
                case DCB_CONNECTOR_POD_VGA      :
                case DCB_CONNECTOR_VGA          : args->v0.type = NVIF_CONN_V0_VGA; break;

                /* TV */
                case DCB_CONNECTOR_TV_0         :
                case DCB_CONNECTOR_TV_1         :
                case DCB_CONNECTOR_TV_2         :
                case DCB_CONNECTOR_TV_SCART     :
                case DCB_CONNECTOR_TV_SCART_D   :
                case DCB_CONNECTOR_TV_DTERM     :
                case DCB_CONNECTOR_POD_TV_3     :
                case DCB_CONNECTOR_POD_TV_1     :
                case DCB_CONNECTOR_POD_TV_0     :
                case DCB_CONNECTOR_TV_3         : args->v0.type = NVIF_CONN_V0_TV; break;

                /* DVI */
                case DCB_CONNECTOR_DVI_I_TV_1   :
                case DCB_CONNECTOR_DVI_I_TV_0   :
                case DCB_CONNECTOR_DVI_I_TV_2   :
                case DCB_CONNECTOR_DVI_ADC      :
                case DCB_CONNECTOR_DMS59_0      :
                case DCB_CONNECTOR_DMS59_1      :
                case DCB_CONNECTOR_DVI_I        : args->v0.type = NVIF_CONN_V0_DVI_I; break;
                case DCB_CONNECTOR_TMDS         :
                case DCB_CONNECTOR_DVI_D        : args->v0.type = NVIF_CONN_V0_DVI_D; break;

                /* LVDS */
                case DCB_CONNECTOR_LVDS         : args->v0.type = NVIF_CONN_V0_LVDS; break;
                case DCB_CONNECTOR_LVDS_SPWG    : args->v0.type = NVIF_CONN_V0_LVDS_SPWG; break;

                /* DP */
                case DCB_CONNECTOR_DMS59_DP0    :
                case DCB_CONNECTOR_DMS59_DP1    :
                case DCB_CONNECTOR_DP           :
                case DCB_CONNECTOR_mDP          :
                case DCB_CONNECTOR_USB_C        : args->v0.type = NVIF_CONN_V0_DP; break;
                case DCB_CONNECTOR_eDP          : args->v0.type = NVIF_CONN_V0_EDP; break;

                /* HDMI */
                case DCB_CONNECTOR_HDMI_0       :
                case DCB_CONNECTOR_HDMI_1       :
                case DCB_CONNECTOR_HDMI_C       : args->v0.type = NVIF_CONN_V0_HDMI; break;

                /*
                 * Dock & unused outputs.
                 * BNC, SPDIF, WFD, and detached LVDS go here.
                 */
                default:
                        nvkm_warn(&disp->engine.subdev,
                                  "unimplemented connector type 0x%02x\n",
                                  conn->info.type);
                        args->v0.type = NVIF_CONN_V0_VGA;
                        ret = -EINVAL;
                        break;
                }

                nvkm_object_ctor(&nvkm_uconn, oclass, &conn->object);
                *pobject = &conn->object;
                ret = 0;
        }
        spin_unlock(&disp->client.lock);
        return ret;
}