Xinqi Bao's Git

patch: swallow
[dwm.git] / dwm.c
diff --git a/dwm.c b/dwm.c
index ab16c75..f266a37 100644 (file)
--- a/dwm.c
+++ b/dwm.c
 #include <X11/extensions/Xinerama.h>
 #endif /* XINERAMA */
 #include <X11/Xft/Xft.h>
+#include <X11/Xlib-xcb.h>
+#include <xcb/res.h>
+#ifdef __OpenBSD__
+#include <sys/sysctl.h>
+#include <kvm.h>
+#endif /* __OpenBSD */
 
 #include "drw.h"
 #include "util.h"
@@ -92,9 +98,11 @@ struct Client {
        int basew, baseh, incw, inch, maxw, maxh, minw, minh;
        int bw, oldbw;
        unsigned int tags;
-       int isfixed, isfloating, isurgent, neverfocus, oldstate, isfullscreen;
+       int isfixed, isfloating, isurgent, neverfocus, oldstate, isfullscreen, isterminal, noswallow;
+       pid_t pid;
        Client *next;
        Client *snext;
+       Client *swallowing;
        Monitor *mon;
        Window win;
 };
@@ -138,6 +146,8 @@ typedef struct {
        const char *title;
        unsigned int tags;
        int isfloating;
+       int isterminal;
+       int noswallow;
        int monitor;
 } Rule;
 
@@ -169,6 +179,7 @@ static void focus(Client *c);
 static void focusin(XEvent *e);
 static void focusmon(const Arg *arg);
 static void focusstack(const Arg *arg);
+static Atom getatomprop(Client *c, Atom prop);
 static int getrootptr(int *x, int *y);
 static long getstate(Window w);
 static int gettextprop(Window w, Atom atom, char *text, unsigned int size);
@@ -216,15 +227,15 @@ static void toggleview(const Arg *arg);
 static void unfocus(Client *c, int setfocus);
 static void unmanage(Client *c, int destroyed);
 static void unmapnotify(XEvent *e);
-static int updategeom(void);
 static void updatebarpos(Monitor *m);
 static void updatebars(void);
 static void updateclientlist(void);
+static int updategeom(void);
 static void updatenumlockmask(void);
 static void updatesizehints(Client *c);
 static void updatestatus(void);
-static void updatewindowtype(Client *c);
 static void updatetitle(Client *c);
+static void updatewindowtype(Client *c);
 static void updatewmhints(Client *c);
 static void view(const Arg *arg);
 static Client *wintoclient(Window w);
@@ -234,6 +245,12 @@ static int xerrordummy(Display *dpy, XErrorEvent *ee);
 static int xerrorstart(Display *dpy, XErrorEvent *ee);
 static void zoom(const Arg *arg);
 
+static pid_t getparentprocess(pid_t p);
+static int isdescprocess(pid_t p, pid_t c);
+static Client *swallowingclient(Window w);
+static Client *termforwin(const Client *c);
+static pid_t winpid(Window w);
+
 /* variables */
 static const char broken[] = "broken";
 static char stext[256];
@@ -268,6 +285,8 @@ static Drw *drw;
 static Monitor *mons, *selmon;
 static Window root, wmcheckwin;
 
+static xcb_connection_t *xcon;
+
 /* configuration, allows nested code to access above variables */
 #include "config.h"
 
@@ -297,6 +316,8 @@ applyrules(Client *c)
                && (!r->class || strstr(class, r->class))
                && (!r->instance || strstr(instance, r->instance)))
                {
+                       c->isterminal = r->isterminal;
+                       c->noswallow  = r->noswallow;
                        c->isfloating = r->isfloating;
                        c->tags |= r->tags;
                        for (m = mons; m && m->num != r->monitor; m = m->next);
@@ -413,6 +434,53 @@ attachstack(Client *c)
        c->mon->stack = c;
 }
 
