/*
 *     Copyright 2017 Couchbase, Inc.
 *
 *   Licensed under the Apache License, Version 2.0 (the "License");
 *   you may not use this file except in compliance with the License.
 *   You may obtain a copy of the License at
 *
 *       http://www.apache.org/licenses/LICENSE-2.0
 *
 *   Unless required by applicable law or agreed to in writing, software
 *   distributed under the License is distributed on an "AS IS" BASIS,
 *   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 *   See the License for the specific language governing permissions and
 *   limitations under the License.
 */

#define LCB_NO_DEPR_CXX_CTORS

#include "common/my_inttypes.h"
#include "config.h"
#include <sys/types.h>
#include <libcouchbase/couchbase.h>
#include <libcouchbase/api3.h>
#include <iostream>
#include <map>
#include <cassert>
#include <cstdio>
#include <cerrno>
#include <stdexcept>
#include <sstream>
#include "common/options.h"
#include "common/histogram.h"

#include "linenoise/linenoise.h"

static std::string get_resp_key(const lcb_RESPBASE *resp)
{
    if (!resp->nkey) {
        return "";
    }
    return std::string((const char *)resp->key, resp->nkey);
}

extern "C" {
void subdoc_callback(lcb_t, int cbtype, const lcb_RESPBASE *rb)
{
    const lcb_RESPSUBDOC *resp = (const lcb_RESPSUBDOC *)rb;
    lcb_SDENTRY cur;
    std::string key = get_resp_key(rb);

    if (rb->rc == LCB_SUCCESS || rb->rc == LCB_SUBDOC_MULTI_FAILURE) {
        fprintf(stderr, "%-20s CAS=0x%" PRIx64 "\n", key.c_str(), resp->cas);
    } else {
        fprintf(stderr, "%-20s %s (0x%x)\n", key.c_str(), lcb_strerror(NULL, rb->rc), rb->rc);
        const char *ctx = lcb_resp_get_error_context(cbtype, rb);
        if (ctx != NULL) {
            fprintf(stderr, "%-20s %s\n", "", ctx);
        }
        const char *ref = lcb_resp_get_error_ref(cbtype, rb);
        if (ref != NULL) {
            fprintf(stderr, "%-20s Ref: %s\n", "", ref);
        }
    }
    size_t vii = 0, oix = 0;
    while (lcb_sdresult_next(resp, &cur, &vii)) {
        int index = oix++;

        if (cbtype == LCB_CALLBACK_SDMUTATE) {
            index = cur.index;
        }
        printf("%d. Size=%lu, RC=0x%02x %s\n", index, (unsigned long)cur.nvalue, cur.status,
               lcb_strerror(NULL, cur.status));
        fflush(stdout);
        if (cur.nvalue > 0) {
            fwrite(cur.value, 1, cur.nvalue, stdout);
            printf("\n");
            fflush(stdout);
        }
    }
}
}

#define CBCSUBDOC_HISTORY_FILENAME ".cbcsubdoc_history"

using namespace cbc;
using namespace cliopts;

static void do_or_die(lcb_error_t rc, std::string msg = "")
{
    if (rc != LCB_SUCCESS) {
        std::stringstream ss;
        if (!msg.empty()) {
            ss << msg << ". ";
        }
        ss << "(0x" << std::hex << rc << ") " << lcb_strerror(NULL, rc);
        throw std::runtime_error(ss.str());
    }
}

static lcb_t instance = NULL;
static Histogram hg;

class Configuration
{
  public:
    Configuration()
    {
    }

    ~Configuration()
    {
    }

    void addToParser(Parser &parser)
    {
        m_params.addToParser(parser);
    }

    void processOptions()
    {
    }

    void fillCropts(lcb_create_st &opts)
    {
        m_params.fillCropts(opts);
    }
    lcb_error_t doCtls()
    {
        return m_params.doCtls(instance);
    }
    bool useTimings()
    {
        return m_params.useTimings();
    }
    bool shouldDump()
    {
        return m_params.shouldDump();
    }

  private:
    ConnParams m_params;
};

static Configuration config;

