swaybar: rename state to bar

1 files changed, 185 insertions, 0 deletions

diff --git a/swaybar/bar.c b/swaybar/bar.c
new file mode 100644
index 00000000..fc01863b
--- /dev/null
+++ b/swaybar/bar.c
@@ -0,0 +1,185 @@
+#include <stdlib.h>
+#include <unistd.h>
+#include <fcntl.h>
+#include <errno.h>
+#include <sys/types.h>
+#include <sys/wait.h>
+
+#include "ipc-client.h"
+#include "list.h"
+#include "log.h"
+#include "ipc.h"
+#include "render.h"
+#include "config.h"
+#include "status_line.h"
+#include "bar.h"
+
+static void bar_init(struct bar *bar) {
+        bar->config = init_config();
+        bar->status = init_status_line();
+        bar->output = malloc(sizeof(struct output));
+        bar->output->window = NULL;
+        bar->output->registry = NULL;
+        bar->output->workspaces = create_list();
+        bar->output->name = NULL;
+}
+
+static void spawn_status_cmd_proc(struct bar *bar) {
+        if (bar->config->status_command) {
+                int pipefd[2];
+                pipe(pipefd);
+                bar->status_command_pid = fork();
+                if (bar->status_command_pid == 0) {
+                        close(pipefd[0]);
+                        dup2(pipefd[1], STDOUT_FILENO);
+                        close(pipefd[1]);
+                        char *const cmd[] = {
+                                "sh",
+                                "-c",
+                                bar->config->status_command,
+                                NULL,
+                        };
+                        execvp(cmd[0], cmd);
+                        return;
+                }
+
+                close(pipefd[1]);
+                bar->status_read_fd = pipefd[0];
+                fcntl(bar->status_read_fd, F_SETFL, O_NONBLOCK);
+        }
+}
+
+
+void bar_setup(struct bar *bar, const char *socket_path, const char *bar_id, int desired_output) {
+        /* initialize bar with default values */
+        bar_init(bar);
+
+        bar->output->registry = registry_poll();
+
+        if (!bar->output->registry->desktop_shell) {
+                sway_abort("swaybar requires the compositor to support the desktop-shell extension.");
+        }
+
+        /* connect to sway ipc */
+        bar->ipc_socketfd = ipc_open_socket(socket_path);
+        bar->ipc_event_socketfd = ipc_open_socket(socket_path);
+
+        ipc_bar_init(bar, desired_output, bar_id);
+
+        struct output_state *output = bar->output->registry->outputs->items[desired_output];
+
+        bar->output->window = window_setup(bar->output->registry, output->width, 30, false);
+        if (!bar->output->window) {
+                sway_abort("Failed to create window.");
+        }
+        desktop_shell_set_panel(bar->output->registry->desktop_shell, output->output, bar->output->window->surface);
+        desktop_shell_set_panel_position(bar->output->registry->desktop_shell, bar->config->position);
+
+        /* set font */
+        bar->output->window->font = bar->config->font;
+
+        /* set window height */
+        set_window_height(bar->output->window, bar->config->height);
+
+        /* spawn status command */
+        spawn_status_cmd_proc(bar);
+}
+
+void bar_run(struct bar *bar) {
+        fd_set readfds;
+        int activity;
+        bool dirty = true;
+
+        while (1) {
+                if (dirty) {
+                        struct output *output = bar->output;
+                        if (window_prerender(output->window) && output->window->cairo) {
+                                render(output, bar->config, bar->status);
+                                window_render(output->window);
+                                if (wl_display_dispatch(output->registry->display) == -1) {
+                                        break;
+                                }
+                        }
+                }
+
+                dirty = false;
+                FD_ZERO(&readfds);
+                FD_SET(bar->ipc_event_socketfd, &readfds);
+                FD_SET(bar->status_read_fd, &readfds);
+
+                activity = select(FD_SETSIZE, &readfds, NULL, NULL, NULL);
+                if (activity < 0) {
+                        sway_log(L_ERROR, "polling failed: %d", errno);
+                }
+
+                if (FD_ISSET(bar->ipc_event_socketfd, &readfds)) {
+                        sway_log(L_DEBUG, "Got IPC event.");
+                        dirty = handle_ipc_event(bar);
+                }
+
+                if (bar->config->status_command && FD_ISSET(bar->status_read_fd, &readfds)) {
+                        sway_log(L_DEBUG, "Got update from status command.");
+                        dirty = handle_status_line(bar);
+                }
+        }
+}
+
+void free_workspaces(list_t *workspaces) {
+        int i;
+        for (i = 0; i < workspaces->length; ++i) {
+                struct workspace *ws = workspaces->items[i];
+                free(ws->name);
+                free(ws);
+        }
+        list_free(workspaces);
+}
+
+static void free_output(struct output *output) {
+        window_teardown(output->window);
+        if (output->registry) {
+                registry_teardown(output->registry);
+        }
+
+        free(output->name);
+
+        if (output->workspaces) {
+                free_workspaces(output->workspaces);
+        }
+
+        free(output);
+}
+
+static void terminate_status_command(pid_t pid) {
+        if (pid) {
+                // terminate status_command process
+                int ret = kill(pid, SIGTERM);
+                if (ret != 0) {
+                        sway_log(L_ERROR, "Unable to terminate status_command [pid: %d]", pid);
+                } else {
+                        int status;
+                        waitpid(pid, &status, 0);
+                }
+        }
+}
+
+void bar_teardown(struct bar *bar) {
+        free_config(bar->config);
+        free_output(bar->output);
+        free_status_line(bar->status);
+
+        /* close sockets/pipes */
+        if (bar->status_read_fd) {
+                close(bar->status_read_fd);
+        }
+
+        if (bar->ipc_socketfd) {
+                close(bar->ipc_socketfd);
+        }
+
+        if (bar->ipc_event_socketfd) {
+                close(bar->ipc_event_socketfd);
+        }
+
+        /* terminate status command process */
+        terminate_status_command(bar->status_command_pid);
+}