chiark - git - mdw - misc/blob - xtitle.c

   1 #include <errno.h>
   2 #include <unistd.h>
   3 #include <termios.h>
   4 #include <stdio.h>
   5 #include <stdlib.h>
   6 #include <string.h>
   7 #include <fcntl.h>
   8
   9 #ifdef BASH_BUILTIN
  10 #include "bash/config.h"
  11 #include "bash/shell.h"
  12 #include "bash/builtins.h"
  13 #include "bash/builtins/common.h"
  14 #include "bash/builtins/bashgetopt.h"
  15 #endif
  16
  17 #ifdef BASH_BUILTIN
  18 int xtitle_builtin(WORD_LIST *list)
  19 #else
  20 int main(int argc, char *argv[])
  21 #endif
  22 {
  23   int query = 0;
  24   int fd;
  25   int openned = 0;
  26
  27 #ifdef BASH_BUILTIN
  28   reset_internal_getopt();
  29 #endif
  30   for (;;) {
  31 #ifdef BASH_BUILTIN
  32     int i;
  33     i = internal_getopt(list, "q");
  34 #else
  35     int i = getopt(argc, argv, "q");
  36 #endif
  37     if (i < 0)
  38       break;
  39     switch (i) {
  40       case 'q':
  41         query = 1;
  42         break;
  43       default:
  44 #ifdef BASH_BUILTIN
  45         builtin_usage();
  46 #else
  47         fprintf(stderr, "usage: xtitle [-q] [string]\n");
  48 #endif
  49         return (1);
  50     }
  51   }
  52
  53 #ifdef BASH_BUILTIN
  54   if (!query && loptend == 0) {
  55 #else
  56   if (!query && optind == argc) {
  57 #endif
  58     fprintf(stderr, "xtitle: no string to set\n");
  59     return (1);
  60   }
  61
  62   {
  63     char *t = getenv("TERM");
  64     if (!t || strncmp(t, "xterm", 5))
  65       return (0);
  66   }
  67
  68   if (isatty(0))
  69     fd = 0;
  70   else {
  71     fd = open("/dev/tty", O_RDWR);
  72     if (fd < 0) {
  73       fprintf(stderr, "xtitle: couldn't open terminal: %s", strerror(errno));
  74       return (1);
  75     }
  76     openned = 1;
  77   }
  78
  79   if (!query) {
  80 #ifdef BASH_BUILTIN
  81     WORD_LIST *l = loptend;
  82     char sp = ' ';
  83     write(fd, "\33]0;", 4);
  84     while (l) {
  85       write(fd, l->word->word, strlen(l->word->word));
  86       if (l->next)
  87         write(fd, &sp, 1);
  88       l = l->next;
  89     }
  90     write(fd, "\7", 2);
  91 #else
  92     int i;
  93     char sp = ' ';
  94     write(fd, "\33]0;", 4);
  95     for (i = optind; i < argc; i++) {
  96       write(fd, argv[i], strlen(argv[i]));
  97       if (i < argc - 1)
  98         write(fd, &sp, 1);
  99     }
 100     write(fd, "\7", 2);
 101 #endif
 102   } else {
 103     struct termios o, n;
 104     char hack;
 105     int state = 0;
 106
 107     tcgetattr(fd, &o);
 108     n = o;
 109     n.c_iflag &= ~(IGNBRK|BRKINT|PARMRK|ISTRIP
 110                    |INLCR|IGNCR|ICRNL|IXON);
 111     n.c_lflag &= ~(ECHO|ECHONL|ICANON|ISIG|IEXTEN);
 112     n.c_cflag &= ~(CSIZE|PARENB);
 113     n.c_cflag |= CS8;
 114     tcsetattr(fd, TCSAFLUSH, &n);
 115     write(fd, "\33[21t", 5);
 116
 117     while (state != -1) {
 118       if (read(fd, &hack, 1) < 1)
 119         break;
 120       switch (state) {
 121         case 0:
 122           if (hack == '\33') state = 1;
 123           break;
 124         case 1:
 125           if (hack == ']') state = 2; else state = 0;
 126           break;
 127         case 2:
 128           if (hack == 'l') state = 3; else state = 0;
 129           break;
 130         case 3:
 131           if (hack == '\33') state = 4; else putchar(hack);
 132           break;
 133         case 4:
 134           if (hack == '\\') { state = -1; putchar('\n'); }
 135           else { putchar('\33'); putchar(hack); state = 3; }
 136           break;
 137       }
 138     }
 139
 140     tcsetattr(fd, TCSAFLUSH, &o);
 141   }
 142
 143   if (openned)
 144     close(fd);
 145
 146   return (0);
 147 }
 148
 149
 150 #ifdef BASH_BUILTIN
 151
 152 static char *xtitle_doc[] = {
 153   "Either set or read the title of the current xterm window.  With the",
 154   "-q option, writes the current xterm title to standard output.  Without",
 155   "the -q option, sets the xterm title to be the arguments given,",
 156   "separated by space characters.  [By Mark Wooding, mdw@nsict.org]",
 157   0
 158 };
 159
 160 struct builtin xtitle_struct = {
 161   "xtitle",
 162   xtitle_builtin,
 163   BUILTIN_ENABLED,
 164   xtitle_doc,
 165   "xtitle [-q] [arguments]",
 166   0
 167 };
 168
 169 #endif