path: root/src/cli.c

#include <stdio.h>
#include <string.h>
#include <assert.h>
#include <stdlib.h>

#include "ioline.h"
#include "cli.h"
#include "module.h"
#include "sink.h"
#include "source.h"
#include "client.h"
#include "sinkinput.h"
#include "sourceoutput.h"
#include "tokenizer.h"
#include "strbuf.h"
#include "namereg.h"

struct pa_cli {
    struct pa_core *core;
    struct pa_ioline *line;

    void (*eof_callback)(struct pa_cli *c, void *userdata);
    void *userdata;

    struct pa_client *client;
};

struct command {
    const char *name;
    void (*proc) (struct pa_cli *cli, struct pa_tokenizer*t);
    const char *help;
    unsigned args;
};

static void line_callback(struct pa_ioline *line, const char *s, void *userdata);

static void pa_cli_command_exit(struct pa_cli *c, struct pa_tokenizer *t);
static void pa_cli_command_help(struct pa_cli *c, struct pa_tokenizer *t);
static void pa_cli_command_modules(struct pa_cli *c, struct pa_tokenizer *t);
static void pa_cli_command_clients(struct pa_cli *c, struct pa_tokenizer *t);
static void pa_cli_command_sinks(struct pa_cli *c, struct pa_tokenizer *t);
static void pa_cli_command_sources(struct pa_cli *c, struct pa_tokenizer *t);
static void pa_cli_command_sink_inputs(struct pa_cli *c, struct pa_tokenizer *t);
static void pa_cli_command_source_outputs(struct pa_cli *c, struct pa_tokenizer *t);
static void pa_cli_command_stat(struct pa_cli *c, struct pa_tokenizer *t);
static void pa_cli_command_info(struct pa_cli *c, struct pa_tokenizer *t);
static void pa_cli_command_load(struct pa_cli *c, struct pa_tokenizer *t);
static void pa_cli_command_unload(struct pa_cli *c, struct pa_tokenizer *t);
static void pa_cli_command_sink_volume(struct pa_cli *c, struct pa_tokenizer *t);
static void pa_cli_command_sink_input_volume(struct pa_cli *c, struct pa_tokenizer *t);

static const struct command commands[] = {
    { "exit",                    pa_cli_command_exit,              "Terminate the daemon",         1 },
    { "help",                    pa_cli_command_help,              "Show this help",               1 },
    { "modules",                 pa_cli_command_modules,           "List loaded modules",          1 },
    { "sinks",                   pa_cli_command_sinks,             "List loaded sinks",            1 },
    { "sources",                 pa_cli_command_sources,           "List loaded sources",          1 },
    { "clients",                 pa_cli_command_clients,           "List loaded clients",          1 },
    { "sink_inputs",             pa_cli_command_sink_inputs,       "List sink inputs",             1 },
    { "source_outputs",          pa_cli_command_source_outputs,    "List source outputs",          1 },
    { "stat",                    pa_cli_command_stat,              "Show memory block statistics", 1 },
    { "info",                    pa_cli_command_info,              "Show comprehensive status",    1 },
    { "load",                    pa_cli_command_load,              "Load a module (args: name, arguments)",                     3},
    { "unload",                  pa_cli_command_unload,            "Unload a module (args: index)",                             2},
    { "sink_volume",             pa_cli_command_sink_volume,       "Set the volume of a sink (args: sink, volume)",             3},
    { "sink_input_volume",       pa_cli_command_sink_input_volume, "Set the volume of a sink input (args: sink input, volume)", 3},
    { NULL, NULL, NULL, 0 }
};

static const char prompt[] = ">>> ";

struct pa_cli* pa_cli_new(struct pa_core *core, struct pa_iochannel *io) {
    char cname[256];
    struct pa_cli *c;
    assert(io);

    c = malloc(sizeof(struct pa_cli));
    assert(c);
    c->core = core;
    c->line = pa_ioline_new(io);
    assert(c->line);

    c->userdata = NULL;
    c->eof_callback = NULL;

    pa_iochannel_peer_to_string(io, cname, sizeof(cname));
    c->client = pa_client_new(core, "CLI", cname);
    assert(c->client);
    
