/*
 * Copyright (c) 2025 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 <gtest/gtest.h>
#include "modifier_test_base.h"
#include "modifiers_test_utils.h"
#include "core/interfaces/native/utility/ace_engine_types.h"
#include "core/interfaces/native/utility/converter.h"
#include "core/interfaces/native/utility/reverse_converter.h"
#include "core/components/search/search_theme.h"
#include "core/components/text_field/textfield_theme.h"
#include "core/components/theme/icon_theme.h"

#include "test/unittest/capi/utils/custom_node_builder_test_helper.h"
#include "core/components_ng/pattern/text_field/text_field_pattern.h"

namespace OHOS::Ace::NG {

using namespace testing;
using namespace testing::ext;
using namespace Converter;

const auto ATTRIBUTE_KEYBOARD_APPEARANCE_NAME = "keyboardAppearance";
const auto ATTRIBUTE_KEYBOARD_APPEARANCE_DEFAULT_VALUE = "0";
const auto ATTRIBUTE_AUTOCAPITALIZATION_MODE_NAME = "autocapitalizationMode";
const auto ATTRIBUTE_AUTOCAPITALIZATION_MODE_DEFAULT_VALUE = "AutoCapitalizationMode.NONE";

std::vector<std::tuple<std::string, Opt_AutoCapitalizationMode, std::string>>
    testFixtureEnumAutoCapitalizationModeTestPlan = {
        { "AutoCapitalizationMode.NONE", Converter::ArkValue<Opt_AutoCapitalizationMode>(AutoCapitalizationMode::NONE),
            "AutoCapitalizationMode.NONE" },
        { "AutoCapitalizationMode.WORDS",
            Converter::ArkValue<Opt_AutoCapitalizationMode>(AutoCapitalizationMode::WORDS),
            "AutoCapitalizationMode.WORDS" },
        { "AutoCapitalizationMode.SENTENCES",
            Converter::ArkValue<Opt_AutoCapitalizationMode>(AutoCapitalizationMode::SENTENCES),
            "AutoCapitalizationMode.SENTENCES" },
        { "AutoCapitalizationMode.ALL_CHARACTERS",
            Converter::ArkValue<Opt_AutoCapitalizationMode>(AutoCapitalizationMode::ALL_CHARACTERS),
            "AutoCapitalizationMode.ALL_CHARACTERS" },
        { "AutoCapitalizationMode.INVALID",
            Converter::ArkValue<Opt_AutoCapitalizationMode>(Converter::INVALID_ENUM_VAL<Ark_AutoCapitalizationMode>),
            ATTRIBUTE_AUTOCAPITALIZATION_MODE_DEFAULT_VALUE },
    };

class SearchModifierTest2 : public ModifierTestBase<GENERATED_ArkUISearchModifier,
                               &GENERATED_ArkUINodeModifiers::getSearchModifier, GENERATED_ARKUI_SEARCH> {
public:
    static void SetUpTestCase()
    {
        ModifierTestBase::SetUpTestCase();
        SetupTheme<SearchTheme>();
        SetupTheme<TextFieldTheme>();
        SetupTheme<IconTheme>();
    }
};

/*
 * @tc.name: setAutoCapitalizationModeTest
 * @tc.desc:
 * @tc.type: FUNC
 */
HWTEST_F(SearchModifierTest2, setAutoCapitalizationModeTest, TestSize.Level1)
{
    ASSERT_TRUE(modifier_->setAutoCapitalizationMode);
    auto jsonValue = GetJsonValue(node_);
    auto resultStr = GetAttrValue<std::string>(jsonValue, ATTRIBUTE_AUTOCAPITALIZATION_MODE_NAME);
    EXPECT_THAT(resultStr, Eq(ATTRIBUTE_AUTOCAPITALIZATION_MODE_DEFAULT_VALUE))
                                        << "Method: setAutoCapitalizationMode, attribute: keyboardAppearance";
    auto checkValue = [this](const std::string& input, const std::string& expectedStr,
                          const Opt_AutoCapitalizationMode& value) {
        modifier_->setAutoCapitalizationMode(node_, &value);
        auto jsonValue = GetJsonValue(node_);
        auto resultStr = GetAttrValue<std::string>(jsonValue, ATTRIBUTE_AUTOCAPITALIZATION_MODE_NAME);
        EXPECT_THAT(resultStr, Eq(expectedStr)) << "Input value is: " << input
                                        << ", method: setAutoCapitalizationMode, attribute: keyboardAppearance";
    };
    for (auto& [input, value, expected] : testFixtureEnumAutoCapitalizationModeTestPlan) {
        checkValue(input, expected, value);
    }
}

/*
 * @tc.name: setCustomKeyboardTestCustomNodeBuilder
 * @tc.desc:
 * @tc.type: FUNC
 */
