cmd-set-option.c

   1 /* $OpenBSD$ */
   2
   3 /*
   4  * Copyright (c) 2007 Nicholas Marriott <nicm@users.sourceforge.net>
   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
  21 #include <stdlib.h>
  22 #include <string.h>
  23
  24 #include "tmux.h"
  25
  26 /*
  27  * Set an option.
  28  */
  29
  30 int     cmd_set_option_exec(struct cmd *, struct cmd_ctx *);
  31
  32 int     cmd_set_option_find(const char *, const struct options_table_entry **,
  33             const struct options_table_entry **);
  34
  35 int     cmd_set_option_unset(struct cmd *, struct cmd_ctx *,
  36             const struct options_table_entry *, struct options *,
  37             const char *);
  38 int     cmd_set_option_set(struct cmd *, struct cmd_ctx *,
  39             const struct options_table_entry *, struct options *,
  40             const char *);
  41
  42 struct options_entry *cmd_set_option_string(struct cmd *, struct cmd_ctx *,
  43             const struct options_table_entry *, struct options *,
  44             const char *);
  45 struct options_entry *cmd_set_option_number(struct cmd *, struct cmd_ctx *,
  46             const struct options_table_entry *, struct options *,
  47             const char *);
  48 struct options_entry *cmd_set_option_key(struct cmd *, struct cmd_ctx *,
  49             const struct options_table_entry *, struct options *,
  50             const char *);
  51 struct options_entry *cmd_set_option_colour(struct cmd *, struct cmd_ctx *,
  52             const struct options_table_entry *, struct options *,
  53             const char *);
  54 struct options_entry *cmd_set_option_attributes(struct cmd *, struct cmd_ctx *,
  55             const struct options_table_entry *, struct options *,
  56             const char *);
  57 struct options_entry *cmd_set_option_flag(struct cmd *, struct cmd_ctx *,
  58             const struct options_table_entry *, struct options *,
  59             const char *);
  60 struct options_entry *cmd_set_option_choice(struct cmd *, struct cmd_ctx *,
  61             const struct options_table_entry *, struct options *,
  62             const char *);
  63
  64 const struct cmd_entry cmd_set_option_entry = {
  65         "set-option", "set",
  66         "agst:uw", 1, 2,
  67         "[-agsuw] [-t target-session|target-window] option [value]",
  68         0,
  69         NULL,
  70         NULL,
  71         cmd_set_option_exec
  72 };
  73
  74 const struct cmd_entry cmd_set_window_option_entry = {
  75         "set-window-option", "setw",
  76         "agt:u", 1, 2,
  77         "[-agu] " CMD_TARGET_WINDOW_USAGE " option [value]",
  78         0,
  79         NULL,
  80         NULL,
  81         cmd_set_option_exec
  82 };
  83
  84 /* Look for an option in all three tables. */
  85 int
  86 cmd_set_option_find(
  87     const char *optstr, const struct options_table_entry **table,
  88     const struct options_table_entry **oe)
  89 {
  90         static const struct options_table_entry *tables[] = {
  91                 server_options_table,
  92                 window_options_table,
  93                 session_options_table
  94         };
  95         const struct options_table_entry        *oe_loop;
  96         u_int                                    i;
  97
  98         for (i = 0; i < nitems(tables); i++) {
  99                 for (oe_loop = tables[i]; oe_loop->name != NULL; oe_loop++) {
 100                         if (strncmp(oe_loop->name, optstr, strlen(optstr)) != 0)
 101                                 continue;
 102
 103                         /* If already found, ambiguous. */
 104                         if (*oe != NULL)
 105                                 return (-1);
 106                         *oe = oe_loop;
 107                         *table = tables[i];
 108
 109                         /* Bail now if an exact match. */
 110                         if (strcmp((*oe)->name, optstr) == 0)
 111                                 break;
 112                 }
 113         }
 114         return (0);
 115 }
 116
 117 int
 118 cmd_set_option_exec(struct cmd *self, struct cmd_ctx *ctx)
 119 {
 120         struct args                             *args = self->args;
 121         const struct options_table_entry        *table, *oe;
 122         struct session                          *s;
 123         struct winlink                          *wl;
 124         struct client                           *c;
 125         struct options                          *oo;
 126         const char                              *optstr, *valstr;
 127         u_int                                    i;
 128
 129         /* Get the option name and value. */
 130         optstr = args->argv[0];
 131         if (*optstr == '\0') {
 132                 ctx->error(ctx, "invalid option");
 133                 return (-1);
 134         }
 135         if (args->argc < 2)
 136                 valstr = NULL;
 137         else
 138                 valstr = args->argv[1];
 139
 140         /* Find the option entry, try each table. */
 141         table = oe = NULL;
 142         if (cmd_set_option_find(optstr, &table, &oe) != 0) {
 143                 ctx->error(ctx, "ambiguous option: %s", optstr);
 144                 return (-1);
 145         }
 146         if (oe == NULL) {
 147                 ctx->error(ctx, "unknown option: %s", optstr);
 148                 return (-1);
 149         }
 150
 151         /* Work out the tree from the table. */
 152         if (table == server_options_table)
 153                 oo = &global_options;
 154         else if (table == window_options_table) {
 155                 if (args_has(self->args, 'g'))
 156                         oo = &global_w_options;
 157                 else {
 158                         wl = cmd_find_window(ctx, args_get(args, 't'), NULL);
 159                         if (wl == NULL)
 160                                 return (-1);
 161                         oo = &wl->window->options;
 162                 }
 163         } else if (table == session_options_table) {
 164                 if (args_has(self->args, 'g'))
 165                         oo = &global_s_options;
 166                 else {
 167                         s = cmd_find_session(ctx, args_get(args, 't'), 0);
 168                         if (s == NULL)
 169                                 return (-1);
 170                         oo = &s->options;
 171                 }
 172         } else {
 173                 ctx->error(ctx, "unknown table");
 174                 return (-1);
 175         }
 176
 177         /* Unset or set the option. */
 178         if (args_has(args, 'u')) {
 179                 if (cmd_set_option_unset(self, ctx, oe, oo, valstr) != 0)
 180                         return (-1);
 181         } else {
 182                 if (cmd_set_option_set(self, ctx, oe, oo, valstr) != 0)
 183                         return (-1);
 184         }
 185
 186         /* Update sizes and redraw. May not need it but meh. */
 187         recalculate_sizes();
 188         for (i = 0; i < ARRAY_LENGTH(&clients); i++) {
 189                 c = ARRAY_ITEM(&clients, i);
 190                 if (c != NULL && c->session != NULL)
 191                         server_redraw_client(c);
 192         }
 193
 194         return (0);
 195 }
 196
 197 /* Unset an option. */
 198 int
 199 cmd_set_option_unset(struct cmd *self, struct cmd_ctx *ctx,
 200     const struct options_table_entry *oe, struct options *oo, const char *value)
 201 {
 202         struct args     *args = self->args;
 203
 204         if (args_has(args, 'g')) {
 205                 ctx->error(ctx, "can't unset global option: %s", oe->name);
 206                 return (-1);
 207         }
 208         if (value != NULL) {
 209                 ctx->error(ctx, "value passed to unset option: %s", oe->name);
 210                 return (-1);
 211         }
 212
 213         options_remove(oo, oe->name);
 214         ctx->info(ctx, "unset option: %s", oe->name);
 215         return (0);
 216 }
 217
 218 /* Set an option. */
 219 int
 220 cmd_set_option_set(struct cmd *self, struct cmd_ctx *ctx,
 221     const struct options_table_entry *oe, struct options *oo, const char *value)
 222 {
 223         struct options_entry    *o;
 224         const char              *s;
 225
 226         if (oe->type != OPTIONS_TABLE_FLAG && value == NULL) {
 227                 ctx->error(ctx, "empty value");
 228                 return (-1);
 229         }
 230
 231         o = NULL;
 232         switch (oe->type) {
 233         case OPTIONS_TABLE_STRING:
 234                 o = cmd_set_option_string(self, ctx, oe, oo, value);
 235                 break;
 236         case OPTIONS_TABLE_NUMBER:
 237                 o = cmd_set_option_number(self, ctx, oe, oo, value);
 238                 break;
 239         case OPTIONS_TABLE_KEY:
 240                 o = cmd_set_option_key(self, ctx, oe, oo, value);
 241                 break;
 242         case OPTIONS_TABLE_COLOUR:
 243                 o = cmd_set_option_colour(self, ctx, oe, oo, value);
 244                 break;
 245         case OPTIONS_TABLE_ATTRIBUTES:
 246                 o = cmd_set_option_attributes(self, ctx, oe, oo, value);
 247                 break;
 248         case OPTIONS_TABLE_FLAG:
 249                 o = cmd_set_option_flag(self, ctx, oe, oo, value);
 250                 break;
 251         case OPTIONS_TABLE_CHOICE:
 252                 o = cmd_set_option_choice(self, ctx, oe, oo, value);
 253                 break;
 254         }
 255         if (o == NULL)
 256                 return (-1);
 257
 258         s = options_table_print_entry(oe, o);
 259         ctx->info(ctx, "set option: %s -> %s", oe->name, s);
 260         return (0);
 261 }
 262
 263 /* Set a string option. */
 264 struct options_entry *
 265 cmd_set_option_string(struct cmd *self, unused struct cmd_ctx *ctx,
 266     const struct options_table_entry *oe, struct options *oo, const char *value)
 267 {
 268         struct args     *args = self->args;
 269         struct options_entry    *o;
 270         char                    *oldval, *newval;
 271
 272         if (args_has(args, 'a')) {
 273                 oldval = options_get_string(oo, oe->name);
 274                 xasprintf(&newval, "%s%s", oldval, value);
 275         } else
 276                 newval = xstrdup(value);
 277
 278         o = options_set_string(oo, oe->name, "%s", newval);
 279
 280         xfree(newval);
 281         return (o);
 282 }
 283
 284 /* Set a number option. */
 285 struct options_entry *
 286 cmd_set_option_number(unused struct cmd *self, struct cmd_ctx *ctx,
 287     const struct options_table_entry *oe, struct options *oo, const char *value)
 288 {
 289         long long        ll;
 290         const char      *errstr;
 291
 292         ll = strtonum(value, oe->minimum, oe->maximum, &errstr);
 293         if (errstr != NULL) {
 294                 ctx->error(ctx, "value is %s: %s", errstr, value);
 295                 return (NULL);
 296         }
 297
 298         return (options_set_number(oo, oe->name, ll));
 299 }
 300
 301 /* Set a key option. */
 302 struct options_entry *
 303 cmd_set_option_key(unused struct cmd *self, struct cmd_ctx *ctx,
 304     const struct options_table_entry *oe, struct options *oo, const char *value)
 305 {
 306         int     key;
 307
 308         if ((key = key_string_lookup_string(value)) == KEYC_NONE) {
 309                 ctx->error(ctx, "bad key: %s", value);
 310                 return (NULL);
 311         }
 312
 313         return (options_set_number(oo, oe->name, key));
 314 }
 315
 316 /* Set a colour option. */
 317 struct options_entry *
 318 cmd_set_option_colour(unused struct cmd *self, struct cmd_ctx *ctx,
 319     const struct options_table_entry *oe, struct options *oo, const char *value)
 320 {
 321         int     colour;
 322
 323         if ((colour = colour_fromstring(value)) == -1) {
 324                 ctx->error(ctx, "bad colour: %s", value);
 325                 return (NULL);
 326         }
 327
 328         return (options_set_number(oo, oe->name, colour));
 329 }
 330
 331 /* Set an attributes option. */
 332 struct options_entry *
 333 cmd_set_option_attributes(unused struct cmd *self, struct cmd_ctx *ctx,
 334     const struct options_table_entry *oe, struct options *oo, const char *value)
 335 {
 336         int     attr;
 337
 338         if ((attr = attributes_fromstring(value)) == -1) {
 339                 ctx->error(ctx, "bad attributes: %s", value);
 340                 return (NULL);
 341         }
 342
 343         return (options_set_number(oo, oe->name, attr));
 344 }
 345
 346 /* Set a flag option. */
 347 struct options_entry *
 348 cmd_set_option_flag(unused struct cmd *self, struct cmd_ctx *ctx,
 349     const struct options_table_entry *oe, struct options *oo, const char *value)
 350 {
 351         int     flag;
 352
 353         if (value == NULL || *value == '\0')
 354                 flag = !options_get_number(oo, oe->name);
 355         else {
 356                 if ((value[0] == '1' && value[1] == '\0') ||
 357                     strcasecmp(value, "on") == 0 ||
 358                     strcasecmp(value, "yes") == 0)
 359                         flag = 1;
 360                 else if ((value[0] == '0' && value[1] == '\0') ||
 361                     strcasecmp(value, "off") == 0 ||
 362                     strcasecmp(value, "no") == 0)
 363                         flag = 0;
 364                 else {
 365                         ctx->error(ctx, "bad value: %s", value);
 366                         return (NULL);
 367                 }
 368         }
 369
 370         return (options_set_number(oo, oe->name, flag));
 371 }
 372
 373 /* Set a choice option. */
 374 struct options_entry *
 375 cmd_set_option_choice(unused struct cmd *self, struct cmd_ctx *ctx,
 376     const struct options_table_entry *oe, struct options *oo, const char *value)
 377 {
 378         const char      **choicep;
 379         int               n, choice = -1;
 380
 381         n = 0;
 382         for (choicep = oe->choices; *choicep != NULL; choicep++) {
 383                 n++;
 384                 if (strncmp(*choicep, value, strlen(value)) != 0)
 385                         continue;
 386
 387                 if (choice != -1) {
 388                         ctx->error(ctx, "ambiguous value: %s", value);
 389                         return (NULL);
 390                 }
 391                 choice = n - 1;
 392         }
 393         if (choice == -1) {
 394                 ctx->error(ctx, "unknown value: %s", value);
 395                 return (NULL);
 396         }
 397
 398         return (options_set_number(oo, oe->name, choice));
 399 }