/**
 * Copyright (c) 2021 OceanBase
 * OceanBase CE is licensed under Mulan PubL v2.
 * You can use this software according to the terms and conditions of the Mulan PubL v2.
 * You may obtain a copy of Mulan PubL v2 at:
 *          http://license.coscl.org.cn/MulanPubL-2.0
 * THIS SOFTWARE IS PROVIDED ON AN "AS IS" BASIS, WITHOUT WARRANTIES OF ANY KIND,
 * EITHER EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO NON-INFRINGEMENT,
 * MERCHANTABILITY OR FIT FOR A PARTICULAR PURPOSE.
 * See the Mulan PubL v2 for more details.
 */

#define USING_LOG_PREFIX SQL_ENG
#include <string.h>
#include "sql/parser/ob_item_type.h"
#include "sql/engine/expr/ob_expr_utl_i18n.h"
#include "sql/engine/expr/ob_expr_hex.h"
#include "sql/engine/expr/ob_expr_util.h"
#include "sql/engine/expr/ob_expr_operator.h"
#include "share/object/ob_obj_cast.h"
#include "lib/oblog/ob_log.h"
#include "sql/session/ob_sql_session_info.h"
using namespace oceanbase::common;

namespace oceanbase {
namespace sql {

// UTL_I18N.STRING_TO_RAW begin
ObExprUtlI18nStringToRaw::ObExprUtlI18nStringToRaw(ObIAllocator& alloc)
    : ObStringExprOperator(alloc, T_FUN_SYS_UTL_I18N_STRING_TO_RAW, N_UTL_I18N_STRING_TO_RAW, 2)
{}

ObExprUtlI18nStringToRaw::~ObExprUtlI18nStringToRaw()
{}

int ObExprUtlI18nStringToRaw::calc_result_type2(
    ObExprResType& type, ObExprResType& type1, ObExprResType& type2, common::ObExprTypeCtx& type_ctx) const
{
  UNUSED(type_ctx);
  int ret = OB_SUCCESS;

  if (ob_is_null(type1.get_type())) {
    type.set_null();
  } else {
    type1.set_calc_type(common::ObVarcharType);
    type.set_raw();
    type.set_collation_level(common::CS_LEVEL_NUMERIC);
    if (ob_is_null(type2.get_type())) {
    } else {
      type2.set_calc_type(common::ObVarcharType);
    }
  }

  return ret;
}

int ObExprUtlI18nStringToRaw::calc(common::ObObj& result, const common::ObObj& obj1, const common::ObObj& obj2,
    common::ObCastCtx& cast_ctx, const ObSQLSessionInfo* session_info)
{
  int ret = OB_SUCCESS;

  if (obj1.is_null()) {
    result.set_null();
  } else if (OB_UNLIKELY(!obj1.is_varchar())) {
    ret = OB_ERR_UNEXPECTED;
    LOG_WARN("input is not varchar", K(ret), K(obj1), K(obj2));
  } else {
    if (OB_ISNULL(cast_ctx.allocator_v2_)) {
      ret = OB_INVALID_ARGUMENT;
      LOG_WARN("allocator in cast ctx is NULL", K(ret));
    } else if (CS_TYPE_INVALID == obj1.get_collation_type()) {
      ret = OB_ERR_UNEXPECTED;
      LOG_WARN("collation type of input is invalid", K(ret), K(obj1));
    } else {
      ObCollationType dest_collation = CS_TYPE_INVALID;
      ObString dest_str;
      if (obj2.is_null()) {
        dest_collation = session_info->get_nls_collation();
      } else {  // obj2.is_varchar()
        dest_collation =
            ObCharset::get_default_collation_oracle(ObCharset::charset_type_by_name_oracle(obj2.get_string()));
      }
      if (OB_FAIL(ret)) {
      } else if (CS_TYPE_INVALID != dest_collation) {
        if (OB_FAIL(ObExprUtil::convert_string_collation(
                obj1.get_string(), obj1.get_collation_type(), dest_str, dest_collation, *cast_ctx.allocator_v2_))) {
          LOG_WARN("fail to convert string collation", K(ret), K(obj1), K(dest_str), K(dest_collation));
        } else {
          result.set_raw(dest_str.ptr(), dest_str.length());
        }
      } else {
        result.set_null();
      }
    }
  }

  return ret;
}

int ObExprUtlI18nStringToRaw::calc_result2(
    common::ObObj& result, const common::ObObj& obj1, const common::ObObj& obj2, common::ObExprCtx& expr_ctx) const
{
  int ret = OB_SUCCESS;
  const ObSQLSessionInfo* session_info = NULL;

  if (OB_ISNULL(session_info = expr_ctx.my_session_)) {
    ret = OB_ERR_UNEXPECTED;
    LOG_WARN("session info is null", K(ret));
  } else if (OB_ISNULL(expr_ctx.calc_buf_)) {
    ret = OB_NOT_INIT;
    LOG_WARN("varchar buffer not init", K(ret));
  } else {
    EXPR_DEFINE_CAST_CTX(expr_ctx, CM_NONE);
    if (OB_FAIL(calc(result, obj1, obj2, cast_ctx, session_info))) {
      LOG_WARN("fail to calc", K(ret), K(obj1), K(obj2));
    }
  }

  return ret;
}
// UTL_I18N.STRING_TO_RAW end

// UTL_I18N.RAW_TO_CHAR begin
ObExprUtlI18nRawToChar::ObExprUtlI18nRawToChar(ObIAllocator& alloc)
    : ObStringExprOperator(alloc, T_FUN_SYS_UTL_I18N_RAW_TO_CHAR, N_UTL_I18N_RAW_TO_CHAR, 2)
{}

ObExprUtlI18nRawToChar::~ObExprUtlI18nRawToChar()
{}

int ObExprUtlI18nRawToChar::calc_result_type2(
    ObExprResType& type, ObExprResType& type1, ObExprResType& type2, common::ObExprTypeCtx& type_ctx) const
{
  UNUSED(type_ctx);
  int ret = OB_SUCCESS;
  const ObBasicSessionInfo* session_info = NULL;
  if (OB_ISNULL(session_info = type_ctx.get_session())) {
    ret = OB_ERR_UNEXPECTED;
    LOG_WARN("session info is null", K(ret));
  } else if (ob_is_null(type1.get_type())) {
    type.set_null();
  } else if (ob_is_string_type(type1.get_type()) || ob_is_raw(type1.get_type())) {
    type.set_varchar();
    type.set_collation_level(common::CS_LEVEL_COERCIBLE);
    type.set_collation_type(session_info->get_nls_collation());
    if (ob_is_null(type2.get_type())) {
    } else {
      type2.set_calc_type(common::ObVarcharType);
    }
  } else {
    ret = OB_ERR_WRONG_FUNC_ARGUMENTS_TYPE;
    LOG_WARN("wrong type of arguments in function", K(ret), K(type1));
    ObString func_name("RAW_TO_CHAR");
    LOG_USER_ERROR(OB_ERR_WRONG_FUNC_ARGUMENTS_TYPE, func_name.length(), func_name.ptr());
  }

  return ret;
}

int ObExprUtlI18nRawToChar::calc(common::ObObj& result, const common::ObObj& obj1, const common::ObObj& obj2,
    common::ObCastCtx& cast_ctx, const ObSQLSessionInfo* session_info)
{
  int ret = OB_SUCCESS;

  if (obj1.is_null()) {
    result.set_null();
  } else if (OB_UNLIKELY(!obj1.is_raw() && !obj1.is_string_type())) {
    ret = OB_ERR_UNEXPECTED;
    LOG_WARN("wrong number or types of arguments in function", K(ret), K(obj1));
  } else if (OB_ISNULL(cast_ctx.allocator_v2_)) {
    ret = OB_INVALID_ARGUMENT;
    LOG_WARN("allocator in cast ctx is NULL", K(ret));
  } else {
    ObObj tmp_obj;
    ObString src_str;
    ObString dest_str;
    ObCollationType src_collation = CS_TYPE_INVALID;
    if (obj1.is_raw()) {
      src_str = obj1.get_string();
    } else {
      if (OB_FAIL(ObHexUtils::hextoraw(obj1, cast_ctx, tmp_obj))) {
        LOG_WARN("fail to hextoraw", K(ret), K(obj1));
      } else {
        src_str = tmp_obj.get_string();
      }
    }
    if (OB_FAIL(ret)) {
    } else if (obj2.is_null()) {
      src_collation = session_info->get_nls_collation();
    } else {  // obj2.is_varchar()
      src_collation =
          ObCharset::get_default_collation_oracle(ObCharset::charset_type_by_name_oracle(obj2.get_string()));
    }
    if (OB_FAIL(ret)) {
    } else if (CS_TYPE_INVALID != src_collation) {
      if (OB_FAIL(ObExprUtil::convert_string_collation(
              src_str, src_collation, dest_str, session_info->get_nls_collation(), *cast_ctx.allocator_v2_))) {
        LOG_WARN("fail to convert string collation", K(ret), K(obj1), K(src_collation));
      } else {
        result.set_varchar(dest_str.ptr(), dest_str.length());
        result.set_collation_type(session_info->get_nls_collation());
      }
    } else {
      result.set_null();
    }
  }

  return ret;
}

int ObExprUtlI18nRawToChar::calc_result2(
    common::ObObj& result, const common::ObObj& obj1, const common::ObObj& obj2, common::ObExprCtx& expr_ctx) const
{
  int ret = OB_SUCCESS;
  const ObSQLSessionInfo* session_info = NULL;

  if (OB_ISNULL(session_info = expr_ctx.my_session_)) {
    ret = OB_ERR_UNEXPECTED;
    LOG_WARN("session info is null", K(ret));
  } else if (OB_ISNULL(expr_ctx.calc_buf_)) {
    ret = OB_NOT_INIT;
    LOG_WARN("varchar buffer not init", K(ret));
  } else {
    EXPR_DEFINE_CAST_CTX(expr_ctx, CM_NONE);
    if (OB_FAIL(calc(result, obj1, obj2, cast_ctx, session_info))) {
      LOG_WARN("fail to calc", K(ret), K(obj1), K(obj2));
    }
  }

  return ret;
}
// UTL_I18N.RAW_TO_CHAR end

}  // namespace sql
}  // namespace oceanbase