/*      $NetBSD: akbd.c,v 1.49 2025/01/13 16:17:36 riastradh Exp $      */

/*
 * Copyright (C) 1998   Colin Wood
 * All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 * 1. Redistributions of source code must retain the above copyright
 *    notice, this list of conditions and the following disclaimer.
 * 2. Redistributions in binary form must reproduce the above copyright
 *    notice, this list of conditions and the following disclaimer in the
 *    documentation and/or other materials provided with the distribution.
 * 3. All advertising materials mentioning features or use of this software
 *    must display the following acknowledgement:
 *      This product includes software developed by Colin Wood.
 * 4. The name of the author may not be used to endorse or promote products
 *    derived from this software without specific prior written permission.
 *
 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
 * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
 * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
 * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
 * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
 * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 */

#include <sys/cdefs.h>
__KERNEL_RCSID(0, "$NetBSD: akbd.c,v 1.49 2025/01/13 16:17:36 riastradh Exp $");

#include <sys/param.h>
#include <sys/device.h>
#include <sys/fcntl.h>
#include <sys/poll.h>
#include <sys/select.h>
#include <sys/proc.h>
#include <sys/signalvar.h>
#include <sys/systm.h>

#include <dev/wscons/wsconsio.h>
#include <dev/wscons/wskbdvar.h>
#include <dev/wscons/wsksymdef.h>
#include <dev/wscons/wsksymvar.h>
#include <dev/ofw/openfirm.h>

#include <dev/adb/adb_keymap.h>

#include <machine/autoconf.h>
#define KEYBOARD_ARRAY
#include <machine/keyboard.h>

#include <macppc/dev/adbvar.h>
#include <macppc/dev/aedvar.h>
#include <macppc/dev/akbdvar.h>
#include <macppc/dev/pm_direct.h>

#include "aed.h"

/*
 * Function declarations.
 */
static int      akbdmatch(device_t, cfdata_t, void *);
static void     akbdattach(device_t, device_t, void *);
static void     kbd_processevent(adb_event_t *event, struct akbd_softc *);
#ifdef notyet
static u_char   getleds(int);
static int      setleds(struct akbd_softc *, u_char);
static void     blinkleds(struct akbd_softc *);
#endif

/* Driver definition. */
CFATTACH_DECL_NEW(akbd, sizeof(struct akbd_softc),
    akbdmatch, akbdattach, NULL, NULL);

extern struct cfdriver akbd_cd;

int akbd_enable(void *, int);
void akbd_set_leds(void *, int);
int akbd_ioctl(void *, u_long, void *, int, struct lwp *);

struct wskbd_accessops akbd_accessops = {
        akbd_enable,
        akbd_set_leds,
        akbd_ioctl,
};

void akbd_cngetc(void *, u_int *, int *);
void akbd_cnpollc(void *, int);

struct wskbd_consops akbd_consops = {
        akbd_cngetc,
        akbd_cnpollc,
};

struct wskbd_mapdata akbd_keymapdata = {
        akbd_keydesctab,
#ifdef AKBD_LAYOUT
        AKBD_LAYOUT,
#else
        KB_US,
#endif
};

static int akbd_is_console;
static int akbd_console_attached;
static int pcmcia_soft_eject;

static int
akbdmatch(device_t parent, cfdata_t cf, void *aux)
{
        struct adb_attach_args *aa_args = aux;

        if (aa_args->origaddr == ADBADDR_KBD)
                return 1;
        else
                return 0;
}