    pa_ioline_set_callback(c->line, line_callback, c);
    pa_ioline_puts(c->line, "Welcome to polypaudio! Use \"help\" for usage information.\n");
    pa_ioline_puts(c->line, prompt);
    
    return c;
}

void pa_cli_free(struct pa_cli *c) {
    assert(c);
    pa_ioline_free(c->line);
    pa_client_free(c->client);
    free(c);
}

static void line_callback(struct pa_ioline *line, const char *s, void *userdata) {
    struct pa_cli *c = userdata;
    const char *cs;
    const char delimiter[] = " \t\n\r";
    assert(line && c);

    if (!s) {
        fprintf(stderr, "CLI client exited\n");
        if (c->eof_callback)
            c->eof_callback(c, c->userdata);

        return;
    }

    cs = s+strspn(s, delimiter);
    if (*cs && *cs != '#') {
        const struct command*command;
        int unknown = 1;
        size_t l;
        
        l = strcspn(s, delimiter);

        for (command = commands; command->name; command++) 
            if (strlen(command->name) == l && !strncmp(s, command->name, l)) {
                struct pa_tokenizer *t = pa_tokenizer_new(s, command->args);
                assert(t);
                command->proc(c, t);
                pa_tokenizer_free(t);
                unknown = 0;
                break;
            }

        if (unknown)
            pa_ioline_puts(line, "Unknown command\n");
    }
    
    pa_ioline_puts(c->line, prompt);
}

void pa_cli_set_eof_callback(struct pa_cli *c, void (*cb)(struct pa_cli*c, void *userdata), void *userdata) {
    assert(c && cb);
    c->eof_callback = cb;
    c->userdata = userdata;
}

static void pa_cli_command_exit(struct pa_cli *c, struct pa_tokenizer *t) {
    assert(c && c->core && c->core->mainloop && t);
    c->core->mainloop->quit(c->core->mainloop, 0);
}

static void pa_cli_command_help(struct pa_cli *c, struct pa_tokenizer *t) {
    const struct command*command;
    struct pa_strbuf *pa_strbuf;
    char *p;
    assert(c && t);

    pa_strbuf = pa_strbuf_new();
    assert(pa_strbuf);

    pa_strbuf_puts(pa_strbuf, "Available commands:\n");
    
    for (command = commands; command->name; command++)
        pa_strbuf_printf(pa_strbuf, "    %-20s %s\n", command->name, command->help);

    pa_ioline_puts(c->line, p = pa_strbuf_tostring_free(pa_strbuf));
    free(p);
}

static void pa_cli_command_modules(struct pa_cli *c, struct pa_tokenizer *t) {
    char *s;
    assert(c && t);
    s = pa_module_list_to_string(c->core);
    assert(s);
    pa_ioline_puts(c->line, s);
    free(s);
}

static void pa_cli_command_clients(struct pa_cli *c, struct pa_tokenizer *t) {
    char *s;
    assert(c && t);
    s = pa_client_list_to_string(c->core);
    assert(s);
    pa_ioline_puts(c->line, s);
    free(s);
}

static void pa_cli_command_sinks(struct pa_cli *c, struct pa_tokenizer *t) {
    char *s;
    assert(c && t);
    s = pa_sink_list_to_string(c->core);
    assert(s);
    pa_ioline_puts(c->line, s);
    free(s);
}

static void pa_cli_command_sources(struct pa_cli *c, struct pa_tokenizer *t) {
    char *s;
    assert(c && t);
    s = pa_source_list_to_string(c->core);
    assert(s);
    pa_ioline_puts(c->line, s);
    free(s);
}

static void pa_cli_command_sink_inputs(struct pa_cli *c, struct pa_tokenizer *t) {
    char *s;
    assert(c && t);
    s = pa_sink_input_list_to_string(c->core);
    assert(s);
    pa_ioline_puts(c->line, s);
    free(s);
}

static void pa_cli_command_source_outputs(struct pa_cli *c, struct pa_tokenizer *t) {
    char *s;
    assert(c && t);
    s = pa_source_output_list_to_string(c->core);
    assert(s);
    pa_ioline_puts(c->line, s);
    free(s);
}