+void
+swallow(Client *p, Client *c)
+{
+
+       if (c->noswallow || c->isterminal)
+               return;
+       if (c->noswallow && !swallowfloating && c->isfloating)
+               return;
+
+       detach(c);
+       detachstack(c);
+
+       setclientstate(c, WithdrawnState);
+       XUnmapWindow(dpy, p->win);
+
+       p->swallowing = c;
+       c->mon = p->mon;
+
+       Window w = p->win;
+       p->win = c->win;
+       c->win = w;
+       updatetitle(p);
+       XMoveResizeWindow(dpy, p->win, p->x, p->y, p->w, p->h);
+       arrange(p->mon);
+       configure(p);
+       updateclientlist();
+}
+
+void
+unswallow(Client *c)
+{
+       c->win = c->swallowing->win;
+
+       free(c->swallowing);
+       c->swallowing = NULL;
+
+       /* unfullscreen the client */
+       setfullscreen(c, 0);
+       updatetitle(c);
+       arrange(c->mon);
+       XMapWindow(dpy, c->win);
+       XMoveResizeWindow(dpy, c->win, c->x, c->y, c->w, c->h);
+       setclientstate(c, NormalState);
+       focus(NULL);
+       arrange(c->mon);
+}
+
 void
 buttonpress(XEvent *e)
 {
@@ -439,7 +507,7 @@ buttonpress(XEvent *e)
                        arg.ui = 1 << i;
                } else if (ev->x < x + blw)
                        click = ClkLtSymbol;
-               else if (ev->x > selmon->ww - TEXTW(stext))
+               else if (ev->x > selmon->ww - (int)TEXTW(stext))
                        click = ClkStatusText;
                else
                        click = ClkWinTitle;
@@ -486,6 +554,7 @@ cleanup(void)
                drw_cur_free(drw, cursor[i]);
        for (i = 0; i < LENGTH(colors); i++)
                free(scheme[i]);
+       free(scheme);
        XDestroyWindow(dpy, wmcheckwin);
        drw_free(drw);
        XSync(dpy, False);
@@ -652,6 +721,9 @@ destroynotify(XEvent *e)
 
        if ((c = wintoclient(ev->window)))
                unmanage(c, 1);
+
+       else if ((c = swallowingclient(ev->window)))
+               unmanage(c->swallowing, 1);
 }
 
 void
