/*      $NetBSD: keymacro.c,v 1.25 2025/01/03 00:40:08 rillig Exp $     */

/*-
* Copyright (c) 1992, 1993
*      The Regents of the University of California.  All rights reserved.
*
* This code is derived from software contributed to Berkeley by
* Christos Zoulas of Cornell University.
*
* 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. Neither the name of the University nor the names of its contributors
*    may be used to endorse or promote products derived from this software
*    without specific prior written permission.
*
* THIS SOFTWARE IS PROVIDED BY THE REGENTS 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 REGENTS 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 "config.h"
#if !defined(lint) && !defined(SCCSID)
#if 0
static char sccsid[] = "@(#)key.c       8.1 (Berkeley) 6/4/93";
#else
__RCSID("$NetBSD: keymacro.c,v 1.25 2025/01/03 00:40:08 rillig Exp $");
#endif
#endif /* not lint && not SCCSID */

/*
* keymacro.c: This module contains the procedures for maintaining
*             the extended-key map.
*
*      An extended-key (key) is a sequence of keystrokes introduced
*      with a sequence introducer and consisting of an arbitrary
*      number of characters.  This module maintains a map (the
*      el->el_keymacro.map)
*      to convert these extended-key sequences into input strs
*      (XK_STR) or editor functions (XK_CMD).
*
*      Warning:
*        If key is a substr of some other keys, then the longer
*        keys are lost!!  That is, if the keys "abcd" and "abcef"
*        are in el->el_keymacro.map, adding the key "abc" will cause
*        the first two definitions to be lost.
*
*      Restrictions:
*      -------------
*      1) It is not possible to have one key that is a
*         substr of another.
*/
#include <stdlib.h>
#include <string.h>

#include "el.h"
#include "fcns.h"

/*
* The Nodes of the el->el_keymacro.map.  The el->el_keymacro.map is a
* linked list of these node elements
*/
struct keymacro_node_t {
       wchar_t          ch;            /* single character of key       */
       int              type;          /* node type                     */
       keymacro_value_t val;           /* command code or pointer to str,  */
                                       /* if this is a leaf             */
       struct keymacro_node_t *next;   /* ptr to next char of this key  */
       struct keymacro_node_t *sibling;/* ptr to another key with same prefix*/
};

static int               node_trav(EditLine *, keymacro_node_t *, wchar_t *,
   keymacro_value_t *);
static int               node__try(EditLine *, keymacro_node_t *,
   const wchar_t *, keymacro_value_t *, int);
static keymacro_node_t  *node__get(wint_t);
static void              node__free(keymacro_node_t *);
static void              node__put(EditLine *, keymacro_node_t *);
static int               node__delete(EditLine *, keymacro_node_t **,
   const wchar_t *);
static int               node_lookup(EditLine *, const wchar_t *,
   keymacro_node_t *, size_t);
static int               node_enum(EditLine *, keymacro_node_t *, size_t);

#define KEY_BUFSIZ      EL_BUFSIZ


/* keymacro_init():
*      Initialize the key maps
*/
libedit_private int
keymacro_init(EditLine *el)
{

       el->el_keymacro.buf = el_calloc(KEY_BUFSIZ,
           sizeof(*el->el_keymacro.buf));
       if (el->el_keymacro.buf == NULL)
               return -1;
       el->el_keymacro.map = NULL;
       keymacro_reset(el);
       return 0;
}

/* keymacro_end():
*      Free the key maps
*/
libedit_private void
keymacro_end(EditLine *el)
{

       el_free(el->el_keymacro.buf);
       el->el_keymacro.buf = NULL;
       node__free(el->el_keymacro.map);
}


/* keymacro_map_cmd():
*      Associate cmd with a key value
*/
libedit_private keymacro_value_t *
keymacro_map_cmd(EditLine *el, int cmd)
{

       el->el_keymacro.val.cmd = (el_action_t) cmd;
       return &el->el_keymacro.val;
}


