/* $OpenBSD$ */

/*
* Copyright (c) 2010 Nicholas Marriott <[email protected]>
*
* Permission to use, copy, modify, and distribute this software for any
* purpose with or without fee is hereby granted, provided that the above
* copyright notice and this permission notice appear in all copies.
*
* THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
* WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
* MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
* ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
* WHATSOEVER RESULTING FROM LOSS OF MIND, USE, DATA OR PROFITS, WHETHER
* IN AN ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING
* OUT OF OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
*/

#include <sys/types.h>

#include <stdlib.h>
#include <string.h>

#include "tmux.h"

/* Table mapping ACS entries to UTF-8. */
struct tty_acs_entry {
       u_char           key;
       const char      *string;
};
static const struct tty_acs_entry tty_acs_table[] = {
       { '+', "\342\206\222" },        /* arrow pointing right */
       { ',', "\342\206\220" },        /* arrow pointing left */
       { '-', "\342\206\221" },        /* arrow pointing up */
       { '.', "\342\206\223" },        /* arrow pointing down */
       { '0', "\342\226\256" },        /* solid square block */
       { '`', "\342\227\206" },        /* diamond */
       { 'a', "\342\226\222" },        /* checker board (stipple) */
       { 'b', "\342\220\211" },
       { 'c', "\342\220\214" },
       { 'd', "\342\220\215" },
       { 'e', "\342\220\212" },
       { 'f', "\302\260" },            /* degree symbol */
       { 'g', "\302\261" },            /* plus/minus */
       { 'h', "\342\220\244" },
       { 'i', "\342\220\213" },
       { 'j', "\342\224\230" },        /* lower right corner */
       { 'k', "\342\224\220" },        /* upper right corner */
       { 'l', "\342\224\214" },        /* upper left corner */
       { 'm', "\342\224\224" },        /* lower left corner */
       { 'n', "\342\224\274" },        /* large plus or crossover */
       { 'o', "\342\216\272" },        /* scan line 1 */
       { 'p', "\342\216\273" },        /* scan line 3 */
       { 'q', "\342\224\200" },        /* horizontal line */
       { 'r', "\342\216\274" },        /* scan line 7 */
       { 's', "\342\216\275" },        /* scan line 9 */
       { 't', "\342\224\234" },        /* tee pointing right */
       { 'u', "\342\224\244" },        /* tee pointing left */
       { 'v', "\342\224\264" },        /* tee pointing up */
       { 'w', "\342\224\254" },        /* tee pointing down */
       { 'x', "\342\224\202" },        /* vertical line */
       { 'y', "\342\211\244" },        /* less-than-or-equal-to */
       { 'z', "\342\211\245" },        /* greater-than-or-equal-to */
       { '{', "\317\200" },            /* greek pi */
       { '|', "\342\211\240" },        /* not-equal */
       { '}', "\302\243" },            /* UK pound sign */
       { '~', "\302\267" }             /* bullet */
};

/* Table mapping UTF-8 to ACS entries. */
struct tty_acs_reverse_entry {
       const char      *string;
       u_char           key;
};
static const struct tty_acs_reverse_entry tty_acs_reverse2[] = {
       { "\302\267", '~' }
};
static const struct tty_acs_reverse_entry tty_acs_reverse3[] = {
       { "\342\224\200", 'q' },
       { "\342\224\201", 'q' },
       { "\342\224\202", 'x' },
       { "\342\224\203", 'x' },
       { "\342\224\214", 'l' },
       { "\342\224\217", 'k' },
       { "\342\224\220", 'k' },
       { "\342\224\223", 'l' },
       { "\342\224\224", 'm' },
       { "\342\224\227", 'm' },
       { "\342\224\230", 'j' },
       { "\342\224\233", 'j' },
       { "\342\224\234", 't' },
       { "\342\224\243", 't' },
       { "\342\224\244", 'u' },
       { "\342\224\253", 'u' },
       { "\342\224\263", 'w' },
       { "\342\224\264", 'v' },
       { "\342\224\273", 'v' },
       { "\342\224\274", 'n' },
       { "\342\225\213", 'n' },
       { "\342\225\220", 'q' },
       { "\342\225\221", 'x' },
       { "\342\225\224", 'l' },
       { "\342\225\227", 'k' },
       { "\342\225\232", 'm' },
       { "\342\225\235", 'j' },
       { "\342\225\240", 't' },
       { "\342\225\243", 'u' },
       { "\342\225\246", 'w' },
       { "\342\225\251", 'v' },
       { "\342\225\254", 'n' },
};

/* UTF-8 double borders. */
static const struct utf8_data tty_acs_double_borders_list[] = {
       { "", 0, 0, 0 },
       { "\342\225\221", 0, 3, 1 }, /* U+2551 */
       { "\342\225\220", 0, 3, 1 }, /* U+2550 */
       { "\342\225\224", 0, 3, 1 }, /* U+2554 */
       { "\342\225\227", 0, 3, 1 }, /* U+2557 */
       { "\342\225\232", 0, 3, 1 }, /* U+255A */
       { "\342\225\235", 0, 3, 1 }, /* U+255D */
       { "\342\225\246", 0, 3, 1 }, /* U+2566 */
       { "\342\225\251", 0, 3, 1 }, /* U+2569 */
       { "\342\225\240", 0, 3, 1 }, /* U+2560 */
       { "\342\225\243", 0, 3, 1 }, /* U+2563 */
       { "\342\225\254", 0, 3, 1 }, /* U+256C */
       { "\302\267",     0, 2, 1 }  /* U+00B7 */
};

