diff options
Diffstat (limited to 'swaybar/bar.c')
| -rw-r--r-- | swaybar/bar.c | 185 |
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); +} |