@@ -695,17 +767,20 @@ dirtomon(int dir)
 void
 drawbar(Monitor *m)
 {
-       int x, w, sw = 0;
+       int x, w, tw = 0;
        int boxs = drw->fonts->h / 9;
        int boxw = drw->fonts->h / 6 + 2;
        unsigned int i, occ = 0, urg = 0;
        Client *c;
 
+       if (!m->showbar)
+               return;
+
        /* draw status first so it can be overdrawn by tags later */
        if (m == selmon) { /* status is only drawn on selected monitor */
                drw_setscheme(drw, scheme[SchemeNorm]);
-               sw = TEXTW(stext) - lrpad + 2; /* 2px right padding */
-               drw_text(drw, m->ww - sw, 0, sw, bh, 0, stext, 0);
+               tw = TEXTW(stext) - lrpad + 2; /* 2px right padding */
+               drw_text(drw, m->ww - tw, 0, tw, bh, 0, stext, 0);
        }
 
        for (c = m->clients; c; c = c->next) {
@@ -728,7 +803,7 @@ drawbar(Monitor *m)
        drw_setscheme(drw, scheme[SchemeNorm]);
        x = drw_text(drw, x, 0, w, bh, lrpad / 2, m->ltsymbol, 0);
 
-       if ((w = m->ww - sw - x) > bh) {
+       if ((w = m->ww - tw - x) > bh) {
                if (m->sel) {
                        drw_setscheme(drw, scheme[m == selmon ? SchemeSel : SchemeNorm]);
                        drw_text(drw, x, 0, w, bh, lrpad / 2, m->sel->name, 0);
@@ -834,7 +909,7 @@ focusstack(const Arg *arg)
 {
        Client *c = NULL, *i;
 
-       if (!selmon->sel)
+       if (!selmon->sel || (selmon->sel->isfullscreen && lockfullscreen))
                return;
        if (arg->i > 0) {
                for (c = selmon->sel->next; c && !ISVISIBLE(c); c = c->next);
@@ -1017,12 +1092,13 @@ killclient(const Arg *arg)
 void
 manage(Window w, XWindowAttributes *wa)
 {
-       Client *c, *t = NULL;
+       Client *c, *t = NULL, *term = NULL;
        Window trans = None;
        XWindowChanges wc;
 
        c = ecalloc(1, sizeof(Client));
        c->win = w;
+       c->pid = winpid(w);
        /* geometry */
        c->x = c->oldx = wa->x;
        c->y = c->oldy = wa->y;
@@ -1037,6 +1113,7 @@ manage(Window w, XWindowAttributes *wa)
        } else {
                c->mon = selmon;
                applyrules(c);
+               term = termforwin(c);
        }
 
        if (c->x + WIDTH(c) > c->mon->mx + c->mon->mw)
@@ -1073,6 +1150,8 @@ manage(Window w, XWindowAttributes *wa)
        c->mon->sel = c;
        arrange(c->mon);
        XMapWindow(dpy, c->win);
+       if (term)
+               swallow(term, c);
        focus(NULL);
 }
 
@@ -1520,7 +1599,7 @@ setmfact(const Arg *arg)
        if (!arg || !selmon->lt[selmon->sellt]->arrange)
                return;
        f = arg->f < 1.0 ? arg->f + selmon->mfact : arg->f - 1.0;
-       if (f < 0.1 || f > 0.9)
+       if (f < 0.05 || f > 0.95)
                return;
        selmon->mfact = f;
        arrange(selmon);
@@ -1688,11 +1767,13 @@ tile(Monitor *m)
                if (i < m->nmaster) {
                        h = (m->wh - my) / (MIN(n, m->nmaster) - i);
                        resize(c, m->wx, m->wy + my, mw - (2*c->bw), h - (2*c->bw), 0);
-                       my += HEIGHT(c);
+                       if (my + HEIGHT(c) < m->wh)
+                               my += HEIGHT(c);
                } else {
                        h = (m->wh - ty) / (n - i);
                        resize(c, m->wx + mw, m->wy + ty, m->ww - mw - (2*c->bw), h - (2*c->bw), 0);
-                       ty += HEIGHT(c);
+                       if (ty + HEIGHT(c) < m->wh)
+                               ty += HEIGHT(c);
                }
 }
 
@@ -1765,6 +1846,20 @@ unmanage(Client *c, int destroyed)
        Monitor *m = c->mon;
        XWindowChanges wc;
 
+       if (c->swallowing) {
+               unswallow(c);
+               return;
+       }
+
+       Client *s = swallowingclient(c->win);
+       if (s) {
+               free(s->swallowing);
+               s->swallowing = NULL;
+               arrange(m);
+               focus(NULL);
+               return;
+       }
+
        detach(c);
        detachstack(c);
        if (!destroyed) {
@@ -1779,9 +1874,12 @@ unmanage(Client *c, int destroyed)
                XUngrabServer(dpy);
        }
        free(c);
-       focus(NULL);
-       updateclientlist();
-       arrange(m);
+
+       if (!s) {
+               arrange(m);
+               focus(NULL);
+               updateclientlist();
+       }
 }
 
 void
@@ -1984,6 +2082,14 @@ updatesizehints(Client *c)
        c->isfixed = (c->maxw && c->maxh && c->maxw == c->minw && c->maxh == c->minh);
 }
 
+void
+updatestatus(void)
+{
+       if (!gettextprop(root, XA_WM_NAME, stext, sizeof(stext)))
+               strcpy(stext, "dwm-"VERSION);
+       drawbar(selmon);
+}
+
 void
 updatetitle(Client *c)
 {
@@ -1993,14 +2099,6 @@ updatetitle(Client *c)
                strcpy(c->name, broken);
 }
 
-void
-updatestatus(void)
-{
-       if (!gettextprop(root, XA_WM_NAME, stext, sizeof(stext)))
-               strcpy(stext, "dwm-"VERSION);
-       drawbar(selmon);
-}
-
 void
 updatewindowtype(Client *c)
 {
@@ -2044,6 +2142,136 @@ view(const Arg *arg)
        arrange(selmon);
 }
 
+pid_t
+winpid(Window w)
+{
+
+       pid_t result = 0;
+
+#ifdef __linux__
+       xcb_res_client_id_spec_t spec = {0};
+       spec.client = w;
+       spec.mask = XCB_RES_CLIENT_ID_MASK_LOCAL_CLIENT_PID;
+
+       xcb_generic_error_t *e = NULL;
+       xcb_res_query_client_ids_cookie_t c = xcb_res_query_client_ids(xcon, 1, &spec);
+       xcb_res_query_client_ids_reply_t *r = xcb_res_query_client_ids_reply(xcon, c, &e);
+
+       if (!r)
+               return (pid_t)0;
+
+       xcb_res_client_id_value_iterator_t i = xcb_res_query_client_ids_ids_iterator(r);
+       for (; i.rem; xcb_res_client_id_value_next(&i)) {
+               spec = i.data->spec;
+               if (spec.mask & XCB_RES_CLIENT_ID_MASK_LOCAL_CLIENT_PID) {
+                       uint32_t *t = xcb_res_client_id_value_value(i.data);
+                       result = *t;
+                       break;
+               }
+       }
+
+       free(r);
+
+       if (result == (pid_t)-1)
+               result = 0;
+
+#endif /* __linux__ */
+
+#ifdef __OpenBSD__
+        Atom type;
+        int format;
+        unsigned long len, bytes;
+        unsigned char *prop;
+        pid_t ret;
+
+        if (XGetWindowProperty(dpy, w, XInternAtom(dpy, "_NET_WM_PID", 0), 0, 1, False, AnyPropertyType, &type, &format, &len, &bytes, &prop) != Success || !prop)
+               return 0;
+
+        ret = *(pid_t*)prop;
+        XFree(prop);
+        result = ret;
+
+#endif /* __OpenBSD__ */
+       return result;
+}
+
+pid_t
+getparentprocess(pid_t p)
+{
+       unsigned int v = 0;
+
+#ifdef __linux__
+       FILE *f;
+       char buf[256];
+       snprintf(buf, sizeof(buf) - 1, "/proc/%u/stat", (unsigned)p);
+
+       if (!(f = fopen(buf, "r")))
+               return 0;
+
+       fscanf(f, "%*u %*s %*c %u", &v);
+       fclose(f);
+#endif /* __linux__*/
+
+#ifdef __OpenBSD__
+       int n;
+       kvm_t *kd;
+       struct kinfo_proc *kp;
+
+       kd = kvm_openfiles(NULL, NULL, NULL, KVM_NO_FILES, NULL);
+       if (!kd)
+               return 0;
+
+       kp = kvm_getprocs(kd, KERN_PROC_PID, p, sizeof(*kp), &n);
+       v = kp->p_ppid;
+#endif /* __OpenBSD__ */
+
+       return (pid_t)v;
+}
+
+int
+isdescprocess(pid_t p, pid_t c)
+{
+       while (p != c && c != 0)
+               c = getparentprocess(c);
+
+       return (int)c;
+}
+
+Client *
+termforwin(const Client *w)
+{
+       Client *c;
+       Monitor *m;
+
+       if (!w->pid || w->isterminal)
+               return NULL;
+
+       for (m = mons; m; m = m->next) {
+               for (c = m->clients; c; c = c->next) {
+                       if (c->isterminal && !c->swallowing && c->pid && isdescprocess(c->pid, w->pid))
+                               return c;
+               }
+       }
+
+       return NULL;
+}
+
+Client *
+swallowingclient(Window w)
+{
+       Client *c;
+       Monitor *m;
+
+       for (m = mons; m; m = m->next) {
+               for (c = m->clients; c; c = c->next) {
+                       if (c->swallowing && c->swallowing->win == w)
+                               return c;
+               }
+       }
+
+       return NULL;
+}
+
 Client *
 wintoclient(Window w)
 {
@@ -2135,8 +2363,14 @@ main(int argc, char *argv[])
                fputs("warning: no locale support\n", stderr);
        if (!(dpy = XOpenDisplay(NULL)))
                die("dwm: cannot open display");
+       if (!(xcon = XGetXCBConnection(dpy)))
+               die("dwm: cannot get xcb connection\n");
        checkotherwm();
        setup();
+#ifdef __OpenBSD__
+       if (pledge("stdio rpath proc exec ps", NULL) == -1)
+               die("pledge");
+#endif /* __OpenBSD__ */
        scan();
        run();
        cleanup();