static const char *handlers_sorted[] = {"help",
                                        "get",
                                        "set",
                                        "exists",
                                        "remove",
                                        "replace",
                                        "array-insert",
                                        "array-add-first",
                                        "array-add-last",
                                        "array-add-unique",
                                        "dict-add",
                                        "dict-upsert",
                                        "counter",
                                        "size",
                                        NULL};

static void command_completion(const char *buf, linenoiseCompletions *lc)
{
    size_t nbuf = strlen(buf);
    for (const char **cur = handlers_sorted; *cur; cur++) {
        if (memcmp(buf, *cur, nbuf) == 0) {
            linenoiseAddCompletion(lc, *cur);
        }
    }
}

namespace subdoc
{
class Handler;
}

static std::map< std::string, subdoc::Handler * > handlers;

namespace subdoc
{
#define HANDLER_DESCRIPTION(s)                                                                                         \
    const char *description() const                                                                                    \
    {                                                                                                                  \
        return s;                                                                                                      \
    }
#define HANDLER_USAGE(s)                                                                                               \
    const char *usagestr() const                                                                                       \
    {                                                                                                                  \
        return s;                                                                                                      \
    }

class Handler
{
  public:
    Handler(const char *name) : parser(name)
    {
        if (name != NULL) {
            cmdname = name;
        }
        parser.default_settings.error_noexit = 1;
        parser.default_settings.help_noexit = 1;
    }

    virtual ~Handler()
    {
    }
    virtual const char *description() const
    {
        return NULL;
    }
    virtual const char *usagestr() const
    {
        return NULL;
    }
    void execute(int argc, char **argv)
    {
        parser.reset();
        parser.default_settings.argstring = usagestr();
        parser.default_settings.shortdesc = description();
        addOptions();
        parser.parse(argc, argv, true);
        run();
    }

  protected:
    virtual const std::string &getLoneArg(bool required = false)
    {
        static std::string empty("");

        const std::vector< std::string > &args = parser.getRestArgs();
        if (args.empty() || args.size() != 1) {
            if (required) {
                throw BadArg("Command requires single argument");
            }
            return empty;
        }
        return args[0];
    }

    virtual const std::string &getRequiredArg()
    {
        return getLoneArg(true);
    }

    virtual void addOptions()
    {
    }

    virtual void run() = 0;

    void splitNameValue(std::string &arg, const char **name, size_t *nname, const char **value, size_t *nvalue)
    {
        size_t sep = arg.find("=");
        if (sep == std::string::npos) {
            throw BadArg("Name and value have to be separated with '='");
        }

        const char *k = arg.c_str();
        size_t nk = sep;
        for (size_t j = nk - 1; j > 0; j--, nk--) {
            if (k[j] != ' ' && k[j] != '\t') {
                break;
            }
        }
        if (nk == 0) {
            throw BadArg("Name cannot be empty");
        }

        *name = k;
        *nname = nk;
        *value = arg.c_str() + sep + 1;
        *nvalue = arg.size() - sep - 1;
    }

    cliopts::Parser parser;
    std::string cmdname;
};

class LookupHandler : public Handler
{
  public:
    HANDLER_USAGE("[OPTIONS...] KEY...")

    LookupHandler(const char *name, lcb_SUBDOCOP opcode, const char *description_)
        : Handler(name), m_opcode(opcode), m_description(description_), o_paths("path"), o_xattrs("xattr"),
          o_deleted("deleted")
    {
        o_paths.abbrev('p').argdesc("PATH").description("JSON path in the document");
        o_xattrs.abbrev('x').argdesc("PATH").description("Access XATTR path (exentnded attributes)");
        o_deleted.abbrev('d').description("Access XATTR attributes of deleted documents");
    }

    const char *description() const
    {
        return m_description;
    }

  protected:
    void addOptions()
    {
        Handler::addOptions();
        parser.addOption(o_paths.reset());
        parser.addOption(o_xattrs.reset());
        parser.addOption(o_deleted.reset());
    }

