// Licensed to the Apache Software Foundation (ASF) under one
// or more contributor license agreements.  See the NOTICE file
// distributed with this work for additional information
// regarding copyright ownership.  The ASF licenses this file
// to you 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.

#include "testutil/array_utils.h"

#include "common/status.h"
#include "exprs/anyval_util.h"
#include "gen_cpp/olap_file.pb.h"
#include "runtime/collection_value.h"
#include "runtime/free_pool.hpp"
#include "runtime/mem_pool.h"
#include "runtime/memory/mem_tracker.h"
#include "udf/udf_internal.h"

namespace doris {

using TypeDesc = FunctionContext::TypeDesc;

void ArrayUtils::prepare_context(FunctionContext& context, MemPool& mem_pool,
                                 const ColumnPB& column_pb) {
    auto function_type_desc = create_function_type_desc(column_pb);
    context.impl()->_return_type = function_type_desc;
    context.impl()->_pool = new FreePool(&mem_pool);
}

TypeDesc ArrayUtils::create_function_type_desc(const ColumnPB& column_pb) {
    TypeDesc type_desc;
    type_desc.len = column_pb.length();
    type_desc.precision = column_pb.precision();
    type_desc.scale = column_pb.frac();
    if (column_pb.type() == "ARRAY") {
        type_desc.type = FunctionContext::TYPE_ARRAY;
    } else if (column_pb.type() == "BOOLEAN") {
        type_desc.type = FunctionContext::TYPE_BOOLEAN;
    } else if (column_pb.type() == "TINYINT") {
        type_desc.type = FunctionContext::TYPE_TINYINT;
    } else if (column_pb.type() == "SMALLINT") {
        type_desc.type = FunctionContext::TYPE_SMALLINT;
    } else if (column_pb.type() == "INT") {
        type_desc.type = FunctionContext::TYPE_INT;
    } else if (column_pb.type() == "BIGINT") {
        type_desc.type = FunctionContext::TYPE_BIGINT;
    } else if (column_pb.type() == "LARGEINT") {
        type_desc.type = FunctionContext::TYPE_LARGEINT;
    } else if (column_pb.type() == "FLOAT") {
        type_desc.type = FunctionContext::TYPE_FLOAT;
    } else if (column_pb.type() == "DOUBLE") {
        type_desc.type = FunctionContext::TYPE_DOUBLE;
    } else if (column_pb.type() == "CHAR") {
        type_desc.type = FunctionContext::TYPE_CHAR;
    } else if (column_pb.type() == "VARCHAR") {
        type_desc.type = FunctionContext::TYPE_VARCHAR;
    } else if (column_pb.type() == "STRING") {
        type_desc.type = FunctionContext::TYPE_STRING;
    } else if (column_pb.type() == "DATE") {
        type_desc.type = FunctionContext::TYPE_DATE;
    } else if (column_pb.type() == "DATETIME") {
        type_desc.type = FunctionContext::TYPE_DATETIME;
    } else if (column_pb.type().compare(0, 7, "DECIMAL") == 0) {
        type_desc.type = FunctionContext::TYPE_DECIMALV2;
    } else {
        DCHECK(false) << "Failed to create function type descriptor.";
    }
    for (const auto& sub_column_pb : column_pb.children_columns()) {
        type_desc.children.push_back(create_function_type_desc(sub_column_pb));
    }
    return type_desc;
}

} // namespace doris