Skip to content

Commit

Permalink
Add *almost* wayland support. See #2
Browse files Browse the repository at this point in the history
  • Loading branch information
@fm16191
fm16191 committed Jun 18, 2023
1 parent ad05468 commit 20fa4a7
Show file tree
Hide file tree
Showing 3 changed files with 978 additions and 0 deletions.
3 changes: 3 additions & 0 deletions status.sh
View file
Original file line number Diff line number Diff line change
Expand Up @@ -5,6 +5,9 @@ path=$(dirname "$(realpath $0)")
if [ ! -z $HYPRLAND_INSTANCE_SIGNATURE ]; then
echo "Mode : Wayland/hyprland"
name="hyprland_status"
elif [ $XDG_SESSION_TYPE = "wayland" ]; then
echo "Mode : Wayland";
name="wayland_status"
else
echo "Mode : Xorg";
name="status"
Expand Down
360 changes: 360 additions & 0 deletions wayland_logger.c
View file
Original file line number Diff line number Diff line change
@@ -0,0 +1,360 @@
#define _POSIX_C_SOURCE 200809L
#include "wlr-foreign-toplevel-management-unstable-v1-client-protocol.h"

// This first attempt to run activity-logger on wayland was built on top of
// https://gitlab.freedesktop.org/wlroots/wlroots/-/blob/master/examples/foreign-toplevel.c
// In the future, this file should be able to compile itself.
// For the moment, it can only be compiled by adding it to the wlroots/examples folder
// and then compiling it via meson/ninja.

// Any suggestions, PR are much appreciated !

#include <getopt.h>
#include <stdbool.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <sys/time.h>
#include <time.h>
#include <wayland-client.h>

#define WLR_FOREIGN_TOPLEVEL_MANAGEMENT_VERSION 3

struct timeval time_milisec;
FILE *fd;
static char fd_filepath[1024];
static char buffer[4096];
static char last[4096];

enum toplevel_state_field {
TOPLEVEL_STATE_MAXIMIZED = (1 << 0),
TOPLEVEL_STATE_MINIMIZED = (1 << 1),
TOPLEVEL_STATE_ACTIVATED = (1 << 2),
TOPLEVEL_STATE_FULLSCREEN = (1 << 3),
TOPLEVEL_STATE_INVALID = (1 << 4),
};

static const uint32_t no_parent = (uint32_t)-1;
static struct wl_output *pref_output = NULL;
static uint32_t pref_output_id = UINT32_MAX;

struct toplevel_state {
char *title;
char *app_id;

uint32_t state;
uint32_t parent_id;
};

static void copy_state(struct toplevel_state *current, struct toplevel_state *pending)
{
if (current->title && pending->title) {
free(current->title);
}
if (current->app_id && pending->app_id) {
free(current->app_id);
}

if (pending->title) {
current->title = pending->title;
pending->title = NULL;
}
if (pending->app_id) {
current->app_id = pending->app_id;
pending->app_id = NULL;
}

if (!(pending->state & TOPLEVEL_STATE_INVALID)) {
current->state = pending->state;
}

current->parent_id = pending->parent_id;

pending->state = TOPLEVEL_STATE_INVALID;
}

static uint32_t global_id = 0;
struct toplevel_v1 {
struct wl_list link;
struct zwlr_foreign_toplevel_handle_v1 *zwlr_toplevel;

uint32_t id;
struct toplevel_state current, pending;
};

static void print_toplevel(struct toplevel_v1 *toplevel, bool print_endl)
{
gettimeofday(&time_milisec, NULL);
sprintf(buffer, "[%05u] [%s] %s\n", toplevel->id, toplevel->current.app_id ?: "(nil)",
toplevel->current.title ?: "(nil)");

if (strcmp(last, buffer)) { // if different
fd = fopen(fd_filepath, "a");
fprintf(fd, "[%ld-%03lu] %s", time_milisec.tv_sec, time_milisec.tv_usec % 1000, buffer);
fflush(fd);
fclose(fd);
}
strcpy(last, buffer);
}

static void print_toplevel_state(struct toplevel_v1 *toplevel, bool print_endl)
{
// void
}

static void finish_toplevel_state(struct toplevel_state *state)
{
free(state->title);
free(state->app_id);
}

static void toplevel_handle_title(void *data, struct zwlr_foreign_toplevel_handle_v1 *zwlr_toplevel,
const char *title)
{
struct toplevel_v1 *toplevel = data;
free(toplevel->pending.title);
toplevel->pending.title = strdup(title);
}