    void run()
    {
        lcb_error_t err;

        const std::vector< std::string > &keys = parser.getRestArgs();
        if (keys.empty()) {
            throw BadArg("At least one key has to be specified");
        }
        std::vector< std::string > paths = o_paths.result();
        std::vector< std::string > xattrs = o_xattrs.result();

        if (m_opcode != LCB_SDCMD_GET) {
            if (paths.empty() && xattrs.empty()) {
                throw BadArg("At least one path has to be specified");
            }
        }

        lcb_sched_enter(instance);
        for (size_t ii = 0; ii < keys.size(); ++ii) {
            const std::string &key = keys[ii];
            lcb_CMDSUBDOC cmd = {0};

            LCB_KREQ_SIMPLE(&cmd.key, key.c_str(), key.size());
            cmd.nspecs = 0;
            std::vector< lcb_SDSPEC > specs;

            if (o_xattrs.passed()) {
                for (size_t i = 0; i < xattrs.size(); i++) {
                    lcb_SDSPEC spec = lcb_SDSPEC();
                    spec.sdcmd = m_opcode;
                    spec.options |= LCB_SDSPEC_F_XATTRPATH;
                    if (o_deleted.passed()) {
                        spec.options |= LCB_SDSPEC_F_XATTR_DELETED_OK;
                    }
                    LCB_SDSPEC_SET_PATH(&spec, xattrs[i].c_str(), xattrs[i].size());
                    specs.push_back(spec);
                }
            }
            if (o_paths.passed()) {
                for (size_t i = 0; i < paths.size(); i++) {
                    lcb_SDSPEC spec = lcb_SDSPEC();
                    spec.sdcmd = m_opcode;
                    LCB_SDSPEC_SET_PATH(&spec, paths[i].c_str(), paths[i].size());
                    specs.push_back(spec);
                }
            } else if (m_opcode == LCB_SDCMD_GET) {
                lcb_SDSPEC spec = lcb_SDSPEC();
                spec.sdcmd = LCB_SDCMD_GET_FULLDOC;
                specs.push_back(spec);
            }
            cmd.specs = specs.data();
            cmd.nspecs = specs.size();
            err = lcb_subdoc3(instance, this, &cmd);
            if (err != LCB_SUCCESS) {
                throw LcbError(err, "Failed to schedule " + cmdname + " command");
            }
        }
        lcb_sched_leave(instance);
        err = lcb_wait(instance);
        if (err != LCB_SUCCESS) {
            throw LcbError(err, "Failed to execute " + cmdname + " command");
        }
    }

  protected:
    lcb_SUBDOCOP m_opcode;
    const char *m_description;

    cliopts::ListOption o_paths;
    cliopts::ListOption o_xattrs;
    cliopts::BoolOption o_deleted;
};

class RemoveHandler : public Handler
{
  public:
    HANDLER_DESCRIPTION("Remove path in the item on the server")
    HANDLER_USAGE("[OPTIONS...] KEY...")

    RemoveHandler(const char *name = "remove") : Handler(name), o_paths("path"), o_xattrs("xattr")
    {
        o_paths.abbrev('p').argdesc("PATH").description(
            "JSON path in the document. When skipped, the operation applied to full document.");
        o_xattrs.abbrev('x').argdesc("PATH").description("Access XATTR path (exentnded attributes)");
    }

  protected:
    void addOptions()
    {
        Handler::addOptions();
        parser.addOption(o_paths.reset());
        parser.addOption(o_xattrs.reset());
    }

