JasonWoof Got questions, comments, patches, etc.? Contact Jason Woofenden
8347dbe254d8f3c83cf124e672134b5cbd89f609
[st.git] / st.c
1 /* See LICENSE for licence details. */
2 #define _XOPEN_SOURCE 600
3 #include <ctype.h>
4 #include <errno.h>
5 #include <fcntl.h>
6 #include <limits.h>
7 #include <locale.h>
8 #include <stdarg.h>
9 #include <stdio.h>
10 #include <stdlib.h>
11 #include <string.h>
12 #include <signal.h>
13 #include <sys/ioctl.h>
14 #include <sys/select.h>
15 #include <sys/stat.h>
16 #include <sys/types.h>
17 #include <sys/wait.h>
18 #include <unistd.h>
19 #include <X11/Xatom.h>
20 #include <X11/Xlib.h>
21 #include <X11/Xutil.h>
22 #include <X11/cursorfont.h>
23 #include <X11/keysym.h>
24
25 #include <sys/time.h>
26 #include <time.h>
27
28 #if   defined(__linux)
29  #include <pty.h>
30 #elif defined(__OpenBSD__) || defined(__NetBSD__) || defined(__APPLE__)
31  #include <util.h>
32 #elif defined(__FreeBSD__) || defined(__DragonFly__)
33  #include <libutil.h>
34 #endif
35
36 #define USAGE \
37         "st-" VERSION ", (c) 2010 st engineers\n" \
38         "usage: st [-t title] [-c class] [-v] [-e cmd]\n"
39
40 /* Arbitrary sizes */
41 #define ESC_TITLE_SIZ 256
42 #define ESC_BUF_SIZ   256
43 #define ESC_ARG_SIZ   16
44 #define DRAW_BUF_SIZ  1024
45 #define UTF_SIZ       4
46
47 #define SERRNO strerror(errno)
48 #define MIN(a, b)  ((a) < (b) ? (a) : (b))
49 #define MAX(a, b)  ((a) < (b) ? (b) : (a))
50 #define LEN(a)     (sizeof(a) / sizeof(a[0]))
51 #define DEFAULT(a, b)     (a) = (a) ? (a) : (b)
52 #define BETWEEN(x, a, b)  ((a) <= (x) && (x) <= (b))
53 #define LIMIT(x, a, b)    (x) = (x) < (a) ? (a) : (x) > (b) ? (b) : (x)
54 #define ATTRCMP(a, b) ((a).mode != (b).mode || (a).fg != (b).fg || (a).bg != (b).bg)
55 #define IS_SET(flag) (term.mode & (flag))
56 #define TIMEDIFF(t1, t2) ((t1.tv_sec-t2.tv_sec)*1000 + (t1.tv_usec-t2.tv_usec)/1000)
57 #define X2COL(x) (((x) - BORDER)/xw.cw)
58 #define Y2ROW(y) (((y) - BORDER)/xw.ch)
59
60 /* Attribute, Cursor, Character state, Terminal mode, Screen draw mode */
61 enum { ATTR_NULL=0 , ATTR_REVERSE=1 , ATTR_UNDERLINE=2, ATTR_BOLD=4, ATTR_GFX=8 };
62 enum { CURSOR_UP, CURSOR_DOWN, CURSOR_LEFT, CURSOR_RIGHT,
63        CURSOR_SAVE, CURSOR_LOAD };
64 enum { CURSOR_DEFAULT = 0, CURSOR_HIDE = 1, CURSOR_WRAPNEXT = 2 };
65 enum { GLYPH_SET=1, GLYPH_DIRTY=2 };
66 enum { MODE_WRAP=1, MODE_INSERT=2, MODE_APPKEYPAD=4, MODE_ALTSCREEN=8,
67        MODE_CRLF=16, MODE_MOUSE=32 };
68 enum { ESC_START=1, ESC_CSI=2, ESC_OSC=4, ESC_TITLE=8, ESC_ALTCHARSET=16 };
69 enum { WIN_VISIBLE=1, WIN_REDRAW=2, WIN_FOCUSED=4 };
70
71 #undef B0
72 enum { B0=1, B1=2, B2=4, B3=8, B4=16, B5=32, B6=64, B7=128 };
73
74 typedef struct {
75         char c[UTF_SIZ];     /* character code */
76         char mode;  /* attribute flags */
77         int fg;     /* foreground      */
78         int bg;     /* background      */
79         char state; /* state flags     */
80 } Glyph;
81
82 typedef Glyph* Line;
83
84 typedef struct {
85         Glyph attr;      /* current char attributes */
86         int x;
87         int y;
88         char state;
89 } TCursor;
90
91 /* CSI Escape sequence structs */
92 /* ESC '[' [[ [<priv>] <arg> [;]] <mode>] */
93 typedef struct {
94         char buf[ESC_BUF_SIZ]; /* raw string */
95         int len;               /* raw string length */
96         char priv;
97         int arg[ESC_ARG_SIZ];
98         int narg;              /* nb of args */
99         char mode;
100 } CSIEscape;
101
102 /* Internal representation of the screen */
103 typedef struct {
104         int row;        /* nb row */
105         int col;        /* nb col */
106         Line* line;     /* screen */
107         Line* alt;      /* alternate screen */
108         TCursor c;      /* cursor */
109         int top;        /* top    scroll limit */
110         int bot;        /* bottom scroll limit */
111         int mode;       /* terminal mode flags */
112         int esc;        /* escape state flags */
113         char title[ESC_TITLE_SIZ];
114         int titlelen;
115 } Term;
116
117 /* Purely graphic info */
118 typedef struct {
119         Display* dpy;
120         Colormap cmap;
121         Window win;
122         Pixmap buf;
123         XIM xim;
124         XIC xic;
125         int scr;
126         int w;  /* window width */
127         int h;  /* window height */
128         int bufw; /* pixmap width  */
129         int bufh; /* pixmap height */
130         int ch; /* char height */
131         int cw; /* char width  */
132         char state; /* focus, redraw, visible */
133 } XWindow;
134
135 typedef struct {
136         KeySym k;
137         unsigned int mask;
138         char s[ESC_BUF_SIZ];
139 } Key;
140
141 /* Drawing Context */
142 typedef struct {
143         unsigned long col[256];
144         GC gc;
145         struct {
146                 int ascent;
147                 int descent;
148                 short lbearing;
149                 short rbearing;
150                 XFontSet set;
151         } font, bfont;
152 } DC;
153
154 /* TODO: use better name for vars... */
155 typedef struct {
156         int mode;
157         int bx, by;
158         int ex, ey;
159         struct {int x, y;} b, e;
160         char *clip;
161         Atom xtarget;
162         struct timeval tclick1;
163         struct timeval tclick2;
164 } Selection;
165
166 #include "config.h"
167
168 static void die(const char*, ...);
169 static void draw(void);
170 static void drawregion(int, int, int, int);
171 static void execsh(void);
172 static void sigchld(int);
173 static void run(void);
174
175 static void csidump(void);
176 static void csihandle(void);
177 static void csiparse(void);
178 static void csireset(void);
179
180 static void tclearregion(int, int, int, int);
181 static void tcursor(int);
182 static void tdeletechar(int);
183 static void tdeleteline(int);
184 static void tinsertblank(int);
185 static void tinsertblankline(int);
186 static void tmoveto(int, int);
187 static void tnew(int, int);
188 static void tnewline(int);
189 static void tputtab(void);
190 static void tputc(char*);
191 static void treset(void);
192 static int tresize(int, int);
193 static void tscrollup(int, int);
194 static void tscrolldown(int, int);
195 static void tsetattr(int*, int);
196 static void tsetchar(char*);
197 static void tsetscroll(int, int);
198 static void tswapscreen(void);
199
200 static void ttynew(void);
201 static void ttyread(void);
202 static void ttyresize(int, int);
203 static void ttywrite(const char *, size_t);
204
205 static void xdraws(char *, Glyph, int, int, int, int);
206 static void xhints(void);
207 static void xclear(int, int, int, int);
208 static void xdrawcursor(void);
209 static void xinit(void);
210 static void xloadcols(void);
211 static void xseturgency(int);
212 static void xsetsel(char*);
213 static void xresize(int, int);
214
215 static void expose(XEvent *);
216 static void visibility(XEvent *);
217 static void unmap(XEvent *);
218 static char* kmap(KeySym, unsigned int state);
219 static void kpress(XEvent *);
220 static void resize(XEvent *);
221 static void focus(XEvent *);
222 static void brelease(XEvent *);
223 static void bpress(XEvent *);
224 static void bmotion(XEvent *);
225 static void selnotify(XEvent *);
226 static void selrequest(XEvent *);
227
228 static void selinit(void);
229 static inline int selected(int, int);
230 static void selcopy(void);
231 static void selpaste();
232
233 static int utf8decode(char *, long *);
234 static int utf8encode(long *, char *);
235 static int utf8size(char *);
236 static int isfullutf8(char *, int);
237
238 static void (*handler[LASTEvent])(XEvent *) = {
239         [KeyPress] = kpress,
240         [ConfigureNotify] = resize,
241         [VisibilityNotify] = visibility,
242         [UnmapNotify] = unmap,
243         [Expose] = expose,
244         [FocusIn] = focus,
245         [FocusOut] = focus,
246         [MotionNotify] = bmotion,
247         [ButtonPress] = bpress,
248         [ButtonRelease] = brelease,
249         [SelectionNotify] = selnotify,
250         [SelectionRequest] = selrequest,
251 };
252
253 /* Globals */
254 static DC dc;
255 static XWindow xw;
256 static Term term;
257 static CSIEscape escseq;
258 static int cmdfd;
259 static pid_t pid;
260 static Selection sel;
261 static char **opt_cmd  = NULL;
262 static char *opt_title = NULL;
263 static char *opt_class = NULL;
264
265 int
266 utf8decode(char *s, long *u) {
267         unsigned char c;
268         int i, n, rtn;
269
270         rtn = 1;
271         c = *s;
272         if(~c&B7) { /* 0xxxxxxx */
273                 *u = c;
274                 return rtn;
275         } else if((c&(B7|B6|B5)) == (B7|B6)) { /* 110xxxxx */
276                 *u = c&(B4|B3|B2|B1|B0);
277                 n = 1;
278         } else if((c&(B7|B6|B5|B4)) == (B7|B6|B5)) { /* 1110xxxx */
279                 *u = c&(B3|B2|B1|B0);
280                 n = 2;
281         } else if((c&(B7|B6|B5|B4|B3)) == (B7|B6|B5|B4)) { /* 11110xxx */
282                 *u = c&(B2|B1|B0);
283                 n = 3;
284         } else
285                 goto invalid;
286         for(i=n,++s; i>0; --i,++rtn,++s) {
287                 c = *s;
288                 if((c&(B7|B6)) != B7) /* 10xxxxxx */
289                         goto invalid;
290                 *u <<= 6;
291                 *u |= c&(B5|B4|B3|B2|B1|B0);
292         }
293         if((n == 1 && *u < 0x80) ||
294            (n == 2 && *u < 0x800) ||
295            (n == 3 && *u < 0x10000) ||
296            (*u >= 0xD800 && *u <= 0xDFFF))
297                 goto invalid;
298         return rtn;
299 invalid:
300         *u = 0xFFFD;
301         return rtn;
302 }
303
304 int
305 utf8encode(long *u, char *s) {
306         unsigned char *sp;
307         unsigned long uc;
308         int i, n;
309
310         sp = (unsigned char*) s;
311         uc = *u;
312         if(uc < 0x80) {
313                 *sp = uc; /* 0xxxxxxx */
314                 return 1;
315         } else if(*u < 0x800) {
316                 *sp = (uc >> 6) | (B7|B6); /* 110xxxxx */
317                 n = 1;
318         } else if(uc < 0x10000) {
319                 *sp = (uc >> 12) | (B7|B6|B5); /* 1110xxxx */
320                 n = 2;
321         } else if(uc <= 0x10FFFF) {
322                 *sp = (uc >> 18) | (B7|B6|B5|B4); /* 11110xxx */
323                 n = 3;
324         } else {
325                 goto invalid;
326         }
327         for(i=n,++sp; i>0; --i,++sp)
328                 *sp = ((uc >> 6*(i-1)) & (B5|B4|B3|B2|B1|B0)) | B7; /* 10xxxxxx */
329         return n+1;
330 invalid:
331         /* U+FFFD */
332         *s++ = '\xEF';
333         *s++ = '\xBF';
334         *s = '\xBD';
335         return 3;
336 }
337
338 /* use this if your buffer is less than UTF_SIZ, it returns 1 if you can decode
339    UTF-8 otherwise return 0 */
340 int
341 isfullutf8(char *s, int b) {
342         unsigned char *c1, *c2, *c3;
343
344         c1 = (unsigned char *) s;
345         c2 = (unsigned char *) ++s;
346         c3 = (unsigned char *) ++s;
347         if(b < 1)
348                 return 0;
349         else if((*c1&(B7|B6|B5)) == (B7|B6) && b == 1)
350                 return 0;
351         else if((*c1&(B7|B6|B5|B4)) == (B7|B6|B5) &&
352             ((b == 1) ||
353             ((b == 2) && (*c2&(B7|B6)) == B7)))
354                 return 0;
355         else if((*c1&(B7|B6|B5|B4|B3)) == (B7|B6|B5|B4) &&
356             ((b == 1) ||
357             ((b == 2) && (*c2&(B7|B6)) == B7) ||
358             ((b == 3) && (*c2&(B7|B6)) == B7 && (*c3&(B7|B6)) == B7)))
359                 return 0;
360         else
361                 return 1;
362 }
363
364 int
365 utf8size(char *s) {
366         unsigned char c = *s;
367
368         if(~c&B7)
369                 return 1;
370         else if((c&(B7|B6|B5)) == (B7|B6))
371                 return 2;
372         else if((c&(B7|B6|B5|B4)) == (B7|B6|B5))
373                 return 3;
374         else
375                 return 4;
376 }
377
378 void
379 selinit(void) {
380         sel.tclick1.tv_sec = 0;
381         sel.tclick1.tv_usec = 0;
382         sel.mode = 0;
383         sel.bx = -1;
384         sel.clip = NULL;
385         sel.xtarget = XInternAtom(xw.dpy, "UTF8_STRING", 0);
386         if(sel.xtarget == None)
387                 sel.xtarget = XA_STRING;
388 }
389
390 static inline int
391 selected(int x, int y) {
392         if(sel.ey == y && sel.by == y) {
393                 int bx = MIN(sel.bx, sel.ex);
394                 int ex = MAX(sel.bx, sel.ex);
395                 return BETWEEN(x, bx, ex);
396         }
397         return ((sel.b.y < y&&y < sel.e.y) || (y==sel.e.y && x<=sel.e.x))
398                 || (y==sel.b.y && x>=sel.b.x && (x<=sel.e.x || sel.b.y!=sel.e.y));
399 }
400
401 void
402 getbuttoninfo(XEvent *e, int *b, int *x, int *y) {
403         if(b)
404                 *b = e->xbutton.button;
405
406         *x = X2COL(e->xbutton.x);
407         *y = Y2ROW(e->xbutton.y);
408         sel.b.x = sel.by < sel.ey ? sel.bx : sel.ex;
409         sel.b.y = MIN(sel.by, sel.ey);
410         sel.e.x = sel.by < sel.ey ? sel.ex : sel.bx;
411         sel.e.y = MAX(sel.by, sel.ey);
412 }
413
414 void
415 mousereport(XEvent *e) {
416         int x = X2COL(e->xbutton.x);
417         int y = Y2ROW(e->xbutton.y);
418         int button = e->xbutton.button;
419         int state = e->xbutton.state;
420         char buf[] = { '\033', '[', 'M', 0, 32+x+1, 32+y+1 };
421         
422         if(!IS_SET(MODE_MOUSE))
423                 return;
424         
425         /* from urxvt */
426         if(e->xbutton.type == ButtonRelease || button == AnyButton)
427                 button = 3;
428         else {
429                 button -= Button1;
430                 if(button >= 3)
431                         button += 64 - 3;
432         }
433         
434         buf[3] = 32 + button + (state & ShiftMask ? 4 : 0)
435                 + (state & Mod4Mask    ? 8  : 0)
436                 + (state & ControlMask ? 16 : 0);
437         
438         ttywrite(buf, sizeof(buf));
439 }
440
441 void
442 bpress(XEvent *e) {
443         mousereport(e);
444         sel.mode = 1;
445         sel.ex = sel.bx = X2COL(e->xbutton.x);
446         sel.ey = sel.by = Y2ROW(e->xbutton.y);
447 }
448
449 void
450 selcopy(void) {
451         char *str, *ptr;
452         int x, y, sz, sl, ls = 0;
453
454         if(sel.bx == -1)
455                 str = NULL;
456         else {
457                 sz = (term.col+1) * (sel.e.y-sel.b.y+1) * UTF_SIZ;
458                 ptr = str = malloc(sz);
459                 for(y = 0; y < term.row; y++) {
460                         for(x = 0; x < term.col; x++)
461                                 if(term.line[y][x].state & GLYPH_SET && (ls = selected(x, y))) {
462                                         sl = utf8size(term.line[y][x].c);
463                                         memcpy(ptr, term.line[y][x].c, sl);
464                                         ptr += sl;
465                                 }
466                         if(ls && y < sel.e.y)
467                                 *ptr++ = '\n';
468                 }
469                 *ptr = 0;
470         }
471         xsetsel(str);
472 }
473
474 void
475 selnotify(XEvent *e) {
476         unsigned long nitems;
477         unsigned long ofs, rem;
478         int format;
479         unsigned char *data;
480         Atom type;
481
482         ofs = 0;
483         do {
484                 if(XGetWindowProperty(xw.dpy, xw.win, XA_PRIMARY, ofs, BUFSIZ/4,
485                                         False, AnyPropertyType, &type, &format,
486                                         &nitems, &rem, &data)) {
487                         fprintf(stderr, "Clipboard allocation failed\n");
488                         return;
489                 }
490                 ttywrite((const char *) data, nitems * format / 8);
491                 XFree(data);
492                 /* number of 32-bit chunks returned */
493                 ofs += nitems * format / 32;
494         } while(rem > 0);
495 }
496
497 void
498 selpaste() {
499         XConvertSelection(xw.dpy, XA_PRIMARY, sel.xtarget, XA_PRIMARY, xw.win, CurrentTime);
500 }
501
502 void
503 selrequest(XEvent *e) {
504         XSelectionRequestEvent *xsre;
505         XSelectionEvent xev;
506         Atom xa_targets;
507
508         xsre = (XSelectionRequestEvent *) e;
509         xev.type = SelectionNotify;
510         xev.requestor = xsre->requestor;
511         xev.selection = xsre->selection;
512         xev.target = xsre->target;
513         xev.time = xsre->time;
514         /* reject */
515         xev.property = None;
516
517         xa_targets = XInternAtom(xw.dpy, "TARGETS", 0);
518         if(xsre->target == xa_targets) {
519                 /* respond with the supported type */
520                 Atom string = sel.xtarget;
521                 XChangeProperty(xsre->display, xsre->requestor, xsre->property,
522                                 XA_ATOM, 32, PropModeReplace,
523                                 (unsigned char *) &string, 1);
524                 xev.property = xsre->property;
525         } else if(xsre->target == sel.xtarget) {
526                 XChangeProperty(xsre->display, xsre->requestor, xsre->property,
527                                 xsre->target, 8, PropModeReplace,
528                                 (unsigned char *) sel.clip, strlen(sel.clip));
529                 xev.property = xsre->property;
530         }
531
532         /* all done, send a notification to the listener */
533         if(!XSendEvent(xsre->display, xsre->requestor, True, 0, (XEvent *) &xev))
534                 fprintf(stderr, "Error sending SelectionNotify event\n");
535 }
536
537 void
538 xsetsel(char *str) {
539         /* register the selection for both the clipboard and the primary */
540         Atom clipboard;
541
542         free(sel.clip);
543         sel.clip = str;
544
545         XSetSelectionOwner(xw.dpy, XA_PRIMARY, xw.win, CurrentTime);
546
547         clipboard = XInternAtom(xw.dpy, "CLIPBOARD", 0);
548         XSetSelectionOwner(xw.dpy, clipboard, xw.win, CurrentTime);
549
550         XFlush(xw.dpy);
551 }
552
553 void
554 brelease(XEvent *e) {
555         int b;
556
557         sel.mode = 0;
558         getbuttoninfo(e, &b, &sel.ex, &sel.ey);
559         mousereport(e);
560         if(sel.bx == sel.ex && sel.by == sel.ey) {
561                 sel.bx = -1;
562                 if(b == 2)
563                         selpaste();
564                 else if(b == 1) {
565                         struct timeval now;
566                         gettimeofday(&now, NULL);
567
568                         if(TIMEDIFF(now, sel.tclick2) <= TRIPLECLICK_TIMEOUT) {
569                                 /* triple click on the line */
570                                 sel.b.x = sel.bx = 0;
571                                 sel.e.x = sel.ex = term.col;
572                                 sel.b.y = sel.e.y = sel.ey;
573                                 selcopy();
574                         } else if(TIMEDIFF(now, sel.tclick1) <= DOUBLECLICK_TIMEOUT) {
575                                 /* double click to select word */
576                                 sel.bx = sel.ex;
577                                 while(term.line[sel.ey][sel.bx-1].state & GLYPH_SET &&
578                                           term.line[sel.ey][sel.bx-1].c[0] != ' ') sel.bx--;
579                                 sel.b.x = sel.bx;
580                                 while(term.line[sel.ey][sel.ex+1].state & GLYPH_SET &&
581                                           term.line[sel.ey][sel.ex+1].c[0] != ' ') sel.ex++;
582                                 sel.e.x = sel.ex;
583                                 sel.b.y = sel.e.y = sel.ey;
584                                 selcopy();
585                         }
586                 }
587         } else if(b == 1)
588                 selcopy();
589         memcpy(&sel.tclick2, &sel.tclick1, sizeof(struct timeval));
590         gettimeofday(&sel.tclick1, NULL);
591         draw();
592 }
593
594 void
595 bmotion(XEvent *e) {
596         if(sel.mode) {
597                 int oldey = sel.ey,
598                         oldex = sel.ex;
599                 getbuttoninfo(e, NULL, &sel.ex, &sel.ey);
600
601                 if(oldey != sel.ey || oldex != sel.ex) {
602                         int starty = MIN(oldey, sel.ey);
603                         int endy = MAX(oldey, sel.ey);
604                         drawregion(0, (starty > 0 ? starty : 0), term.col, (sel.ey < term.row ? endy+1 : term.row));
605                 }
606         }
607 }
608
609 void
610 die(const char *errstr, ...) {
611         va_list ap;
612
613         va_start(ap, errstr);
614         vfprintf(stderr, errstr, ap);
615         va_end(ap);
616         exit(EXIT_FAILURE);
617 }
618
619 void
620 execsh(void) {
621         char **args;
622         char *envshell = getenv("SHELL");
623
624         DEFAULT(envshell, "sh");
625         putenv("TERM="TNAME);
626         args = opt_cmd ? opt_cmd : (char*[]){envshell, "-i", NULL};
627         execvp(args[0], args);
628         exit(EXIT_FAILURE);
629 }
630
631 void
632 sigchld(int a) {
633         int stat = 0;
634         if(waitpid(pid, &stat, 0) < 0)
635                 die("Waiting for pid %hd failed: %s\n", pid, SERRNO);
636         if(WIFEXITED(stat))
637                 exit(WEXITSTATUS(stat));
638         else
639                 exit(EXIT_FAILURE);
640 }
641
642 void
643 ttynew(void) {
644         int m, s;
645         
646         /* seems to work fine on linux, openbsd and freebsd */
647         struct winsize w = {term.row, term.col, 0, 0};
648         if(openpty(&m, &s, NULL, NULL, &w) < 0)
649                 die("openpty failed: %s\n", SERRNO);
650
651         switch(pid = fork()) {
652         case -1:
653                 die("fork failed\n");
654                 break;
655         case 0:
656                 setsid(); /* create a new process group */
657                 dup2(s, STDIN_FILENO);
658                 dup2(s, STDOUT_FILENO);
659                 dup2(s, STDERR_FILENO);
660                 if(ioctl(s, TIOCSCTTY, NULL) < 0)
661                         die("ioctl TIOCSCTTY failed: %s\n", SERRNO);
662                 close(s);
663                 close(m);
664                 execsh();
665                 break;
666         default:
667                 close(s);
668                 cmdfd = m;
669                 signal(SIGCHLD, sigchld);
670         }
671 }
672
673 void
674 dump(char c) {
675         static int col;
676         fprintf(stderr, " %02x '%c' ", c, isprint(c)?c:'.');
677         if(++col % 10 == 0)
678                 fprintf(stderr, "\n");
679 }
680
681 void
682 ttyread(void) {
683         static char buf[BUFSIZ];
684         static int buflen = 0;
685         char *ptr;
686         char s[UTF_SIZ];
687         int charsize; /* size of utf8 char in bytes */
688         long utf8c;
689         int ret;
690
691         /* append read bytes to unprocessed bytes */
692         if((ret = read(cmdfd, buf+buflen, LEN(buf)-buflen)) < 0)
693                 die("Couldn't read from shell: %s\n", SERRNO);
694
695         /* process every complete utf8 char */
696         buflen += ret;
697         ptr = buf;
698         while(buflen >= UTF_SIZ || isfullutf8(ptr,buflen)) {
699                 charsize = utf8decode(ptr, &utf8c);
700                 utf8encode(&utf8c, s);
701                 tputc(s);
702                 ptr    += charsize;
703                 buflen -= charsize;
704         }
705
706         /* keep any uncomplete utf8 char for the next call */
707         memmove(buf, ptr, buflen);
708 }
709
710 void
711 ttywrite(const char *s, size_t n) {
712         if(write(cmdfd, s, n) == -1)
713                 die("write error on tty: %s\n", SERRNO);
714 }
715
716 void
717 ttyresize(int x, int y) {
718         struct winsize w;
719
720         w.ws_row = term.row;
721         w.ws_col = term.col;
722         w.ws_xpixel = w.ws_ypixel = 0;
723         if(ioctl(cmdfd, TIOCSWINSZ, &w) < 0)
724                 fprintf(stderr, "Couldn't set window size: %s\n", SERRNO);
725 }
726
727 void
728 tcursor(int mode) {
729         static TCursor c;
730
731         if(mode == CURSOR_SAVE)
732                 c = term.c;
733         else if(mode == CURSOR_LOAD)
734                 term.c = c, tmoveto(c.x, c.y);
735 }
736
737 void
738 treset(void) {
739         term.c = (TCursor){{
740                 .mode = ATTR_NULL,
741                 .fg = DefaultFG,
742                 .bg = DefaultBG
743         }, .x = 0, .y = 0, .state = CURSOR_DEFAULT};
744         
745         term.top = 0, term.bot = term.row - 1;
746         term.mode = MODE_WRAP;
747         tclearregion(0, 0, term.col-1, term.row-1);
748 }
749
750 void
751 tnew(int col, int row) {
752         /* set screen size */
753         term.row = row, term.col = col;
754         term.line = malloc(term.row * sizeof(Line));
755         term.alt  = malloc(term.row * sizeof(Line));
756         for(row = 0 ; row < term.row; row++) {
757                 term.line[row] = malloc(term.col * sizeof(Glyph));
758                 term.alt [row] = malloc(term.col * sizeof(Glyph));
759         }
760         /* setup screen */
761         treset();
762 }
763
764 void
765 tswapscreen(void) {
766         Line* tmp = term.line;
767         term.line = term.alt;
768         term.alt = tmp;
769         term.mode ^= MODE_ALTSCREEN;
770 }
771
772 void
773 tscrolldown(int orig, int n) {
774         int i;
775         Line temp;
776         
777         LIMIT(n, 0, term.bot-orig+1);
778
779         tclearregion(0, term.bot-n+1, term.col-1, term.bot);
780         
781         for(i = term.bot; i >= orig+n; i--) {
782                 temp = term.line[i];
783                 term.line[i] = term.line[i-n];
784                 term.line[i-n] = temp;
785         }
786 }
787
788 void
789 tscrollup(int orig, int n) {
790         int i;
791         Line temp;
792         LIMIT(n, 0, term.bot-orig+1);
793         
794         tclearregion(0, orig, term.col-1, orig+n-1);
795         
796         for(i = orig; i <= term.bot-n; i++) {
797                  temp = term.line[i];
798                  term.line[i] = term.line[i+n];
799                  term.line[i+n] = temp;
800         }
801 }
802
803 void
804 tnewline(int first_col) {
805         int y = term.c.y;
806         if(y == term.bot)
807                 tscrollup(term.top, 1);
808         else
809                 y++;
810         tmoveto(first_col ? 0 : term.c.x, y);
811 }
812
813 void
814 csiparse(void) {
815         /* int noarg = 1; */
816         char *p = escseq.buf;
817
818         escseq.narg = 0;
819         if(*p == '?')
820                 escseq.priv = 1, p++;
821         
822         while(p < escseq.buf+escseq.len) {
823                 while(isdigit(*p)) {
824                         escseq.arg[escseq.narg] *= 10;
825                         escseq.arg[escseq.narg] += *p++ - '0'/*, noarg = 0 */;
826                 }
827                 if(*p == ';' && escseq.narg+1 < ESC_ARG_SIZ)
828                         escseq.narg++, p++;
829                 else {
830                         escseq.mode = *p;
831                         escseq.narg++;
832                         return;
833                 }
834         }
835 }
836
837 void
838 tmoveto(int x, int y) {
839         LIMIT(x, 0, term.col-1);
840         LIMIT(y, 0, term.row-1);
841         term.c.state &= ~CURSOR_WRAPNEXT;
842         term.c.x = x;
843         term.c.y = y;
844 }
845
846 void
847 tsetchar(char *c) {
848         term.line[term.c.y][term.c.x] = term.c.attr;
849         memcpy(term.line[term.c.y][term.c.x].c, c, UTF_SIZ);
850         term.line[term.c.y][term.c.x].state |= GLYPH_SET;
851 }
852
853 void
854 tclearregion(int x1, int y1, int x2, int y2) {
855         int x, y, temp;
856
857         if(x1 > x2)
858                 temp = x1, x1 = x2, x2 = temp;
859         if(y1 > y2)
860                 temp = y1, y1 = y2, y2 = temp;
861
862         LIMIT(x1, 0, term.col-1);
863         LIMIT(x2, 0, term.col-1);
864         LIMIT(y1, 0, term.row-1);
865         LIMIT(y2, 0, term.row-1);
866
867         for(y = y1; y <= y2; y++)
868                 for(x = x1; x <= x2; x++)
869                         term.line[y][x].state = 0;
870 }
871
872 void
873 tdeletechar(int n) {
874         int src = term.c.x + n;
875         int dst = term.c.x;
876         int size = term.col - src;
877
878         if(src >= term.col) {
879                 tclearregion(term.c.x, term.c.y, term.col-1, term.c.y);
880                 return;
881         }
882         memmove(&term.line[term.c.y][dst], &term.line[term.c.y][src], size * sizeof(Glyph));
883         tclearregion(term.col-n, term.c.y, term.col-1, term.c.y);
884 }
885
886 void
887 tinsertblank(int n) {
888         int src = term.c.x;
889         int dst = src + n;
890         int size = term.col - dst;
891
892         if(dst >= term.col) {
893                 tclearregion(term.c.x, term.c.y, term.col-1, term.c.y);
894                 return;
895         }
896         memmove(&term.line[term.c.y][dst], &term.line[term.c.y][src], size * sizeof(Glyph));
897         tclearregion(src, term.c.y, dst - 1, term.c.y);
898 }
899
900 void
901 tinsertblankline(int n) {
902         if(term.c.y < term.top || term.c.y > term.bot)
903                 return;
904
905         tscrolldown(term.c.y, n);
906 }
907
908 void
909 tdeleteline(int n) {
910         if(term.c.y < term.top || term.c.y > term.bot)
911                 return;
912
913         tscrollup(term.c.y, n);
914 }
915
916 void
917 tsetattr(int *attr, int l) {
918         int i;
919
920         for(i = 0; i < l; i++) {
921                 switch(attr[i]) {
922                 case 0:
923                         term.c.attr.mode &= ~(ATTR_REVERSE | ATTR_UNDERLINE | ATTR_BOLD);
924                         term.c.attr.fg = DefaultFG;
925                         term.c.attr.bg = DefaultBG;
926                         break;
927                 case 1:
928                         term.c.attr.mode |= ATTR_BOLD;
929                         break;
930                 case 4:
931                         term.c.attr.mode |= ATTR_UNDERLINE;
932                         break;
933                 case 7:
934                         term.c.attr.mode |= ATTR_REVERSE;
935                         break;
936                 case 22:
937                         term.c.attr.mode &= ~ATTR_BOLD;
938                         break;
939                 case 24:
940                         term.c.attr.mode &= ~ATTR_UNDERLINE;
941                         break;
942                 case 27:
943                         term.c.attr.mode &= ~ATTR_REVERSE;
944                         break;
945                 case 38:
946                         if(i + 2 < l && attr[i + 1] == 5) {
947                                 i += 2;
948                                 if(BETWEEN(attr[i], 0, 255))
949                                         term.c.attr.fg = attr[i];
950                                 else
951                                         fprintf(stderr, "erresc: bad fgcolor %d\n", attr[i]);
952                         }
953                         else
954                                 fprintf(stderr, "erresc: gfx attr %d unknown\n", attr[i]);
955                         break;
956                 case 39:
957                         term.c.attr.fg = DefaultFG;
958                         break;
959                 case 48:
960                         if(i + 2 < l && attr[i + 1] == 5) {
961                                 i += 2;
962                                 if(BETWEEN(attr[i], 0, 255))
963                                         term.c.attr.bg = attr[i];
964                                 else
965                                         fprintf(stderr, "erresc: bad bgcolor %d\n", attr[i]);
966                         }
967                         else
968                                 fprintf(stderr, "erresc: gfx attr %d unknown\n", attr[i]);
969                         break;
970                 case 49:
971                         term.c.attr.bg = DefaultBG;
972                         break;
973                 default:
974                         if(BETWEEN(attr[i], 30, 37))
975                                 term.c.attr.fg = attr[i] - 30;
976                         else if(BETWEEN(attr[i], 40, 47))
977                                 term.c.attr.bg = attr[i] - 40;
978                         else if(BETWEEN(attr[i], 90, 97))
979                                 term.c.attr.fg = attr[i] - 90 + 8;
980                         else if(BETWEEN(attr[i], 100, 107))
981                                 term.c.attr.fg = attr[i] - 100 + 8;
982                         else
983                                 fprintf(stderr, "erresc: gfx attr %d unknown\n", attr[i]), csidump();
984                         
985                         break;
986                 }
987         }
988 }
989
990 void
991 tsetscroll(int t, int b) {
992         int temp;
993
994         LIMIT(t, 0, term.row-1);
995         LIMIT(b, 0, term.row-1);
996         if(t > b) {
997                 temp = t;
998                 t = b;
999                 b = temp;
1000         }
1001         term.top = t;
1002         term.bot = b;
1003 }
1004
1005 void
1006 csihandle(void) {
1007         switch(escseq.mode) {
1008         default:
1009         unknown:
1010                 fprintf(stderr, "erresc: unknown csi ");
1011                 csidump();
1012                 /* die(""); */
1013                 break;
1014         case '@': /* ICH -- Insert <n> blank char */
1015                 DEFAULT(escseq.arg[0], 1);
1016                 tinsertblank(escseq.arg[0]);
1017                 break;
1018         case 'A': /* CUU -- Cursor <n> Up */
1019         case 'e':
1020                 DEFAULT(escseq.arg[0], 1);
1021                 tmoveto(term.c.x, term.c.y-escseq.arg[0]);
1022                 break;
1023         case 'B': /* CUD -- Cursor <n> Down */
1024                 DEFAULT(escseq.arg[0], 1);
1025                 tmoveto(term.c.x, term.c.y+escseq.arg[0]);
1026                 break;
1027         case 'C': /* CUF -- Cursor <n> Forward */
1028         case 'a':
1029                 DEFAULT(escseq.arg[0], 1);
1030                 tmoveto(term.c.x+escseq.arg[0], term.c.y);
1031                 break;
1032         case 'D': /* CUB -- Cursor <n> Backward */
1033                 DEFAULT(escseq.arg[0], 1);
1034                 tmoveto(term.c.x-escseq.arg[0], term.c.y);
1035                 break;
1036         case 'E': /* CNL -- Cursor <n> Down and first col */
1037                 DEFAULT(escseq.arg[0], 1);
1038                 tmoveto(0, term.c.y+escseq.arg[0]);
1039                 break;
1040         case 'F': /* CPL -- Cursor <n> Up and first col */
1041                 DEFAULT(escseq.arg[0], 1);
1042                 tmoveto(0, term.c.y-escseq.arg[0]);
1043                 break;
1044         case 'G': /* CHA -- Move to <col> */
1045         case '`': /* XXX: HPA -- same? */
1046                 DEFAULT(escseq.arg[0], 1);
1047                 tmoveto(escseq.arg[0]-1, term.c.y);
1048                 break;
1049         case 'H': /* CUP -- Move to <row> <col> */
1050         case 'f': /* XXX: HVP -- same? */
1051                 DEFAULT(escseq.arg[0], 1);
1052                 DEFAULT(escseq.arg[1], 1);
1053                 tmoveto(escseq.arg[1]-1, escseq.arg[0]-1);
1054                 break;
1055         /* XXX: (CSI n I) CHT -- Cursor Forward Tabulation <n> tab stops */
1056         case 'J': /* ED -- Clear screen */
1057                 switch(escseq.arg[0]) {
1058                 case 0: /* below */
1059                         tclearregion(term.c.x, term.c.y, term.col-1, term.c.y);
1060                         if(term.c.y < term.row-1)
1061                                 tclearregion(0, term.c.y+1, term.col-1, term.row-1);
1062                         break;
1063                 case 1: /* above */
1064                         if(term.c.y > 1)
1065                                 tclearregion(0, 0, term.col-1, term.c.y-1);
1066                         tclearregion(0, term.c.y, term.c.x, term.c.y);
1067                         break;
1068                 case 2: /* all */
1069                         tclearregion(0, 0, term.col-1, term.row-1);
1070                         break;
1071                 default:
1072                         goto unknown;
1073                 }
1074                 break;
1075         case 'K': /* EL -- Clear line */
1076                 switch(escseq.arg[0]) {
1077                 case 0: /* right */
1078                         tclearregion(term.c.x, term.c.y, term.col-1, term.c.y);
1079                         break;
1080                 case 1: /* left */
1081                         tclearregion(0, term.c.y, term.c.x, term.c.y);
1082                         break;
1083                 case 2: /* all */
1084                         tclearregion(0, term.c.y, term.col-1, term.c.y);
1085                         break;
1086                 }
1087                 break;
1088         case 'S': /* SU -- Scroll <n> line up */
1089                 DEFAULT(escseq.arg[0], 1);
1090                 tscrollup(term.top, escseq.arg[0]);
1091                 break;
1092         case 'T': /* SD -- Scroll <n> line down */
1093                 DEFAULT(escseq.arg[0], 1);
1094                 tscrolldown(term.top, escseq.arg[0]);
1095                 break;
1096         case 'L': /* IL -- Insert <n> blank lines */
1097                 DEFAULT(escseq.arg[0], 1);
1098                 tinsertblankline(escseq.arg[0]);
1099                 break;
1100         case 'l': /* RM -- Reset Mode */
1101                 if(escseq.priv) {
1102                         switch(escseq.arg[0]) {
1103                         case 1:
1104                                 term.mode &= ~MODE_APPKEYPAD;
1105                                 break;
1106                         case 5: /* TODO: DECSCNM -- Remove reverse video */
1107                                 break;
1108                         case 7:
1109                                 term.mode &= ~MODE_WRAP;
1110                                 break;
1111                         case 12: /* att610 -- Stop blinking cursor (IGNORED) */
1112                                 break;
1113                         case 20:
1114                                 term.mode &= ~MODE_CRLF;
1115                                 break;
1116                         case 25:
1117                                 term.c.state |= CURSOR_HIDE;
1118                                 break;
1119                         case 1000: /* disable X11 xterm mouse reporting */
1120                                 term.mode &= ~MODE_MOUSE;
1121                                 break;
1122                         case 1049: /* = 1047 and 1048 */
1123                         case 1047:
1124                                 if(IS_SET(MODE_ALTSCREEN)) {
1125                                         tclearregion(0, 0, term.col-1, term.row-1);
1126                                         tswapscreen();
1127                                 }
1128                                 if(escseq.arg[0] == 1047)
1129                                         break;
1130                         case 1048:
1131                                 tcursor(CURSOR_LOAD);
1132                                 break;
1133                         default:
1134                                 goto unknown;
1135                         }
1136                 } else {
1137                         switch(escseq.arg[0]) {
1138                         case 4:
1139                                 term.mode &= ~MODE_INSERT;
1140                                 break;
1141                         default:
1142                                 goto unknown;
1143                         }
1144                 }
1145                 break;
1146         case 'M': /* DL -- Delete <n> lines */
1147                 DEFAULT(escseq.arg[0], 1);
1148                 tdeleteline(escseq.arg[0]);
1149                 break;
1150         case 'X': /* ECH -- Erase <n> char */
1151                 DEFAULT(escseq.arg[0], 1);
1152                 tclearregion(term.c.x, term.c.y, term.c.x + escseq.arg[0], term.c.y);
1153                 break;
1154         case 'P': /* DCH -- Delete <n> char */
1155                 DEFAULT(escseq.arg[0], 1);
1156                 tdeletechar(escseq.arg[0]);
1157                 break;
1158         /* XXX: (CSI n Z) CBT -- Cursor Backward Tabulation <n> tab stops */
1159         case 'd': /* VPA -- Move to <row> */
1160                 DEFAULT(escseq.arg[0], 1);
1161                 tmoveto(term.c.x, escseq.arg[0]-1);
1162                 break;
1163         case 'h': /* SM -- Set terminal mode */
1164                 if(escseq.priv) {
1165                         switch(escseq.arg[0]) {
1166                         case 1:
1167                                 term.mode |= MODE_APPKEYPAD;
1168                                 break;
1169                         case 5: /* DECSCNM -- Reverve video */
1170                                 /* TODO: set REVERSE on the whole screen (f) */
1171                                 break;
1172                         case 7:
1173                                 term.mode |= MODE_WRAP;
1174                                 break;
1175                         case 20:
1176                                 term.mode |= MODE_CRLF;
1177                                 break;
1178                         case 12: /* att610 -- Start blinking cursor (IGNORED) */
1179                                  /* fallthrough for xterm cvvis = CSI [ ? 12 ; 25 h */
1180                                 if(escseq.narg > 1 && escseq.arg[1] != 25)
1181                                         break;
1182                         case 25:
1183                                 term.c.state &= ~CURSOR_HIDE;
1184                                 break;
1185                         case 1000: /* enable X11 xterm mouse reporting */
1186                                 term.mode |= MODE_MOUSE;
1187                                 break;
1188                         case 1049: /* = 1047 and 1048 */
1189                         case 1047:
1190                                 if(IS_SET(MODE_ALTSCREEN))
1191                                         tclearregion(0, 0, term.col-1, term.row-1);
1192                                 else
1193                                         tswapscreen();
1194                                 if(escseq.arg[0] == 1047)
1195                                         break;
1196                         case 1048:
1197                                 tcursor(CURSOR_SAVE);
1198                                 break;
1199                         default: goto unknown;
1200                         }
1201                 } else {
1202                         switch(escseq.arg[0]) {
1203                         case 4:
1204                                 term.mode |= MODE_INSERT;
1205                                 break;
1206                         default: goto unknown;
1207                         }
1208                 };
1209                 break;
1210         case 'm': /* SGR -- Terminal attribute (color) */
1211                 tsetattr(escseq.arg, escseq.narg);
1212                 break;
1213         case 'r': /* DECSTBM -- Set Scrolling Region */
1214                 if(escseq.priv)
1215                         goto unknown;
1216                 else {
1217                         DEFAULT(escseq.arg[0], 1);
1218                         DEFAULT(escseq.arg[1], term.row);
1219                         tsetscroll(escseq.arg[0]-1, escseq.arg[1]-1);
1220                         tmoveto(0, 0);
1221                 }
1222                 break;
1223         case 's': /* DECSC -- Save cursor position (ANSI.SYS) */
1224                 tcursor(CURSOR_SAVE);
1225                 break;
1226         case 'u': /* DECRC -- Restore cursor position (ANSI.SYS) */
1227                 tcursor(CURSOR_LOAD);
1228                 break;
1229         }
1230 }
1231
1232 void
1233 csidump(void) {
1234         int i;
1235         printf("ESC [ %s", escseq.priv ? "? " : "");
1236         if(escseq.narg)
1237                 for(i = 0; i < escseq.narg; i++)
1238                         printf("%d ", escseq.arg[i]);
1239         if(escseq.mode)
1240                 putchar(escseq.mode);
1241         putchar('\n');
1242 }
1243
1244 void
1245 csireset(void) {
1246         memset(&escseq, 0, sizeof(escseq));
1247 }
1248
1249 void
1250 tputtab(void) {
1251         int space = TAB - term.c.x % TAB;
1252         tmoveto(term.c.x + space, term.c.y);
1253 }
1254
1255 void
1256 tputc(char *c) {
1257         char ascii = *c;
1258         if(term.esc & ESC_START) {
1259                 if(term.esc & ESC_CSI) {
1260                         escseq.buf[escseq.len++] = ascii;
1261                         if(BETWEEN(ascii, 0x40, 0x7E) || escseq.len >= ESC_BUF_SIZ) {
1262                                 term.esc = 0;
1263                                 csiparse(), csihandle();
1264                         }
1265                         /* TODO: handle other OSC */
1266                 } else if(term.esc & ESC_OSC) {
1267                         if(ascii == ';') {
1268                                 term.titlelen = 0;
1269                                 term.esc = ESC_START | ESC_TITLE;
1270                         }
1271                 } else if(term.esc & ESC_TITLE) {
1272                         if(ascii == '\a' || term.titlelen+1 >= ESC_TITLE_SIZ) {
1273                                 term.esc = 0;
1274                                 term.title[term.titlelen] = '\0';
1275                                 XStoreName(xw.dpy, xw.win, term.title);
1276                         } else {
1277                                 term.title[term.titlelen++] = ascii;
1278                         }
1279                 } else if(term.esc & ESC_ALTCHARSET) {
1280                         switch(ascii) {
1281                         case '0': /* Line drawing crap */
1282                                 term.c.attr.mode |= ATTR_GFX;
1283                                 break;
1284                         case 'B': /* Back to regular text */
1285                                 term.c.attr.mode &= ~ATTR_GFX;
1286                                 break;
1287                         default:
1288                                 fprintf(stderr, "esc unhandled charset: ESC ( %c\n", ascii);
1289                         }
1290                         term.esc = 0;
1291                 } else {
1292                         switch(ascii) {
1293                         case '[':
1294                                 term.esc |= ESC_CSI;
1295                                 break;
1296                         case ']':
1297                                 term.esc |= ESC_OSC;
1298                                 break;
1299                         case '(':
1300                                 term.esc |= ESC_ALTCHARSET;
1301                                 break;
1302                         case 'D': /* IND -- Linefeed */
1303                                 if(term.c.y == term.bot)
1304                                         tscrollup(term.top, 1);
1305                                 else
1306                                         tmoveto(term.c.x, term.c.y+1);
1307                                 term.esc = 0;
1308                                 break;
1309                         case 'E': /* NEL -- Next line */
1310                                 tnewline(1); /* always go to first col */
1311                                 term.esc = 0;
1312                                 break;
1313                         case 'M': /* RI -- Reverse index */
1314                                 if(term.c.y == term.top)
1315                                         tscrolldown(term.top, 1);
1316                                 else
1317                                         tmoveto(term.c.x, term.c.y-1);
1318                                 term.esc = 0;
1319                                 break;
1320                         case 'c': /* RIS -- Reset to inital state */
1321                                 treset();
1322                                 term.esc = 0;
1323                                 break;
1324                         case '=': /* DECPAM -- Application keypad */
1325                                 term.mode |= MODE_APPKEYPAD;
1326                                 term.esc = 0;
1327                                 break;
1328                         case '>': /* DECPNM -- Normal keypad */
1329                                 term.mode &= ~MODE_APPKEYPAD;
1330                                 term.esc = 0;
1331                                 break;
1332                         case '7': /* DECSC -- Save Cursor */
1333                                 tcursor(CURSOR_SAVE);
1334                                 term.esc = 0;
1335                                 break;
1336                         case '8': /* DECRC -- Restore Cursor */
1337                                 tcursor(CURSOR_LOAD);
1338                                 term.esc = 0;
1339                                 break;
1340                         default:
1341                                 fprintf(stderr, "erresc: unknown sequence ESC 0x%02X '%c'\n",
1342                                     (unsigned char) ascii, isprint(ascii)?ascii:'.');
1343                                 term.esc = 0;
1344                         }
1345                 }
1346         } else {
1347                 switch(ascii) {
1348                 case '\t':
1349                         tputtab();
1350                         break;
1351                 case '\b':
1352                         tmoveto(term.c.x-1, term.c.y);
1353                         break;
1354                 case '\r':
1355                         tmoveto(0, term.c.y);
1356                         break;
1357                 case '\f':
1358                 case '\v':
1359                 case '\n':
1360                         /* go to first col if the mode is set */
1361                         tnewline(IS_SET(MODE_CRLF));
1362                         break;
1363                 case '\a':
1364                         if(!(xw.state & WIN_FOCUSED))
1365                                 xseturgency(1);
1366                         break;
1367                 case '\033':
1368                         csireset();
1369                         term.esc = ESC_START;
1370                         break;
1371                 default:
1372                         if(IS_SET(MODE_WRAP) && term.c.state & CURSOR_WRAPNEXT)
1373                                 tnewline(1); /* always go to first col */
1374                         tsetchar(c);
1375                         if(term.c.x+1 < term.col)
1376                                 tmoveto(term.c.x+1, term.c.y);
1377                         else
1378                                 term.c.state |= CURSOR_WRAPNEXT;
1379                 }
1380         }
1381 }
1382
1383 int
1384 tresize(int col, int row) {
1385         int i, x;
1386         int minrow = MIN(row, term.row);
1387         int mincol = MIN(col, term.col);
1388         int slide = term.c.y - row + 1;
1389
1390         if(col < 1 || row < 1)
1391                 return 0;
1392
1393         /* free unneeded rows */
1394         i = 0;
1395         if(slide > 0) {
1396                 /* slide screen to keep cursor where we expect it -
1397                  * tscrollup would work here, but we can optimize to
1398                  * memmove because we're freeing the earlier lines */
1399                 for(/* i = 0 */; i < slide; i++) {
1400                         free(term.line[i]);
1401                         free(term.alt[i]);
1402                 }
1403                 memmove(term.line, term.line + slide, row * sizeof(Line));
1404                 memmove(term.alt, term.alt + slide, row * sizeof(Line));
1405         }
1406         for(i += row; i < term.row; i++) {
1407                 free(term.line[i]);
1408                 free(term.alt[i]);
1409         }
1410
1411         /* resize to new height */
1412         term.line = realloc(term.line, row * sizeof(Line));
1413         term.alt  = realloc(term.alt,  row * sizeof(Line));
1414
1415         /* resize each row to new width, zero-pad if needed */
1416         for(i = 0; i < minrow; i++) {
1417                 term.line[i] = realloc(term.line[i], col * sizeof(Glyph));
1418                 term.alt[i]  = realloc(term.alt[i],  col * sizeof(Glyph));
1419                 for(x = mincol; x < col; x++) {
1420                         term.line[i][x].state = 0;
1421                         term.alt[i][x].state = 0;
1422                 }
1423         }
1424
1425         /* allocate any new rows */
1426         for(/* i == minrow */; i < row; i++) {
1427                 term.line[i] = calloc(col, sizeof(Glyph));
1428                 term.alt [i] = calloc(col, sizeof(Glyph));
1429         }
1430         
1431         /* update terminal size */
1432         term.col = col, term.row = row;
1433         /* make use of the LIMIT in tmoveto */
1434         tmoveto(term.c.x, term.c.y);
1435         /* reset scrolling region */
1436         tsetscroll(0, row-1);
1437         return (slide > 0);
1438 }
1439
1440 void
1441 xresize(int col, int row) {
1442         Pixmap newbuf;
1443         int oldw, oldh;
1444
1445         oldw = xw.bufw;
1446         oldh = xw.bufh;
1447         xw.bufw = MAX(1, col * xw.cw);
1448         xw.bufh = MAX(1, row * xw.ch);
1449         newbuf = XCreatePixmap(xw.dpy, xw.win, xw.bufw, xw.bufh, XDefaultDepth(xw.dpy, xw.scr));
1450         XCopyArea(xw.dpy, xw.buf, newbuf, dc.gc, 0, 0, xw.bufw, xw.bufh, 0, 0);
1451         XFreePixmap(xw.dpy, xw.buf);
1452         XSetForeground(xw.dpy, dc.gc, dc.col[DefaultBG]);
1453         if(xw.bufw > oldw)
1454                 XFillRectangle(xw.dpy, newbuf, dc.gc, oldw, 0,
1455                                 xw.bufw-oldw, MIN(xw.bufh, oldh));
1456         else if(xw.bufw < oldw && (BORDER > 0 || xw.w > xw.bufw))
1457                 XClearArea(xw.dpy, xw.win, BORDER+xw.bufw, BORDER,
1458                                 xw.w-xw.bufh-BORDER, BORDER+MIN(xw.bufh, oldh),
1459                                 False);
1460         if(xw.bufh > oldh)
1461                 XFillRectangle(xw.dpy, newbuf, dc.gc, 0, oldh,
1462                                 xw.bufw, xw.bufh-oldh);
1463         else if(xw.bufh < oldh && (BORDER > 0 || xw.h > xw.bufh))
1464                 XClearArea(xw.dpy, xw.win, BORDER, BORDER+xw.bufh,
1465                                 xw.w-2*BORDER, xw.h-xw.bufh-BORDER,
1466                                 False);
1467         xw.buf = newbuf;
1468 }
1469
1470 void
1471 xloadcols(void) {
1472         int i, r, g, b;
1473         XColor color;
1474         unsigned long white = WhitePixel(xw.dpy, xw.scr);
1475
1476         for(i = 0; i < 16; i++) {
1477                 if(!XAllocNamedColor(xw.dpy, xw.cmap, colorname[i], &color, &color)) {
1478                         dc.col[i] = white;
1479                         fprintf(stderr, "Could not allocate color '%s'\n", colorname[i]);
1480                 } else
1481                         dc.col[i] = color.pixel;
1482         }
1483
1484         /* same colors as xterm */
1485         for(r = 0; r < 6; r++)
1486                 for(g = 0; g < 6; g++)
1487                         for(b = 0; b < 6; b++) {
1488                                 color.red = r == 0 ? 0 : 0x3737 + 0x2828 * r;
1489                                 color.green = g == 0 ? 0 : 0x3737 + 0x2828 * g;
1490                                 color.blue = b == 0 ? 0 : 0x3737 + 0x2828 * b;
1491                                 if(!XAllocColor(xw.dpy, xw.cmap, &color)) {
1492                                         dc.col[i] = white;
1493                                         fprintf(stderr, "Could not allocate color %d\n", i);
1494                                 } else
1495                                         dc.col[i] = color.pixel;
1496                                 i++;
1497                         }
1498
1499         for(r = 0; r < 24; r++, i++) {
1500                 color.red = color.green = color.blue = 0x0808 + 0x0a0a * r;
1501                 if (!XAllocColor(xw.dpy, xw.cmap, &color)) {
1502                         dc.col[i] = white;
1503                         fprintf(stderr, "Could not allocate color %d\n", i);
1504                 } else
1505                         dc.col[i] = color.pixel;
1506         }
1507 }
1508
1509 void
1510 xclear(int x1, int y1, int x2, int y2) {
1511         XSetForeground(xw.dpy, dc.gc, dc.col[DefaultBG]);
1512         XFillRectangle(xw.dpy, xw.buf, dc.gc,
1513                        x1 * xw.cw, y1 * xw.ch,
1514                        (x2-x1+1) * xw.cw, (y2-y1+1) * xw.ch);
1515 }
1516
1517 void
1518 xhints(void) {
1519         XClassHint class = {opt_class ? opt_class : TNAME, TNAME};
1520         XWMHints wm = {.flags = InputHint, .input = 1};
1521         XSizeHints size = {
1522                 .flags = PSize | PResizeInc | PBaseSize,
1523                 .height = xw.h,
1524                 .width = xw.w,
1525                 .height_inc = xw.ch,
1526                 .width_inc = xw.cw,
1527                 .base_height = 2*BORDER,
1528                 .base_width = 2*BORDER,
1529         };
1530         XSetWMProperties(xw.dpy, xw.win, NULL, NULL, NULL, 0, &size, &wm, &class);
1531 }
1532
1533 XFontSet
1534 xinitfont(char *fontstr) {
1535         XFontSet set;
1536         char *def, **missing;
1537         int n;
1538
1539         missing = NULL;
1540         set = XCreateFontSet(xw.dpy, fontstr, &missing, &n, &def);
1541         if(missing) {
1542                 while(n--)
1543                         fprintf(stderr, "st: missing fontset: %s\n", missing[n]);
1544                 XFreeStringList(missing);
1545         }
1546         return set;
1547 }
1548
1549 void
1550 xgetfontinfo(XFontSet set, int *ascent, int *descent, short *lbearing, short *rbearing) {
1551         XFontStruct **xfonts;
1552         char **font_names;
1553         int i, n;
1554
1555         *ascent = *descent = *lbearing = *rbearing = 0;
1556         n = XFontsOfFontSet(set, &xfonts, &font_names);
1557         for(i = 0; i < n; i++) {
1558                 *ascent = MAX(*ascent, (*xfonts)->ascent);
1559                 *descent = MAX(*descent, (*xfonts)->descent);
1560                 *lbearing = MAX(*lbearing, (*xfonts)->min_bounds.lbearing);
1561                 *rbearing = MAX(*rbearing, (*xfonts)->max_bounds.rbearing);
1562                 xfonts++;
1563         }
1564 }
1565
1566 void
1567 initfonts(char *fontstr, char *bfontstr) {
1568         if((dc.font.set = xinitfont(fontstr)) == NULL ||
1569            (dc.bfont.set = xinitfont(bfontstr)) == NULL)
1570                 die("Can't load font %s\n", dc.font.set ? BOLDFONT : FONT);
1571         xgetfontinfo(dc.font.set, &dc.font.ascent, &dc.font.descent,
1572             &dc.font.lbearing, &dc.font.rbearing);
1573         xgetfontinfo(dc.bfont.set, &dc.bfont.ascent, &dc.bfont.descent,
1574             &dc.bfont.lbearing, &dc.bfont.rbearing);
1575 }
1576
1577 void
1578 xinit(void) {
1579         XSetWindowAttributes attrs;
1580         Cursor cursor;
1581
1582         if(!(xw.dpy = XOpenDisplay(NULL)))
1583                 die("Can't open display\n");
1584         xw.scr = XDefaultScreen(xw.dpy);
1585         
1586         /* font */
1587         initfonts(FONT, BOLDFONT);
1588
1589         /* XXX: Assuming same size for bold font */
1590         xw.cw = dc.font.rbearing - dc.font.lbearing;
1591         xw.ch = dc.font.ascent + dc.font.descent;
1592
1593         /* colors */
1594         xw.cmap = XDefaultColormap(xw.dpy, xw.scr);
1595         xloadcols();
1596
1597         /* window - default size */
1598         xw.bufh = 24 * xw.ch;
1599         xw.bufw = 80 * xw.cw;
1600         xw.h = xw.bufh + 2*BORDER;
1601         xw.w = xw.bufw + 2*BORDER;
1602
1603         attrs.background_pixel = dc.col[DefaultBG];
1604         attrs.border_pixel = dc.col[DefaultBG];
1605         attrs.bit_gravity = NorthWestGravity;
1606         attrs.event_mask = FocusChangeMask | KeyPressMask
1607                 | ExposureMask | VisibilityChangeMask | StructureNotifyMask
1608                 | PointerMotionMask | ButtonPressMask | ButtonReleaseMask;
1609         attrs.colormap = xw.cmap;
1610
1611         xw.win = XCreateWindow(xw.dpy, XRootWindow(xw.dpy, xw.scr), 0, 0,
1612                         xw.w, xw.h, 0, XDefaultDepth(xw.dpy, xw.scr), InputOutput,
1613                         XDefaultVisual(xw.dpy, xw.scr),
1614                         CWBackPixel | CWBorderPixel | CWBitGravity | CWEventMask
1615                         | CWColormap,
1616                         &attrs);
1617         xw.buf = XCreatePixmap(xw.dpy, xw.win, xw.bufw, xw.bufh, XDefaultDepth(xw.dpy, xw.scr));
1618
1619
1620         /* input methods */
1621         xw.xim = XOpenIM(xw.dpy, NULL, NULL, NULL);
1622         xw.xic = XCreateIC(xw.xim, XNInputStyle, XIMPreeditNothing
1623                                            | XIMStatusNothing, XNClientWindow, xw.win,
1624                                            XNFocusWindow, xw.win, NULL);
1625         /* gc */
1626         dc.gc = XCreateGC(xw.dpy, xw.win, 0, NULL);
1627         
1628         /* white cursor, black outline */
1629         cursor = XCreateFontCursor(xw.dpy, XC_xterm);
1630         XDefineCursor(xw.dpy, xw.win, cursor);
1631         XRecolorCursor(xw.dpy, cursor,
1632                 &(XColor){.red = 0xffff, .green = 0xffff, .blue = 0xffff},
1633                 &(XColor){.red = 0x0000, .green = 0x0000, .blue = 0x0000});
1634
1635         XMapWindow(xw.dpy, xw.win);
1636         xhints();
1637         XStoreName(xw.dpy, xw.win, opt_title ? opt_title : "st");
1638         XSync(xw.dpy, 0);
1639 }
1640
1641 void
1642 xdraws(char *s, Glyph base, int x, int y, int charlen, int bytelen) {
1643         unsigned long xfg, xbg;
1644         int winx = x*xw.cw, winy = y*xw.ch + dc.font.ascent, width = charlen*xw.cw;
1645         int i;
1646
1647         if(base.mode & ATTR_REVERSE)
1648                 xfg = dc.col[base.bg], xbg = dc.col[base.fg];
1649         else
1650                 xfg = dc.col[base.fg], xbg = dc.col[base.bg];
1651
1652         XSetBackground(xw.dpy, dc.gc, xbg);
1653         XSetForeground(xw.dpy, dc.gc, xfg);
1654
1655         if(base.mode & ATTR_GFX) {
1656                 for(i = 0; i < bytelen; i++) {
1657                         char c = gfx[(unsigned int)s[i] % 256];
1658                         if(c)
1659                                 s[i] = c;
1660                         else if(s[i] > 0x5f)
1661                                 s[i] -= 0x5f;
1662                 }
1663         }
1664
1665         XmbDrawImageString(xw.dpy, xw.buf, base.mode & ATTR_BOLD ? dc.bfont.set : dc.font.set,
1666                 dc.gc, winx, winy, s, bytelen);
1667         
1668         if(base.mode & ATTR_UNDERLINE)
1669                 XDrawLine(xw.dpy, xw.buf, dc.gc, winx, winy+1, winx+width-1, winy+1);
1670 }
1671
1672 void
1673 xdrawcursor(void) {
1674         static int oldx = 0;
1675         static int oldy = 0;
1676         int sl;
1677         Glyph g = {{' '}, ATTR_NULL, DefaultBG, DefaultCS, 0};
1678         
1679         LIMIT(oldx, 0, term.col-1);
1680         LIMIT(oldy, 0, term.row-1);
1681         
1682         if(term.line[term.c.y][term.c.x].state & GLYPH_SET)
1683                 memcpy(g.c, term.line[term.c.y][term.c.x].c, UTF_SIZ);
1684
1685         /* remove the old cursor */
1686         if(term.line[oldy][oldx].state & GLYPH_SET) {
1687                 sl = utf8size(term.line[oldy][oldx].c);
1688                 xdraws(term.line[oldy][oldx].c, term.line[oldy][oldx], oldx, oldy, 1, sl);
1689         } else
1690                 xclear(oldx, oldy, oldx, oldy);
1691         
1692         /* draw the new one */
1693         if(!(term.c.state & CURSOR_HIDE) && (xw.state & WIN_FOCUSED)) {
1694                 sl = utf8size(g.c);
1695                 xdraws(g.c, g, term.c.x, term.c.y, 1, sl);
1696                 oldx = term.c.x, oldy = term.c.y;
1697         }
1698 }
1699
1700 void
1701 draw() {
1702         drawregion(0, 0, term.col, term.row);
1703 }
1704
1705 void
1706 drawregion(int x1, int y1, int x2, int y2) {
1707         int ic, ib, x, y, ox, sl;
1708         Glyph base, new;
1709         char buf[DRAW_BUF_SIZ];
1710
1711         if(!(xw.state & WIN_VISIBLE))
1712                 return;
1713
1714         xclear(x1, y1, x2-1, y2-1);
1715         for(y = y1; y < y2; y++) {
1716                 base = term.line[y][0];
1717                 ic = ib = ox = 0;
1718                 for(x = x1; x < x2; x++) {
1719                         new = term.line[y][x];
1720                         if(sel.bx != -1 && *(new.c) && selected(x, y))
1721                                 new.mode ^= ATTR_REVERSE;
1722                         if(ib > 0 && (!(new.state & GLYPH_SET) || ATTRCMP(base, new) ||
1723                                         ib >= DRAW_BUF_SIZ-UTF_SIZ)) {
1724                                 xdraws(buf, base, ox, y, ic, ib);
1725                                 ic = ib = 0;
1726                         }
1727                         if(new.state & GLYPH_SET) {
1728                                 if(ib == 0) {
1729                                         ox = x;
1730                                         base = new;
1731                                 }
1732                                 sl = utf8size(new.c);
1733                                 memcpy(buf+ib, new.c, sl);
1734                                 ib += sl;
1735                                 ++ic;
1736                         }
1737                 }
1738                 if(ib > 0)
1739                         xdraws(buf, base, ox, y, ic, ib);
1740         }
1741         xdrawcursor();
1742         XCopyArea(xw.dpy, xw.buf, xw.win, dc.gc, 0, 0, xw.bufw, xw.bufh, BORDER, BORDER);
1743 }
1744
1745 void
1746 expose(XEvent *ev) {
1747         XExposeEvent *e = &ev->xexpose;
1748         if(xw.state & WIN_REDRAW) {
1749                 if(!e->count) {
1750                         xw.state &= ~WIN_REDRAW;
1751                         draw();
1752                 }
1753         } else
1754                 XCopyArea(xw.dpy, xw.buf, xw.win, dc.gc, e->x-BORDER, e->y-BORDER,
1755                                 e->width, e->height, e->x, e->y);
1756 }
1757
1758 void
1759 visibility(XEvent *ev) {
1760         XVisibilityEvent *e = &ev->xvisibility;
1761         if(e->state == VisibilityFullyObscured)
1762                 xw.state &= ~WIN_VISIBLE;
1763         else if(!(xw.state & WIN_VISIBLE))
1764                 /* need a full redraw for next Expose, not just a buf copy */
1765                 xw.state |= WIN_VISIBLE | WIN_REDRAW;
1766 }
1767
1768 void
1769 unmap(XEvent *ev) {
1770         xw.state &= ~WIN_VISIBLE;
1771 }
1772
1773 void
1774 xseturgency(int add) {
1775         XWMHints *h = XGetWMHints(xw.dpy, xw.win);
1776         h->flags = add ? (h->flags | XUrgencyHint) : (h->flags & ~XUrgencyHint);
1777         XSetWMHints(xw.dpy, xw.win, h);
1778         XFree(h);
1779 }
1780
1781 void
1782 focus(XEvent *ev) {
1783         if(ev->type == FocusIn) {
1784                 xw.state |= WIN_FOCUSED;
1785                 xseturgency(0);
1786         } else
1787                 xw.state &= ~WIN_FOCUSED;
1788         draw();
1789 }
1790
1791 char*
1792 kmap(KeySym k, unsigned int state) {
1793         int i;
1794         for(i = 0; i < LEN(key); i++)
1795                 if(key[i].k == k && (key[i].mask == 0 || key[i].mask & state))
1796                         return (char*)key[i].s;
1797         return NULL;
1798 }
1799
1800 void
1801 kpress(XEvent *ev) {
1802         XKeyEvent *e = &ev->xkey;
1803         KeySym ksym;
1804         char buf[32];
1805         char *customkey;
1806         int len;
1807         int meta;
1808         int shift;
1809         Status status;
1810
1811         meta = e->state & Mod1Mask;
1812         shift = e->state & ShiftMask;
1813         len = XmbLookupString(xw.xic, e, buf, sizeof(buf), &ksym, &status);
1814         
1815         /* 1. custom keys from config.h */
1816         if((customkey = kmap(ksym, e->state)))
1817                 ttywrite(customkey, strlen(customkey));
1818         /* 2. hardcoded (overrides X lookup) */
1819         else
1820                 switch(ksym) {
1821                 case XK_Up:
1822                 case XK_Down:
1823                 case XK_Left:
1824                 case XK_Right:
1825                         /* XXX: shift up/down doesn't work */
1826                         sprintf(buf, "\033%c%c", IS_SET(MODE_APPKEYPAD) ? 'O' : '[', (shift ? "dacb":"DACB")[ksym - XK_Left]);
1827                         ttywrite(buf, 3);
1828                         break;
1829                 case XK_Insert:
1830                         if(shift)
1831                                 selpaste();
1832                         break;
1833                 case XK_Return:
1834                         if(IS_SET(MODE_CRLF))
1835                                 ttywrite("\r\n", 2);
1836                         else
1837                                 ttywrite("\r", 1);
1838                         break;
1839                         /* 3. X lookup  */
1840                 default:
1841                         if(len > 0) {
1842                                 if(meta && len == 1)
1843                                         ttywrite("\033", 1);
1844                                 ttywrite(buf, len);
1845                         }
1846                         break;
1847                 }
1848 }
1849
1850 void
1851 resize(XEvent *e) {
1852         int col, row;
1853         
1854         if(e->xconfigure.width == xw.w && e->xconfigure.height == xw.h)
1855                 return;
1856         
1857         xw.w = e->xconfigure.width;
1858         xw.h = e->xconfigure.height;
1859         col = (xw.w - 2*BORDER) / xw.cw;
1860         row = (xw.h - 2*BORDER) / xw.ch;
1861         if(col == term.col && row == term.row)
1862                 return;
1863         if(tresize(col, row))
1864                 draw();
1865         ttyresize(col, row);
1866         xresize(col, row);
1867 }
1868
1869 void
1870 run(void) {
1871         XEvent ev;
1872         fd_set rfd;
1873         int xfd = XConnectionNumber(xw.dpy);
1874
1875         for(;;) {
1876                 FD_ZERO(&rfd);
1877                 FD_SET(cmdfd, &rfd);
1878                 FD_SET(xfd, &rfd);
1879                 if(select(MAX(xfd, cmdfd)+1, &rfd, NULL, NULL, NULL) < 0) {
1880                         if(errno == EINTR)
1881                                 continue;
1882                         die("select failed: %s\n", SERRNO);
1883                 }
1884                 if(FD_ISSET(cmdfd, &rfd)) {
1885                         ttyread();
1886                         draw();
1887                 }
1888                 while(XPending(xw.dpy)) {
1889                         XNextEvent(xw.dpy, &ev);
1890                         if(XFilterEvent(&ev, xw.win))
1891                                 continue;
1892                         if(handler[ev.type])
1893                                 (handler[ev.type])(&ev);
1894                 }
1895         }
1896 }
1897
1898 int
1899 main(int argc, char *argv[]) {
1900         int i;
1901         
1902         for(i = 1; i < argc; i++) {
1903                 switch(argv[i][0] != '-' || argv[i][2] ? -1 : argv[i][1]) {
1904                 case 't':
1905                         if(++i < argc) opt_title = argv[i];
1906                         break;
1907                 case 'c':
1908                         if(++i < argc) opt_class = argv[i];
1909                         break;
1910                 case 'e':
1911                         if(++i < argc) opt_cmd = &argv[i];
1912                         break;
1913                 case 'v':
1914                 default:
1915                         die(USAGE);
1916                 }
1917                 /* -e eats every remaining arguments */
1918                 if(opt_cmd)
1919                         break;
1920         }
1921         setlocale(LC_CTYPE, "");
1922         tnew(80, 24);
1923         ttynew();
1924         xinit();
1925         selinit();
1926         run();
1927         return 0;
1928 }