venowm.c

#include <stdio.h>
#include <string.h>
#include <unistd.h>
#include <stdlib.h>
#include <wayland-server.h>
#include <signal.h>
#include <wait.h>

// #include <xkbcommon/xkbcommon.h>
#include <linux/input.h>

#include "venowm.h"
#include "split.h"
#include "screen.h"
#include "window.h"
#include "workspace.h"
#include "bindings.h"
#include "libvenowm.h"

// backend_t, needed for keybindings
static backend_t *be;

// venowm global variables
workspace_t *g_workspace;

screen_t **g_screens;
size_t g_screens_size;
size_t g_nscreens;

workspace_t **g_workspaces;
size_t g_workspaces_size;
size_t g_nworkspaces;

void sigchld_handler(int signum){
    logmsg("handled sigchld\n");
    (void)signum;
    int wstatus;
    wait(&wstatus);
    if(WIFEXITED(wstatus)){
        // exited normally, now we can safely check exit code
        int exit_code = WEXITSTATUS(wstatus);
        if(exit_code == 0){
            // nothing went wrong!
            logmsg("handled sigchld 0\n");
            return;
        }
    }
    // TODO: tell the user if something went wrong
}

int compositor_main(){
    int retval = 0;
    int err;


    // set the SIGCHLD handler
    signal(SIGCHLD, sigchld_handler);

    be = backend_new();
    if(!be){;
        return 99;
    }

    INIT_PTR(g_workspaces, g_workspaces_size, g_nworkspaces, 8, err);
    if(err){
        retval = 99;
        goto cu_backend;
    }

    // allocate some workspaces
    for(size_t i = 0; i <= 5; i ++){
        workspace_t *new = workspace_new(be);
        if(!new){
            retval = 99;
            goto cu_workspaces;
        }
        int err;
        APPEND_PTR(g_workspaces, g_workspaces_size, g_nworkspaces, new, err);
        if(err){
            workspace_free(new);
            retval = 99;
            goto cu_workspaces;
        }
    }

    // set first workspace
    g_workspace = g_workspaces[0];

    INIT_PTR(g_screens, g_screens_size, g_nscreens, 4, err);
    if(err){
        retval = 99;
        goto cu_workspaces;
    }

    if(add_bindings(be)) goto cu_workspaces;

    backend_run(be);

    logmsg("post run\n");

//cu_screens:
    // screens are freed by the pre-destroy-screen handler
    FREE_PTR(g_screens, g_screens_size, g_nscreens);
cu_workspaces:
    // but we have to manually free workspaces
    for(size_t i = 0; i < g_nworkspaces; i++){
        workspace_free(g_workspaces[i]);
    }
    FREE_PTR(g_workspaces, g_workspaces_size, g_nworkspaces);
cu_backend:
    backend_free(be);
    logmsg("exiting from main: %d\n", retval);
    return retval;
}

// function pointer from libvenowm_t
typedef int (*libvenowm_func_t)(struct venowm*, bool);

int command_main(libvenowm_func_t func){
    struct venowm *v = venowm_create();
    if(!v){
        fprintf(stderr, "failed to create venowm client\n");
        return 1;
    }

    int ret = venowm_connect(v, NULL);
    if(ret < 0){
        fprintf(stderr, "%s\n", venowm_errmsg(v));
        return 1;
    }

    ret = func(v, true);
    if(ret < 0){
        fprintf(stderr, "%s\n", venowm_errmsg(v));
        return 1;
    }

    venowm_destroy(v);

    return 0;
}

int launch_main(int argc, char **argv){
    struct venowm *v = venowm_create();
    if(!v){
        fprintf(stderr, "failed to create venowm client\n");
        return 1;
    }

    int ret = venowm_connect(v, NULL);
    if(ret < 0){
        fprintf(stderr, "%s\n", venowm_errmsg(v));
        return 1;
    }

    ret = venowm_launch(v, argc, argv);
    if(ret < 0){
        fprintf(stderr, "%s\n", venowm_errmsg(v));
        return 1;
    }

    venowm_destroy(v);

    return 0;
}

int main(int argc, char **argv){
    if(argc < 2){
        return compositor_main();
    }
    if(strcmp(argv[1], "focus-up") == 0){
        return command_main(venowm_focus_up);
    }
    if(strcmp(argv[1], "focus-down") == 0){
        return command_main(venowm_focus_down);
    }
    if(strcmp(argv[1], "focus-left") == 0){
        return command_main(venowm_focus_left);
    }
    if(strcmp(argv[1], "focus-right") == 0){
        return command_main(venowm_focus_right);
    }
    if(strcmp(argv[1], "launch") == 0){
        if(argc > 2){
            return launch_main(argc - 2, &argv[2]);
        }
    }
    fprintf(stderr,
        "usage: venowm\n"
        "usage: venowm focus_up\n"
        "usage: venowm focus_down\n"
        "usage: venowm focus_left\n"
        "usage: venowm focus_right\n"
        "usage: venowm launch ...\n"
    );
    return 1;
}