    void run()
    {
        lcb_error_t err;

        const std::vector< std::string > &keys = parser.getRestArgs();
        if (keys.empty()) {
            throw BadArg("At least one key has to be specified");
        }
        std::vector< std::string > paths = o_paths.result();
        std::vector< std::string > xattrs = o_xattrs.result();

        lcb_sched_enter(instance);
        for (size_t ii = 0; ii < keys.size(); ++ii) {
            const std::string &key = keys[ii];
            lcb_CMDSUBDOC cmd = {0};

            LCB_KREQ_SIMPLE(&cmd.key, key.c_str(), key.size());
            std::vector< lcb_SDSPEC > specs;

            if (o_xattrs.passed()) {
                for (size_t i = 0; i < xattrs.size(); i++) {
                    lcb_SDSPEC spec = lcb_SDSPEC();
                    spec.sdcmd = LCB_SDCMD_REMOVE;
                    spec.options |= LCB_SDSPEC_F_XATTRPATH;
                    LCB_SDSPEC_SET_PATH(&spec, xattrs[i].c_str(), xattrs[i].size());
                    specs.push_back(spec);
                }
            }
            if (o_paths.passed()) {
                for (size_t i = 0; i < paths.size(); i++) {
                    lcb_SDSPEC spec = lcb_SDSPEC();
                    spec.sdcmd = LCB_SDCMD_REMOVE;
                    LCB_SDSPEC_SET_PATH(&spec, paths[i].c_str(), paths[i].size());
                    specs.push_back(spec);
                }
            } else {
                lcb_SDSPEC spec = lcb_SDSPEC();
                spec.sdcmd = LCB_SDCMD_REMOVE_FULLDOC;
                specs.push_back(spec);
            }
            cmd.specs = specs.data();
            cmd.nspecs = specs.size();
            err = lcb_subdoc3(instance, this, &cmd);
            if (err != LCB_SUCCESS) {
                throw LcbError(err, "Failed to schedule remove command");
            }
        }
        lcb_sched_leave(instance);
        err = lcb_wait(instance);
        if (err != LCB_SUCCESS) {
            throw LcbError(err, "Failed to execute remove");
        }
    }

  protected:
    cliopts::ListOption o_paths;
    cliopts::ListOption o_xattrs;
};

class UpsertHandler : public Handler
{
  public:
    HANDLER_DESCRIPTION("Store document on the server")
    HANDLER_USAGE("[OPTIONS...] KEY VALUE")

    UpsertHandler(const char *name = "upsert") : Handler(name), o_xattrs("xattr"), o_expiry("expiry")
    {
        o_xattrs.abbrev('x').argdesc("PATH=VALUE").description("Store XATTR path (exentnded attributes)");
        o_expiry.abbrev('e').argdesc("TIME").description(
            "Expiration time in seconds. Relative (up to 30 days) or absolute (as Unix timestamp)");
    }

  protected:
    void addOptions()
    {
        Handler::addOptions();
        parser.addOption(o_xattrs.reset());
        parser.addOption(o_expiry.reset());
    }

    void run()
    {
        lcb_error_t err;

        const std::vector< std::string > &args = parser.getRestArgs();
        if (args.size() != 2) {
            throw BadArg("Exactly two arguments required: KEY and VALUE");
        }
        // currently it is not possible to upsert document without XATTRs
        // so lets allocate "_cbc" object with some useful stuff
        std::string ver = "\"" LCB_CLIENT_ID "\"";
        std::string path = "_cbc.version";

        std::string key = args[0];
        std::string value = args[1];
        std::vector< std::pair< std::string, std::string > > xattrs = o_xattrs.result();

        std::vector< lcb_SDSPEC > specs;
        lcb_CMDSUBDOC cmd = {0};

        LCB_KREQ_SIMPLE(&cmd.key, key.c_str(), key.size());
        cmd.cmdflags = LCB_CMDSUBDOC_F_UPSERT_DOC;

        if (o_xattrs.passed()) {
            for (size_t i = 0; i < xattrs.size(); i++) {
                lcb_SDSPEC spec = lcb_SDSPEC();
                spec.sdcmd = LCB_SDCMD_DICT_UPSERT;
                spec.options = LCB_SDSPEC_F_XATTRPATH | LCB_SDSPEC_F_MKINTERMEDIATES;
                LCB_SDSPEC_SET_PATH(&spec, xattrs[i].first.c_str(), xattrs[i].first.size());
                LCB_SDSPEC_SET_VALUE(&spec, xattrs[i].second.c_str(), xattrs[i].second.size());
                specs.push_back(spec);
            }
        } else {
            lcb_SDSPEC spec = lcb_SDSPEC();
            spec.sdcmd = LCB_SDCMD_DICT_UPSERT;
            spec.options = LCB_SDSPEC_F_XATTRPATH | LCB_SDSPEC_F_MKINTERMEDIATES;
            LCB_SDSPEC_SET_PATH(&spec, path.c_str(), path.size());
            LCB_SDSPEC_SET_VALUE(&spec, ver.c_str(), ver.size());
            specs.push_back(spec);
        }
        {
            lcb_SDSPEC spec = lcb_SDSPEC();
            spec.sdcmd = LCB_SDCMD_SET_FULLDOC;
            LCB_SDSPEC_SET_VALUE(&spec, value.c_str(), value.size());
            specs.push_back(spec);
        }
        cmd.specs = specs.data();
        cmd.nspecs = specs.size();
        if (o_expiry.passed()) {
            cmd.exptime = o_expiry.result();
        }

        lcb_sched_enter(instance);
        err = lcb_subdoc3(instance, this, &cmd);
        if (err != LCB_SUCCESS) {
            throw LcbError(err, "Failed to schedule upsert command");
        }
        lcb_sched_leave(instance);

        err = lcb_wait(instance);
        if (err != LCB_SUCCESS) {
            throw LcbError(err, "Failed to execute upsert");
        }
    }

