]> git.lizzy.rs Git - bspwm.git/blobdiff - bspwm.c
Add new contributor
[bspwm.git] / bspwm.c
diff --git a/bspwm.c b/bspwm.c
index de6a6f13ae304c3b79cb5b47d51a1a0cf47c91b3..249a7d8d35c477be920e68424690091ada8b6b6e 100644 (file)
--- a/bspwm.c
+++ b/bspwm.c
 #include <xcb/xcb.h>
 #include <xcb/xcb_event.h>
 #include <xcb/xcb_ewmh.h>
-#include "helpers.h"
+#include <xcb/randr.h>
 #include "types.h"
 #include "settings.h"
 #include "messages.h"
 #include "rules.h"
 #include "events.h"
 #include "common.h"
-#include "utils.h"
+#include "helpers.h"
+#include "window.h"
 #include "bspwm.h"
 #include "tree.h"
 #include "ewmh.h"
@@ -29,107 +30,268 @@ void quit(void)
     running = false;
 }
 
-int register_events(void)
+void cleanup(void)
 {
-    xcb_generic_error_t *err;
-    uint32_t values[] = {XCB_EVENT_MASK_SUBSTRUCTURE_REDIRECT | XCB_EVENT_MASK_SUBSTRUCTURE_NOTIFY | XCB_EVENT_MASK_PROPERTY_CHANGE | XCB_EVENT_MASK_BUTTON_PRESS};
-    err = xcb_request_check(dpy, xcb_change_window_attributes_checked(dpy, screen->root, XCB_CW_EVENT_MASK, values));
-    if (err != NULL)
-        return 1;
-    return 0;
+    while (mon_head != NULL)
+        remove_monitor(mon_head);
+    while (rule_head != NULL)
+        remove_rule(rule_head);
+    free(frozen_pointer);
 }
 
+void register_events(void)
+{
+    uint32_t values[] = {ROOT_EVENT_MASK};
+    xcb_generic_error_t *e = xcb_request_check(dpy, xcb_change_window_attributes_checked(dpy, root, XCB_CW_EVENT_MASK, values));
+    if (e != NULL) {
+        xcb_disconnect(dpy);
+        err("Another window manager is already running.\n");
+    }
+}
 
