pager.c

   1 #include "cache.h"
   2 #include "config.h"
   3 #include "run-command.h"
   4 #include "sigchain.h"
   5 #include "alias.h"
   6
   7 #ifndef DEFAULT_PAGER
   8 #define DEFAULT_PAGER "less"
   9 #endif
  10
  11 static struct child_process pager_process = CHILD_PROCESS_INIT;
  12 static const char *pager_program;
  13
  14 static void wait_for_pager(int in_signal)
  15 {
  16         if (!in_signal) {
  17                 fflush(stdout);
  18                 fflush(stderr);
  19         }
  20         /* signal EOF to pager */
  21         close(1);
  22         close(2);
  23         if (in_signal)
  24                 finish_command_in_signal(&pager_process);
  25         else
  26                 finish_command(&pager_process);
  27 }
  28
  29 static void wait_for_pager_atexit(void)
  30 {
  31         wait_for_pager(0);
  32 }
  33
  34 static void wait_for_pager_signal(int signo)
  35 {
  36         wait_for_pager(1);
  37         sigchain_pop(signo);
  38         raise(signo);
  39 }
  40
  41 static int core_pager_config(const char *var, const char *value, void *data)
  42 {
  43         if (!strcmp(var, "core.pager"))
  44                 return git_config_string(&pager_program, var, value);
  45         return 0;
  46 }
  47
  48 const char *git_pager(int stdout_is_tty)
  49 {
  50         const char *pager;
  51
  52         if (!stdout_is_tty)
  53                 return NULL;
  54
  55         pager = getenv("GIT_PAGER");
  56         if (!pager) {
  57                 if (!pager_program)
  58                         read_early_config(core_pager_config, NULL);
  59                 pager = pager_program;
  60         }
  61         if (!pager)
  62                 pager = getenv("PAGER");
  63         if (!pager)
  64                 pager = DEFAULT_PAGER;
  65         if (!*pager || !strcmp(pager, "cat"))
  66                 pager = NULL;
  67
  68         return pager;
  69 }
  70
  71 static void setup_pager_env(struct argv_array *env)
  72 {
  73         const char **argv;
  74         int i;
  75         char *pager_env = xstrdup(PAGER_ENV);
  76         int n = split_cmdline(pager_env, &argv);
  77
  78         if (n < 0)
  79                 die("malformed build-time PAGER_ENV: %s",
  80                         split_cmdline_strerror(n));
  81
  82         for (i = 0; i < n; i++) {
  83                 char *cp = strchr(argv[i], '=');
  84
  85                 if (!cp)
  86                         die("malformed build-time PAGER_ENV");
  87
  88                 *cp = '\0';
  89                 if (!getenv(argv[i])) {
  90                         *cp = '=';
  91                         argv_array_push(env, argv[i]);
  92                 }
  93         }
  94         free(pager_env);
  95         free(argv);
  96 }
  97
  98 void prepare_pager_args(struct child_process *pager_process, const char *pager)
  99 {
 100         argv_array_push(&pager_process->args, pager);
 101         pager_process->use_shell = 1;
 102         setup_pager_env(&pager_process->env_array);
 103         pager_process->trace2_child_class = "pager";
 104 }
 105
 106 void setup_pager(void)
 107 {
 108         const char *pager = git_pager(isatty(1));
 109
 110         if (!pager)
 111                 return;
 112
 113         /*
 114          * After we redirect standard output, we won't be able to use an ioctl
 115          * to get the terminal size. Let's grab it now, and then set $COLUMNS
 116          * to communicate it to any sub-processes.
 117          */
 118         {
 119                 char buf[64];
 120                 xsnprintf(buf, sizeof(buf), "%d", term_columns());
 121                 setenv("COLUMNS", buf, 0);
 122         }
 123
 124         setenv("GIT_PAGER_IN_USE", "true", 1);
 125
 126         /* spawn the pager */
 127         prepare_pager_args(&pager_process, pager);
 128         pager_process.in = -1;
 129         argv_array_push(&pager_process.env_array, "GIT_PAGER_IN_USE");
 130         if (start_command(&pager_process))
 131                 return;
 132
 133         /* original process continues, but writes to the pipe */
 134         dup2(pager_process.in, 1);
 135         if (isatty(2))
 136                 dup2(pager_process.in, 2);
 137         close(pager_process.in);
 138
 139         /* this makes sure that the parent terminates after the pager */
 140         sigchain_push_common(wait_for_pager_signal);
 141         atexit(wait_for_pager_atexit);
 142 }
 143
 144 int pager_in_use(void)
 145 {
 146         return git_env_bool("GIT_PAGER_IN_USE", 0);
 147 }
 148
 149 /*
 150  * Return cached value (if set) or $COLUMNS environment variable (if
 151  * set and positive) or ioctl(1, TIOCGWINSZ).ws_col (if positive),
 152  * and default to 80 if all else fails.
 153  */
 154 int term_columns(void)
 155 {
 156         static int term_columns_at_startup;
 157
 158         char *col_string;
 159         int n_cols;
 160
 161         if (term_columns_at_startup)
 162                 return term_columns_at_startup;
 163
 164         term_columns_at_startup = 80;
 165
 166         col_string = getenv("COLUMNS");
 167         if (col_string && (n_cols = atoi(col_string)) > 0)
 168                 term_columns_at_startup = n_cols;
 169 #ifdef TIOCGWINSZ
 170         else {
 171                 struct winsize ws;
 172                 if (!ioctl(1, TIOCGWINSZ, &ws) && ws.ws_col)
 173                         term_columns_at_startup = ws.ws_col;
 174         }
 175 #endif
 176
 177         return term_columns_at_startup;
 178 }
 179
 180 /*
 181  * How many columns do we need to show this number in decimal?
 182  */
 183 int decimal_width(uintmax_t number)
 184 {
 185         int width;
 186
 187         for (width = 1; number >= 10; width++)
 188                 number /= 10;
 189         return width;
 190 }
 191
 192 struct pager_command_config_data {
 193         const char *cmd;
 194         int want;
 195         char *value;
 196 };
 197
 198 static int pager_command_config(const char *var, const char *value, void *vdata)
 199 {
 200         struct pager_command_config_data *data = vdata;
 201         const char *cmd;
 202
 203         if (skip_prefix(var, "pager.", &cmd) && !strcmp(cmd, data->cmd)) {
 204                 int b = git_parse_maybe_bool(value);
 205                 if (b >= 0)
 206                         data->want = b;
 207                 else {
 208                         data->want = 1;
 209                         data->value = xstrdup(value);
 210                 }
 211         }
 212
 213         return 0;
 214 }
 215
 216 /* returns 0 for "no pager", 1 for "use pager", and -1 for "not specified" */
 217 int check_pager_config(const char *cmd)
 218 {
 219         struct pager_command_config_data data;
 220
 221         data.cmd = cmd;
 222         data.want = -1;
 223         data.value = NULL;
 224
 225         read_early_config(pager_command_config, &data);
 226
 227         if (data.value)
 228                 pager_program = data.value;
 229         return data.want;
 230 }