/*-
 * SPDX-License-Identifier: BSD-2-Clause
 *
 * Copyright (c) 2023-2024 Chelsio Communications, Inc.
 * Written by: John Baldwin <jhb@FreeBSD.org>
 */

#include <err.h>
#include <libnvmf.h>
#include <stdlib.h>
#include <sysexits.h>
#include <unistd.h>

#include "nvmecontrol.h"

static struct options {
        const char *dev;
} opt = {
        .dev = NULL
};

static const struct args args[] = {
        { arg_string, &opt.dev, "controller-id|namespace-id|SubNQN" },
        { arg_none, NULL, NULL },
};

static void
disconnect(const struct cmd *f, int argc, char *argv[])
{
        int     error, fd;
        char    *path;

        if (arg_parse(argc, argv, f))
                return;
        if (nvmf_nqn_valid(opt.dev)) {
                error = nvmf_disconnect_host(opt.dev);
                if (error != 0)
                        errc(EX_IOERR, error, "failed to disconnect from %s",
                            opt.dev);
        } else {
                open_dev(opt.dev, &fd, 1, 1);
                get_nsid(fd, &path, NULL);
                close(fd);

                error = nvmf_disconnect_host(path);
                if (error != 0)
                        errc(EX_IOERR, error, "failed to disconnect from %s",
                            path);
        }

        exit(0);
}

static void
disconnect_all(const struct cmd *f __unused, int argc __unused,
    char *argv[] __unused)
{
        int     error;

        error = nvmf_disconnect_all();
        if (error != 0)
                errc(EX_IOERR, error,
                    "failed to disconnect from remote controllers");

        exit(0);
}

static struct cmd disconnect_cmd = {
        .name = "disconnect",
        .fn = disconnect,
        .descr = "Disconnect from a fabrics controller",
        .args = args,
};

static struct cmd disconnect_all_cmd = {
        .name = "disconnect-all",
        .fn = disconnect_all,
        .descr = "Disconnect from all fabrics controllers",
};

CMD_COMMAND(disconnect_cmd);
CMD_COMMAND(disconnect_all_cmd);