-void handle_zombie(int sig)
+bool import_monitors(void)
 {
-    while (waitpid(-1, NULL, WNOHANG) > 0)
-        ;
-    signal(sig, handle_zombie);
+    PUTS("import monitors");
+    xcb_randr_get_screen_resources_current_reply_t *sres = xcb_randr_get_screen_resources_current_reply(dpy, xcb_randr_get_screen_resources_current(dpy, root), NULL);
+    if (sres == NULL)
+        return false;
+
+    int len = xcb_randr_get_screen_resources_current_outputs_length(sres);
+    xcb_randr_output_t *outputs = xcb_randr_get_screen_resources_current_outputs(sres);
+
+    xcb_randr_get_output_info_cookie_t cookies[len];
+    for (int i = 0; i < len; i++)
+        cookies[i] = xcb_randr_get_output_info(dpy, outputs[i], XCB_CURRENT_TIME);
+
+    for (monitor_t *m = mon_head; m != NULL; m = m->next)
+        m->wired = false;
+
+    monitor_t *mm = NULL;
+    unsigned int num = 0;
+
+    for (int i = 0; i < len; i++) {
+        xcb_randr_get_output_info_reply_t *info = xcb_randr_get_output_info_reply(dpy, cookies[i], NULL);
+        if (info != NULL && info->crtc != XCB_NONE) {
+
+            xcb_randr_get_crtc_info_reply_t *cir = xcb_randr_get_crtc_info_reply(dpy, xcb_randr_get_crtc_info(dpy, info->crtc, XCB_CURRENT_TIME), NULL);
+            if (cir != NULL) {
+                xcb_rectangle_t rect = (xcb_rectangle_t) {cir->x, cir->y, cir->width, cir->height};
+                mm = get_monitor_by_id(outputs[i]);
+                if (mm != NULL) {
+                    mm->rectangle = rect;
+                    for (desktop_t *d = mm->desk_head; d != NULL; d = d->next)
+                        for (node_t *n = first_extrema(d->root); n != NULL; n = next_leaf(n, d->root))
+                            fit_monitor(mm, n->client);
+                    arrange(mm, mm->desk);
+                    mm->wired = true;
+                    PRINTF("update monitor %s (0x%X)\n", mm->name, mm->id);
+                } else {
+                    mm = add_monitor(&rect);
+                    char *name = (char *)xcb_randr_get_output_info_name(info);
+                    size_t name_len = MIN(sizeof(mm->name), (size_t)xcb_randr_get_output_info_name_length(info));
+                    strncpy(mm->name, name, name_len);
+                    mm->name[name_len] = '\0';
+                    mm->id = outputs[i];
+                    add_desktop(mm, make_desktop(NULL));
+                    PRINTF("add monitor %s (0x%X)\n", mm->name, mm->id);
+                }
+                num++;
+            }
+            free(cir);
+        }
+        free(info);
+    }
+
+    monitor_t *m = mon_head;
+    while (m != NULL) {
+        monitor_t *next = m->next;
+        if (!m->wired) {
+            PRINTF("remove monitor %s (0x%X)\n", m->name, m->id);
+            merge_monitors(m, mm);
+            remove_monitor(m);
+        }
+        m = next;
+    }
+
+    free(sres);
+    update_motion_recorder();
+    return (num_monitors > 0);
+}
+
+void init(void)
+{
+    num_monitors = num_desktops = num_clients = 0;
+    monitor_uid = desktop_uid = client_uid = rule_uid = 0;
+    mon = last_mon = mon_head = mon_tail = NULL;
+    rule_head = rule_tail = NULL;
+    status_fifo = NULL;
+    randr_base = 0;
+    visible = true;
+    exit_status = 0;
 }
 
