/*
 * Copyright (c) 2021 Huawei Device Co., Ltd.
 * Licensed 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 "script_function.h"
#include "script_context.h"
#include "script_interpreter.h"
#include "script_manager.h"
#include "script_param.h"
#include "script_utils.h"

using namespace std;

namespace Uscript {
ScriptFunction::~ScriptFunction()
{
    delete params_;
    delete statements_;
}

UScriptValuePtr ScriptFunction::Execute(ScriptInterpreter &inter,
    UScriptContextPtr context, ScriptParams *inputParams)
{
    INTERPRETER_LOGI(inter, context, "ScriptFunction execute %s", functionName_.c_str());
    UScriptContextPtr funcContext = std::make_shared<UScriptInterpretContext>();
    if (funcContext == nullptr) {
        USCRIPT_LOGE("[interpreter-%d] Fail to create context for function %s",
            inter.GetInstanceId(), functionName_.c_str());
        return std::make_shared<ErrorValue>(USCRIPT_ERROR_CREATE_OBJ);
    }

    if (inputParams == nullptr || params_ == nullptr) {
        if (params_ != nullptr || inputParams != nullptr) {
            USCRIPT_LOGE("[interpreter-%d] ScriptFunction::Execute param not match %s",
                inter.GetInstanceId(), functionName_.c_str());
            return std::make_shared<ErrorValue>(USCRIPT_ERROR_INTERPRET);
        }
    } else {
        if (params_->GetParams().size() != inputParams->GetParams().size()) {
            USCRIPT_LOGE("[interpreter-%d] ScriptFunction::Execute param not match %s",
                inter.GetInstanceId(), functionName_.c_str());
            return std::make_shared<ErrorValue>(USCRIPT_ERROR_INTERPRET);
        }

        size_t index = 0;
        std::vector<std::string> paramNames = GetParamNames(inter, context);
        for (auto expression : inputParams->GetParams()) {
            UScriptValuePtr var = expression->Execute(inter, context);
            if (var == nullptr || var->GetValueType() == UScriptValue::VALUE_TYPE_ERROR) {
                USCRIPT_LOGE("[interpreter-%d] ScriptFunction::Execute fail to computer param %s",
                    inter.GetInstanceId(), functionName_.c_str());
                return std::make_shared<ErrorValue>(USCRIPT_NOTEXIST_INSTRUCTION);
            }
            if (index >= paramNames.size()) {
                USCRIPT_LOGE("[interpreter-%d] ScriptFunction::Execute invalid index %zu param %s",
                    inter.GetInstanceId(), index, functionName_.c_str());
                return std::make_shared<ErrorValue>(USCRIPT_NOTEXIST_INSTRUCTION);
            }
            funcContext->UpdateVariables(inter, var, paramNames, index);
        }
    }
    UScriptStatementResult result = statements_->Execute(inter, funcContext);
    INTERPRETER_LOGI(inter, context, "ScriptFunction execute %s result %s", functionName_.c_str(),
        UScriptStatementResult::ScriptToString(&result).c_str());
    return result.GetResultValue();
}

std::vector<std::string> ScriptFunction::GetParamNames(ScriptInterpreter &inter,
    UScriptContextPtr context) const
{
    int32_t ret;
    std::vector<std::string> names;
    for (auto expression : params_->GetParams()) {
        std::string varName;
        ret = IdentifierExpression::GetIdentifierName(expression, varName);
        if (ret != USCRIPT_SUCCESS) {
            INTERPRETER_LOGE(inter, context, "Fail to get param name %s", functionName_.c_str());
            return names;
        }
        names.push_back(varName);
    }
    return names;
}
} // namespace Uscript