/*      $NetBSD: wskbd.c,v 1.5 2021/01/31 22:45:47 rillig Exp $ */

/*-
* Copyright (c) 2003 The NetBSD Foundation, Inc.
* All rights reserved.
*
* This code is derived from software contributed to The NetBSD Foundation
* by David Laight.
*
* 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.
*
* THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS
* ``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 FOUNDATION OR CONTRIBUTORS
* 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>
__RCSID("$NetBSD: wskbd.c,v 1.5 2021/01/31 22:45:47 rillig Exp $");

#include <unistd.h>
#include <stdlib.h>
#include <fcntl.h>
#include <sys/ioctl.h>
#include <dev/wscons/wsconsio.h>
#include <dev/wscons/wsksymdef.h>

#include "defs.h"
#include "menu_defs.h"
#include "msg_defs.h"
void save_kb_encoding(void);

/* wscons setup for sysinst */

static const char *kbd_name = 0;
static int kb_default = 0;

struct kb_types {
       kbd_t           kb_encoding;
       const char      *kb_enc_txt;
       const char      *kb_name;
};

/* Types and names of keyboards, maybe the names should be translated... */
static const struct kb_types kb_types[] = {
#define KB_sysinst(tag, tagf, value, cc, ccf, country) \
       {tag | tagf, cc ccf, country},
KB_ENC_FUN(KB_sysinst)
{KB_US | KB_COLEMAK, "us" ".colemak", "US-Colemak"},
{KB_US | KB_DVORAK, "us" ".dvorak", "US-Dvorak"}
};

static int
set_kb_encoding(menudesc *m, void *arg)
{
       int fd = *(int *)arg;
       const struct kb_types *kbt = kb_types + m->cursel;

       if (kbt->kb_encoding != KB_USER) {
               ioctl(fd, WSKBDIO_SETENCODING, &kbt->kb_encoding);
               kbd_name = kbt->kb_enc_txt;
       }
       return 1;
}

static void
set_kb_default(menudesc *m, void *arg)
{
       m->cursel = kb_default;
}

void
get_kb_encoding(void)
{
       int fd;
       unsigned int i;
       int kb_menu;
       kbd_t kbdencoding;
       menu_ent opt[__arraycount(kb_types)];
       const char *dflt = msg_string(MSG_kb_default);

       /*
        * Check if we are running on a wscons keyboard at all,
        * do not bother to try changing the layout if not.
        */
       if (ioctl(0, WSKBDIO_GTYPE,  &i) == -1)
               return;

       memset(opt, 0, sizeof(opt));
       fd = open("/dev/wskbd0", O_WRONLY);
       if (fd < 0)
               return;
       if (ioctl(fd, WSKBDIO_GETENCODING, &kbdencoding) >=  0) {
               memset(&opt, 0, sizeof opt);
               for (i = 0; i < __arraycount(opt); i++) {
                       if (kb_types[i].kb_encoding == KB_USER)
                               opt[0].opt_name = MSG_unchanged;
                       else {
                               opt[i].opt_name = kb_types[i].kb_name;
                               if (strcmp(kb_types[i].kb_name, dflt) == 0)
                                       kb_default = i;
                       }
                       opt[i].opt_action = set_kb_encoding;
               }
               kb_menu = new_menu(MSG_Keyboard_type, opt, __arraycount(opt),
                       -1, -8, 0, 0,
                       MC_SCROLL | MC_NOEXITOPT,
                       set_kb_default, NULL, NULL, NULL, NULL);
               if (kb_menu != -1) {
                       msg_display(MSG_hello);
                       process_menu(kb_menu, &fd);
                       free_menu(kb_menu);
               }
       }
       close(fd);
}

void
save_kb_encoding(void)
{
       const char *tp = target_prefix();

       if (kbd_name == NULL)
               return;
       /*
        * Put the keyboard encoding into the wscons.conf file. Either:
        * 1) replace an existing line
        * 2) replace a commented out line
        * or
        * 3) add a line to the end of the file
        */
       run_program(0, "sed -an -e 'H;$!d;g'"
           " -e 's/\\nencoding [a-zA-Z0-9.]*\\n/\\\nencoding %s\\\n/; t done'"
           " -e 's/\\n#encoding [a-zA-Z0-9.]*\\n/\\\nencoding %s\\\n/; t done'"
           " -e 's/$/\\\nencoding %s/'"
           " -e ':done'"
           " -e 'w %s/etc/wscons.conf' %s/etc/wscons.conf",
           kbd_name, kbd_name, kbd_name, tp, tp);
}