2011-01-04 00:42:46 +00:00
|
|
|
/* $OpenBSD$ */
|
|
|
|
|
|
|
|
/*
|
2016-01-19 15:59:12 +00:00
|
|
|
* Copyright (c) 2010 Nicholas Marriott <nicholas.marriott@gmail.com>
|
2011-01-04 00:42:46 +00:00
|
|
|
*
|
|
|
|
* 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>
|
2013-11-17 20:19:36 +00:00
|
|
|
#include <unistd.h>
|
2017-01-18 10:00:50 +00:00
|
|
|
#include <vis.h>
|
2011-01-04 00:42:46 +00:00
|
|
|
|
|
|
|
#include "tmux.h"
|
|
|
|
|
2013-05-31 12:19:34 +00:00
|
|
|
/*
|
|
|
|
* Manipulate command arguments.
|
|
|
|
*/
|
|
|
|
|
2015-08-29 23:19:52 +00:00
|
|
|
struct args_entry {
|
|
|
|
u_char flag;
|
|
|
|
char *value;
|
|
|
|
RB_ENTRY(args_entry) entry;
|
|
|
|
};
|
|
|
|
|
2016-10-10 13:54:47 +00:00
|
|
|
static struct args_entry *args_find(struct args *, u_char);
|
2013-05-31 12:19:34 +00:00
|
|
|
|
2016-10-10 13:54:47 +00:00
|
|
|
static int args_cmp(struct args_entry *, struct args_entry *);
|
|
|
|
RB_GENERATE_STATIC(args_tree, args_entry, entry, args_cmp);
|
2013-05-31 12:19:34 +00:00
|
|
|
|
|
|
|
/* Arguments tree comparison function. */
|
2016-10-10 13:54:47 +00:00
|
|
|
static int
|
2013-05-31 12:19:34 +00:00
|
|
|
args_cmp(struct args_entry *a1, struct args_entry *a2)
|
|
|
|
{
|
|
|
|
return (a1->flag - a2->flag);
|
|
|
|
}
|
|
|
|
|
|
|
|
/* Find a flag in the arguments tree. */
|
2016-10-10 13:54:47 +00:00
|
|
|
static struct args_entry *
|
2013-05-31 12:19:34 +00:00
|
|
|
args_find(struct args *args, u_char ch)
|
|
|
|
{
|
|
|
|
struct args_entry entry;
|
|
|
|
|
|
|
|
entry.flag = ch;
|
|
|
|
return (RB_FIND(args_tree, &args->tree, &entry));
|
|
|
|
}
|
|
|
|
|
2011-01-04 00:42:46 +00:00
|
|
|
/* Parse an argv and argc into a new argument set. */
|
|
|
|
struct args *
|
|
|
|
args_parse(const char *template, int argc, char **argv)
|
|
|
|
{
|
|
|
|
struct args *args;
|
|
|
|
int opt;
|
|
|
|
|
|
|
|
args = xcalloc(1, sizeof *args);
|
|
|
|
|
|
|
|
optreset = 1;
|
|
|
|
optind = 1;
|
|
|
|
|
|
|
|
while ((opt = getopt(argc, argv, template)) != -1) {
|
2013-05-31 12:19:34 +00:00
|
|
|
if (opt < 0)
|
2011-01-04 00:42:46 +00:00
|
|
|
continue;
|
2014-01-15 11:44:18 +00:00
|
|
|
if (opt == '?' || strchr(template, opt) == NULL) {
|
2013-05-31 12:19:34 +00:00
|
|
|
args_free(args);
|
2011-01-04 00:42:46 +00:00
|
|
|
return (NULL);
|
|
|
|
}
|
2013-05-31 12:19:34 +00:00
|
|
|
args_set(args, opt, optarg);
|
2011-01-04 00:42:46 +00:00
|
|
|
}
|
|
|
|
argc -= optind;
|
|
|
|
argv += optind;
|
|
|
|
|
|
|
|
args->argc = argc;
|
|
|
|
args->argv = cmd_copy_argv(argc, argv);
|
|
|
|
|
|
|
|
return (args);
|
|
|
|
}
|
|
|
|
|
|
|
|
/* Free an arguments set. */
|
|
|
|
void
|
|
|
|
args_free(struct args *args)
|
|
|
|
{
|
2013-05-31 12:19:34 +00:00
|
|
|
struct args_entry *entry;
|
|
|
|
struct args_entry *entry1;
|
2011-01-04 00:42:46 +00:00
|
|
|
|
|
|
|
cmd_free_argv(args->argc, args->argv);
|
|
|
|
|
2013-05-31 12:19:34 +00:00
|
|
|
RB_FOREACH_SAFE(entry, args_tree, &args->tree, entry1) {
|
|
|
|
RB_REMOVE(args_tree, &args->tree, entry);
|
|
|
|
free(entry->value);
|
|
|
|
free(entry);
|
|
|
|
}
|
2011-01-04 00:42:46 +00:00
|
|
|
|
2012-07-10 11:53:01 +00:00
|
|
|
free(args);
|
2011-01-04 00:42:46 +00:00
|
|
|
}
|
|
|
|
|
2015-11-27 15:06:43 +00:00
|
|
|
/* Add to string. */
|
|
|
|
static void printflike(3, 4)
|
|
|
|
args_print_add(char **buf, size_t *len, const char *fmt, ...)
|
|
|
|
{
|
|
|
|
va_list ap;
|
|
|
|
char *s;
|
|
|
|
size_t slen;
|
|
|
|
|
|
|
|
va_start(ap, fmt);
|
|
|
|
slen = xvasprintf(&s, fmt, ap);
|
|
|
|
va_end(ap);
|
|
|
|
|
|
|
|
*len += slen;
|
|
|
|
*buf = xrealloc(*buf, *len);
|
|
|
|
|
|
|
|
strlcat(*buf, s, *len);
|
|
|
|
free(s);
|
|
|
|
}
|
|
|
|
|
2011-01-04 00:42:46 +00:00
|
|
|
/* Print a set of arguments. */
|
2015-11-27 15:06:43 +00:00
|
|
|
char *
|
|
|
|
args_print(struct args *args)
|
2011-01-04 00:42:46 +00:00
|
|
|
{
|
2015-11-27 15:06:43 +00:00
|
|
|
size_t len;
|
2017-01-18 10:00:50 +00:00
|
|
|
char *buf, *escaped;
|
|
|
|
int i, flags;
|
2013-05-31 12:19:34 +00:00
|
|
|
struct args_entry *entry;
|
2017-01-18 10:00:50 +00:00
|
|
|
static const char quoted[] = " #\"';$";
|
2011-01-04 00:42:46 +00:00
|
|
|
|
2015-11-27 15:06:43 +00:00
|
|
|
len = 1;
|
|
|
|
buf = xcalloc(1, len);
|
2011-01-04 00:42:46 +00:00
|
|
|
|
|
|
|
/* Process the flags first. */
|
2013-05-31 12:19:34 +00:00
|
|
|
RB_FOREACH(entry, args_tree, &args->tree) {
|
|
|
|
if (entry->value != NULL)
|
2011-01-04 00:42:46 +00:00
|
|
|
continue;
|
|
|
|
|
2015-11-27 15:06:43 +00:00
|
|
|
if (*buf == '\0')
|
|
|
|
args_print_add(&buf, &len, "-");
|
|
|
|
args_print_add(&buf, &len, "%c", entry->flag);
|
2011-01-04 00:42:46 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
/* Then the flags with arguments. */
|
2013-05-31 12:19:34 +00:00
|
|
|
RB_FOREACH(entry, args_tree, &args->tree) {
|
|
|
|
if (entry->value == NULL)
|
2011-01-04 00:42:46 +00:00
|
|
|
continue;
|
|
|
|
|
2015-11-27 15:06:43 +00:00
|
|
|
if (*buf != '\0')
|
|
|
|
args_print_add(&buf, &len, " -%c ", entry->flag);
|
|
|
|
else
|
|
|
|
args_print_add(&buf, &len, "-%c ", entry->flag);
|
2017-01-18 10:00:50 +00:00
|
|
|
|
|
|
|
flags = VIS_OCTAL|VIS_TAB|VIS_NL;
|
|
|
|
if (entry->value[strcspn(entry->value, quoted)] != '\0')
|
|
|
|
flags |= VIS_DQ;
|
2017-01-18 10:08:05 +00:00
|
|
|
utf8_stravis(&escaped, entry->value, flags);
|
2017-01-18 10:00:50 +00:00
|
|
|
if (flags & VIS_DQ)
|
|
|
|
args_print_add(&buf, &len, "\"%s\"", escaped);
|
2011-01-04 00:42:46 +00:00
|
|
|
else
|
2017-01-18 10:00:50 +00:00
|
|
|
args_print_add(&buf, &len, "%s", escaped);
|
|
|
|
free(escaped);
|
2011-01-04 00:42:46 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
/* And finally the argument vector. */
|
|
|
|
for (i = 0; i < args->argc; i++) {
|
2015-11-27 15:06:43 +00:00
|
|
|
if (*buf != '\0')
|
|
|
|
args_print_add(&buf, &len, " ");
|
2017-01-18 10:00:50 +00:00
|
|
|
|
|
|
|
flags = VIS_OCTAL|VIS_TAB|VIS_NL;
|
|
|
|
if (args->argv[i][strcspn(args->argv[i], quoted)] != '\0')
|
|
|
|
flags |= VIS_DQ;
|
2017-01-18 10:08:05 +00:00
|
|
|
utf8_stravis(&escaped, args->argv[i], flags);
|
2017-01-18 10:00:50 +00:00
|
|
|
if (flags & VIS_DQ)
|
|
|
|
args_print_add(&buf, &len, "\"%s\"", escaped);
|
2011-01-04 00:42:46 +00:00
|
|
|
else
|
2017-01-18 10:00:50 +00:00
|
|
|
args_print_add(&buf, &len, "%s", escaped);
|
|
|
|
free(escaped);
|
2011-01-04 00:42:46 +00:00
|
|
|
}
|
|
|
|
|
2015-11-27 15:06:43 +00:00
|
|
|
return (buf);
|
2011-01-04 00:42:46 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
/* Return if an argument is present. */
|
|
|
|
int
|
|
|
|
args_has(struct args *args, u_char ch)
|
|
|
|
{
|
2017-08-23 09:14:21 +00:00
|
|
|
return (args_find(args, ch) != NULL);
|
2011-01-04 00:42:46 +00:00
|
|
|
}
|
|
|
|
|
2013-05-31 12:19:34 +00:00
|
|
|
/* Set argument value in the arguments tree. */
|
Rewrite of choose mode, both to simplify and tidy the code and to add
some modern features.
Now the common code is in mode-tree.c, which provides an API used by the
three modes now separated into window-{buffer,client,tree}.c. Buffer
mode shows buffers, client mode clients and tree mode a tree of
sessions, windows and panes.
Each mode has a common set of key bindings plus a few that are specific
to the mode. Other changes are:
- each mode has a preview pane: for buffers this is the buffer content
(very useful), for others it is a preview of the pane;
- items may be sorted in different ways ('O' key);
- multiple items may be tagged and an operation applied to all of them
(for example, to delete multiple buffers at once);
- in tree mode a command may be run on the selected item (session,
window, pane) or on tagged items (key ':');
- displayed items may be filtered in tree mode by using a format (this
is used to implement find-window) (key 'f');
- the custom format (-F) for the display is no longer available;
- shortcut keys change from 0-9, a-z, A-Z which was always a bit weird
with keys used for other uses to 0-9, M-a to M-z.
Now that the code is simpler, other improvements will come later.
Primary key bindings for each mode are documented under the commands in
the man page (choose-buffer, choose-client, choose-tree).
Parts written by Thomas Adam.
2017-05-30 21:44:59 +00:00
|
|
|
void
|
2011-01-04 00:42:46 +00:00
|
|
|
args_set(struct args *args, u_char ch, const char *value)
|
|
|
|
{
|
2013-05-31 12:19:34 +00:00
|
|
|
struct args_entry *entry;
|
|
|
|
|
|
|
|
/* Replace existing argument. */
|
|
|
|
if ((entry = args_find(args, ch)) != NULL) {
|
|
|
|
free(entry->value);
|
2014-01-09 13:51:57 +00:00
|
|
|
entry->value = NULL;
|
|
|
|
} else {
|
|
|
|
entry = xcalloc(1, sizeof *entry);
|
|
|
|
entry->flag = ch;
|
|
|
|
RB_INSERT(args_tree, &args->tree, entry);
|
2013-05-31 12:19:34 +00:00
|
|
|
}
|
|
|
|
|
2011-01-30 12:09:30 +00:00
|
|
|
if (value != NULL)
|
2013-05-31 12:19:34 +00:00
|
|
|
entry->value = xstrdup(value);
|
2011-01-04 00:42:46 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
/* Get argument value. Will be NULL if it isn't present. */
|
|
|
|
const char *
|
|
|
|
args_get(struct args *args, u_char ch)
|
|
|
|
{
|
2013-05-31 12:19:34 +00:00
|
|
|
struct args_entry *entry;
|
|
|
|
|
|
|
|
if ((entry = args_find(args, ch)) == NULL)
|
|
|
|
return (NULL);
|
|
|
|
return (entry->value);
|
2011-01-04 00:42:46 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
/* Convert an argument value to a number. */
|
|
|
|
long long
|
2013-05-31 12:19:34 +00:00
|
|
|
args_strtonum(struct args *args, u_char ch, long long minval, long long maxval,
|
|
|
|
char **cause)
|
2011-01-04 00:42:46 +00:00
|
|
|
{
|
2013-05-31 12:19:34 +00:00
|
|
|
const char *errstr;
|
|
|
|
long long ll;
|
|
|
|
struct args_entry *entry;
|
2011-01-04 00:42:46 +00:00
|
|
|
|
2013-05-31 12:19:34 +00:00
|
|
|
if ((entry = args_find(args, ch)) == NULL) {
|
2011-01-04 00:42:46 +00:00
|
|
|
*cause = xstrdup("missing");
|
|
|
|
return (0);
|
|
|
|
}
|
|
|
|
|
2013-05-31 12:19:34 +00:00
|
|
|
ll = strtonum(entry->value, minval, maxval, &errstr);
|
2011-01-04 00:42:46 +00:00
|
|
|
if (errstr != NULL) {
|
|
|
|
*cause = xstrdup(errstr);
|
|
|
|
return (0);
|
|
|
|
}
|
|
|
|
|
|
|
|
*cause = NULL;
|
|
|
|
return (ll);
|
|
|
|
}
|