cmd-set-environment.c

   1 /* $OpenBSD$ */
   2
   3 /*
   4  * Copyright (c) 2009 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
  21 #include <stdlib.h>
  22 #include <string.h>
  23
  24 #include "tmux.h"
  25
  26 /*
  27  * Set an environment variable.
  28  */
  29
  30 static enum cmd_retval  cmd_set_environment_exec(struct cmd *,
  31                             struct cmdq_item *);
  32
  33 const struct cmd_entry cmd_set_environment_entry = {
  34         .name = "set-environment",
  35         .alias = "setenv",
  36
  37         .args = { "Fhgrt:u", 1, 2, NULL },
  38         .usage = "[-Fhgru] " CMD_TARGET_SESSION_USAGE " name [value]",
  39
  40         .target = { 't', CMD_FIND_SESSION, CMD_FIND_CANFAIL },
  41
  42         .flags = CMD_AFTERHOOK,
  43         .exec = cmd_set_environment_exec
  44 };
  45
  46 static enum cmd_retval
  47 cmd_set_environment_exec(struct cmd *self, struct cmdq_item *item)
  48 {
  49         struct args             *args = cmd_get_args(self);
  50         struct cmd_find_state   *target = cmdq_get_target(item);
  51         struct environ          *env;
  52         const char              *name = args_string(args, 0), *value;
  53         const char              *tflag;
  54         char                    *expanded = NULL;
  55         enum cmd_retval          retval = CMD_RETURN_NORMAL;
  56
  57         if (*name == '\0') {
  58                 cmdq_error(item, "empty variable name");
  59                 return (CMD_RETURN_ERROR);
  60         }
  61         if (strchr(name, '=') != NULL) {
  62                 cmdq_error(item, "variable name contains =");
  63                 return (CMD_RETURN_ERROR);
  64         }
  65
  66         if (args_count(args) < 2)
  67                 value = NULL;
  68         else
  69                 value = args_string(args, 1);
  70         if (value != NULL && args_has(args, 'F')) {
  71                 expanded = format_single_from_target(item, value);
  72                 value = expanded;
  73         }
  74         if (args_has(args, 'g'))
  75                 env = global_environ;
  76         else {
  77                 if (target->s == NULL) {
  78                         tflag = args_get(args, 't');
  79                         if (tflag != NULL)
  80                                 cmdq_error(item, "no such session: %s", tflag);
  81                         else
  82                                 cmdq_error(item, "no current session");
  83                         retval = CMD_RETURN_ERROR;
  84                         goto out;
  85                 }
  86                 env = target->s->environ;
  87         }
  88
  89         if (args_has(args, 'u')) {
  90                 if (value != NULL) {
  91                         cmdq_error(item, "can't specify a value with -u");
  92                         retval = CMD_RETURN_ERROR;
  93                         goto out;
  94                 }
  95                 environ_unset(env, name);
  96         } else if (args_has(args, 'r')) {
  97                 if (value != NULL) {
  98                         cmdq_error(item, "can't specify a value with -r");
  99                         retval = CMD_RETURN_ERROR;
 100                         goto out;
 101                 }
 102                 environ_clear(env, name);
 103         } else {
 104                 if (value == NULL) {
 105                         cmdq_error(item, "no value specified");
 106                         retval = CMD_RETURN_ERROR;
 107                         goto out;
 108                 }
 109
 110                 if (args_has(args, 'h'))
 111                         environ_set(env, name, ENVIRON_HIDDEN, "%s", value);
 112                 else
 113                         environ_set(env, name, 0, "%s", value);
 114         }
 115
 116 out:
 117         free(expanded);
 118         return (retval);
 119 }