-void setup(int default_screen)
+void setup(void)
 {
-    signal(SIGCHLD, handle_zombie);
+    init();
     ewmh_init();
-    screen = screen_of_display(dpy, default_screen);
-    if (!screen)
-        die("error: cannot aquire screen\n");
+    screen = xcb_setup_roots_iterator(xcb_get_setup(dpy)).data;
+    if (screen == NULL)
+        err("Can't acquire the default screen.\n");
+    root = screen->root;
+    register_events();
 
     screen_width = screen->width_in_pixels;
     screen_height = screen->height_in_pixels;
+    root_depth = screen->root_depth;
+
+    uint32_t mask = XCB_CW_EVENT_MASK;
+    uint32_t values[] = {XCB_EVENT_MASK_POINTER_MOTION};
+    motion_recorder = xcb_generate_id(dpy);
+    xcb_create_window(dpy, XCB_COPY_FROM_PARENT, motion_recorder, root, 0, 0, screen_width, screen_height, 0, XCB_WINDOW_CLASS_INPUT_ONLY, XCB_COPY_FROM_PARENT, mask, values);
+
+    xcb_atom_t net_atoms[] = {ewmh->_NET_SUPPORTED,
+                              ewmh->_NET_DESKTOP_NAMES,
+                              ewmh->_NET_NUMBER_OF_DESKTOPS,
+                              ewmh->_NET_CURRENT_DESKTOP,
+                              ewmh->_NET_CLIENT_LIST,
+                              ewmh->_NET_ACTIVE_WINDOW,
+                              ewmh->_NET_WM_DESKTOP,
+                              ewmh->_NET_WM_STATE,
+                              ewmh->_NET_WM_STATE_FULLSCREEN,
+                              ewmh->_NET_WM_STATE_DEMANDS_ATTENTION,
+                              ewmh->_NET_WM_WINDOW_TYPE,
+                              ewmh->_NET_WM_WINDOW_TYPE_DOCK,
+                              ewmh->_NET_WM_WINDOW_TYPE_NOTIFICATION,
+                              ewmh->_NET_WM_WINDOW_TYPE_DIALOG,
+                              ewmh->_NET_WM_WINDOW_TYPE_UTILITY,
+                              ewmh->_NET_WM_WINDOW_TYPE_TOOLBAR};
+
+    xcb_ewmh_set_supported(ewmh, default_screen, LENGTH(net_atoms), net_atoms);
+
+    xcb_intern_atom_reply_t *iar = xcb_intern_atom_reply(dpy, xcb_intern_atom(dpy, 0, strlen("_COMPTON_SHADOW"), "_COMPTON_SHADOW"), NULL);
+
+    if (iar != NULL) {
+        compton_shadow = iar->atom;
+        free(iar);
+    }
 
-    xcb_atom_t net_atoms[] = {ewmh._NET_SUPPORTED, ewmh._NET_WM_STATE_FULLSCREEN, ewmh._NET_WM_STATE, ewmh._NET_ACTIVE_WINDOW};
-
-    xcb_ewmh_set_supported(&ewmh, default_screen, LENGTH(net_atoms), net_atoms);
-    xcb_ewmh_set_wm_name(&ewmh, screen->root, LENGTH(WM_NAME), WM_NAME);
-
-    desk = make_desktop(DESK_NAME);
-    last_desk = NULL;
-    desk_head = desk;
-    desk_tail = desk;
-    rule_head = make_rule();
+    const xcb_query_extension_reply_t *qep = xcb_get_extension_data(dpy, &xcb_randr_id);
+    if (qep->present && import_monitors()) {
+        randr = true;
+        randr_base = qep->first_event;
+        xcb_randr_select_input(dpy, root, XCB_RANDR_NOTIFY_MASK_SCREEN_CHANGE);
+    } else {
+        randr = false;
+        warn("Couldn't retrieve monitors via RandR.\n");
+        xcb_rectangle_t rect = (xcb_rectangle_t) {0, 0, screen_width, screen_height};
+        monitor_t *m = add_monitor(&rect);
+        add_desktop(m, make_desktop(NULL));
+    }
 
-    split_mode = MODE_AUTOMATIC;
+    ewmh_update_number_of_desktops();
+    ewmh_update_desktop_names();
+    ewmh_update_current_desktop();
+    frozen_pointer = make_pointer_state();
 }
 