/* keymacro_map_str():
*      Associate str with a key value
*/
libedit_private keymacro_value_t *
keymacro_map_str(EditLine *el, wchar_t *str)
{

       el->el_keymacro.val.str = str;
       return &el->el_keymacro.val;
}


/* keymacro_reset():
*      Takes all nodes on el->el_keymacro.map and puts them on free list.
*      Then initializes el->el_keymacro.map with arrow keys
*      [Always bind the ansi arrow keys?]
*/
libedit_private void
keymacro_reset(EditLine *el)
{

       node__put(el, el->el_keymacro.map);
       el->el_keymacro.map = NULL;
       return;
}


/* keymacro_get():
*      Calls the recursive function with entry point el->el_keymacro.map
*      Looks up *ch in map and then reads characters until a
*      complete match is found or a mismatch occurs. Returns the
*      type of the match found (XK_STR or XK_CMD).
*      Returns NULL in val.str and XK_STR for no match.
*      Returns XK_NOD for end of file or read error.
*      The last character read is returned in *ch.
*/
libedit_private int
keymacro_get(EditLine *el, wchar_t *ch, keymacro_value_t *val)
{

       return node_trav(el, el->el_keymacro.map, ch, val);
}


/* keymacro_add():
*      Adds key to the el->el_keymacro.map and associates the value in
*      val with it. If key is already is in el->el_keymacro.map, the new
*      code is applied to the existing key. Ntype specifies if code is a
*      command, an out str or a unix command.
*/
libedit_private void
keymacro_add(EditLine *el, const wchar_t *key, keymacro_value_t *val,
   int ntype)
{

       if (key[0] == '\0') {
               (void) fprintf(el->el_errfile,
                   "keymacro_add: Null extended-key not allowed.\n");
               return;
       }
       if (ntype == XK_CMD && val->cmd == ED_SEQUENCE_LEAD_IN) {
               (void) fprintf(el->el_errfile,
                   "keymacro_add: sequence-lead-in command not allowed\n");
               return;
       }
       if (el->el_keymacro.map == NULL)
               /* tree is initially empty.  Set up new node to match key[0] */
               el->el_keymacro.map = node__get(key[0]);
                       /* it is properly initialized */

       /* Now recurse through el->el_keymacro.map */
       (void) node__try(el, el->el_keymacro.map, key, val, ntype);
       return;
}


/* keymacro_clear():
*
*/
libedit_private void
keymacro_clear(EditLine *el, el_action_t *map, const wchar_t *in)
{
       if (*in > N_KEYS) /* can't be in the map */
               return;
       if ((map[(unsigned char)*in] == ED_SEQUENCE_LEAD_IN) &&
           ((map == el->el_map.key &&
           el->el_map.alt[(unsigned char)*in] != ED_SEQUENCE_LEAD_IN) ||
           (map == el->el_map.alt &&
           el->el_map.key[(unsigned char)*in] != ED_SEQUENCE_LEAD_IN)))
               (void) keymacro_delete(el, in);
}


/* keymacro_delete():
*      Delete the key and all longer keys staring with key, if
*      they exists.
*/
libedit_private int
keymacro_delete(EditLine *el, const wchar_t *key)
{

       if (key[0] == '\0') {
               (void) fprintf(el->el_errfile,
                   "keymacro_delete: Null extended-key not allowed.\n");
               return -1;
       }
       if (el->el_keymacro.map == NULL)
               return 0;

       (void) node__delete(el, &el->el_keymacro.map, key);
       return 0;
}


/* keymacro_print():
*      Print the binding associated with key key.
*      Print entire el->el_keymacro.map if null
*/
libedit_private void
keymacro_print(EditLine *el, const wchar_t *key)
{

       /* do nothing if el->el_keymacro.map is empty and null key specified */
       if (el->el_keymacro.map == NULL && *key == 0)
               return;

       el->el_keymacro.buf[0] = '"';
       if (node_lookup(el, key, el->el_keymacro.map, (size_t)1) <= -1)
               /* key is not bound */
               (void) fprintf(el->el_errfile, "Unbound extended key \"%ls"
                   "\"\n", key);
       return;
}


