#include "acconfig.h"
#include "auth/Auth.h"
#include "auth/Crypto.h"
#include "auth/KeyRing.h"
#include "common/ceph_argparse.h"
#include "common/version.h"
#include "common/PluginRegistry.h"
#include "compressor/snappy/CompressionPluginSnappy.h"
#include "compressor/zlib/CompressionPluginZlib.h"
#include "compressor/zstd/CompressionPluginZstd.h"
#include "erasure-code/ErasureCodePlugin.h"
#if __x86_64__ && defined(HAVE_BETTER_YASM_ELF64)
#include "erasure-code/isa/ErasureCodePluginIsa.h"
#endif
#include "erasure-code/jerasure/ErasureCodePluginJerasure.h"
#include "erasure-code/jerasure/jerasure_init.h"
#include "erasure-code/lrc/ErasureCodePluginLrc.h"
#include "erasure-code/shec/ErasureCodePluginShec.h"
#include "include/cephd/libcephd.h"
#include "global/global_context.h"
#include "global/global_init.h"
#include "objclass/objclass.h"
#include "osd/OSD.h"
#include "osd/ClassHandler.h"

// forward declarations of RADOS class init functions
CLS_INIT(cephfs);
CLS_INIT(hello);
CLS_INIT(journal);
CLS_INIT(kvs);
CLS_INIT(lock);
CLS_INIT(log);
CLS_INIT(lua);
CLS_INIT(numops);
CLS_INIT(rbd);
CLS_INIT(refcount);
CLS_INIT(replica_log);
CLS_INIT(rgw);
CLS_INIT(statelog);
CLS_INIT(timeindex);
CLS_INIT(user);
CLS_INIT(version);

extern "C" void cephd_version(int *pmajor, int *pminor, int *ppatch)
{
  if (pmajor)
    *pmajor = LIBCEPHD_VER_MAJOR;
  if (pminor)
    *pminor = LIBCEPHD_VER_MINOR;
  if (ppatch)
    *ppatch = LIBCEPHD_VER_PATCH;
}

extern "C" const char *ceph_version(int *pmajor, int *pminor, int *ppatch)
{
  int major, minor, patch;
  const char *v = ceph_version_to_str();

  int n = sscanf(v, "%d.%d.%d", &major, &minor, &patch);
  if (pmajor)
    *pmajor = (n >= 1) ? major : 0;
  if (pminor)
    *pminor = (n >= 2) ? minor : 0;
  if (ppatch)
    *ppatch = (n >= 3) ? patch : 0;
  return v;
}

extern "C" int cephd_generate_fsid(char *buf, size_t len)
{
    if (len < sizeof("b06ad912-70d7-4263-a5ff-011462a5929a")) {
        return -ERANGE;
    }

    uuid_d fsid;
    fsid.generate_random();
    fsid.print(buf);

    return 0;
}

extern "C" int cephd_generate_secret_key(char *buf, size_t len)
{
    CephInitParameters iparams(CEPH_ENTITY_TYPE_MON);
    CephContext *cct = common_preinit(iparams, CODE_ENVIRONMENT_LIBRARY, 0);
    cct->_conf->apply_changes(NULL);
    cct->init_crypto();

    CryptoKey key;
    key.create(cct, CEPH_CRYPTO_AES);

    cct->put();

    string keystr;
    key.encode_base64(keystr);
    if (keystr.length() >= len) {
        return -ERANGE;
    }
    strcpy(buf, keystr.c_str());
    return keystr.length();
}

