#include <sys/ipc.h>
#include <sys/msg.h>
#include <string.h>
#include <stdio.h>
#include <stdlib.h>
#include <stdbool.h>

#include "mangoapp_proto.h"

static void help_and_quit() {
    fprintf(stderr, "Usage: mangohudctl [set|toggle] attribute [value]\n");
    fprintf(stderr, "       mangohudctl reload-cfg\n");
    fprintf(stderr, "Attributes:\n");
    fprintf(stderr, "   no_display      hides or shows hud\n");
    fprintf(stderr, "   log_session     handles logging status\n");
    fprintf(stderr, "   reload_config   reloads the config\n");
    fprintf(stderr, "Accepted values:\n");
    fprintf(stderr, "   true\n");
    fprintf(stderr, "   false\n");
    fprintf(stderr, "   1\n");
    fprintf(stderr, "   0\n");
    exit(1);
}

static bool str_to_bool(const char *value)
{
    if (strcasecmp(value, "true") == 0 || strcmp(value, "1") == 0)
        return true;
    else if (strcasecmp(value, "false") == 0 || strcmp(value, "0") == 0)
        return false;

    /* invalid boolean, display a nice error message saying that */
    fprintf(stderr, "The value '%s' is not an accepted boolean. Use 0/1 or true/false\n", value);
    exit(1);
}

int main(int argc, char *argv[])
{
    /* Set up message queue */
    int key = ftok("mangoapp", 65);
    int msgid = msgget(key, 0666 | IPC_CREAT);
    /* Create the message that we will send to mangohud */
    struct mangoapp_ctrl_msgid1_v1 ctrl_msg = {
        .hdr.msg_type = 2,
        .hdr.ctrl_msg_type = 1,
        .hdr.version = 1,
    };
    uint8_t value;

    if (argc <= 2)
        help_and_quit();

    if (strcmp(argv[1], "set") == 0) {
        if (argc != 4)
            help_and_quit();

        value = str_to_bool(argv[3]) ? 1 : 2;
    } else if (strcmp(argv[1], "toggle") == 0) {
        if (argc != 3)
            help_and_quit();

        value = 3;
    } else {
        help_and_quit();
    }

    if (strcmp(argv[2], "no_display") == 0)
        ctrl_msg.no_display = value;
    else if (strcmp(argv[2], "log_session") == 0)
        ctrl_msg.log_session = value;
    else if (strcmp(argv[2], "reload_config") == 0)
        ctrl_msg.reload_config = value;
    else
        help_and_quit();

    msgsnd(msgid, &ctrl_msg, sizeof(struct mangoapp_ctrl_msgid1_v1), IPC_NOWAIT);

    return 0;
}