/* node_trav():
*      recursively traverses node in tree until match or mismatch is
*      found.  May read in more characters.
*/
static int
node_trav(EditLine *el, keymacro_node_t *ptr, wchar_t *ch,
   keymacro_value_t *val)
{

       if (ptr->ch == *ch) {
               /* match found */
               if (ptr->next) {
                       /* key not complete so get next char */
                       if (el_wgetc(el, ch) != 1)
                               return XK_NOD;
                       return node_trav(el, ptr->next, ch, val);
               } else {
                       *val = ptr->val;
                       if (ptr->type != XK_CMD)
                               *ch = '\0';
                       return ptr->type;
               }
       } else {
               /* no match found here */
               if (ptr->sibling) {
                       /* try next sibling */
                       return node_trav(el, ptr->sibling, ch, val);
               } else {
                       /* no next sibling -- mismatch */
                       val->str = NULL;
                       return XK_STR;
               }
       }
}


/* node__try():
*      Find a node that matches *str or allocate a new one
*/
static int
node__try(EditLine *el, keymacro_node_t *ptr, const wchar_t *str,
   keymacro_value_t *val, int ntype)
{

       if (ptr->ch != *str) {
               keymacro_node_t *xm;

               for (xm = ptr; xm->sibling != NULL; xm = xm->sibling)
                       if (xm->sibling->ch == *str)
                               break;
               if (xm->sibling == NULL)
                       xm->sibling = node__get(*str);  /* setup new node */
               ptr = xm->sibling;
       }
       if (*++str == '\0') {
               /* we're there */
               if (ptr->next != NULL) {
                       node__put(el, ptr->next);
                               /* lose longer keys with this prefix */
                       ptr->next = NULL;
               }
               switch (ptr->type) {
               case XK_CMD:
               case XK_NOD:
                       break;
               case XK_STR:
                       if (ptr->val.str)
                               el_free(ptr->val.str);
                       break;
               default:
                       EL_ABORT((el->el_errfile, "Bad XK_ type %d\n",
                           ptr->type));
               }

               switch (ptr->type = ntype) {
               case XK_CMD:
                       ptr->val = *val;
                       break;
               case XK_STR:
                       if ((ptr->val.str = wcsdup(val->str)) == NULL)
                               return -1;
                       break;
               default:
                       EL_ABORT((el->el_errfile, "Bad XK_ type %d\n", ntype));
               }
       } else {
               /* still more chars to go */
               if (ptr->next == NULL)
                       ptr->next = node__get(*str);    /* setup new node */
               (void) node__try(el, ptr->next, str, val, ntype);
       }
       return 0;
}


/* node__delete():
*      Delete node that matches str
*/
static int
node__delete(EditLine *el, keymacro_node_t **inptr, const wchar_t *str)
{
       keymacro_node_t *ptr;
       keymacro_node_t *prev_ptr = NULL;

       ptr = *inptr;

       if (ptr->ch != *str) {
               keymacro_node_t *xm;

               for (xm = ptr; xm->sibling != NULL; xm = xm->sibling)
                       if (xm->sibling->ch == *str)
                               break;
               if (xm->sibling == NULL)
                       return 0;
               prev_ptr = xm;
               ptr = xm->sibling;
       }
       if (*++str == '\0') {
               /* we're there */
               if (prev_ptr == NULL)
                       *inptr = ptr->sibling;
               else
                       prev_ptr->sibling = ptr->sibling;
               ptr->sibling = NULL;
               node__put(el, ptr);
               return 1;
       } else if (ptr->next != NULL &&
           node__delete(el, &ptr->next, str) == 1) {
               if (ptr->next != NULL)
                       return 0;
               if (prev_ptr == NULL)
                       *inptr = ptr->sibling;
               else
                       prev_ptr->sibling = ptr->sibling;
               ptr->sibling = NULL;
               node__put(el, ptr);
               return 1;
       } else {
               return 0;
       }
}


