New version 1.2.3.

[misc] / xtitle.c

#include <errno.h>
#include <unistd.h>
#include <termios.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <fcntl.h>

int main(int argc, char *argv[])
{
  int query = 0;
  int fd;
  int openned = 0;
  int i;

  for (;;) {
    i = getopt(argc, argv, "q");
    if (i < 0)
      break;
    switch (i) {
      case 'q':
	query = 1;
	break;
      default:
	fprintf(stderr, "usage: xtitle [-q] [STRING]\n");
	return (1);
    }
  }

  if (!query && optind == argc) {
    fprintf(stderr, "xtitle: no string to set\n");
    return (1);
  }

  {
    char *t = getenv("TERM");
    if (!t || strncmp(t, "xterm", 5))
      return (0);
  }

  if (isatty(0))
    fd = 0;
  else {
    fd = open("/dev/tty", O_RDWR);
    if (fd < 0) {
      fprintf(stderr, "xtitle: couldn't open terminal: %s", strerror(errno));
      return (1);
    }
    openned = 1;
  }

  if (!query) {
    char sp = ' ';
    write(fd, "\33]0;", 4);
    for (i = optind; i < argc; i++) {
      write(fd, argv[i], strlen(argv[i]));
      if (i < argc - 1)
	write(fd, &sp, 1);
    }
    write(fd, "\7", 2);
  } else {
    struct termios o, n;
    char hack;
    int state = 0;

    tcgetattr(fd, &o);
    n = o;
    n.c_iflag &= ~(IGNBRK|BRKINT|PARMRK|ISTRIP
		   |INLCR|IGNCR|ICRNL|IXON);
    n.c_lflag &= ~(ECHO|ECHONL|ICANON|ISIG|IEXTEN);
    n.c_cflag &= ~(CSIZE|PARENB);
    n.c_cflag |= CS8;
    tcsetattr(fd, TCSAFLUSH, &n);
    write(fd, "\33[21t", 5);

    while (state != -1) {
      if (read(fd, &hack, 1) < 1)
	break;
      switch (state) {
	case 0:
	  if (hack == '\33') state = 1;
	  break;
	case 1:
	  if (hack == ']') state = 2; else state = 0;
	  break;
	case 2:
	  if (hack == 'l') state = 3; else state = 0;
	  break;
	case 3:
	  if (hack == '\33') state = 4; else putchar(hack);
	  break;
	case 4:
	  if (hack == '\\') { state = -1; putchar('\n'); }
	  else { putchar('\33'); putchar(hack); state = 3; }
	  break;
      }
    }

    tcsetattr(fd, TCSAFLUSH, &o);
  }

  if (openned)
    close(fd);

  return (0);
}