// load the embedded plugins. This is safe to call multiple
// times in the same process
void cephd_preload_embedded_plugins()
{
  int r;

  // load erasure coding plugins
  {
    ErasureCodePlugin* plugin;
    ErasureCodePluginRegistry& reg = ErasureCodePluginRegistry::instance();
    Mutex::Locker l(reg.lock);
    reg.disable_dlclose = true;

    // initialize jerasure (and gf-complete)
    int w[] = { 4, 8, 16, 32 };
    r = jerasure_init(4, w);
    assert(r == 0);

    plugin = new ErasureCodePluginJerasure();
    r = reg.add("jerasure", plugin);
    if (r == -EEXIST) {
      delete plugin;
    }
    assert(r == 0);

    plugin = new ErasureCodePluginLrc();
    r = reg.add("lrc", plugin);
    if (r == -EEXIST) {
      delete plugin;
    }
    assert(r == 0);

    plugin = new ErasureCodePluginShec();
    r = reg.add("shec", plugin);
    if (r == -EEXIST) {
      delete plugin;
    }
    assert(r == 0);

#if __x86_64__ && defined(HAVE_BETTER_YASM_ELF64)
    plugin = new ErasureCodePluginIsa();
    r = reg.add("isa", plugin);
    if (r == -EEXIST) {
      delete plugin;
    }
    assert(r == 0);
#endif
  }

  // now load the compression plugins
  {
    Plugin *plugin;
    PluginRegistry *reg = g_ceph_context->get_plugin_registry();
    Mutex::Locker l(reg->lock);
    reg->disable_dlclose = true;

    plugin = new CompressionPluginSnappy(g_ceph_context);
    r = reg->add("compressor", "snappy", plugin);
    if (r == -EEXIST) {
      delete plugin;
    }
    assert(r == 0);

    plugin = new CompressionPluginZlib(g_ceph_context);
    r = reg->add("compressor", "zlib", plugin);
    if (r == -EEXIST) {
      delete plugin;
    }
    assert(r == 0);

    plugin = new CompressionPluginZstd(g_ceph_context);
    r = reg->add("compressor", "zstd", plugin);
    if (r == -EEXIST) {
      delete plugin;
    }
    assert(r == 0);
  }
}

void cephd_preload_rados_classes(OSD *osd)
{
  // intialize RADOS classes
  {
    ClassHandler  *class_handler = osd->class_handler;
    Mutex::Locker l(class_handler->mutex);

#ifdef WITH_CEPHFS
    class_handler->add_embedded_class("cephfs");
    cephfs_cls_init();
#endif
    class_handler->add_embedded_class("hello");
    hello_cls_init();
    class_handler->add_embedded_class("journal");
    journal_cls_init();
#ifdef WITH_KVS
    class_handler->add_embedded_class("kvs");
    kvs_cls_init();
#endif
    class_handler->add_embedded_class("lock");
    lock_cls_init();
    class_handler->add_embedded_class("log");
    log_cls_init();
    class_handler->add_embedded_class("lua");
    lua_cls_init();
    class_handler->add_embedded_class("numops");
    numops_cls_init();
#ifdef WITH_RBD
    class_handler->add_embedded_class("rbd");
    rbd_cls_init();
#endif
    class_handler->add_embedded_class("refcount");
    refcount_cls_init();
    class_handler->add_embedded_class("replica_log");
    replica_log_cls_init();
#ifdef WITH_RADOSGW
    class_handler->add_embedded_class("rgw");
    rgw_cls_init();
#endif
    class_handler->add_embedded_class("statelog");
    statelog_cls_init();
    class_handler->add_embedded_class("timeindex");
    timeindex_cls_init();
    class_handler->add_embedded_class("user");
    user_cls_init();
    class_handler->add_embedded_class("version");
    version_cls_init();
  }
}

extern "C" int cephd_mon(int argc, const char **argv);
extern "C" int cephd_osd(int argc, const char **argv);
extern "C" int cephd_mds(int argc, const char **argv);
extern "C" int cephd_rgw(int argc, const char **argv);
extern "C" int cephd_rgw_admin(int argc, const char **argv);

int cephd_run_mon(int argc, const char **argv)
{
    return cephd_mon(argc, argv);
}

int cephd_run_osd(int argc, const char **argv)
{
    return cephd_osd(argc, argv);
}

int cephd_run_mds(int argc, const char **argv)
{
    return cephd_mds(argc, argv);
}


int cephd_run_rgw(int argc, const char **argv)
{
    return cephd_rgw(argc, argv);
}

int cephd_run_rgw_admin(int argc, const char **argv)
{
    return cephd_rgw_admin(argc, argv);
}