/* UTF-8 heavy borders. */
static const struct utf8_data tty_acs_heavy_borders_list[] = {
       { "", 0, 0, 0 },
       { "\342\224\203", 0, 3, 1 }, /* U+2503 */
       { "\342\224\201", 0, 3, 1 }, /* U+2501 */
       { "\342\224\217", 0, 3, 1 }, /* U+250F */
       { "\342\224\223", 0, 3, 1 }, /* U+2513 */
       { "\342\224\227", 0, 3, 1 }, /* U+2517 */
       { "\342\224\233", 0, 3, 1 }, /* U+251B */
       { "\342\224\263", 0, 3, 1 }, /* U+2533 */
       { "\342\224\273", 0, 3, 1 }, /* U+253B */
       { "\342\224\243", 0, 3, 1 }, /* U+2523 */
       { "\342\224\253", 0, 3, 1 }, /* U+252B */
       { "\342\225\213", 0, 3, 1 }, /* U+254B */
       { "\302\267",     0, 2, 1 }  /* U+00B7 */
};

/* UTF-8 rounded borders. */
static const struct utf8_data tty_acs_rounded_borders_list[] = {
       { "", 0, 0, 0 },
       { "\342\224\202", 0, 3, 1 }, /* U+2502 */
       { "\342\224\200", 0, 3, 1 }, /* U+2500 */
       { "\342\225\255", 0, 3, 1 }, /* U+256D */
       { "\342\225\256", 0, 3, 1 }, /* U+256E */
       { "\342\225\260", 0, 3, 1 }, /* U+2570 */
       { "\342\225\257", 0, 3, 1 }, /* U+256F */
       { "\342\224\263", 0, 3, 1 }, /* U+2533 */
       { "\342\224\273", 0, 3, 1 }, /* U+253B */
       { "\342\224\234", 0, 3, 1 }, /* U+2524 */
       { "\342\224\244", 0, 3, 1 }, /* U+251C */
       { "\342\225\213", 0, 3, 1 }, /* U+254B */
       { "\302\267",     0, 2, 1 }  /* U+00B7 */
};

/* Get cell border character for double style. */
const struct utf8_data *
tty_acs_double_borders(int cell_type)
{
       return (&tty_acs_double_borders_list[cell_type]);
}

/* Get cell border character for heavy style. */
const struct utf8_data *
tty_acs_heavy_borders(int cell_type)
{
       return (&tty_acs_heavy_borders_list[cell_type]);
}

/* Get cell border character for rounded style. */
const struct utf8_data *
tty_acs_rounded_borders(int cell_type)
{
       return (&tty_acs_rounded_borders_list[cell_type]);
}

static int
tty_acs_cmp(const void *key, const void *value)
{
       const struct tty_acs_entry      *entry = value;
       int                              test = *(const u_char *)key;

       return (test - entry->key);
}

static int
tty_acs_reverse_cmp(const void *key, const void *value)
{
       const struct tty_acs_reverse_entry      *entry = value;
       const char                              *test = key;

       return (strcmp(test, entry->string));
}

/* Should this terminal use ACS instead of UTF-8 line drawing? */
int
tty_acs_needed(struct tty *tty)
{
       if (tty == NULL)
               return (0);

       /*
        * If the U8 flag is present, it marks whether a terminal supports
        * UTF-8 and ACS together.
        *
        * If it is present and zero, we force ACS - this gives users a way to
        * turn off UTF-8 line drawing.
        *
        * If it is nonzero, we can fall through to the default and use UTF-8
        * line drawing on UTF-8 terminals.
        */
       if (tty_term_has(tty->term, TTYC_U8) &&
           tty_term_number(tty->term, TTYC_U8) == 0)
               return (1);

       if (tty->client->flags & CLIENT_UTF8)
               return (0);
       return (1);
}

/* Retrieve ACS to output as UTF-8. */
const char *
tty_acs_get(struct tty *tty, u_char ch)
{
       const struct tty_acs_entry      *entry;

       /* Use the ACS set instead of UTF-8 if needed. */
       if (tty_acs_needed(tty)) {
               if (tty->term->acs[ch][0] == '\0')
                       return (NULL);
               return (&tty->term->acs[ch][0]);
       }

       /* Otherwise look up the UTF-8 translation. */
       entry = bsearch(&ch, tty_acs_table, nitems(tty_acs_table),
           sizeof tty_acs_table[0], tty_acs_cmp);
       if (entry == NULL)
               return (NULL);
       return (entry->string);
}

/* Reverse UTF-8 into ACS. */
int
tty_acs_reverse_get(__unused struct tty *tty, const char *s, size_t slen)
{
       const struct tty_acs_reverse_entry      *table, *entry;
       u_int                                    items;

       if (slen == 2) {
               table = tty_acs_reverse2;
               items = nitems(tty_acs_reverse2);
       } else if (slen == 3) {
               table = tty_acs_reverse3;
               items = nitems(tty_acs_reverse3);
       } else
               return (-1);
       entry = bsearch(s, table, items, sizeof table[0], tty_acs_reverse_cmp);
       if (entry == NULL)
               return (-1);
       return (entry->key);
}