static void toplevel_handle_app_id(void *data,
struct zwlr_foreign_toplevel_handle_v1 *zwlr_toplevel,
const char *app_id)
{
struct toplevel_v1 *toplevel = data;
free(toplevel->pending.app_id);
toplevel->pending.app_id = strdup(app_id);
}

static void toplevel_handle_output_enter(void *data,
struct zwlr_foreign_toplevel_handle_v1 *zwlr_toplevel,
struct wl_output *output)
{
struct toplevel_v1 *toplevel = data;
print_toplevel(toplevel, false);
printf(" enter output %u\n", (uint32_t)(size_t)wl_output_get_user_data(output));
}

static void toplevel_handle_output_leave(void *data,
struct zwlr_foreign_toplevel_handle_v1 *zwlr_toplevel,
struct wl_output *output)
{
struct toplevel_v1 *toplevel = data;
print_toplevel(toplevel, false);
printf(" leave output %u\n", (uint32_t)(size_t)wl_output_get_user_data(output));
}

static uint32_t array_to_state(struct wl_array *array)
{
uint32_t state = 0;
uint32_t *entry;
wl_array_for_each(entry, array)
{
if (*entry == ZWLR_FOREIGN_TOPLEVEL_HANDLE_V1_STATE_MAXIMIZED)
state |= TOPLEVEL_STATE_MAXIMIZED;
if (*entry == ZWLR_FOREIGN_TOPLEVEL_HANDLE_V1_STATE_MINIMIZED)
state |= TOPLEVEL_STATE_MINIMIZED;
if (*entry == ZWLR_FOREIGN_TOPLEVEL_HANDLE_V1_STATE_ACTIVATED)
state |= TOPLEVEL_STATE_ACTIVATED;
if (*entry == ZWLR_FOREIGN_TOPLEVEL_HANDLE_V1_STATE_FULLSCREEN)
state |= TOPLEVEL_STATE_FULLSCREEN;
}

return state;
}

static void toplevel_handle_state(void *data, struct zwlr_foreign_toplevel_handle_v1 *zwlr_toplevel,
struct wl_array *state)
{
struct toplevel_v1 *toplevel = data;
toplevel->pending.state = array_to_state(state);
}

static struct zwlr_foreign_toplevel_manager_v1 *toplevel_manager = NULL;
static struct wl_list toplevel_list;

static void toplevel_handle_parent(void *data,
struct zwlr_foreign_toplevel_handle_v1 *zwlr_toplevel,
struct zwlr_foreign_toplevel_handle_v1 *zwlr_parent)
{
struct toplevel_v1 *toplevel = data;
toplevel->pending.parent_id = no_parent;
if (zwlr_parent) {
struct toplevel_v1 *toplevel_tmp;
wl_list_for_each(toplevel_tmp, &toplevel_list, link)
{
if (toplevel_tmp->zwlr_toplevel == zwlr_parent) {
toplevel->pending.parent_id = toplevel_tmp->id;
break;
}
}
if (toplevel->pending.parent_id == no_parent) {
fprintf(stderr, "Cannot find parent toplevel!\n");
}
}
}

static void toplevel_handle_done(void *data, struct zwlr_foreign_toplevel_handle_v1 *zwlr_toplevel)
{
struct toplevel_v1 *toplevel = data;
bool state_changed = toplevel->current.state != toplevel->pending.state;

copy_state(&toplevel->current, &toplevel->pending);

print_toplevel(toplevel, !state_changed);
if (state_changed) {
print_toplevel_state(toplevel, true);
}
}

static void toplevel_handle_closed(void *data,
struct zwlr_foreign_toplevel_handle_v1 *zwlr_toplevel)
{
struct toplevel_v1 *toplevel = data;
print_toplevel(toplevel, false);

zwlr_foreign_toplevel_handle_v1_destroy(zwlr_toplevel);
finish_toplevel_state(&toplevel->current);
finish_toplevel_state(&toplevel->pending);
free(toplevel);
}

static const struct zwlr_foreign_toplevel_handle_v1_listener toplevel_impl = {
.title = toplevel_handle_title,
.app_id = toplevel_handle_app_id,
.output_enter = toplevel_handle_output_enter,
.output_leave = toplevel_handle_output_leave,
.state = toplevel_handle_state,
.done = toplevel_handle_done,
.closed = toplevel_handle_closed,
.parent = toplevel_handle_parent
};

