[BACK]Return to popup.c CVS log [TXT][DIR] Up to [local] / src / usr.bin / tmux

Annotation of src/usr.bin/tmux/popup.c, Revision 1.23

1.23    ! nicm        1: /* $OpenBSD: popup.c,v 1.22 2021/03/02 10:56:45 nicm Exp $ */
1.1       nicm        2:
                      3: /*
                      4:  * Copyright (c) 2020 Nicholas Marriott <nicholas.marriott@gmail.com>
                      5:  *
                      6:  * Permission to use, copy, modify, and distribute this software for any
                      7:  * purpose with or without fee is hereby granted, provided that the above
                      8:  * copyright notice and this permission notice appear in all copies.
                      9:  *
                     10:  * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
                     11:  * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
                     12:  * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
                     13:  * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
                     14:  * WHATSOEVER RESULTING FROM LOSS OF MIND, USE, DATA OR PROFITS, WHETHER
                     15:  * IN AN ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING
                     16:  * OUT OF OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
                     17:  */
                     18:
                     19: #include <sys/types.h>
                     20: #include <sys/wait.h>
                     21:
1.17      nicm       22: #include <paths.h>
1.1       nicm       23: #include <signal.h>
                     24: #include <stdlib.h>
                     25: #include <string.h>
1.17      nicm       26: #include <unistd.h>
1.1       nicm       27:
                     28: #include "tmux.h"
                     29:
                     30: struct popup_data {
                     31:        struct client            *c;
                     32:        struct cmdq_item         *item;
                     33:        int                       flags;
                     34:
                     35:        struct screen             s;
                     36:        struct job               *job;
                     37:        struct input_ctx         *ictx;
                     38:        int                       status;
1.14      nicm       39:        popup_close_cb            cb;
                     40:        void                     *arg;
1.1       nicm       41:
1.23    ! nicm       42:        /* Current position and size. */
1.1       nicm       43:        u_int                     px;
                     44:        u_int                     py;
                     45:        u_int                     sx;
                     46:        u_int                     sy;
                     47:
1.23    ! nicm       48:        /* Preferred position and size. */
        !            49:        u_int                     ppx;
        !            50:        u_int                     ppy;
        !            51:        u_int                     psx;
        !            52:        u_int                     psy;
        !            53:
1.1       nicm       54:        enum { OFF, MOVE, SIZE }  dragging;
                     55:        u_int                     dx;
                     56:        u_int                     dy;
                     57:
                     58:        u_int                     lx;
                     59:        u_int                     ly;
                     60:        u_int                     lb;
                     61: };
                     62:
1.17      nicm       63: struct popup_editor {
                     64:        char                    *path;
                     65:        popup_finish_edit_cb     cb;
                     66:        void                    *arg;
                     67: };
                     68:
1.1       nicm       69: static void
1.15      nicm       70: popup_redraw_cb(const struct tty_ctx *ttyctx)
                     71: {
                     72:        struct popup_data       *pd = ttyctx->arg;
                     73:
                     74:        pd->c->flags |= CLIENT_REDRAWOVERLAY;
                     75: }
                     76:
                     77: static int
                     78: popup_set_client_cb(struct tty_ctx *ttyctx, struct client *c)
                     79: {
                     80:        struct popup_data       *pd = ttyctx->arg;
                     81:
1.16      nicm       82:        if (c != pd->c)
                     83:                return (0);
1.15      nicm       84:        if (pd->c->flags & CLIENT_REDRAWOVERLAY)
1.16      nicm       85:                return (0);
1.15      nicm       86:
                     87:        ttyctx->bigger = 0;
                     88:        ttyctx->wox = 0;
                     89:        ttyctx->woy = 0;
                     90:        ttyctx->wsx = c->tty.sx;
                     91:        ttyctx->wsy = c->tty.sy;
                     92:
                     93:        ttyctx->xoff = ttyctx->rxoff = pd->px + 1;
                     94:        ttyctx->yoff = ttyctx->ryoff = pd->py + 1;
                     95:
                     96:        return (1);
                     97: }
                     98:
                     99: static void
                    100: popup_init_ctx_cb(struct screen_write_ctx *ctx, struct tty_ctx *ttyctx)
                    101: {
                    102:        struct popup_data       *pd = ctx->arg;
                    103:
                    104:        ttyctx->redraw_cb = popup_redraw_cb;
                    105:        ttyctx->set_client_cb = popup_set_client_cb;
                    106:        ttyctx->arg = pd;
                    107: }
                    108:
                    109: static struct screen *