static void
akbdattach(device_t parent, device_t self, void *aux)
{
        ADBSetInfoBlock adbinfo;
        struct akbd_softc *sc = device_private(self);
        struct adb_attach_args *aa_args = aux;
        int error, kbd_done;
        short cmd;
        u_char buffer[9];
        struct wskbddev_attach_args a;

        /* ohare based models have soft ejectable card slot. */
        if (OF_finddevice("/bandit/ohare") != -1)
                pcmcia_soft_eject = 1;

        sc->origaddr = aa_args->origaddr;
        sc->adbaddr = aa_args->adbaddr;
        sc->handler_id = aa_args->handler_id;

        sc->sc_leds = (u_int8_t)0x00;   /* initially off */

        adbinfo.siServiceRtPtr = (Ptr)kbd_adbcomplete;
        adbinfo.siDataAreaAddr = (void *)sc;

        switch (sc->handler_id) {
        case ADB_STDKBD:
                printf("standard keyboard\n");
                break;
        case ADB_ISOKBD:
                printf("standard keyboard (ISO layout)\n");
                break;
        case ADB_EXTKBD:
                cmd = ADBTALK(sc->adbaddr, 1);
                kbd_done =
                    (adb_op_sync((Ptr)buffer, NULL, (Ptr)0, cmd) == 0);

                /* Ignore Logitech MouseMan/Trackman pseudo keyboard */
                if (kbd_done && buffer[1] == 0x9a && buffer[2] == 0x20) {
                        printf("Mouseman (non-EMP) pseudo keyboard\n");
                        adbinfo.siServiceRtPtr = (Ptr)0;
                        adbinfo.siDataAreaAddr = (Ptr)0;
                } else if (kbd_done && buffer[1] == 0x9a && buffer[2] == 0x21) {
                        printf("Trackman (non-EMP) pseudo keyboard\n");
                        adbinfo.siServiceRtPtr = (Ptr)0;
                        adbinfo.siDataAreaAddr = (Ptr)0;
                } else {
                        printf("extended keyboard\n");
#ifdef notyet
                        blinkleds(sc);
#endif
                }
                break;
        case ADB_EXTISOKBD:
                printf("extended keyboard (ISO layout)\n");
#ifdef notyet
                blinkleds(sc);
#endif
                break;
        case ADB_KBDII:
                printf("keyboard II\n");
                break;
        case ADB_ISOKBDII:
                printf("keyboard II (ISO layout)\n");
                break;
        case ADB_PBKBD:
                printf("PowerBook keyboard\n");
                break;
        case ADB_PBISOKBD:
                printf("PowerBook keyboard (ISO layout)\n");
                break;
        case ADB_ADJKPD:
                printf("adjustable keypad\n");
                break;
        case ADB_ADJKBD:
                printf("adjustable keyboard\n");
                break;
        case ADB_ADJISOKBD:
                printf("adjustable keyboard (ISO layout)\n");
                break;
        case ADB_ADJJAPKBD:
                printf("adjustable keyboard (Japanese layout)\n");
                break;
        case ADB_PBEXTISOKBD:
                printf("PowerBook extended keyboard (ISO layout)\n");
                break;
        case ADB_PBEXTJAPKBD:
                printf("PowerBook extended keyboard (Japanese layout)\n");
                break;
        case ADB_JPKBDII:
                printf("keyboard II (Japanese layout)\n");
                break;
        case ADB_PBEXTKBD:
                printf("PowerBook extended keyboard\n");
                break;
        case ADB_DESIGNKBD:
                printf("extended keyboard\n");
#ifdef notyet
                blinkleds(sc);
#endif
                break;
        case ADB_PBJPKBD:
                printf("PowerBook keyboard (Japanese layout)\n");
                break;
        case ADB_PBG3KBD:
                printf("PowerBook G3 keyboard\n");
                break;
        case ADB_PBG3JPKBD:
                printf("PowerBook G3 keyboard (Japanese layout)\n");
                break;
        default:
                printf("mapped device (%d)\n", sc->handler_id);
                break;
        }
        error = SetADBInfo(&adbinfo, sc->adbaddr);
#ifdef ADB_DEBUG
        if (adb_debug)
                printf("akbd: returned %d from SetADBInfo\n", error);
#endif

        if (akbd_is_console && !akbd_console_attached) {
                wskbd_cnattach(&akbd_consops, sc, &akbd_keymapdata);
                akbd_console_attached = 1;
        }

        a.console = akbd_is_console;
        a.keymap = &akbd_keymapdata;
        a.accessops = &akbd_accessops;
        a.accesscookie = sc;

        sc->sc_wskbddev = config_found(self, &a, wskbddevprint, CFARGS_NONE);
}

/*
 * Handle putting the keyboard data received from the ADB into
 * an ADB event record.
 */
void
kbd_adbcomplete(uint8_t *buffer, uint8_t *data_area, int adb_command)
{
        adb_event_t event;
        struct akbd_softc *ksc;
        int adbaddr;
#ifdef ADB_DEBUG
        int i;

        if (adb_debug)
                printf("adb: transaction completion\n");
#endif

        adbaddr = ADB_CMDADDR(adb_command);
        ksc = (struct akbd_softc *)data_area;

        event.addr = adbaddr;
        event.hand_id = ksc->handler_id;
        event.def_addr = ksc->origaddr;
        event.byte_count = buffer[0];
        memcpy(event.bytes, buffer + 1, event.byte_count);

#ifdef ADB_DEBUG
        if (adb_debug) {
                printf("akbd: from %d at %d (org %d) %d:", event.addr,
                    event.hand_id, event.def_addr, buffer[0]);
                for (i = 1; i <= buffer[0]; i++)
                        printf(" %x", buffer[i]);
                printf("\n");
        }
#endif

        microtime(&event.timestamp);

        kbd_processevent(&event, ksc);
}