  protected:
    cliopts::PairListOption o_xattrs;
    cliopts::UIntOption o_expiry;
};

class MutationHandler : public Handler
{
  public:
    HANDLER_USAGE("[OPTIONS...] KEY...")

    MutationHandler(const char *name, lcb_SUBDOCOP opcode, const char *description_, bool enable_intermediates = true)
        : Handler(name), m_opcode(opcode), m_description(description_), o_paths("path"), o_xattrs("xattr"),
          o_expiry("expiry"), o_intermediates("intermediates"), o_upsert("upsert"),
          m_enable_intermediates(enable_intermediates)
    {
        o_paths.abbrev('p').argdesc("PATH=VALUE").description("JSON path in the document");
        o_xattrs.abbrev('x').argdesc("PATH=VALUE").description("XATTR path (exentnded attributes)");
        o_expiry.abbrev('e').argdesc("TIME").description(
            "Expiration time in seconds. Relative (up to 30 days) or absolute (as Unix timestamp)");
        o_intermediates.abbrev('i').description("Create intermediate paths");
        o_upsert.abbrev('u').description("Create document if it doesn't exist");
    }

    const char *description() const
    {
        return m_description;
    }

  protected:
    void addOptions()
    {
        Handler::addOptions();
        parser.addOption(o_xattrs.reset());
        parser.addOption(o_paths.reset());
        parser.addOption(o_expiry.reset());
        parser.addOption(o_upsert.reset());
        if (m_enable_intermediates) {
            parser.addOption(o_intermediates.reset());
        }
    }

