/* -*- c-basic-offset: 2 -*- */
/*
  Copyright(C) 2013-2018 Brazil
  Copyright(C) 2018-2020 Sutou Kouhei <kou@clear-code.com>

  This library is free software; you can redistribute it and/or
  modify it under the terms of the GNU Lesser General Public
  License version 2.1 as published by the Free Software Foundation.

  This library is distributed in the hope that it will be useful,
  but WITHOUT ANY WARRANTY; without even the implied warranty of
  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  Lesser General Public License for more details.

  You should have received a copy of the GNU Lesser General Public
  License along with this library; if not, write to the Free Software
  Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301  USA
*/

#include "../grn_ctx_impl.h"

#ifdef GRN_WITH_MRUBY
#include <mruby.h>
#include <mruby/class.h>
#include <mruby/data.h>
#include <mruby/error.h>
#include <mruby/variable.h>
#include <mruby/string.h>

#include "../grn_encoding.h"
#include "../grn_mrb.h"
#include "mrb_ctx.h"
#include "mrb_bulk.h"
#include "mrb_converter.h"

static mrb_value
ctx_class_instance(mrb_state *mrb, mrb_value klass)
{
  grn_ctx *ctx = (grn_ctx *)mrb->ud;
  mrb_value mrb_ctx;
  mrb_sym iv_name;

  iv_name = mrb_intern_lit(mrb, "@instance");
  mrb_ctx = mrb_iv_get(mrb, klass, iv_name);
  if (mrb_nil_p(mrb_ctx)) {
    struct RBasic *raw_mrb_ctx;
    raw_mrb_ctx = mrb_obj_alloc(mrb, MRB_TT_DATA, mrb_class_ptr(klass));
    mrb_ctx = mrb_obj_value(raw_mrb_ctx);
    DATA_PTR(mrb_ctx) = ctx;
    mrb_iv_set(mrb, klass, iv_name, mrb_ctx);
  }

  return mrb_ctx;
}

static mrb_value
ctx_array_reference(mrb_state *mrb, mrb_value self)
{
  grn_ctx *ctx = (grn_ctx *)mrb->ud;
  mrb_value mrb_id_or_name;
  grn_obj *object;

  mrb_get_args(mrb, "o", &mrb_id_or_name);

  if (mrb_nil_p(mrb_id_or_name)) {
    return mrb_nil_value();
  }

  if (mrb_fixnum_p(mrb_id_or_name)) {
    grn_id id = mrb_fixnum(mrb_id_or_name);
    object = grn_ctx_at(ctx, id);
  } else {
    mrb_value mrb_name;
    mrb_name = mrb_convert_type(mrb, mrb_id_or_name,
                                MRB_TT_STRING, "String", "to_str");
    object = grn_ctx_get(ctx,
                         RSTRING_PTR(mrb_name),
                         RSTRING_LEN(mrb_name));
  }

  return grn_mrb_value_from_grn_obj(mrb, object);
}

static mrb_value
ctx_get_rc(mrb_state *mrb, mrb_value self)
{
  grn_ctx *ctx = (grn_ctx *)mrb->ud;

  return mrb_fixnum_value(ctx->rc);
}

static mrb_value
ctx_set_rc(mrb_state *mrb, mrb_value self)
{
  grn_ctx *ctx = (grn_ctx *)mrb->ud;
  mrb_int rc;

  mrb_get_args(mrb, "i", &rc);
  ctx->rc = rc;

  return mrb_fixnum_value(ctx->rc);
}

static mrb_value
ctx_get_error_level(mrb_state *mrb, mrb_value self)
{
  grn_ctx *ctx = (grn_ctx *)mrb->ud;

  return mrb_fixnum_value(ctx->errlvl);
}

static mrb_value
ctx_set_error_level(mrb_state *mrb, mrb_value self)
{
  grn_ctx *ctx = (grn_ctx *)mrb->ud;
  mrb_int error_level;

  mrb_get_args(mrb, "i", &error_level);
  ctx->errlvl = error_level;

  return mrb_fixnum_value(ctx->errlvl);
}

static mrb_value
ctx_get_error_file(mrb_state *mrb, mrb_value self)
{
  grn_ctx *ctx = (grn_ctx *)mrb->ud;

  return mrb_str_new_cstr(mrb, ctx->errfile);
}

static mrb_value
ctx_set_error_file(mrb_state *mrb, mrb_value self)
{
  grn_ctx *ctx = (grn_ctx *)mrb->ud;
  mrb_value error_file;

  mrb_get_args(mrb, "S", &error_file);
  mrb_iv_set(mrb, self, mrb_intern_lit(mrb, "@error_file"), error_file);
  ctx->errfile = mrb_string_value_cstr(mrb, &error_file);

  return error_file;
}

static mrb_value
ctx_get_error_line(mrb_state *mrb, mrb_value self)
{
  grn_ctx *ctx = (grn_ctx *)mrb->ud;

  return mrb_fixnum_value(ctx->errline);
}

static mrb_value
ctx_set_error_line(mrb_state *mrb, mrb_value self)
{
  grn_ctx *ctx = (grn_ctx *)mrb->ud;
  mrb_int error_line;

  mrb_get_args(mrb, "i", &error_line);
  ctx->errline = error_line;

  return mrb_fixnum_value(ctx->errline);
}