/*
 * Given a keyboard ADB event, record the keycodes and call the key
 * repeat handler, optionally passing the event through the mouse
 * button emulation handler first.
 */
static void
kbd_processevent(adb_event_t *event, struct akbd_softc *ksc)
{
        adb_event_t new_event;

        new_event = *event;
        new_event.u.k.key = event->bytes[0];
        new_event.bytes[1] = 0xff;
#if NAED > 0
        int result;

        if ((result = aed_input(&new_event)) != 0)
                return;
#endif
        kbd_intr(&new_event);
#if NAED > 0
        aed_input(&new_event);
#endif
        if (event->bytes[1] != 0xff) {
                new_event.u.k.key = event->bytes[1];
                new_event.bytes[0] = event->bytes[1];
                new_event.bytes[1] = 0xff;
                kbd_intr(&new_event);
#if NAED > 0
                aed_input(&new_event);
#endif
        }

}

#ifdef notyet
/*
 * Get the actual hardware LED state and convert it to softc format.
 */
static u_char
getleds(int addr)
{
        short cmd;
        u_char buffer[9], leds;

        leds = 0x00;    /* all off */
        buffer[0] = 0;

        /* talk R2 */
        cmd = ADBTALK(addr, 2);
        if (adb_op_sync((Ptr)buffer, (Ptr)0, (Ptr)0, cmd) == 0 &&
            buffer[0] > 0)
                leds = ~(buffer[2]) & 0x07;

        return (leds);
}

/*
 * Set the keyboard LED's.
 *
 * Automatically translates from ioctl/softc format to the
 * actual keyboard register format
 */
static int
setleds(struct akbd_softc *ksc, u_char leds)
{
        int addr;
        short cmd;
        u_char buffer[9];

        if ((leds & 0x07) == (ksc->sc_leds & 0x07))
                return (0);

        addr = ksc->adbaddr;
        buffer[0] = 0;

        cmd = ADBTALK(addr, 2);
        if (adb_op_sync((Ptr)buffer, (Ptr)0, (Ptr)0, cmd) || buffer[0] == 0)
                return (EIO);

        leds = ~leds & 0x07;
        buffer[2] &= 0xf8;
        buffer[2] |= leds;

        cmd = ADBLISTEN(addr, 2);
        adb_op_sync((Ptr)buffer, (Ptr)0, (Ptr)0, cmd);

        cmd = ADBTALK(addr, 2);
        if (adb_op_sync((Ptr)buffer, (Ptr)0, (Ptr)0, cmd) || buffer[0] == 0)
                return (EIO);

        ksc->sc_leds = ~((u_int8_t)buffer[2]) & 0x07;

        if ((buffer[2] & 0xf8) != leds)
                return (EIO);
        else
                return (0);
}

/*
 * Toggle all of the LED's on and off, just for show.
 */
static void
blinkleds(struct akbd_softc *ksc)
{
        int addr, i;
        u_char blinkleds, origleds;

        addr = ksc->adbaddr;
        origleds = getleds(addr);
        blinkleds = LED_NUMLOCK | LED_CAPSLOCK | LED_SCROLL_LOCK;

        (void)setleds(ksc, blinkleds);

        for (i = 0; i < 10000; i++)
                delay(50);

        /* make sure that we restore the LED settings */
        i = 10;
        do {
                (void)setleds(ksc, (u_char)0x00);
        } while (setleds(ksc, (u_char)0x00) && (i-- > 0));

        return;
}
#endif

int
akbd_enable(void *v, int on)
{
        return 0;
}

void
akbd_set_leds(void *v, int on)
{
}

int
akbd_ioctl(void *v, u_long cmd, void *data, int flag, struct lwp *l)
{
#ifdef WSDISPLAY_COMPAT_RAWKBD
        struct akbd_softc *sc = (struct akbd_softc *) v;
#endif

        switch (cmd) {

        case WSKBDIO_GTYPE:
                *(int *)data = WSKBD_TYPE_ADB;
                return 0;
        case WSKBDIO_SETLEDS:
                return 0;
        case WSKBDIO_GETLEDS:
                *(int *)data = 0;
                return 0;
#ifdef WSDISPLAY_COMPAT_RAWKBD
        case WSKBDIO_SETMODE:
                sc->sc_rawkbd = *(int *)data == WSKBD_RAW;
                return 0;
#endif
        }
        /* kbdioctl(...); */

        return EPASSTHROUGH;
}