    void run()
    {
        lcb_error_t err;

        const std::vector< std::string > &keys = parser.getRestArgs();
        if (keys.empty()) {
            throw BadArg("At least one key has to be specified");
        }
        std::vector< std::pair< std::string, std::string > > paths = o_paths.result();
        std::vector< std::pair< std::string, std::string > > xattrs = o_xattrs.result();

        if (xattrs.empty() && paths.empty()) {
            throw BadArg("At least one path has to be specified");
        }

        lcb_sched_enter(instance);

        for (size_t ii = 0; ii < keys.size(); ++ii) {
            const std::string &key = keys[ii];
            lcb_CMDSUBDOC cmd = {0};
            std::vector< lcb_SDSPEC > specs;

            LCB_KREQ_SIMPLE(&cmd.key, key.c_str(), key.size());
            if (o_upsert.passed()) {
                cmd.cmdflags = LCB_CMDSUBDOC_F_UPSERT_DOC;
            }
            if (o_xattrs.passed()) {
                for (size_t i = 0; i < xattrs.size(); i++) {
                    lcb_SDSPEC spec = lcb_SDSPEC();
                    spec.sdcmd = m_opcode;
                    spec.options = LCB_SDSPEC_F_XATTRPATH;
                    if (o_intermediates.passed()) {
                        spec.options |= LCB_SDSPEC_F_MKINTERMEDIATES;
                    }
                    LCB_SDSPEC_SET_PATH(&spec, xattrs[i].first.c_str(), xattrs[i].first.size());
                    LCB_SDSPEC_SET_VALUE(&spec, xattrs[i].second.c_str(), xattrs[i].second.size());
                    specs.push_back(spec);
                }
            }
            if (o_paths.passed()) {
                for (size_t i = 0; i < paths.size(); i++) {
                    lcb_SDSPEC spec = lcb_SDSPEC();
                    spec.sdcmd = m_opcode;
                    spec.options = 0;
                    if (o_intermediates.passed()) {
                        spec.options |= LCB_SDSPEC_F_MKINTERMEDIATES;
                    }
                    LCB_SDSPEC_SET_PATH(&spec, paths[i].first.c_str(), paths[i].first.size());
                    LCB_SDSPEC_SET_VALUE(&spec, paths[i].second.c_str(), paths[i].second.size());
                    specs.push_back(spec);
                }
            }
            cmd.specs = specs.data();
            cmd.nspecs = specs.size();
            if (o_expiry.passed()) {
                cmd.exptime = o_expiry.result();
            }
            err = lcb_subdoc3(instance, this, &cmd);
            if (err != LCB_SUCCESS) {
                throw LcbError(err, "Failed to schedule " + cmdname + " command");
            }
        }
        lcb_sched_leave(instance);

        err = lcb_wait(instance);
        if (err != LCB_SUCCESS) {
            throw LcbError(err, "Failed to execute " + cmdname + " command");
        }
    }

  protected:
    lcb_SUBDOCOP m_opcode;
    const char *m_description;

    cliopts::PairListOption o_paths;
    cliopts::PairListOption o_xattrs;
    cliopts::UIntOption o_expiry;
    cliopts::BoolOption o_intermediates;
    cliopts::BoolOption o_upsert;

    bool m_enable_intermediates;
};

class HelpHandler : public Handler
{
  public:
    HANDLER_DESCRIPTION("Show help")
    HelpHandler() : Handler("help")
    {
    }

  protected:
    void run()
    {
        fprintf(stderr, "Usage: <command> [options]\n");
        fprintf(stderr, "command may be:\n");
        for (const char **cur = handlers_sorted; *cur; cur++) {
            const Handler *handler = handlers[*cur];
            fprintf(stderr, "   %-20s", *cur);
            fprintf(stderr, "%s\n", handler->description());
        }
    }
};
}

static void setupHandlers()
{
    handlers["help"] = new subdoc::HelpHandler();
    handlers["get"] = new subdoc::LookupHandler("get", LCB_SDCMD_GET, "Retrieve path from the item on the server");
    handlers["exists"] =
        new subdoc::LookupHandler("exists", LCB_SDCMD_EXISTS, "Check if path exists in the item on the server");
    handlers["exist"] = handlers["exists"];
    handlers["remove"] = new subdoc::RemoveHandler();
    handlers["delete"] = handlers["remove"];
    handlers["upsert"] = new subdoc::UpsertHandler();
    handlers["set"] = handlers["upsert"];
    handlers["dict-upsert"] =
        new subdoc::MutationHandler("dict-upsert", LCB_SDCMD_DICT_UPSERT, "Unconditionally set the value at the path");
    handlers["dict-add"] = new subdoc::MutationHandler(
        "dict-add", LCB_SDCMD_DICT_ADD, "Add the value at the given path, if the given path does not exist");
    handlers["replace"] =
        new subdoc::MutationHandler("replace", LCB_SDCMD_REPLACE, "Replace the value at the specified path", false);
    handlers["array-add-first"] =
        new subdoc::MutationHandler("array-add-first", LCB_SDCMD_ARRAY_ADD_FIRST, "Prepend the value(s) to the array");
    handlers["array-add-last"] =
        new subdoc::MutationHandler("array-add-last", LCB_SDCMD_ARRAY_ADD_LAST, "Append the value(s) to the array");
    handlers["array-add-unique"] = new subdoc::MutationHandler(
        "array-add-unique", LCB_SDCMD_ARRAY_ADD_UNIQUE,
        "Add the value to the array indicated by the path, if the value is not already in the array");
    handlers["array-insert"] = new subdoc::MutationHandler(
        "array-insert", LCB_SDCMD_ARRAY_INSERT,
        "Add the value at the given array index. Path must include index, e.g. `my.list[4]`");
    handlers["counter"] = new subdoc::MutationHandler(
        "counter", LCB_SDCMD_COUNTER,
        "Increment or decrement an existing numeric path. The value must be 64-bit integer");
    handlers["size"] = new subdoc::LookupHandler("size", LCB_SDCMD_GET_COUNT,
                                                 "Count the number of elements in an array or dictionary");
    handlers["get-count"] = handlers["size"];
}