static mrb_value
ctx_get_error_method(mrb_state *mrb, mrb_value self)
{
  grn_ctx *ctx = (grn_ctx *)mrb->ud;

  return mrb_str_new_cstr(mrb, ctx->errfunc);
}

static mrb_value
ctx_set_error_method(mrb_state *mrb, mrb_value self)
{
  grn_ctx *ctx = (grn_ctx *)mrb->ud;
  mrb_value error_method;

  mrb_get_args(mrb, "S", &error_method);
  mrb_iv_set(mrb, self, mrb_intern_lit(mrb, "@error_method"), error_method);
  ctx->errfunc = mrb_string_value_cstr(mrb, &error_method);

  return error_method;
}

static mrb_value
ctx_get_error_message(mrb_state *mrb, mrb_value self)
{
  grn_ctx *ctx = (grn_ctx *)mrb->ud;
  const char *utf8_message;
  mrb_value mrb_message;

  utf8_message = grn_encoding_convert_to_utf8(ctx, ctx->errbuf, -1, NULL);
  mrb_message = mrb_str_new_cstr(mrb, utf8_message);
  grn_encoding_converted_free(ctx, utf8_message);

  return mrb_message;
}

static mrb_value
ctx_set_error_message(mrb_state *mrb, mrb_value self)
{
  grn_ctx *ctx = (grn_ctx *)mrb->ud;
  mrb_value mrb_error_message;
  const char *error_message;
  size_t error_message_size;

  mrb_get_args(mrb, "S", &mrb_error_message);
  error_message =
    grn_encoding_convert_from_utf8(ctx,
                                   RSTRING_PTR(mrb_error_message),
                                   RSTRING_LEN(mrb_error_message),
                                   &error_message_size);
  grn_ctx_log(ctx, "%.*s",
              (int)error_message_size,
              error_message);
  grn_encoding_converted_free(ctx, error_message);

  return mrb_error_message;
}

static mrb_value
ctx_clear_error(mrb_state *mrb, mrb_value self)
{
  grn_ctx *ctx = (grn_ctx *)mrb->ud;

  ERRCLR(ctx);

  return mrb_nil_value();
}

static mrb_value
ctx_get_command_version(mrb_state *mrb, mrb_value self)
{
  grn_ctx *ctx = (grn_ctx *)mrb->ud;

  return mrb_fixnum_value(grn_ctx_get_command_version(ctx));
}

static mrb_value
ctx_set_command_version(mrb_state *mrb, mrb_value self)
{
  grn_ctx *ctx = (grn_ctx *)mrb->ud;
  mrb_int command_version;

  mrb_get_args(mrb, "i", &command_version);
  grn_ctx_set_command_version(ctx, command_version);

  return mrb_fixnum_value(command_version);
}

static mrb_value
ctx_get_output(mrb_state *mrb, mrb_value self)
{
  grn_ctx *ctx = (grn_ctx *)mrb->ud;

  return grn_mrb_value_from_bulk(mrb, ctx->impl->output.buf);
}

static mrb_value
ctx_set_output(mrb_state *mrb, mrb_value self)
{
  grn_ctx *ctx = (grn_ctx *)mrb->ud;
  mrb_value mrb_value_;

  mrb_get_args(mrb, "S", &mrb_value_);
  GRN_TEXT_SET(ctx, ctx->impl->output.buf,
               RSTRING_PTR(mrb_value_),
               RSTRING_LEN(mrb_value_));

  return mrb_value_;
}

static mrb_value
ctx_get_output_type(mrb_state *mrb, mrb_value self)
{
  grn_ctx *ctx = (grn_ctx *)mrb->ud;

  return mrb_fixnum_value(grn_ctx_get_output_type(ctx));
}

static mrb_value
ctx_set_output_type(mrb_state *mrb, mrb_value self)
{
  grn_ctx *ctx = (grn_ctx *)mrb->ud;
  mrb_int output_type;

  mrb_get_args(mrb, "i", &output_type);
  grn_ctx_set_output_type(ctx, output_type);

  return mrb_fixnum_value(output_type);
}

static mrb_value
ctx_get_database(mrb_state *mrb, mrb_value self)
{
  grn_ctx *ctx = (grn_ctx *)mrb->ud;

  return grn_mrb_value_from_grn_obj(mrb, grn_ctx_db(ctx));
}

static mrb_value
ctx_is_opened(mrb_state *mrb, mrb_value self)
{
  grn_ctx *ctx = (grn_ctx *)mrb->ud;
  mrb_int mrb_id;

  mrb_get_args(mrb, "i", &mrb_id);

  return mrb_bool_value(grn_ctx_is_opened(ctx, mrb_id));
}

