#include <unistd.h>
#include <sys/types.h>
#include <sys/wait.h>
#include <stdio.h>
#include <stdlib.h>
#include <getopt.h>
#include <ctype.h>
#include <pwd.h>
#include <grp.h>
#include <errno.h>
#include <libgen.h>
#include <syslog.h>

#include "util.h"
#include "config.h"
#include "ldap.h"
#include "krb5.h"
#include "kadm.h"
#include "ceo.pb-c.h"

char *prog = NULL;

static char *name = NULL;
static char *userid = NULL;

static struct option opts[] = {
    { NULL, 0, NULL, '\0' },
};

const char *default_lib_dir = "/usr/lib/ceod";
const char *lib_dir;

static void usage() {
    fprintf(stderr, "Usage: %s userid clubname\n", prog);
    exit(2);
}

int addclub(void) {
    struct strbuf preq = STRBUF_INIT;
    struct strbuf pret = STRBUF_INIT;
    char cpath[1024];
    char *cargv[] = { "ceoc", "adduser", NULL };
    int ret = 0;

    if (snprintf(cpath, sizeof(cpath), "%s/ceoc", lib_dir) >= sizeof(cpath))
        fatal("path too long");

    Ceo__AddUser req;
    ceo__add_user__init(&req);

    req.username = userid;
    req.realname = name;
    req.type = CEO__ADD_USER__TYPE__CLUB;

    strbuf_grow(&preq, ceo__add_user__get_packed_size(&req));
    strbuf_setlen(&preq, ceo__add_user__pack(&req, (uint8_t *)preq.buf));

    if (spawnvem(cpath, cargv, environ, &preq, &pret, 0))
        return 1;

    Ceo__AddUserResponse *resp = ceo__add_user_response__unpack(NULL,
                                                               pret.len, (uint8_t *)pret.buf);
    if (!resp)
        fatal("failed to unpack response");

    for (int i = 0; i < resp->n_messages; i++) {
        if (resp->messages[i]->status) {
            ret = -1;
            error("%s", resp->messages[i]->message);
        } else {
            notice("%s", resp->messages[i]->message);
        }
    }

    ceo__add_user_response__free_unpacked(resp, NULL);
    strbuf_release(&preq);
    strbuf_release(&pret);

    return ret;
}

int main(int argc, char *argv[]) {
    int opt;
    int ret;

    prog = xstrdup(basename(argv[0]));
    init_log(prog, LOG_PID, LOG_AUTHPRIV, 1);

    configure();

    while ((opt = getopt_long(argc, argv, "", opts, NULL)) != -1) {
        switch (opt) {
            case '?':
                usage();
                break;
            default:
                fatal("error parsing arguments");
        }
    }

    if (argc - optind != 2 && argc - optind != 3)
        usage();

    userid = argv[optind++];
    name = argv[optind++];

    lib_dir = getenv("CEO_LIB_DIR") ?: default_lib_dir;

    ret = addclub();

    free_config();
    free(prog);

    return ret;
}