HWTEST_F(SearchModifierTest2, setCustomKeyboardTestCustomNodeBuilder, TestSize.Level1)
{
    ASSERT_NE(modifier_->setCustomKeyboard, nullptr);
    auto frameNode = reinterpret_cast<FrameNode*>(node_);
    ASSERT_NE(frameNode, nullptr);

    int callsCount = 0;
    CustomNodeBuilderTestHelper<SearchModifierTest2> builderHelper(this, frameNode);

    const auto builder = Converter::ArkUnion<Opt_Union_CustomBuilder_ComponentContentBase, CustomNodeBuilder>(
        Converter::ArkValue<Opt_CustomNodeBuilder>(builderHelper.GetBuilder()).value);
    modifier_->setCustomKeyboard(frameNode, &builder, nullptr);

    auto textFieldChild = AceType::DynamicCast<FrameNode>(frameNode->GetChildren().front());
    ASSERT_NE(textFieldChild, nullptr);
    auto textFieldPattern = textFieldChild->GetPattern<TextFieldPattern>();
    ASSERT_NE(textFieldPattern, nullptr);
    EXPECT_TRUE(textFieldPattern->HasCustomKeyboard());
    EXPECT_TRUE(textFieldPattern->RequestCustomKeyboard());
    EXPECT_EQ(builderHelper.GetCallsCountAsync(), ++callsCount);
}

/*
 * @tc.name: setCustomKeyboardTestCustomNodeBuilderKeyboardOptions
 * @tc.desc:
 * @tc.type: FUNC
 */
HWTEST_F(SearchModifierTest2, setCustomKeyboardTestCustomNodeBuilderKeyboardOptions, TestSize.Level1)
{
    ASSERT_NE(modifier_->setCustomKeyboard, nullptr);
    auto frameNode = reinterpret_cast<FrameNode*>(node_);
    ASSERT_NE(frameNode, nullptr);

    KeyboardOptions keyboardOptions = {.supportAvoidance = true};
    auto optKeyboardOptions = Converter::ArkValue<Opt_KeyboardOptions>(keyboardOptions);

    int callsCount = 0;
    CustomNodeBuilderTestHelper<SearchModifierTest2> builderHelper(this, frameNode);
    const auto builder = Converter::ArkUnion<Opt_Union_CustomBuilder_ComponentContentBase, CustomNodeBuilder>(
        Converter::ArkValue<Opt_CustomNodeBuilder>(builderHelper.GetBuilder()).value);
    modifier_->setCustomKeyboard(node_, &builder, &optKeyboardOptions);

    auto textFieldChild = AceType::DynamicCast<FrameNode>(frameNode->GetChildren().front());
    ASSERT_NE(textFieldChild, nullptr);
    auto textFieldPattern = textFieldChild->GetPattern<TextFieldPattern>();
    ASSERT_NE(textFieldPattern, nullptr);
    EXPECT_TRUE(textFieldPattern->HasCustomKeyboard());
    EXPECT_TRUE(textFieldPattern->RequestCustomKeyboard());
    EXPECT_EQ(builderHelper.GetCallsCountAsync(), ++callsCount);
}

/*
 * @tc.name: setKeyboardAppearanceTestDefaultValues
 * @tc.desc:
 * @tc.type: FUNC
 */
HWTEST_F(SearchModifierTest2, setKeyboardAppearanceTestDefaultValues, TestSize.Level1)
{
    std::unique_ptr<JsonValue> jsonValue = GetJsonValue(node_);
    auto resultStr = GetAttrValue<std::string>(jsonValue, ATTRIBUTE_KEYBOARD_APPEARANCE_NAME);
    EXPECT_THAT(resultStr, Eq(ATTRIBUTE_KEYBOARD_APPEARANCE_DEFAULT_VALUE))
        << "Default value for attribute 'keyboardAppearance'";
}

std::vector<std::tuple<std::string, Opt_KeyboardAppearance, std::string>> testFixtureEnumKeyboardAppearanceTestPlan = {
    { "KeyboardAppearance.NONE_IMMERSIVE",
        Converter::ArkValue<Opt_KeyboardAppearance>(ARK_KEYBOARD_APPEARANCE_NONE_IMMERSIVE), "0" },
    { "KeyboardAppearance.IMMERSIVE",
        Converter::ArkValue<Opt_KeyboardAppearance>(ARK_KEYBOARD_APPEARANCE_IMMERSIVE), "1" },
    { "KeyboardAppearance.LIGHT_IMMERSIVE",
        Converter::ArkValue<Opt_KeyboardAppearance>(ARK_KEYBOARD_APPEARANCE_LIGHT_IMMERSIVE), "2" },
    { "-1", Converter::ArkValue<Opt_KeyboardAppearance>(static_cast<Ark_KeyboardAppearance>(-1)),
        ATTRIBUTE_KEYBOARD_APPEARANCE_DEFAULT_VALUE },
    { "INT_MAX", Converter::ArkValue<Opt_KeyboardAppearance>(static_cast<Ark_KeyboardAppearance>(INT_MAX)),
        ATTRIBUTE_KEYBOARD_APPEARANCE_DEFAULT_VALUE },
};