/* node__put():
*      Puts a tree of nodes onto free list using free(3).
*/
static void
node__put(EditLine *el, keymacro_node_t *ptr)
{
       if (ptr == NULL)
               return;

       if (ptr->next != NULL) {
               node__put(el, ptr->next);
               ptr->next = NULL;
       }
       node__put(el, ptr->sibling);

       switch (ptr->type) {
       case XK_CMD:
       case XK_NOD:
               break;
       case XK_STR:
               if (ptr->val.str != NULL)
                       el_free(ptr->val.str);
               break;
       default:
               EL_ABORT((el->el_errfile, "Bad XK_ type %d\n", ptr->type));
       }
       el_free(ptr);
}


/* node__get():
*      Returns pointer to a keymacro_node_t for ch.
*/
static keymacro_node_t *
node__get(wint_t ch)
{
       keymacro_node_t *ptr;

       ptr = el_malloc(sizeof(*ptr));
       if (ptr == NULL)
               return NULL;
       ptr->ch = ch;
       ptr->type = XK_NOD;
       ptr->val.str = NULL;
       ptr->next = NULL;
       ptr->sibling = NULL;
       return ptr;
}

static void
node__free(keymacro_node_t *k)
{
       if (k == NULL)
               return;
       node__free(k->sibling);
       node__free(k->next);
       el_free(k);
}

/* node_lookup():
*      look for the str starting at node ptr.
*      Print if last node
*/
static int
node_lookup(EditLine *el, const wchar_t *str, keymacro_node_t *ptr,
   size_t cnt)
{
       ssize_t used;

       if (ptr == NULL)
               return -1;      /* cannot have null ptr */

       if (!str || *str == 0) {
               /* no more chars in str.  node_enum from here. */
               (void) node_enum(el, ptr, cnt);
               return 0;
       } else {
               /* If match put this char into el->el_keymacro.buf.  Recurse */
               if (ptr->ch == *str) {
                       /* match found */
                       used = ct_visual_char(el->el_keymacro.buf + cnt,
                           KEY_BUFSIZ - cnt, ptr->ch);
                       if (used == -1)
                               return -1; /* ran out of buffer space */
                       if (ptr->next != NULL)
                               /* not yet at leaf */
                               return (node_lookup(el, str + 1, ptr->next,
                                   (size_t)used + cnt));
                       else {
                           /* next node is null so key should be complete */
                               if (str[1] == 0) {
                                       size_t px = cnt + (size_t)used;
                                       el->el_keymacro.buf[px] = '"';
                                       el->el_keymacro.buf[px + 1] = '\0';
                                       keymacro_kprint(el, el->el_keymacro.buf,
                                           &ptr->val, ptr->type);
                                       return 0;
                               } else
                                       return -1;
                                       /* mismatch -- str still has chars */
                       }
               } else {
                       /* no match found try sibling */
                       if (ptr->sibling)
                               return (node_lookup(el, str, ptr->sibling,
                                   cnt));
                       else
                               return -1;
               }
       }
}