static void
toplevel_manager_handle_toplevel(void *data,
struct zwlr_foreign_toplevel_manager_v1 *toplevel_manager,
struct zwlr_foreign_toplevel_handle_v1 *zwlr_toplevel)
{
struct toplevel_v1 *toplevel = calloc(1, sizeof(struct toplevel_v1));
if (!toplevel) {
fprintf(stderr, "Failed to allocate memory for toplevel\n");
return;
}

toplevel->id = global_id++;
toplevel->zwlr_toplevel = zwlr_toplevel;
toplevel->current.parent_id = no_parent;
toplevel->pending.parent_id = no_parent;
wl_list_insert(&toplevel_list, &toplevel->link);

zwlr_foreign_toplevel_handle_v1_add_listener(zwlr_toplevel, &toplevel_impl, toplevel);
}

static void
toplevel_manager_handle_finished(void *data,
struct zwlr_foreign_toplevel_manager_v1 *toplevel_manager)
{
zwlr_foreign_toplevel_manager_v1_destroy(toplevel_manager);
}

static const struct zwlr_foreign_toplevel_manager_v1_listener toplevel_manager_impl = {
.toplevel = toplevel_manager_handle_toplevel,
.finished = toplevel_manager_handle_finished,
};

struct wl_seat *seat = NULL;
static void handle_global(void *data, struct wl_registry *registry, uint32_t name,
const char *interface, uint32_t version)
{
if (strcmp(interface, wl_output_interface.name) == 0) {
if (name == pref_output_id) {
pref_output = wl_registry_bind(registry, name, &wl_output_interface, version);
}
}
else if (strcmp(interface, zwlr_foreign_toplevel_manager_v1_interface.name) == 0) {
toplevel_manager =
wl_registry_bind(registry, name, &zwlr_foreign_toplevel_manager_v1_interface,
WLR_FOREIGN_TOPLEVEL_MANAGEMENT_VERSION);

wl_list_init(&toplevel_list);
zwlr_foreign_toplevel_manager_v1_add_listener(toplevel_manager, &toplevel_manager_impl,
NULL);
}
else if (strcmp(interface, wl_seat_interface.name) == 0 && seat == NULL) {
seat = wl_registry_bind(registry, name, &wl_seat_interface, version);
}
}

static void handle_global_remove(void *data, struct wl_registry *registry, uint32_t name)
{
// who cares
}

static const struct wl_registry_listener registry_listener = {
.global = handle_global,
.global_remove = handle_global_remove,
};

/* return NULL when id == -1
* exit if the given ID cannot be found in the list of toplevels */
// static struct toplevel_v1 *toplevel_by_id_or_bail(int32_t id)
// {
// if (id == -1) {
// return NULL;
// }

// struct toplevel_v1 *toplevel;
// wl_list_for_each(toplevel, &toplevel_list, link)
// {
// if (toplevel->id == (uint32_t)id) {
// return toplevel;
// }
// }

// fprintf(stderr, "No toplevel with the given id: %d\n", id);
// exit(EXIT_FAILURE);
// }

int main(int argc, char **argv)
{
time_t t = time(NULL);
struct tm *tm = localtime(&t);

char time_sec[12];
strftime(time_sec, 1024, "%d-%m-%Y_%Hh%Mm%S", tm);
sprintf(fd_filepath, "%s.wins", time_sec);

fd = fopen(fd_filepath, "w");
if (fd == NULL) {
printf("File couldn't be opened.\nPlease check folder permissions.\n");
}

char *start_time = asctime(tm);
printf("Starting at %s\n", start_time);
fprintf(fd, "Starting at %s\n", start_time);
fclose(fd);

struct wl_display *display = wl_display_connect(NULL);
if (display == NULL) {
fprintf(stderr, "Failed to create display\n");
return EXIT_FAILURE;
}

struct wl_registry *registry = wl_display_get_registry(display);
wl_registry_add_listener(registry, &registry_listener, NULL);
wl_display_roundtrip(display);

if (toplevel_manager == NULL) {
fprintf(stderr, "wlr-foreign-toplevel not available\n");
return EXIT_FAILURE;
}
// wl_display_roundtrip(display); // load list of toplevels
// wl_display_roundtrip(display); // load toplevel details

wl_display_flush(display);

while (wl_display_dispatch(display) != -1) {
// This space intentionally left blank
}

return EXIT_SUCCESS;
}
Loading

0 comments on commit 20fa4a7

Please sign in to comment.