#include <stdlib.h>
#include <libssh/callbacks.h>
#include <libssh/server.h>

#include "server_ssh.h"
#include "server.h"

struct server_callbacks potd_ssh_callbacks = {
    .on_listen = ssh_on_listen,
    .on_shutdown = ssh_on_shutdown
};

static void set_default_keys(ssh_bind sshbind, int rsa_already_set,
                             int dsa_already_set, int ecdsa_already_set);


int ssh_init_cb(struct server_ctx *ctx)
{
    ctx->server_cbs = potd_ssh_callbacks;

    ssh_init();
    ssh_data *d = (ssh_data *) calloc(1, sizeof(*d));
    d->sshbind = ssh_bind_new();
    ctx->server_dat.data = d;
    if (!d->sshbind)
        return 1;

    set_default_keys(d->sshbind, 0, 0, 0);
    return 0;
}

int ssh_on_listen(struct server_data *data)
{
    return 0;
}

int ssh_on_shutdown(struct server_data *data)
{
    return 0;
}

static void set_default_keys(ssh_bind sshbind, int rsa_already_set,
                             int dsa_already_set, int ecdsa_already_set)
{
    if (!rsa_already_set) {
        ssh_bind_options_set(sshbind, SSH_BIND_OPTIONS_RSAKEY,
                             "./ssh_host_rsa_key");
    }
    if (!dsa_already_set) {
        ssh_bind_options_set(sshbind, SSH_BIND_OPTIONS_DSAKEY,
                             "./ssh_host_dsa_key");
    }
    if (!ecdsa_already_set) {
        ssh_bind_options_set(sshbind, SSH_BIND_OPTIONS_ECDSAKEY,
                             "./ssh_host_ecdsa_key");
    }
}