1.1       nicm      110: popup_mode_cb(struct client *c, u_int *cx, u_int *cy)
                    111: {
                    112:        struct popup_data       *pd = c->overlay_data;
                    113:
                    114:        *cx = pd->px + 1 + pd->s.cx;
                    115:        *cy = pd->py + 1 + pd->s.cy;
1.15      nicm      116:        return (&pd->s);
1.1       nicm      117: }
                    118:
                    119: static int
                    120: popup_check_cb(struct client *c, u_int px, u_int py)
                    121: {
                    122:        struct popup_data       *pd = c->overlay_data;
                    123:
                    124:        if (px < pd->px || px > pd->px + pd->sx - 1)
                    125:                return (1);
                    126:        if (py < pd->py || py > pd->py + pd->sy - 1)
                    127:                return (1);
                    128:        return (0);
                    129: }
                    130:
                    131: static void
                    132: popup_draw_cb(struct client *c, __unused struct screen_redraw_ctx *ctx0)
                    133: {
                    134:        struct popup_data       *pd = c->overlay_data;
                    135:        struct tty              *tty = &c->tty;
                    136:        struct screen            s;
                    137:        struct screen_write_ctx  ctx;
                    138:        u_int                    i, px = pd->px, py = pd->py;
                    139:
                    140:        screen_init(&s, pd->sx, pd->sy, 0);
1.15      nicm      141:        screen_write_start(&ctx, &s);
1.1       nicm      142:        screen_write_clearscreen(&ctx, 8);
1.23    ! nicm      143:
        !           144:        /* Skip drawing popup if the terminal is too small. */
        !           145:        if (pd->sx > 2 && pd->sy > 2) {
        !           146:                screen_write_box(&ctx, pd->sx, pd->sy);
        !           147:                screen_write_cursormove(&ctx, 1, 1, 0);
        !           148:                screen_write_fast_copy(&ctx, &pd->s, 0, 0, pd->sx - 2,
        !           149:                    pd->sy - 2);
        !           150:        }
1.1       nicm      151:        screen_write_stop(&ctx);
                    152:
                    153:        c->overlay_check = NULL;
1.15      nicm      154:        for (i = 0; i < pd->sy; i++){
                    155:                tty_draw_line(tty, &s, 0, i, pd->sx, px, py + i,
                    156:                    &grid_default_cell, NULL);
                    157:        }
1.1       nicm      158:        c->overlay_check = popup_check_cb;
                    159: }
                    160:
                    161: static void
                    162: popup_free_cb(struct client *c)
                    163: {
                    164:        struct popup_data       *pd = c->overlay_data;
                    165:        struct cmdq_item        *item = pd->item;
                    166:
1.14      nicm      167:        if (pd->cb != NULL)
                    168:                pd->cb(pd->status, pd->arg);
                    169:
1.1       nicm      170:        if (item != NULL) {
1.22      nicm      171:                if (cmdq_get_client(item) != NULL &&
1.8       nicm      172:                    cmdq_get_client(item)->session == NULL)
                    173:                        cmdq_get_client(item)->retval = pd->status;
1.1       nicm      174:                cmdq_continue(item);
                    175:        }
                    176:        server_client_unref(pd->c);
                    177:
                    178:        if (pd->job != NULL)
                    179:                job_free(pd->job);
1.22      nicm      180:        input_free(pd->ictx);
1.1       nicm      181:
                    182:        screen_free(&pd->s);
                    183:        free(pd);
                    184: }
                    185:
                    186: static void
1.23    ! nicm      187: popup_resize_cb(struct client *c)
        !           188: {
        !           189:        struct popup_data       *pd = c->overlay_data;
        !           190:        struct tty              *tty = &c->tty;
        !           191:
        !           192:        if (pd == NULL)
        !           193:                return;
        !           194:
        !           195:        /* Adjust position and size. */
        !           196:        if (pd->psy > tty->sy)
        !           197:                pd->sy = tty->sy;
        !           198:        else
        !           199:                pd->sy = pd->psy;
        !           200:        if (pd->psx > tty->sx)
        !           201:                pd->sx = tty->sx;
        !           202:        else
        !           203:                pd->sx = pd->psx;
        !           204:        if (pd->ppy + pd->sy > tty->sy)
        !           205:                pd->py = tty->sy - pd->sy;
        !           206:        else
        !           207:                pd->py = pd->ppy;
        !           208:        if (pd->ppx + pd->sx > tty->sx)
        !           209:                pd->px = tty->sx - pd->sx;
        !           210:        else
        !           211:                pd->px = pd->ppx;
        !           212:
        !           213:        /* Avoid zero size screens. */
        !           214:        if (pd->sx > 2 && pd->sy > 2) {
        !           215:                screen_resize(&pd->s, pd->sx - 2, pd->sy - 2, 0);
        !           216:                if (pd->job != NULL)
        !           217:                        job_resize(pd->job, pd->sx - 2, pd->sy - 2);
        !           218:        }
        !           219: }
        !           220:
        !           221: static void