static void cleanup()
{
    std::map< std::string, subdoc::Handler * >::iterator iter = handlers.begin();

    handlers["exists"] = NULL;
    handlers["delete"] = NULL;
    handlers["set"] = NULL;
    handlers["get-count"] = NULL;

    for (; iter != handlers.end(); iter++) {
        if (iter->second) {
            delete iter->second;
        }
    }

    if (instance) {
        if (config.shouldDump()) {
            lcb_dump(instance, stderr, LCB_DUMP_ALL);
        }
        if (config.useTimings()) {
            hg.write();
        }
        if (instance) {
            lcb_destroy(instance);
        }
    }
}

static void real_main(int argc, char **argv)
{
    std::string history_path = ConnParams::getUserHome() + CBCSUBDOC_HISTORY_FILENAME;
    Parser parser;

    config.addToParser(parser);
    parser.parse(argc, argv);
    config.processOptions();

    lcb_create_st cropts;
    memset(&cropts, 0, sizeof cropts);
    config.fillCropts(cropts);
    do_or_die(lcb_create(&instance, &cropts), "Failed to create connection");
    config.doCtls();
    do_or_die(lcb_connect(instance), "Failed to connect to cluster");
    do_or_die(lcb_wait(instance), "Failed to wait for connection bootstrap");
    do_or_die(lcb_get_bootstrap_status(instance), "Failed to bootstrap");
    if (config.useTimings()) {
        hg.install(instance, stdout);
    }
    setupHandlers();
    std::atexit(cleanup);
    lcb_install_callback3(instance, LCB_CALLBACK_SDLOOKUP, subdoc_callback);
    lcb_install_callback3(instance, LCB_CALLBACK_SDMUTATE, subdoc_callback);

    linenoiseSetCompletionCallback(command_completion);
    linenoiseSetMultiLine(1);
    linenoiseHistoryLoad(history_path.c_str());

    char *line;
    while ((line = linenoise("subdoc> ")) != NULL) {
        if (line[0] != '\0') {
            linenoiseHistoryAdd(line);
            linenoiseHistorySave(history_path.c_str());

            int cmd_argc = 0;
            char **cmd_argv = NULL;
            int rv = cliopts_split_args(line, &cmd_argc, &cmd_argv);
            if (rv) {
                fprintf(stderr, "Invalid input: unterminated single quote\n");
            } else {
                if (rv == 0 && cmd_argc > 0) {
                    char *cmd_name = cmd_argv[0];
                    subdoc::Handler *handler = handlers[cmd_name];
                    if (handler == NULL) {
                        fprintf(stderr, "Unknown command %s\n", cmd_name);
                        subdoc::HelpHandler().execute(cmd_argc, cmd_argv);
                    } else {
                        try {
                            handler->execute(cmd_argc, cmd_argv);
                        } catch (std::exception &err) {
                            fprintf(stderr, "%s\n", err.what());
                        }
                    }
                    free(cmd_argv);
                }
            }
        }
        free(line);
    }
}

int main(int argc, char **argv)
{
    try {
        real_main(argc, argv);
        return 0;
    } catch (std::exception &exc) {
        std::cerr << exc.what() << std::endl;
        exit(EXIT_FAILURE);
    }
}