extern int adb_polling;

void
kbd_passup(struct akbd_softc *sc,int key)
{
        if (sc->sc_polling) {
                if (sc->sc_npolledkeys <
                        (sizeof(sc->sc_polledkeys)/sizeof(unsigned char))) {
                        sc->sc_polledkeys[sc->sc_npolledkeys++] = key;
                }
#ifdef ADB_DEBUG
                else {
                        printf("akbd: dumping polled key 0x%02x\n",key);
                }
#endif
#ifdef WSDISPLAY_COMPAT_RAWKBD
        } else if (sc->sc_rawkbd) {
                char cbuf[2];
                int s;
                int j = 0;
                int c = keyboard[ADBK_KEYVAL(key)][3];

                if (c == 0)                     /* XXX */
                        return;

                if (c & 0x80)
                        cbuf[j++] = 0xe0;

                cbuf[j++] = (c & 0x7f) | (ADBK_PRESS(key)? 0 : 0x80);

                s = spltty();
                wskbd_rawinput(sc->sc_wskbddev, cbuf, j);
                splx(s);
#endif
        } else {
                int press, val;
                int type;

                press = ADBK_PRESS(key);
                val = ADBK_KEYVAL(key);

                type = press ? WSCONS_EVENT_KEY_DOWN : WSCONS_EVENT_KEY_UP;

                wskbd_input(sc->sc_wskbddev, type, val);
        }
}

int
kbd_intr(void *arg)
{
        adb_event_t *event = arg;
        int key;
#ifdef CAPS_IS_CONTROL
        static int shift;
#endif

        struct akbd_softc *sc = device_lookup_private(&akbd_cd, 0);

        key = event->u.k.key;

#ifdef CAPS_IS_CONTROL
        /*
         * Caps lock is weird. The key sequence generated is:
         * press:   down(57) [57]  (LED turns on)
         * release: up(127)  [255]
         * press:   up(127)  [255]
         * release: up(57)   [185] (LED turns off)
         */
        if ((key == 57) || (key == 185))
                shift = 0;

        if (key == 255) {
                if (shift == 0) {
                        key = 185;
                        shift = 1;
                } else {
                        key = 57;
                        shift = 0;
                }
        }
#endif

        switch (key) {
#ifndef CAPS_IS_CONTROL
        case 57:        /* Caps Lock pressed */
        case 185:       /* Caps Lock released */
                key = ADBK_KEYDOWN(ADBK_KEYVAL(key));
                kbd_passup(sc,key);
                key = ADBK_KEYUP(ADBK_KEYVAL(key));
                break;
#endif
        case 245:
                if (pcmcia_soft_eject)
                        pm_eject_pcmcia(0);
                break;
        case 244:
                if (pcmcia_soft_eject)
                        pm_eject_pcmcia(1);
                break;
        }

        kbd_passup(sc,key);

        return 0;
}

int
akbd_cnattach(void)
{

        akbd_is_console = 1;
        return 0;
}

void
akbd_cngetc(void *v, u_int *type, int *data)
{
        int key, press, val;
        int s;
        struct akbd_softc *sc = v;

        s = splhigh();

        KASSERT(sc->sc_polling);
        KASSERT(adb_polling);

        while (sc->sc_npolledkeys == 0) {
                adb_intr(NULL);
                DELAY(10000);                           /* XXX */
        }

        splx(s);

        key = sc->sc_polledkeys[0];
        sc->sc_npolledkeys--;
        memmove(sc->sc_polledkeys,sc->sc_polledkeys+1,
                sc->sc_npolledkeys * sizeof(unsigned char));

        press = ADBK_PRESS(key);
        val = ADBK_KEYVAL(key);

        *data = val;
        *type = press ? WSCONS_EVENT_KEY_DOWN : WSCONS_EVENT_KEY_UP;
}

void
akbd_cnpollc(void *v, int on)
{
        struct akbd_softc *sc = v;
        sc->sc_polling = on;
        if (!on) {
                int i;
                for(i=0;i<sc->sc_npolledkeys;i++) {
                        kbd_passup(sc,sc->sc_polledkeys[i]);
                }
                sc->sc_npolledkeys = 0;
        }
        adb_polling = on;
}