mrb_value
grn_mrb_ctx_to_exception(mrb_state *mrb)
{
  grn_ctx *ctx = (grn_ctx *)mrb->ud;
  grn_mrb_data *data = &(ctx->impl->mrb);
  struct RClass *module = data->module;
  struct RClass *error_class = NULL;
#define MESSAGE_SIZE 4096
  char message[MESSAGE_SIZE];
  const char *utf8_error_message;

  if (ctx->rc == GRN_SUCCESS) {
    return mrb_nil_value();
  }

  utf8_error_message = grn_encoding_convert_to_utf8(ctx, ctx->errbuf, -1, NULL);

  switch (ctx->rc) {
  case GRN_SUCCESS:
    break;
  case GRN_END_OF_DATA:
    error_class = mrb_class_get_under(mrb, module, "EndOfData");
    grn_snprintf(message, MESSAGE_SIZE, MESSAGE_SIZE,
                 "end of data: <%s>(%d)",
                 utf8_error_message, ctx->rc);
    break;
  case GRN_UNKNOWN_ERROR:
    error_class = mrb_class_get_under(mrb, module, "UnknownError");
    grn_snprintf(message, MESSAGE_SIZE, MESSAGE_SIZE,
                 "unknown error: <%s>(%d)",
                 utf8_error_message, ctx->rc);
    break;
  case GRN_OPERATION_NOT_PERMITTED:
    error_class = mrb_class_get_under(mrb, module, "OperationNotPermitted");
    grn_snprintf(message, MESSAGE_SIZE, MESSAGE_SIZE,
                 "operation not permitted: <%s>(%d)",
                 utf8_error_message, ctx->rc);
    break;
  case GRN_NO_SUCH_FILE_OR_DIRECTORY:
    error_class = mrb_class_get_under(mrb, module, "NoSuchFileOrDirectory");
    grn_snprintf(message, MESSAGE_SIZE, MESSAGE_SIZE,
                 "no such file or directory: <%s>(%d)",
                 utf8_error_message, ctx->rc);
    break;
  case GRN_NO_SUCH_PROCESS:
    error_class = mrb_class_get_under(mrb, module, "NoSuchProcess");
    grn_snprintf(message, MESSAGE_SIZE, MESSAGE_SIZE,
                 "no such process: <%s>(%d)",
                 utf8_error_message, ctx->rc);
    break;
  case GRN_INTERRUPTED_FUNCTION_CALL:
    error_class = mrb_class_get_under(mrb, module, "InterruptedFunctionCall");
    grn_snprintf(message, MESSAGE_SIZE, MESSAGE_SIZE,
                 "interrupted function call: <%s>(%d)",
                 utf8_error_message, ctx->rc);
    break;
  case GRN_INPUT_OUTPUT_ERROR:
    error_class = mrb_class_get_under(mrb, module, "InputOutputError");
    grn_snprintf(message, MESSAGE_SIZE, MESSAGE_SIZE,
                 "input output error: <%s>(%d)",
                 utf8_error_message, ctx->rc);
    break;
  case GRN_NO_SUCH_DEVICE_OR_ADDRESS:
    error_class = mrb_class_get_under(mrb, module, "NoSuchDeviceOrAddress");
    grn_snprintf(message, MESSAGE_SIZE, MESSAGE_SIZE,
                 "no such device or address: <%s>(%d)",
                 utf8_error_message, ctx->rc);
    break;
  case GRN_ARG_LIST_TOO_LONG:
    error_class = mrb_class_get_under(mrb, module, "ArgListTooLong");
    grn_snprintf(message, MESSAGE_SIZE, MESSAGE_SIZE,
                 "arg list too long: <%s>(%d)",
                 utf8_error_message, ctx->rc);
    break;
  case GRN_EXEC_FORMAT_ERROR:
    error_class = mrb_class_get_under(mrb, module, "ExecFormatError");
    grn_snprintf(message, MESSAGE_SIZE, MESSAGE_SIZE,
                 "exec format error: <%s>(%d)",
                 utf8_error_message, ctx->rc);
    break;
  case GRN_BAD_FILE_DESCRIPTOR:
    error_class = mrb_class_get_under(mrb, module, "BadFileDescriptor");
    grn_snprintf(message, MESSAGE_SIZE, MESSAGE_SIZE,
                 "bad file descriptor: <%s>(%d)",
                 utf8_error_message, ctx->rc);
    break;
  case GRN_NO_CHILD_PROCESSES:
    error_class = mrb_class_get_under(mrb, module, "NoChildProcesses");
    grn_snprintf(message, MESSAGE_SIZE, MESSAGE_SIZE,
                 "no child processes: <%s>(%d)",
                 utf8_error_message, ctx->rc);
    break;
  case GRN_RESOURCE_TEMPORARILY_UNAVAILABLE:
    error_class = mrb_class_get_under(mrb, module,
                                      "ResourceTemporarilyUnavailable");
    grn_snprintf(message, MESSAGE_SIZE, MESSAGE_SIZE,
                 "resource temporarily unavailable: <%s>(%d)",
                 utf8_error_message, ctx->rc);
    break;
  case GRN_NOT_ENOUGH_SPACE:
    error_class = mrb_class_get_under(mrb, module, "NotEnoughSpace");
    grn_snprintf(message, MESSAGE_SIZE, MESSAGE_SIZE,
                 "not enough space: <%s>(%d)",
                 utf8_error_message, ctx->rc);
    break;
  case GRN_PERMISSION_DENIED:
    error_class = mrb_class_get_under(mrb, module, "PermissionDenied");
    grn_snprintf(message, MESSAGE_SIZE, MESSAGE_SIZE,
                 "permission denied: <%s>(%d)",
                 utf8_error_message, ctx->rc);
    break;
  case GRN_BAD_ADDRESS:
    error_class = mrb_class_get_under(mrb, module, "BadAddress");
    grn_snprintf(message, MESSAGE_SIZE, MESSAGE_SIZE,
                 "bad address: <%s>(%d)",
                 utf8_error_message, ctx->rc);
    break;
  case GRN_RESOURCE_BUSY:
    error_class = mrb_class_get_under(mrb, module, "ResourceBusy");
    grn_snprintf(message, MESSAGE_SIZE, MESSAGE_SIZE,
                 "resource busy: <%s>(%d)",
                 utf8_error_message, ctx->rc);
    break;
  case GRN_FILE_EXISTS:
    error_class = mrb_class_get_under(mrb, module, "FileExists");
    grn_snprintf(message, MESSAGE_SIZE, MESSAGE_SIZE,
                 "file exists: <%s>(%d)",
                 utf8_error_message, ctx->rc);
    break;
  case GRN_IMPROPER_LINK:
    error_class = mrb_class_get_under(mrb, module, "ImproperLink");
    grn_snprintf(message, MESSAGE_SIZE, MESSAGE_SIZE,
                 "improper link: <%s>(%d)",
                 utf8_error_message, ctx->rc);
    break;
  case GRN_NO_SUCH_DEVICE:
    error_class = mrb_class_get_under(mrb, module, "NoSuchDevice");
    grn_snprintf(message, MESSAGE_SIZE, MESSAGE_SIZE,
                 "no such device: <%s>(%d)",
                 utf8_error_message, ctx->rc);
    break;
  case GRN_NOT_A_DIRECTORY:
    error_class = mrb_class_get_under(mrb, module, "NotDirectory");
    grn_snprintf(message, MESSAGE_SIZE, MESSAGE_SIZE,
                 "not directory: <%s>(%d)",
                 utf8_error_message, ctx->rc);
    break;
  case GRN_IS_A_DIRECTORY:
    error_class = mrb_class_get_under(mrb, module, "IsDirectory");
    grn_snprintf(message, MESSAGE_SIZE, MESSAGE_SIZE,
                 "is directory: <%s>(%d)",
                 utf8_error_message, ctx->rc);
    break;
  case GRN_INVALID_ARGUMENT:
    error_class = mrb_class_get_under(mrb, module, "InvalidArgument");
    grn_snprintf(message, MESSAGE_SIZE, MESSAGE_SIZE,
                 "invalid argument: <%s>(%d)",
                 utf8_error_message, ctx->rc);
    break;
  case GRN_TOO_MANY_OPEN_FILES_IN_SYSTEM:
    error_class = mrb_class_get_under(mrb, module, "TooManyOpenFilesInSystem");
    grn_snprintf(message, MESSAGE_SIZE, MESSAGE_SIZE,
                 "too many open files in system: <%s>(%d)",
                 utf8_error_message, ctx->rc);
    break;
  case GRN_TOO_MANY_OPEN_FILES:
    error_class = mrb_class_get_under(mrb, module, "TooManyOpenFiles");
    grn_snprintf(message, MESSAGE_SIZE, MESSAGE_SIZE,
                 "too many open files: <%s>(%d)",
                 utf8_error_message, ctx->rc);
    break;
  case GRN_INAPPROPRIATE_I_O_CONTROL_OPERATION:
    error_class = mrb_class_get_under(mrb, module,
                                      "InappropriateIOControlOperation");
    grn_snprintf(message, MESSAGE_SIZE, MESSAGE_SIZE,
                 "inappropriate IO control operation: <%s>(%d)",
                 utf8_error_message, ctx->rc);
    break;
  case GRN_FILE_TOO_LARGE:
    error_class = mrb_class_get_under(mrb, module, "FileTooLarge");
    grn_snprintf(message, MESSAGE_SIZE, MESSAGE_SIZE,
                 "file too large: <%s>(%d)",
                 utf8_error_message, ctx->rc);
    break;
  case GRN_NO_SPACE_LEFT_ON_DEVICE:
    error_class = mrb_class_get_under(mrb, module, "NoSpaceLeftOnDevice");
    grn_snprintf(message, MESSAGE_SIZE, MESSAGE_SIZE,
                 "no space left on device: <%s>(%d)",
                 utf8_error_message, ctx->rc);
    break;
  case GRN_INVALID_SEEK:
    error_class = mrb_class_get_under(mrb, module, "InvalidSeek");
    grn_snprintf(message, MESSAGE_SIZE, MESSAGE_SIZE,
                 "invalid seek: <%s>(%d)",
                 utf8_error_message, ctx->rc);
    break;
  case GRN_READ_ONLY_FILE_SYSTEM:
    error_class = mrb_class_get_under(mrb, module, "ReadOnlyFileSystem");
    grn_snprintf(message, MESSAGE_SIZE, MESSAGE_SIZE,
                 "read only file system: <%s>(%d)",
                 utf8_error_message, ctx->rc);
    break;
  case GRN_TOO_MANY_LINKS:
    error_class = mrb_class_get_under(mrb, module, "TooManyLinks");
    grn_snprintf(message, MESSAGE_SIZE, MESSAGE_SIZE,
                 "too many links: <%s>(%d)",
                 utf8_error_message, ctx->rc);
    break;
  case GRN_BROKEN_PIPE:
    error_class = mrb_class_get_under(mrb, module, "BrokenPipe");
    grn_snprintf(message, MESSAGE_SIZE, MESSAGE_SIZE,
                 "broken pipe: <%s>(%d)",
                 utf8_error_message, ctx->rc);
    break;
  case GRN_DOMAIN_ERROR:
    error_class = mrb_class_get_under(mrb, module, "DomainError");
    grn_snprintf(message, MESSAGE_SIZE, MESSAGE_SIZE,
                 "domain error: <%s>(%d)",
                 utf8_error_message, ctx->rc);
    break;
  case GRN_RESULT_TOO_LARGE:
    error_class = mrb_class_get_under(mrb, module, "ResultTooLarge");
    grn_snprintf(message, MESSAGE_SIZE, MESSAGE_SIZE,
                 "result too large: <%s>(%d)",
                 utf8_error_message, ctx->rc);
    break;
  case GRN_RESOURCE_DEADLOCK_AVOIDED:
    error_class = mrb_class_get_under(mrb, module, "ResourceDeadlockAvoided");
    grn_snprintf(message, MESSAGE_SIZE, MESSAGE_SIZE,
                 "resource deadlock avoided: <%s>(%d)",
                 utf8_error_message, ctx->rc);
    break;
  case GRN_NO_MEMORY_AVAILABLE:
    error_class = mrb_class_get_under(mrb, module, "NoMemoryAvailable");
    grn_snprintf(message, MESSAGE_SIZE, MESSAGE_SIZE,
                 "no memory available: <%s>(%d)",
                 utf8_error_message, ctx->rc);
    break;
  case GRN_FILENAME_TOO_LONG:
    error_class = mrb_class_get_under(mrb, module, "FilenameTooLong");
    grn_snprintf(message, MESSAGE_SIZE, MESSAGE_SIZE,
                 "filename too long: <%s>(%d)",
                 utf8_error_message, ctx->rc);
    break;
  case GRN_NO_LOCKS_AVAILABLE:
    error_class = mrb_class_get_under(mrb, module, "NoLocksAvailable");
    grn_snprintf(message, MESSAGE_SIZE, MESSAGE_SIZE,
                 "no locks available: <%s>(%d)",
                 utf8_error_message, ctx->rc);
    break;
  case GRN_FUNCTION_NOT_IMPLEMENTED:
    error_class = mrb_class_get_under(mrb, module, "FunctionNotImplemented");
    grn_snprintf(message, MESSAGE_SIZE, MESSAGE_SIZE,
                 "function not implemented: <%s>(%d)",
                 utf8_error_message, ctx->rc);
    break;
  case GRN_DIRECTORY_NOT_EMPTY:
    error_class = mrb_class_get_under(mrb, module, "DirectoryNotEmpty");
    grn_snprintf(message, MESSAGE_SIZE, MESSAGE_SIZE,
                 "directory not empty: <%s>(%d)",
                 utf8_error_message, ctx->rc);
    break;
  case GRN_ILLEGAL_BYTE_SEQUENCE:
    error_class = mrb_class_get_under(mrb, module, "IllegalByteSequence");
    grn_snprintf(message, MESSAGE_SIZE, MESSAGE_SIZE,
                 "illegal byte sequence: <%s>(%d)",
                 utf8_error_message, ctx->rc);
    break;
  case GRN_SOCKET_NOT_INITIALIZED:
    error_class = mrb_class_get_under(mrb, module, "SocketNotInitialized");
    grn_snprintf(message, MESSAGE_SIZE, MESSAGE_SIZE,
                 "socket not initialized: <%s>(%d)",
                 utf8_error_message, ctx->rc);
    break;
  case GRN_OPERATION_WOULD_BLOCK:
    error_class = mrb_class_get_under(mrb, module, "OperationWouldBlock");
    grn_snprintf(message, MESSAGE_SIZE, MESSAGE_SIZE,
                 "operation would block: <%s>(%d)",
                 utf8_error_message, ctx->rc);
    break;
  case GRN_ADDRESS_IS_NOT_AVAILABLE:
    error_class = mrb_class_get_under(mrb, module, "AddressIsNotAvailable");
    grn_snprintf(message, MESSAGE_SIZE, MESSAGE_SIZE,
                 "address is not available: <%s>(%d)",
                 utf8_error_message, ctx->rc);
    break;
  case GRN_NETWORK_IS_DOWN:
    error_class = mrb_class_get_under(mrb, module, "NetworkIsDown");
    grn_snprintf(message, MESSAGE_SIZE, MESSAGE_SIZE,
                 "network is down: <%s>(%d)",
                 utf8_error_message, ctx->rc);
    break;
  case GRN_NO_BUFFER:
    error_class = mrb_class_get_under(mrb, module, "NoBuffer");
    grn_snprintf(message, MESSAGE_SIZE, MESSAGE_SIZE,
                 "no buffer: <%s>(%d)",
                 utf8_error_message, ctx->rc);
    break;
  case GRN_SOCKET_IS_ALREADY_CONNECTED:
    error_class = mrb_class_get_under(mrb, module, "SocketIsAlreadyConnected");
    grn_snprintf(message, MESSAGE_SIZE, MESSAGE_SIZE,
                 "socket is already connected: <%s>(%d)",
                 utf8_error_message, ctx->rc);
    break;
  case GRN_SOCKET_IS_NOT_CONNECTED:
    error_class = mrb_class_get_under(mrb, module, "SocketIsNotConnected");
    grn_snprintf(message, MESSAGE_SIZE, MESSAGE_SIZE,
                 "socket is not connected: <%s>(%d)",
                 utf8_error_message, ctx->rc);
    break;
  case GRN_SOCKET_IS_ALREADY_SHUTDOWNED:
    error_class = mrb_class_get_under(mrb, module, "SocketIsAlreadyShutdowned");
    grn_snprintf(message, MESSAGE_SIZE, MESSAGE_SIZE,
                 "socket is already shutdowned: <%s>(%d)",
                 utf8_error_message, ctx->rc);
    break;
  case GRN_OPERATION_TIMEOUT:
    error_class = mrb_class_get_under(mrb, module, "OperationTimeout");
    grn_snprintf(message, MESSAGE_SIZE, MESSAGE_SIZE,
                 "operation timeout: <%s>(%d)",
                 utf8_error_message, ctx->rc);
    break;
  case GRN_CONNECTION_REFUSED:
    error_class = mrb_class_get_under(mrb, module, "ConnectionRefused");
    grn_snprintf(message, MESSAGE_SIZE, MESSAGE_SIZE,
                 "connection refused: <%s>(%d)",
                 utf8_error_message, ctx->rc);
    break;
  case GRN_RANGE_ERROR:
    error_class = mrb_class_get_under(mrb, module, "RangeError");
    grn_snprintf(message, MESSAGE_SIZE, MESSAGE_SIZE,
                 "range error: <%s>(%d)",
                 utf8_error_message, ctx->rc);
    break;
  case GRN_TOKENIZER_ERROR:
    error_class = mrb_class_get_under(mrb, module, "TokenizerError");
    grn_snprintf(message, MESSAGE_SIZE, MESSAGE_SIZE,
                 "tokenizer error: <%s>(%d)",
                 utf8_error_message, ctx->rc);
    break;
  case GRN_FILE_CORRUPT:
    error_class = mrb_class_get_under(mrb, module, "FileCorrupt");
    grn_snprintf(message, MESSAGE_SIZE, MESSAGE_SIZE,
                 "file corrupt: <%s>(%d)",
                 utf8_error_message, ctx->rc);
    break;
  case GRN_INVALID_FORMAT:
    error_class = mrb_class_get_under(mrb, module, "InvalidFormat");
    grn_snprintf(message, MESSAGE_SIZE, MESSAGE_SIZE,
                 "invalid format: <%s>(%d)",
                 utf8_error_message, ctx->rc);
    break;
  case GRN_OBJECT_CORRUPT:
    error_class = mrb_class_get_under(mrb, module, "ObjectCorrupt");
    grn_snprintf(message, MESSAGE_SIZE, MESSAGE_SIZE,
                 "object corrupt: <%s>(%d)",
                 utf8_error_message, ctx->rc);
    break;
  case GRN_TOO_MANY_SYMBOLIC_LINKS:
    error_class = mrb_class_get_under(mrb, module, "TooManySymbolicLinks");
    grn_snprintf(message, MESSAGE_SIZE, MESSAGE_SIZE,
                 "too many symbolic links: <%s>(%d)",
                 utf8_error_message, ctx->rc);
    break;
  case GRN_NOT_SOCKET:
    error_class = mrb_class_get_under(mrb, module, "NotSocket");
    grn_snprintf(message, MESSAGE_SIZE, MESSAGE_SIZE,
                 "not socket: <%s>(%d)",
                 utf8_error_message, ctx->rc);
    break;
  case GRN_OPERATION_NOT_SUPPORTED:
    error_class = mrb_class_get_under(mrb, module, "OperationNotSupported");
    grn_snprintf(message, MESSAGE_SIZE, MESSAGE_SIZE,
                 "operation not supported: <%s>(%d)",
                 utf8_error_message, ctx->rc);
    break;
  case GRN_ADDRESS_IS_IN_USE:
    error_class = mrb_class_get_under(mrb, module, "AddressIsInUse");
    grn_snprintf(message, MESSAGE_SIZE, MESSAGE_SIZE,
                 "address is in use: <%s>(%d)",
                 utf8_error_message, ctx->rc);
    break;
  case GRN_ZLIB_ERROR:
    error_class = mrb_class_get_under(mrb, module, "ZlibError");
    grn_snprintf(message, MESSAGE_SIZE, MESSAGE_SIZE,
                 "zlib error: <%s>(%d)",
                 utf8_error_message, ctx->rc);
    break;
  case GRN_LZ4_ERROR:
    error_class = mrb_class_get_under(mrb, module, "LZ4Error");
    grn_snprintf(message, MESSAGE_SIZE, MESSAGE_SIZE,
                 "LZ4 error: <%s>(%d)",
                 utf8_error_message, ctx->rc);
    break;
  case GRN_STACK_OVER_FLOW:
    error_class = mrb_class_get_under(mrb, module, "StackOverFlow");
    grn_snprintf(message, MESSAGE_SIZE, MESSAGE_SIZE,
                 "stack over flow: <%s>(%d)",
                 utf8_error_message, ctx->rc);
    break;
  case GRN_SYNTAX_ERROR:
    error_class = mrb_class_get_under(mrb, module, "SyntaxError");
    grn_snprintf(message, MESSAGE_SIZE, MESSAGE_SIZE,
                 "syntax error: <%s>(%d)",
                 utf8_error_message, ctx->rc);
    break;
  case GRN_RETRY_MAX:
    error_class = mrb_class_get_under(mrb, module, "RetryMax");
    grn_snprintf(message, MESSAGE_SIZE, MESSAGE_SIZE,
                 "retry max: <%s>(%d)",
                 utf8_error_message, ctx->rc);
    break;
  case GRN_INCOMPATIBLE_FILE_FORMAT:
    error_class = mrb_class_get_under(mrb, module, "IncompatibleFileFormat");
    grn_snprintf(message, MESSAGE_SIZE, MESSAGE_SIZE,
                 "incompatible file format: <%s>(%d)",
                 utf8_error_message, ctx->rc);
    break;
  case GRN_UPDATE_NOT_ALLOWED:
    error_class = mrb_class_get_under(mrb, module, "UpdateNotAllowed");
    grn_snprintf(message, MESSAGE_SIZE, MESSAGE_SIZE,
                 "update not allowed: <%s>(%d)",
                 utf8_error_message, ctx->rc);
    break;
  case GRN_TOO_SMALL_OFFSET:
    error_class = mrb_class_get_under(mrb, module, "TooSmallOffset");
    grn_snprintf(message, MESSAGE_SIZE, MESSAGE_SIZE,
                 "too small offset: <%s>(%d)",
                 utf8_error_message, ctx->rc);
    break;
  case GRN_TOO_LARGE_OFFSET:
    error_class = mrb_class_get_under(mrb, module, "TooLargeOffset");
    grn_snprintf(message, MESSAGE_SIZE, MESSAGE_SIZE,
                 "too large offset: <%s>(%d)",
                 utf8_error_message, ctx->rc);
    break;
  case GRN_TOO_SMALL_LIMIT:
    error_class = mrb_class_get_under(mrb, module, "TooSmallLimit");
    grn_snprintf(message, MESSAGE_SIZE, MESSAGE_SIZE,
                 "too small limit: <%s>(%d)",
                 utf8_error_message, ctx->rc);
    break;
  case GRN_CAS_ERROR:
    error_class = mrb_class_get_under(mrb, module, "CASError");
    grn_snprintf(message, MESSAGE_SIZE, MESSAGE_SIZE,
                 "CAS error: <%s>(%d)",
                 utf8_error_message, ctx->rc);
    break;
  case GRN_UNSUPPORTED_COMMAND_VERSION:
    error_class = mrb_class_get_under(mrb, module, "UnsupportedCommandVersion");
    grn_snprintf(message, MESSAGE_SIZE, MESSAGE_SIZE,
                 "unsupported command version: <%s>(%d)",
                 utf8_error_message, ctx->rc);
    break;
  case GRN_NORMALIZER_ERROR:
    error_class = mrb_class_get_under(mrb, module, "NormalizerError");
    grn_snprintf(message, MESSAGE_SIZE, MESSAGE_SIZE,
                 "normalizer error: <%s>(%d)",
                 utf8_error_message, ctx->rc);
    break;
  case GRN_TOKEN_FILTER_ERROR:
    error_class = mrb_class_get_under(mrb, module, "TokenFilterError");
    grn_snprintf(message, MESSAGE_SIZE, MESSAGE_SIZE,
                 "token filter error: <%s>(%d)",
                 utf8_error_message, ctx->rc);
    break;
  case GRN_COMMAND_ERROR:
    error_class = mrb_class_get_under(mrb, module, "CommandError");
    grn_snprintf(message, MESSAGE_SIZE, MESSAGE_SIZE,
                 "command error: <%s>(%d)",
                 utf8_error_message, ctx->rc);
    break;
  case GRN_PLUGIN_ERROR:
    error_class = mrb_class_get_under(mrb, module, "PluginError");
    grn_snprintf(message, MESSAGE_SIZE, MESSAGE_SIZE,
                 "plugin error: <%s>(%d)",
                 utf8_error_message, ctx->rc);
    break;
  case GRN_SCORER_ERROR:
    error_class = mrb_class_get_under(mrb, module, "ScorerError");
    grn_snprintf(message, MESSAGE_SIZE, MESSAGE_SIZE,
                 "scorer error: <%s>(%d)",
                 utf8_error_message, ctx->rc);
    break;
  case GRN_CANCEL:
    error_class = mrb_class_get_under(mrb, module, "Cancel");
    grn_snprintf(message, MESSAGE_SIZE, MESSAGE_SIZE,
                 "cancel: <%s>(%d)",
                 utf8_error_message, ctx->rc);
    break;
  case GRN_WINDOW_FUNCTION_ERROR:
    error_class = mrb_class_get_under(mrb, module, "WindowFunctionError");
    grn_snprintf(message, MESSAGE_SIZE, MESSAGE_SIZE,
                 "window function error: <%s>(%d)",
                 utf8_error_message, ctx->rc);
    break;
  case GRN_ZSTD_ERROR:
    error_class = mrb_class_get_under(mrb, module, "ZstdError");
    grn_snprintf(message, MESSAGE_SIZE, MESSAGE_SIZE,
                 "Zstandard error: <%s>(%d)",
                 utf8_error_message, ctx->rc);
    break;
  case GRN_CONNECTION_RESET:
    error_class = mrb_class_get_under(mrb, module, "ConnectionReset");
    grn_snprintf(message, MESSAGE_SIZE, MESSAGE_SIZE,
                 "Connection reset: <%s>(%d)",
                 utf8_error_message, ctx->rc);
    break;
  }

  if (!error_class) {
    error_class = mrb_class_get_under(mrb, module, "Error");
    grn_snprintf(message, MESSAGE_SIZE, MESSAGE_SIZE,
                 "unsupported error: <%s>(%d)",
                 utf8_error_message, ctx->rc);
  }

  grn_encoding_converted_free(ctx, utf8_error_message);
#undef MESSAGE_SIZE

  return mrb_exc_new_str(mrb, error_class, mrb_str_new_cstr(mrb, message));
}