1.1       nicm      222: popup_handle_drag(struct client *c, struct popup_data *pd,
                    223:     struct mouse_event *m)
                    224: {
                    225:        u_int   px, py;
                    226:
                    227:        if (!MOUSE_DRAG(m->b))
                    228:                pd->dragging = OFF;
                    229:        else if (pd->dragging == MOVE) {
                    230:                if (m->x < pd->dx)
                    231:                        px = 0;
                    232:                else if (m->x - pd->dx + pd->sx > c->tty.sx)
                    233:                        px = c->tty.sx - pd->sx;
                    234:                else
                    235:                        px = m->x - pd->dx;
                    236:                if (m->y < pd->dy)
                    237:                        py = 0;
                    238:                else if (m->y - pd->dy + pd->sy > c->tty.sy)
                    239:                        py = c->tty.sy - pd->sy;
                    240:                else
                    241:                        py = m->y - pd->dy;
                    242:                pd->px = px;
                    243:                pd->py = py;
                    244:                pd->dx = m->x - pd->px;
                    245:                pd->dy = m->y - pd->py;
1.23    ! nicm      246:                pd->ppx = px;
        !           247:                pd->ppy = py;
1.1       nicm      248:                server_redraw_client(c);
                    249:        } else if (pd->dragging == SIZE) {
1.7       nicm      250:                if (m->x < pd->px + 3)
1.1       nicm      251:                        return;
1.7       nicm      252:                if (m->y < pd->py + 3)
1.1       nicm      253:                        return;
                    254:                pd->sx = m->x - pd->px;
                    255:                pd->sy = m->y - pd->py;
1.23    ! nicm      256:                pd->psx = pd->sx;
        !           257:                pd->psy = pd->sy;
1.1       nicm      258:
1.20      nicm      259:                screen_resize(&pd->s, pd->sx - 2, pd->sy - 2, 0);
1.22      nicm      260:                if (pd->job != NULL)
1.1       nicm      261:                        job_resize(pd->job, pd->sx - 2, pd->sy - 2);
                    262:                server_redraw_client(c);
                    263:        }
                    264: }
                    265:
                    266: static int
                    267: popup_key_cb(struct client *c, struct key_event *event)
                    268: {
                    269:        struct popup_data       *pd = c->overlay_data;
                    270:        struct mouse_event      *m = &event->m;
1.22      nicm      271:        const char              *buf;
1.6       nicm      272:        size_t                   len;
1.1       nicm      273:
                    274:        if (KEYC_IS_MOUSE(event->key)) {
                    275:                if (pd->dragging != OFF) {
                    276:                        popup_handle_drag(c, pd, m);
                    277:                        goto out;
                    278:                }
                    279:                if (m->x < pd->px ||
                    280:                    m->x > pd->px + pd->sx - 1 ||
                    281:                    m->y < pd->py ||
                    282:                    m->y > pd->py + pd->sy - 1) {
                    283:                        if (MOUSE_BUTTONS (m->b) == 1)
                    284:                                return (1);
                    285:                        return (0);
                    286:                }
                    287:                if ((m->b & MOUSE_MASK_META) ||
                    288:                    m->x == pd->px ||
                    289:                    m->x == pd->px + pd->sx - 1 ||
                    290:                    m->y == pd->py ||
                    291:                    m->y == pd->py + pd->sy - 1) {
                    292:                        if (!MOUSE_DRAG(m->b))
                    293:                                goto out;
                    294:                        if (MOUSE_BUTTONS(m->lb) == 0)
                    295:                                pd->dragging = MOVE;
                    296:                        else if (MOUSE_BUTTONS(m->lb) == 2)
                    297:                                pd->dragging = SIZE;
                    298:                        pd->dx = m->lx - pd->px;
                    299:                        pd->dy = m->ly - pd->py;
                    300:                        goto out;
                    301:                }
                    302:        }
                    303:
1.22      nicm      304:        if ((((pd->flags & (POPUP_CLOSEEXIT|POPUP_CLOSEEXITZERO)) == 0) ||
                    305:            pd->job == NULL) &&
                    306:            (event->key == '\033' || event->key == '\003'))
                    307:                return (1);
                    308:        if (pd->job != NULL) {
1.6       nicm      309:                if (KEYC_IS_MOUSE(event->key)) {
                    310:                        /* Must be inside, checked already. */
1.21      nicm      311:                        if (!input_key_get_mouse(&pd->s, m, m->x - pd->px - 1,
                    312:                            m->y - pd->py - 1, &buf, &len))
1.6       nicm      313:                                return (0);
                    314:                        bufferevent_write(job_get_event(pd->job), buf, len);
                    315:                        return (0);
                    316:                }
1.18      nicm      317:                input_key(&pd->s, job_get_event(pd->job), event->key);
1.1       nicm      318:        }
1.22      nicm      319:        return (0);
1.1       nicm      320:
                    321: out:
                    322:        pd->lx = m->x;
                    323:        pd->ly = m->y;
                    324:        pd->lb = m->b;
                    325:        return (0);
                    326: }
                    327:
                    328: static void
                    329: popup_job_update_cb(struct job *job)
                    330: {
                    331:        struct popup_data       *pd = job_get_data(job);
                    332:        struct evbuffer         *evb = job_get_event(job)->input;
1.15      nicm      333:        struct client           *c = pd->c;
1.1       nicm      334:        struct screen           *s = &pd->s;
                    335:        void                    *data = EVBUFFER_DATA(evb);
                    336:        size_t                   size = EVBUFFER_LENGTH(evb);
                    337:
1.15      nicm      338:        if (size == 0)
                    339:                return;
                    340:
                    341:        c->overlay_check = NULL;
                    342:        c->tty.flags &= ~TTY_FREEZE;
                    343:
                    344:        input_parse_screen(pd->ictx, s, popup_init_ctx_cb, pd, data, size);
                    345:
                    346:        c->tty.flags |= TTY_FREEZE;
                    347:        c->overlay_check = popup_check_cb;
                    348:
                    349:        evbuffer_drain(evb, size);
1.1       nicm      350: }
                    351:
                    352: static void
                    353: popup_job_complete_cb(struct job *job)
                    354: {
                    355:        struct popup_data       *pd = job_get_data(job);
                    356:        int                      status;
                    357:
                    358:        status = job_get_status(pd->job);
                    359:        if (WIFEXITED(status))
                    360:                pd->status = WEXITSTATUS(status);
                    361:        else if (WIFSIGNALED(status))
                    362:                pd->status = WTERMSIG(status);
                    363:        else
                    364:                pd->status = 0;
                    365:        pd->job = NULL;
                    366:
1.4       nicm      367:        if ((pd->flags & POPUP_CLOSEEXIT) ||
                    368:            ((pd->flags & POPUP_CLOSEEXITZERO) && pd->status == 0))
1.1       nicm      369:                server_client_clear_overlay(pd->c);
                    370: }
                    371:
                    372: int
                    373: popup_display(int flags, struct cmdq_item *item, u_int px, u_int py, u_int sx,
1.22      nicm      374:     u_int sy, const char *shellcmd, int argc, char **argv, const char *cwd,
                    375:     struct client *c, struct session *s, popup_close_cb cb, void *arg)
