Module: Iodine::Base::CLI

Defined in:

ext/iodine/iodine.c

Class Method Summary

• .parse(desc) ⇒ Object

  Parses the CLI argnumnents, returning the Rack filename (if provided).

Class Method Details

.parse(desc) ⇒ Object

Parses the CLI argnumnents, returning the Rack filename (if provided).

Unknown arguments are ignored.

# File 'ext/iodine/iodine.c', line 289

static VALUE iodine_cli_parse(VALUE self, VALUE desc) {
  (void)self;
  Check_Type(desc, T_STRING);
  VALUE ARGV = rb_get_argv();
  VALUE ret = Qtrue;
  VALUE defaults = iodine_default_args;
  if (!defaults || !ARGV || TYPE(ARGV) != T_ARRAY || TYPE(defaults) != T_HASH) {
    FIO_LOG_ERROR("CLI parsing initialization error "
                  "ARGV=%p, Array?(%d), defaults == %p (%d)",
                  (void *)ARGV, (int)(TYPE(ARGV) == T_ARRAY), (void *)defaults,
                  (int)(TYPE(defaults) == T_HASH));
    return Qnil;
  }
  /* Copy the Ruby ARGV to a C valid ARGV */
  int argc = (int)rb_array_len(ARGV) + 1;
  if (argc <= 1) {
    FIO_LOG_DEBUG("CLI: No arguments to parse...\n");
    return Qnil;
  } else {
    FIO_LOG_DEBUG("Iodine CLI parsing %d arguments", argc);
  }
  char **argv = calloc(argc, sizeof(*argv));
  FIO_ASSERT_ALLOC(argv);
  argv[0] = (char *)"iodine";
  for (int i = 1; i < argc; ++i) {
    VALUE tmp = rb_ary_entry(ARGV, (long)(i - 1));
    if (TYPE(tmp) != T_STRING) {
      FIO_LOG_ERROR("ARGV Array contains a non-String object.");
      ret = Qnil;
      goto finish;
    }
    fio_str_info_s s = IODINE_RSTRINFO(tmp);
    argv[i] = malloc(s.len + 1);
    FIO_ASSERT_ALLOC(argv[i]);
    memcpy(argv[i], s.data, s.len);
    argv[i][s.len] = 0;
  }
  /* Levarage the facil.io CLI library */
  fio_cli_start(
      argc, (const char **)argv, 0, -1, StringValueCStr(desc),
      "\x1B[4mAddress Binding:\x1B[0m", FIO_CLI_TYPE_PRINT,
      "-bind -b -address address to listen to. defaults any available.",
      "-port -p port number to listen to. defaults port 3000", FIO_CLI_TYPE_INT,
      "\n\x1B[4mConcurrency:\x1B[0m", FIO_CLI_TYPE_PRINT,
      "-workers -w number of processes to use.", FIO_CLI_TYPE_INT,
      "-threads -t number of threads per process.", FIO_CLI_TYPE_INT,
      "\n\x1B[4mHTTP Server:\x1B[0m", FIO_CLI_TYPE_PRINT,
      "-public -www public folder, for static file service.",
      "-log -v HTTP request logging.", FIO_CLI_TYPE_BOOL,
      "-keep-alive -k -tout HTTP keep-alive timeout (0..255). Default: 40s",
      FIO_CLI_TYPE_INT, "-ping websocket ping interval (0..255). Default: 40s",
      FIO_CLI_TYPE_INT,
      "-max-body -maxbd HTTP upload limit in Mega-Bytes. Default: 50Mb",
      FIO_CLI_TYPE_INT,
      "-max-header -maxhd header limit per HTTP request in Kb."
      " Default: 32Kb.",
      FIO_CLI_TYPE_INT, "\n\x1B[4mWebSocket Server:\x1B[0m", FIO_CLI_TYPE_PRINT,
      "-max-msg -maxms incoming WebSocket message limit in Kb. "
      "Default: 250Kb",
      FIO_CLI_TYPE_INT, "\n\x1B[4mConnecting Iodine to Redis:\x1B[0m",
      FIO_CLI_TYPE_PRINT,
      "-redis -r an optional Redis URL server address. Default: none.",
      "-redis-ping -rp websocket ping interval (0..255). Default: 5 minutes",
      FIO_CLI_TYPE_INT, "\n\x1B[4mMisc:\x1B[0m", FIO_CLI_TYPE_PRINT,
      "-warmup warm up the application. CAREFUL! iodine might fork.",
      FIO_CLI_TYPE_BOOL,
      "-verbosity -V 0..5 server verbosity level. Default: 4",
      FIO_CLI_TYPE_INT);
  /* copy values from CLI library to iodine */
  if (fio_cli_get("-V")) {
    int level = fio_cli_get_i("-V");
    if (level > 0 && level < 100)
      FIO_LOG_LEVEL = level;
  }
  if (fio_cli_get("-w")) {
    iodine_workers_set(IodineModule, INT2NUM(fio_cli_get_i("-w")));
  }
  if (fio_cli_get("-t")) {
    iodine_threads_set(IodineModule, INT2NUM(fio_cli_get_i("-t")));
  }
  if (fio_cli_get_bool("-v")) {
    rb_hash_aset(defaults, ID2SYM(rb_intern("log")), Qtrue);
  }
  if (fio_cli_get_bool("-warmup")) {
    rb_hash_aset(defaults, ID2SYM(rb_intern("warmup_")), Qtrue);
  }
  if (fio_cli_get("-p")) {
    rb_hash_aset(defaults, ID2SYM(rb_intern("port")),
                 rb_str_new_cstr(fio_cli_get("-p")));
  }
  if (fio_cli_get("-b")) {
    rb_hash_aset(defaults, ID2SYM(rb_intern("address")),
                 rb_str_new_cstr(fio_cli_get("-b")));
  }
  if (fio_cli_get("-www")) {
    rb_hash_aset(defaults, ID2SYM(rb_intern("public")),
                 rb_str_new_cstr(fio_cli_get("-www")));
  }
  if (fio_cli_get("-redis")) {
    rb_hash_aset(defaults, ID2SYM(rb_intern("redis_")),
                 rb_str_new_cstr(fio_cli_get("-redis")));
  }
  if (fio_cli_get("-k")) {
    rb_hash_aset(defaults, ID2SYM(rb_intern("timeout")),
                 INT2NUM(fio_cli_get_i("-k")));
  }
  if (fio_cli_get("-ping")) {
    rb_hash_aset(defaults, ID2SYM(rb_intern("ping")),
                 INT2NUM(fio_cli_get_i("-ping")));
  }
  if (fio_cli_get("-redis-ping")) {
    rb_hash_aset(defaults, ID2SYM(rb_intern("redis_ping_")),
                 INT2NUM(fio_cli_get_i("-redis-ping")));
  }
  if (fio_cli_get("-max-body")) {
    rb_hash_aset(defaults, ID2SYM(rb_intern("max_body")),
                 INT2NUM((fio_cli_get_i("-max-body") * 1024 * 1024)));
  }
  if (fio_cli_get("-max-message")) {
    rb_hash_aset(defaults, ID2SYM(rb_intern("max_msg")),
                 INT2NUM((fio_cli_get_i("-max-message") * 1024)));
  }
  if (fio_cli_get("-max-headers")) {
    rb_hash_aset(defaults, ID2SYM(rb_intern("max_headers")),
                 INT2NUM((fio_cli_get_i("-max-headers") * 1024)));
  }
  if (fio_cli_unnamed_count()) {
    rb_hash_aset(defaults, ID2SYM(rb_intern("filename_")),
                 rb_str_new_cstr(fio_cli_unnamed(0)));
  }

  /* create `filename` String, cleanup and return */
  fio_cli_end();
finish:
  for (int i = 1; i < argc; ++i) {
    free(argv[i]);
  }
  free(argv);
  return ret;
}