-int main(void)
+int main(int argc, char *argv[])
 {
     fd_set descriptors;
-    int sock_fd, ret_fd, dpy_fd, sel, nbr;
+    char socket_path[MAXLEN];
+    char *fifo_path = NULL;
+    status_prefix = NULL;
+    int sock_fd, ret_fd, dpy_fd, sel, n;
     struct sockaddr_un sock_address;
-    char *sock_path;
-    char msg[BUFSIZ], rsp[BUFSIZ];
-
+    size_t rsplen = 0;
+    char msg[BUFSIZ] = {0};
+    char rsp[BUFSIZ] = {0};
     xcb_generic_event_t *event;
+    char opt;
+
+    while ((opt = getopt(argc, argv, "hvs:p:")) != -1) {
+        switch (opt) {
+            case 'h':
+                printf("bspwm [-h|-v|-s PANEL_FIFO|-p PANEL_PREFIX]\n");
+                exit(EXIT_SUCCESS);
+                break;
+            case 'v':
+                printf("%s\n", VERSION);
+                exit(EXIT_SUCCESS);
+                break;
+            case 's':
+                fifo_path = optarg;
+                break;
+            case 'p':
+                status_prefix = optarg;
+                break;
+        }
+    }
 
     running = true;
-
     dpy = xcb_connect(NULL, &default_screen);
 
     if (xcb_connection_has_error(dpy))
-        die("error: cannot open display\n");
+        err("Can't open the default display.\n");
 
-    setup(default_screen);
-
-    if (register_events() == 1) {
-        xcb_disconnect(dpy);
-        die("another WM is already running\n");
-    }
+    setup();
 
     dpy_fd = xcb_get_file_descriptor(dpy);
 
-    sock_path = getenv(SOCK_PATH);
-
-    if (sock_path == NULL)
-        die("BSPWM_SOCKET environment variable is not set\n");
+    char *sp = getenv(SOCKET_ENV_VAR);
+    strncpy(socket_path, (sp == NULL ? DEFAULT_SOCKET_PATH : sp), sizeof(socket_path));
 
     sock_address.sun_family = AF_UNIX;
-    strcpy(sock_address.sun_path, sock_path);
-    unlink(sock_path);
+    strncpy(sock_address.sun_path, socket_path, sizeof(sock_address.sun_path));
+    unlink(socket_path);
 
     sock_fd = socket(AF_UNIX, SOCK_STREAM, 0);
 
-    if (sock_fd == -1) 
-        die("error: could not create socket\n");
+    if (sock_fd == -1)
+        err("Couldn't create the socket.\n");
 
-    bind(sock_fd, (struct sockaddr *) &sock_address, sizeof(sock_address));
-    listen(sock_fd, SOMAXCONN);
+    if (bind(sock_fd, (struct sockaddr *) &sock_address, sizeof(sock_address)) == -1)
+        err("Couldn't bind a name to the socket.\n");
+
+    if (listen(sock_fd, SOMAXCONN) == -1)
+        err("Couldn't listen to the socket.\n");
 
     sel = MAX(sock_fd, dpy_fd) + 1;
 
+    if (fifo_path != NULL) {
+        int fifo_fd = open(fifo_path, O_RDWR | O_NONBLOCK);
+        if (fifo_fd != -1)
+            status_fifo = fdopen(fifo_fd, "w");
+        else
+            warn("Couldn't open status fifo.\n");
+    }
+
     load_settings();
     run_autostart();
-    update_root_dimensions();
-
-    xcb_flush(dpy);
+    ewmh_update_wm_name();
 
     while (running) {
 
+        xcb_flush(dpy);
+
         FD_ZERO(&descriptors);
         FD_SET(sock_fd, &descriptors);
         FD_SET(dpy_fd, &descriptors);
 
-        if (select(sel, &descriptors, NULL, NULL, NULL)) {
+        if (select(sel, &descriptors, NULL, NULL, NULL) > 0) {
+
+            if (FD_ISSET(sock_fd, &descriptors)) {
+                ret_fd = accept(sock_fd, NULL, 0);
+                if (ret_fd > 0 && (n = recv(ret_fd, msg, sizeof(msg), 0)) > 0) {
+                    msg[n] = '\0';
+                    process_message(msg, rsp);
+                    rsplen = strlen(rsp);
+                    if (rsp[rsplen - 1] == '\n')
+                        rsp[--rsplen] = '\0';
+                    send(ret_fd, rsp, rsplen, 0);
+                    close(ret_fd);
+                    rsp[0] = '\0';
+                }
+            }
 
             if (FD_ISSET(dpy_fd, &descriptors)) {
                 while ((event = xcb_poll_for_event(dpy)) != NULL) {
@@ -138,20 +300,20 @@ int main(void)
                 }
             }
 
-            if (FD_ISSET(sock_fd, &descriptors)) {
-                ret_fd = accept(sock_fd, NULL, 0);
-                if (ret_fd > 0 && (nbr = recv(ret_fd, msg, sizeof(msg), 0)) > 0) {
-                    msg[nbr] = '\0';
-                    strcpy(rsp, EMPTY_RESPONSE);
-                    process_message(msg, rsp);
-                    send(ret_fd, rsp, strlen(rsp), 0);
-                    close(ret_fd);
-                }
-            }
         }
+
+        if (xcb_connection_has_error(dpy))
+            err("The server has closed the connection.\n");
     }
 
+    cleanup();
     close(sock_fd);
+    if (status_fifo != NULL)
+        fclose(status_fifo);
+    xcb_ewmh_connection_wipe(ewmh);
+    xcb_destroy_window(dpy, motion_recorder);
+    free(ewmh);
+    xcb_flush(dpy);
     xcb_disconnect(dpy);
-    return 0;
+    return exit_status;
 }