/* node_enum():
*      Traverse the node printing the characters it is bound in buffer
*/
static int
node_enum(EditLine *el, keymacro_node_t *ptr, size_t cnt)
{
       ssize_t used;

       if (cnt >= KEY_BUFSIZ - 5) {    /* buffer too small */
               el->el_keymacro.buf[++cnt] = '"';
               el->el_keymacro.buf[++cnt] = '\0';
               (void) fprintf(el->el_errfile,
                   "Some extended keys too long for internal print buffer");
               (void) fprintf(el->el_errfile, " \"%ls...\"\n",
                   el->el_keymacro.buf);
               return 0;
       }
       if (ptr == NULL) {
#ifdef DEBUG_EDIT
               (void) fprintf(el->el_errfile,
                   "node_enum: BUG!! Null ptr passed\n!");
#endif
               return -1;
       }
       /* put this char at end of str */
       used = ct_visual_char(el->el_keymacro.buf + cnt, KEY_BUFSIZ - cnt,
           ptr->ch);
       if (ptr->next == NULL) {
               /* print this key and function */
               el->el_keymacro.buf[cnt + (size_t)used   ] = '"';
               el->el_keymacro.buf[cnt + (size_t)used + 1] = '\0';
               keymacro_kprint(el, el->el_keymacro.buf, &ptr->val, ptr->type);
       } else
               (void) node_enum(el, ptr->next, cnt + (size_t)used);

       /* go to sibling if there is one */
       if (ptr->sibling)
               (void) node_enum(el, ptr->sibling, cnt);
       return 0;
}


/* keymacro_kprint():
*      Print the specified key and its associated
*      function specified by val
*/
libedit_private void
keymacro_kprint(EditLine *el, const wchar_t *key, keymacro_value_t *val,
   int ntype)
{
       el_bindings_t *fp;
       char unparsbuf[EL_BUFSIZ];
       static const char fmt[] = "%-15s->  %s\n";

       if (val != NULL)
               switch (ntype) {
               case XK_STR:
                       (void) keymacro__decode_str(val->str, unparsbuf,
                           sizeof(unparsbuf),
                           ntype == XK_STR ? "\"\"" : "[]");
                       (void) fprintf(el->el_outfile, fmt,
                           ct_encode_string(key, &el->el_scratch), unparsbuf);
                       break;
               case XK_CMD:
                       for (fp = el->el_map.help; fp->name; fp++)
                               if (val->cmd == fp->func) {
                   wcstombs(unparsbuf, fp->name, sizeof(unparsbuf));
                   unparsbuf[sizeof(unparsbuf) -1] = '\0';
                                       (void) fprintf(el->el_outfile, fmt,
                       ct_encode_string(key, &el->el_scratch), unparsbuf);
                                       break;
                               }
#ifdef DEBUG_KEY
                       if (fp->name == NULL)
                               (void) fprintf(el->el_outfile,
                                   "BUG! Command not found.\n");
#endif

                       break;
               default:
                       EL_ABORT((el->el_errfile, "Bad XK_ type %d\n", ntype));
               }
       else
               (void) fprintf(el->el_outfile, fmt, ct_encode_string(key,
                   &el->el_scratch), "no input");
}


#define ADDC(c) \
       if (b < eb) \
               *b++ = c; \
       else \
               b++
/* keymacro__decode_str():
*      Make a printable version of the ey
*/
libedit_private size_t
keymacro__decode_str(const wchar_t *str, char *buf, size_t len,
   const char *sep)
{
       char *b = buf, *eb = b + len;
       const wchar_t *p;

       b = buf;
       if (sep[0] != '\0') {
               ADDC(sep[0]);
       }
       if (*str == '\0') {
               ADDC('^');
               ADDC('@');
               goto add_endsep;
       }
       for (p = str; *p != 0; p++) {
               wchar_t dbuf[VISUAL_WIDTH_MAX];
               wchar_t *p2 = dbuf;
               ssize_t l = ct_visual_char(dbuf, VISUAL_WIDTH_MAX, *p);
               while (l-- > 0) {
                       ssize_t n = ct_encode_char(b, (size_t)(eb - b), *p2++);
                       if (n == -1) /* ran out of space */
                               goto add_endsep;
                       else
                               b += n;
               }
       }
add_endsep:
       if (sep[0] != '\0' && sep[1] != '\0') {
               ADDC(sep[1]);
       }
       ADDC('\0');
       if ((size_t)(b - buf) >= len)
           buf[len - 1] = '\0';
       return (size_t)(b - buf);
}