/*
 * @tc.name: setKeyboardAppearanceTestValues
 * @tc.desc:
 * @tc.type: FUNC
 */
HWTEST_F(SearchModifierTest2, setKeyboardAppearanceTestValues, TestSize.Level1)
{
    ASSERT_NE(modifier_->setKeyboardAppearance, nullptr);
    auto checkValue = [this](
                        const std::string& input, const std::string& expectedStr, const Opt_KeyboardAppearance& value) {
        Opt_KeyboardAppearance inputValueKeyboardAppearance = value;
        modifier_->setKeyboardAppearance(node_, &inputValueKeyboardAppearance);
        auto jsonValue = GetJsonValue(node_);
        auto resultStr = GetAttrValue<std::string>(jsonValue, ATTRIBUTE_KEYBOARD_APPEARANCE_NAME);
        EXPECT_THAT(resultStr, Eq(expectedStr)) << "Input value is: " << input
                                        << ", method: setKeyboardAppearance, attribute: keyboardAppearance";
    };
    for (auto& [input, value, expected] : testFixtureEnumKeyboardAppearanceTestPlan) {
        checkValue(input, expected, value);
    }
}


namespace Converter {
template<>
ChangeValueInfo Convert(const Ark_EditableTextChangeValue& parameter)
{
    ChangeValueInfo info;
    info.value = Converter::Convert<std::u16string>(parameter.content);
    info.previewText = Converter::Convert<PreviewText>(parameter.previewText.value);
    info.rangeBefore = Converter::Convert<TextRange>(parameter.options.value.rangeBefore);
    info.rangeAfter = Converter::Convert<TextRange>(parameter.options.value.rangeAfter);
    info.oldContent = Converter::Convert<std::u16string>(parameter.options.value.oldContent);
    info.oldPreviewText =  Converter::Convert<PreviewText>(parameter.options.value.oldPreviewText);
    return info;
}
} // namespace Converter

/*
 * @tc.name: setOnWillChangeTest
 * @tc.desc:
 * @tc.type: FUNC
 */
HWTEST_F(SearchModifierTest2, setOnWillChangeTest, TestSize.Level1)
{
    ASSERT_NE(modifier_->setOnWillChange, nullptr);
    auto frameNode = reinterpret_cast<FrameNode*>(node_);
    ASSERT_NE(frameNode, nullptr);
    static std::optional<ChangeValueInfo> checkEvent = std::nullopt;
    auto expectedChangeValueInfo = ChangeValueInfo {
        .value = u"test content",
        .previewText = { .offset = 2, .value = u"previewText" },
        .oldPreviewText = { .offset = 1, .value = u"oldPreviewText" },
        .oldContent = u"oldContent",
        .rangeBefore = { .start = 1, .end = 6 },
        .rangeAfter = { .start = 2, .end = 5 },
    };

    auto inputCallback = [] (Ark_VMContext context, const Ark_Int32 resourceId,
        const Ark_EditableTextChangeValue parameter, const Callback_Boolean_Void continuation) {
        checkEvent = Converter::Convert<ChangeValueInfo>(parameter);
        CallbackHelper(continuation).InvokeSync(Converter::ArkValue<Ark_Boolean>(true));
    };
    auto func = Converter::ArkCallback<Opt_Callback_EditableTextChangeValue_Boolean>(inputCallback);
    modifier_->setOnWillChange(node_, &func);

    auto searchTextField = AceType::DynamicCast<FrameNode>(frameNode->GetChildren().front());
    ASSERT_NE(searchTextField, nullptr);
    auto eventHub = searchTextField->GetEventHub<TextFieldEventHub>();
    ASSERT_NE(eventHub, nullptr);
    auto result = eventHub->FireOnWillChangeEvent(expectedChangeValueInfo);
    EXPECT_TRUE(result);
    ASSERT_TRUE(checkEvent);
    EXPECT_EQ(checkEvent->value, expectedChangeValueInfo.value);
    EXPECT_EQ(checkEvent->previewText.offset, expectedChangeValueInfo.previewText.offset);
    EXPECT_EQ(checkEvent->previewText.value, expectedChangeValueInfo.previewText.value);
    EXPECT_EQ(checkEvent->oldPreviewText.offset, expectedChangeValueInfo.oldPreviewText.offset);
    EXPECT_EQ(checkEvent->oldPreviewText.value, expectedChangeValueInfo.oldPreviewText.value);
    EXPECT_EQ(checkEvent->oldContent, expectedChangeValueInfo.oldContent);
    EXPECT_EQ(checkEvent->rangeBefore.start, expectedChangeValueInfo.rangeBefore.start);
    EXPECT_EQ(checkEvent->rangeBefore.end, expectedChangeValueInfo.rangeBefore.end);
    EXPECT_EQ(checkEvent->rangeAfter.start, expectedChangeValueInfo.rangeAfter.start);
    EXPECT_EQ(checkEvent->rangeAfter.end, expectedChangeValueInfo.rangeAfter.end);
}
} // namespace OHOS::Ace::NG