1.1       nicm      376: {
                    377:        struct popup_data       *pd;
                    378:
                    379:        if (sx < 3 || sy < 3)
                    380:                return (-1);
                    381:        if (c->tty.sx < sx || c->tty.sy < sy)
                    382:                return (-1);
                    383:
                    384:        pd = xcalloc(1, sizeof *pd);
                    385:        pd->item = item;
                    386:        pd->flags = flags;
                    387:
                    388:        pd->c = c;
                    389:        pd->c->references++;
                    390:
1.14      nicm      391:        pd->cb = cb;
                    392:        pd->arg = arg;
1.1       nicm      393:        pd->status = 128 + SIGHUP;
                    394:
                    395:        screen_init(&pd->s, sx - 2, sy - 2, 0);
                    396:
                    397:        pd->px = px;
                    398:        pd->py = py;
                    399:        pd->sx = sx;
                    400:        pd->sy = sy;
                    401:
1.23    ! nicm      402:        pd->ppx = px;
        !           403:        pd->ppy = py;
        !           404:        pd->psx = sx;
        !           405:        pd->psy = sy;
        !           406:
1.22      nicm      407:        pd->job = job_run(shellcmd, argc, argv, s, cwd,
                    408:            popup_job_update_cb, popup_job_complete_cb, NULL, pd,
                    409:            JOB_NOWAIT|JOB_PTY|JOB_KEEPWRITE, pd->sx - 2, pd->sy - 2);
                    410:        pd->ictx = input_init(NULL, job_get_event(pd->job));
1.1       nicm      411:
                    412:        server_client_set_overlay(c, 0, popup_check_cb, popup_mode_cb,
1.23    ! nicm      413:            popup_draw_cb, popup_key_cb, popup_free_cb, popup_resize_cb, pd);
1.17      nicm      414:        return (0);
                    415: }
                    416:
                    417: static void
                    418: popup_editor_free(struct popup_editor *pe)
                    419: {
                    420:        unlink(pe->path);
                    421:        free(pe->path);
                    422:        free(pe);
                    423: }
                    424:
                    425: static void
                    426: popup_editor_close_cb(int status, void *arg)
                    427: {
                    428:        struct popup_editor     *pe = arg;
                    429:        FILE                    *f;
                    430:        char                    *buf = NULL;
                    431:        off_t                    len = 0;
                    432:
                    433:        if (status != 0) {
                    434:                pe->cb(NULL, 0, pe->arg);
                    435:                popup_editor_free(pe);
                    436:                return;
                    437:        }
                    438:
                    439:        f = fopen(pe->path, "r");
                    440:        if (f != NULL) {
                    441:                fseeko(f, 0, SEEK_END);
                    442:                len = ftello(f);
                    443:                fseeko(f, 0, SEEK_SET);
                    444:
                    445:                if (len == 0 ||
                    446:                    (uintmax_t)len > (uintmax_t)SIZE_MAX ||
                    447:                    (buf = malloc(len)) == NULL ||
                    448:                    fread(buf, len, 1, f) != 1) {
                    449:                        free(buf);
                    450:                        buf = NULL;
                    451:                        len = 0;
                    452:                }
                    453:                fclose(f);
                    454:        }
                    455:        pe->cb(buf, len, pe->arg); /* callback now owns buffer */
                    456:        popup_editor_free(pe);
                    457: }
                    458:
                    459: int
                    460: popup_editor(struct client *c, const char *buf, size_t len,
                    461:     popup_finish_edit_cb cb, void *arg)
                    462: {
                    463:        struct popup_editor     *pe;
                    464:        int                      fd;
                    465:        FILE                    *f;
                    466:        char                    *cmd;
                    467:        char                     path[] = _PATH_TMP "tmux.XXXXXXXX";
                    468:        const char              *editor;
                    469:        u_int                    px, py, sx, sy;
                    470:
                    471:        editor = options_get_string(global_options, "editor");
                    472:        if (*editor == '\0')
                    473:                return (-1);
                    474:
                    475:        fd = mkstemp(path);
                    476:        if (fd == -1)
                    477:                return (-1);
                    478:        f = fdopen(fd, "w");
                    479:        if (fwrite(buf, len, 1, f) != 1) {
                    480:                fclose(f);
                    481:                return (-1);
                    482:        }
                    483:        fclose(f);
                    484:
                    485:        pe = xcalloc(1, sizeof *pe);
                    486:        pe->path = xstrdup(path);
                    487:        pe->cb = cb;
                    488:        pe->arg = arg;
                    489:
                    490:        sx = c->tty.sx * 9 / 10;
                    491:        sy = c->tty.sy * 9 / 10;
                    492:        px = (c->tty.sx / 2) - (sx / 2);
                    493:        py = (c->tty.sy / 2) - (sy / 2);
                    494:
                    495:        xasprintf(&cmd, "%s %s", editor, path);
1.22      nicm      496:        if (popup_display(POPUP_CLOSEEXIT, NULL, px, py, sx, sy, cmd, 0, NULL,
                    497:            _PATH_TMP, c, NULL, popup_editor_close_cb, pe) != 0) {
1.17      nicm      498:                popup_editor_free(pe);
                    499:                free(cmd);
                    500:                return (-1);
                    501:        }
                    502:        free(cmd);
1.1       nicm      503:        return (0);
                    504: }