static void pa_cli_command_stat(struct pa_cli *c, struct pa_tokenizer *t) {
    char txt[256];
    assert(c && t);
    snprintf(txt, sizeof(txt), "Memory blocks allocated: %u, total size: %u bytes.\n", pa_memblock_count, pa_memblock_total);
    pa_ioline_puts(c->line, txt);
}

static void pa_cli_command_info(struct pa_cli *c, struct pa_tokenizer *t) {
    assert(c && t);
    pa_cli_command_stat(c, t);
    pa_cli_command_modules(c, t);
    pa_cli_command_sources(c, t);
    pa_cli_command_sinks(c, t);
    pa_cli_command_clients(c, t);
    pa_cli_command_sink_inputs(c, t);
    pa_cli_command_source_outputs(c, t);
}

static void pa_cli_command_load(struct pa_cli *c, struct pa_tokenizer *t) {
    struct pa_module *m;
    const char *name;
    char txt[256];
    assert(c && t);

    if (!(name = pa_tokenizer_get(t, 1))) {
        pa_ioline_puts(c->line, "You need to specfiy the module name and optionally arguments.\n");
        return;
    }
    
    if (!(m = pa_module_load(c->core, name,  pa_tokenizer_get(t, 2)))) {
        pa_ioline_puts(c->line, "Module load failed.\n");
        return;
    }

    snprintf(txt, sizeof(txt), "Module successfully loaded, index: %u.\n", m->index);
    pa_ioline_puts(c->line, txt);
}

static void pa_cli_command_unload(struct pa_cli *c, struct pa_tokenizer *t) {
    struct pa_module *m;
    uint32_t index;
    const char *i;
    char *e;
    assert(c && t);

    if (!(i = pa_tokenizer_get(t, 1))) {
        pa_ioline_puts(c->line, "You need to specfiy the module index.\n");
        return;
    }

    index = (uint32_t) strtoul(i, &e, 10);
    if (*e || !(m = pa_idxset_get_by_index(c->core->modules, index))) {
        pa_ioline_puts(c->line, "Invalid module index.\n");
        return;
    }

    pa_module_unload_request(c->core, m);
}


static void pa_cli_command_sink_volume(struct pa_cli *c, struct pa_tokenizer *t) {
    const char *n, *v;
    char *x = NULL;
    struct pa_sink *sink;
    long volume;

    if (!(n = pa_tokenizer_get(t, 1))) {
        pa_ioline_puts(c->line, "You need to specify a sink either by its name or its index.\n");
        return;
    }

    if (!(v = pa_tokenizer_get(t, 2))) {
        pa_ioline_puts(c->line, "You need to specify a volume >= 0. (0 is muted, 0x100 is normal volume)\n");
        return;
    }

    volume = strtol(v, &x, 0);
    if (!x || *x != 0 || volume < 0) {
        pa_ioline_puts(c->line, "Failed to parse volume.\n");
        return;
    }

    if (!(sink = pa_namereg_get(c->core, n, PA_NAMEREG_SINK))) {
        pa_ioline_puts(c->line, "No sink found by this name or index.\n");
        return;
    }

    sink->volume = (uint32_t) volume;
}

static void pa_cli_command_sink_input_volume(struct pa_cli *c, struct pa_tokenizer *t) {
    const char *n, *v;
    char *x = NULL;
    struct pa_sink_input *si;
    long index, volume;

    if (!(n = pa_tokenizer_get(t, 1))) {
        pa_ioline_puts(c->line, "You need to specify a sink input by its index.\n");
        return;
    }

    index = strtol(n, &x, 0);
    if (!x || *x != 0 || index < 0) {
        pa_ioline_puts(c->line, "Failed to parse index.\n");
        return;
    }

    if (!(v = pa_tokenizer_get(t, 2))) {
        pa_ioline_puts(c->line, "You need to specify a volume >= 0. (0 is muted, 0x100 is normal volume)\n");
        return;
    }

    x = NULL;
    volume = strtol(v, &x, 0);
    if (!x || *x != 0 || volume < 0) {
        pa_ioline_puts(c->line, "Failed to parse volume.\n");
        return;
    }

    if (!(si = pa_idxset_get_by_index(c->core->sink_inputs, (uint32_t) index))) {
        pa_ioline_puts(c->line, "No sink input found with this index.\n");
        return;
    }

    si->volume = (uint32_t) volume;
}