riow: keyboard-controller for rio

1 files changed, 433 insertions, 0 deletions

diff --git a/sys/src/cmd/riow.c b/sys/src/cmd/riow.c
new file mode 100644
index 000000000..787a87ad6
--- /dev/null
+++ b/sys/src/cmd/riow.c
@@ -0,0 +1,433 @@
+#include <u.h>
+#include <libc.h>
+#include <draw.h>
+#include <keyboard.h>
+
+typedef struct W W;
+
+enum {
+	Mmod4 = 1<<0,
+	Mctl = 1<<1,
+	Mshift = 1<<2,
+
+	Step = 16,
+	Stepbig = 64,
+
+	Fvisible = 1<<0,
+	Fcurrent = 1<<1,
+	Fsticky = 1<<2,
+	Ffullscreen = 1<<3,
+};
+
+struct W {
+	int id;
+	Rectangle r;
+	int vd;
+	int flags;
+};
+
+static int vd = 1; /* current virtual desktop */
+static int wsys; /* rios /dev/wsys fd */
+static int mod;
+static W *ws, *wcur;
+static int wsn;
+static int vd2wcur[10] = {-1};
+
+static char *sticky[32] = {
+	"bar",
+	"cat clock",
+	"clock",
+	"faces",
+	"kbmap",
+	"stats",
+	"winwatch",
+	nil,
+};
+
+static int
+wwctl(int id, int mode)
+{
+	char s[64];
+
+	snprint(s, sizeof(s), "/dev/wsys/%d/wctl", id);
+
+	return open(s, mode);
+}
+
+static void
+wsupdate(void)
+{
+	int i, k, n, f, seen, tn, dsn;
+	char s[256], *t[8];
+	W *newws, *w;
+	Dir *ds, *d;
+
+	seek(wsys, 0, 0);
+	if((dsn = dirreadall(wsys, &ds)) < 0)
+		sysfatal("/dev/wsys: %r");
+
+	newws = malloc(sizeof(W)*dsn);
+	wcur = nil;
+	for(i = 0, d = ds, w = newws; i < dsn; i++, d++){
+		if((f = wwctl(atoi(d->name), OREAD)) < 0)
+			continue;
+		n = read(f, s, sizeof(s)-1);
+		close(f);
+		if(n < 12)
+			continue;
+		s[n] = 0;
+		if((tn = tokenize(s, t, nelem(t))) < 6)
+			continue;
+
+		w->id = atoi(d->name);
+		w->r.min.x = atoi(t[0]);
+		w->r.min.y = atoi(t[1]);
+		w->r.max.x = atoi(t[2]);
+		w->r.max.y = atoi(t[3]);
+		w->vd = -1;
+		w->flags = 0;
+
+		/* move over the current state of the window */
+		for(k = 0, seen = 0; k < wsn; k++){
+			if(ws[k].id == w->id){
+				w->vd = ws[k].vd;
+				w->flags = ws[k].flags & ~(Fvisible|Fcurrent);
+				if(w->flags & Ffullscreen)
+					w->r = ws[k].r;
+				seen = 1;
+				break;
+			}
+		}
+
+		/* update current state */
+		for(k = 4; k < tn; k++){
+			if(strcmp(t[k], "current") == 0){
+				w->flags |= Fcurrent;
+				wcur = w;
+				w->vd = vd;
+			}else if(strcmp(t[k], "visible") == 0){
+				w->flags |= Fvisible;
+				w->vd = vd;
+			}
+		}
+
+		if(!seen){
+			/* not seen previously - set the new state for it */
+			w->vd = vd;
+		}
+
+		/* because a different program can run in any window we have to re-read */
+		snprint(s, sizeof(s), "/dev/wsys/%d/label", w->id);
+		w->flags &= ~Fsticky;
+		if((f = open(s, OREAD)) >= 0){
+			n = read(f, s, sizeof(s)-1);
+			close(f);
+			if(n > 0){
+				s[n] = 0;
+				for(k = 0; k < nelem(sticky) && sticky[k] != nil; k++){
+					if(strcmp(sticky[k], s) == 0){
+						w->flags |= Fsticky;
+						break;
+					}
+				}
+			}
+		}
+		w++;
+	}
+
+	free(ds);
+	free(ws);
+	ws = newws;
+	wsn = w - newws;
+}
+
+static void
+spawn(char *s)
+{
+	if(rfork(RFPROC|RFNOWAIT|RFNAMEG|RFENVG|RFCFDG|RFREND) == 0)
+		execl("/bin/rc", "rc", s, nil);
+}
+
+static void
+togglefullscreen(void)
+{
+	int f;
+
+	if(wcur == nil || (f = wwctl(wcur->id, OWRITE)) < 0)
+		return;
+	wcur->flags ^= Ffullscreen;
+	if(wcur->flags & Ffullscreen)
+		fprint(f, "resize -r 0 0 9999 9999");
+	else
+		fprint(f, "resize -r %d %d %d %d", wcur->r.min.x, wcur->r.min.y, wcur->r.max.x, wcur->r.max.y);
+	close(f);
+}
+
+static void
+togglesticky(void)
+{
+	if(wcur != nil)
+		wcur->flags ^= Fsticky;
+}
+
+static void
+vdaction(int nvd)
+{
+	int f, wcurf;
+	W *w;
+
+	if(mod == Mmod4){
+		wcur = nil;
+		wcurf = -1;
+		vd2wcur[vd] = -1;
+		for(w = ws; w < ws+wsn; w++){
+			if((f = wwctl(w->id, OWRITE)) < 0)
+				continue;
+
+			if(w->flags & Fvisible)
+				w->vd = vd;
+			else if(w->vd == vd)
+				w->vd = -1;
+
+			if(w->flags & Fcurrent)
+				vd2wcur[vd] = w->id;
+
+			if(w->vd != nvd && (w->flags & Fsticky) == 0){
+				fprint(f, "hide");
+			}else{
+				fprint(f, "unhide");
+				if(vd2wcur[nvd] == w->id && wcurf < 0){
+					wcur = w;
+					wcurf = f;
+					f = -1;
+				}
+			}
+			if(f >= 0)
+				close(f);
+		}
+		if(wcur != nil){
+			fprint(wcurf, "top");
+			fprint(wcurf, "current");
+			close(wcurf);
+		}
+		vd = nvd;
+		fprint(3, "%d\n", vd);
+	}else if(mod == (Mmod4 | Mshift) && wcur != nil && wcur->vd != nvd){
+		if((f = wwctl(wcur->id, OWRITE)) >= 0){
+			fprint(f, "hide");
+			wcur->vd = nvd;
+			vd2wcur[nvd] = wcur->id; /* bring to the top */
+			wcur = nil;
+			close(f);
+		}
+	}
+}
+
+static void
+arrowaction(int x, int y)
+{
+	int f;
+
+	if(wcur == nil || (f = wwctl(wcur->id, OWRITE)) < 0)
+		return;
+
+	x *= (mod & Mctl) ? Stepbig : Step;
+	y *= (mod & Mctl) ? Stepbig : Step;
+	if((mod & Mshift) == 0)
+		fprint(f, "move -minx %+d -miny %+d", x, y);
+	else
+		fprint(f, "resize -maxx %+d -maxy %+d -minx %+d -miny %+d", x, y, -x, -y);
+	close(f);
+}
+
+static struct {
+	int x, y;
+}cyclectx;
+
+static int
+cyclecmp(void *a_, void *b_)
+{
+	W *a = a_, *b = b_;
+
+	return cyclectx.x*(a->r.min.x - b->r.min.x) + cyclectx.y*(a->r.min.y - b->r.min.y);
+}
+
+static void
+cycleaction(int x, int y)
+{
+	int wcurid, i, f;
+	W *w, *w₀;
+
+	wcurid = wcur == nil ? -1 : wcur->id;
+	cyclectx.x = x;
+	cyclectx.y = y;
+	qsort(ws, wsn, sizeof(*ws), cyclecmp);
+	w₀ = nil;
+	wcur = nil;
+	for(i = 0, w = ws; i < wsn; i++, w++){
+		if(w->id == wcurid){
+			wcur = w;
+			continue;
+		}
+		if((w->flags & Fsticky) != 0 || w->vd != vd)
+			continue;
+		if(w₀ == nil)
+			w₀ = w;
+		if(wcur != nil)
+			break;
+	}
+	if(i >= wsn)
+		w = w₀;
+	if(w == nil || (f = wwctl(w->id, OWRITE)) < 0)
+		return;
+	fprint(f, "top");
+	fprint(f, "current");
+	close(f);
+	wcur = w;
+}
+
+static void
+keyevent(Rune r)
+{
+	wsupdate();
+
+	if(r == '\n')
+		spawn("window");
+	else if(r == 'f')
+		togglefullscreen();
+	else if(r == 's')
+		togglesticky();
+	else if(r >= '0' && r <= '9')
+		vdaction(r - '0');
+	else if(r == Kup)
+		arrowaction(0, -1);
+	else if(r == Kdown)
+		arrowaction(0, 1);
+	else if(r == Kleft)
+		arrowaction(-1, 0);
+	else if(r == Kright)
+		arrowaction(1, 0);
+	else if(r == 'h')
+		cycleaction(-1, 0);
+	else if(r == 'l')
+		cycleaction(1, 0);
+	else if(r == 'j')
+		cycleaction(0, 1);
+	else if(r == 'k')
+		cycleaction(0, -1);
+}
+
+static void
+process(char *s)
+{
+	int n, o, oldmod;
+	char b[128], *p;
+	Rune r;
+
+	if(*s == 'K' && s[1] == 0)
+		mod = 0;
+
+	o = 0;
+	b[o++] = *s;
+	for(p = s+1; *p != 0; p += n){
+		if((n = chartorune(&r, p)) == 1 && r == Runeerror){
+			/* bail out */
+			n = strlen(p);
+			memmove(b+o, p, n);
+			o += n;
+			p += n;
+			break;
+		}
+
+		oldmod = mod;
+
+		if(*s == 'c' && (mod & Mmod4) != 0){
+			keyevent(r);
+			continue;
+		}
+
+		if(*s == 'k'){
+			if(r == Kmod4)
+				mod |= Mmod4;
+			else if(r == Kctl)
+				mod |= Mctl;
+			else if(r == Kshift)
+				mod |= Mshift;
+			else if(r >= '0' && r <= '9' && (mod & (Mshift|Mmod4)) == (Mshift|Mmod4))
+				keyevent(r);
+		}else if(*s == 'K'){
+			if(r == Kmod4)
+				mod &= ~Mmod4;
+			else if(r == Kctl)
+				mod &= ~Mctl;
+			else if(r == Kshift)
+				mod &= ~Mshift;
+		}
+
+		if((oldmod | mod) & Mmod4)
+			continue;
+
+		memmove(b+o, p, n);
+		o += n;
+	}
+
+	/* all runes filtered out - ignore completely */
+	if(o == 1 && p-s > 1)
+		return;
+
+	b[o++] = 0;
+	if(write(1, b, o) != o)
+		exits(nil);
+}
+
+static void
+usage(void)
+{
+	fprint(2, "usage: %s [-s label]\n", argv0);
+	exits("usage");
+}
+
+void
+main(int argc, char **argv)
+{
+	char b[128];
+	int i, j, n;
+
+	for(n = 0; sticky[n] != nil; n++)
+		;
+
+	ARGBEGIN{
+	case 's':
+		if(n >= nelem(sticky))
+			sysfatal("ewwww");
+		sticky[n++] = EARGF(usage());
+		break;
+	default:
+		usage();
+	}ARGEND
+
+	if((wsys = open("/dev/wsys", OREAD)) < 0)
+		sysfatal("%r");
+
+	/* initial state */
+	wsupdate();
+	for(i = 0; i < wsn; i++)
+		ws[i].vd = vd;
+	fprint(3, "%d\n", vd);
+
+	for(i = 0;;){
+		if((n = read(0, b+i, sizeof(b)-i)) <= 0)
+			break;
+		n += i;
+		for(j = 0; j < n; j++){
+			if(b[j] == 0){
+				process(b+i);
+				i = j+1;
+			}
+		}
+		memmove(b, b+i, j-i);
+		i -= j;
+	}
+
+	exits(nil);
+}