void
grn_mrb_ctx_check(mrb_state *mrb)
{
  mrb_value mrb_exception;

  mrb_exception = grn_mrb_ctx_to_exception(mrb);
  if (mrb_nil_p(mrb_exception)) {
    return;
  }

  mrb_exc_raise(mrb, mrb_exception);
}

void
grn_mrb_ctx_init(grn_ctx *ctx)
{
  grn_mrb_data *data = &(ctx->impl->mrb);
  mrb_state *mrb = data->state;
  struct RClass *module = data->module;
  struct RClass *klass;

  klass = mrb_define_class_under(mrb, module, "Context", mrb->object_class);
  MRB_SET_INSTANCE_TT(klass, MRB_TT_DATA);

  mrb_define_class_method(mrb, klass, "instance",
                          ctx_class_instance, MRB_ARGS_NONE());

  mrb_define_method(mrb, klass, "[]", ctx_array_reference, MRB_ARGS_REQ(1));
  mrb_define_method(mrb, klass, "rc", ctx_get_rc, MRB_ARGS_NONE());
  mrb_define_method(mrb, klass, "rc=", ctx_set_rc, MRB_ARGS_REQ(1));
  mrb_define_method(mrb, klass, "error_level", ctx_get_error_level,
                    MRB_ARGS_NONE());
  mrb_define_method(mrb, klass, "error_level=", ctx_set_error_level,
                    MRB_ARGS_REQ(1));
  mrb_define_method(mrb, klass, "error_file", ctx_get_error_file,
                    MRB_ARGS_NONE());
  mrb_define_method(mrb, klass, "error_file=", ctx_set_error_file,
                    MRB_ARGS_REQ(1));
  mrb_define_method(mrb, klass, "error_line", ctx_get_error_line,
                    MRB_ARGS_NONE());
  mrb_define_method(mrb, klass, "error_line=", ctx_set_error_line,
                    MRB_ARGS_REQ(1));
  mrb_define_method(mrb, klass, "error_method", ctx_get_error_method,
                    MRB_ARGS_NONE());
  mrb_define_method(mrb, klass, "error_method=", ctx_set_error_method,
                    MRB_ARGS_REQ(1));
  mrb_define_method(mrb, klass, "error_message", ctx_get_error_message,
                    MRB_ARGS_NONE());
  mrb_define_method(mrb, klass, "error_message=", ctx_set_error_message,
                    MRB_ARGS_REQ(1));
  mrb_define_method(mrb, klass, "clear_error", ctx_clear_error, MRB_ARGS_NONE());
  mrb_define_method(mrb, klass, "command_version",
                    ctx_get_command_version, MRB_ARGS_NONE());
  mrb_define_method(mrb, klass, "command_version=",
                    ctx_set_command_version, MRB_ARGS_REQ(1));

  mrb_define_method(mrb, klass, "output",
                    ctx_get_output, MRB_ARGS_NONE());
  mrb_define_method(mrb, klass, "output=",
                    ctx_set_output, MRB_ARGS_REQ(1));

  mrb_define_method(mrb, klass, "output_type",
                    ctx_get_output_type, MRB_ARGS_NONE());
  mrb_define_method(mrb, klass, "output_type=",
                    ctx_set_output_type, MRB_ARGS_REQ(1));

  mrb_define_method(mrb, klass, "database", ctx_get_database,
                    MRB_ARGS_NONE());

  mrb_define_method(mrb, klass, "opened?", ctx_is_opened,
                    MRB_ARGS_REQ(1));
}
#endif
