/*
 * Copyright (c) 2022 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.
 */

#ifndef UI_TEST_CUSTOM_INPUT_METHOD_H
#define UI_TEST_CUSTOM_INPUT_METHOD_H

#include "common/input_method_manager.h"
#include "components/ui_edit_text.h"
#include "components/ui_label.h"
#include "components/ui_label_button.h"
#include "components/ui_scroll_view.h"
#include "layout/flex_layout.h"
#include "ui_test.h"

namespace OHOS {
enum class KeyboardType {
    LOW_CASE,
    UPPER_CASE,
    NUMBER,
    SYMBOL
};
class CustomInputMethod : public UIView::OnClickListener, public InputMethodManager::InputMethodListener {
public:
    class UIEditTextEx : public UIEditText {
        // override the view type, so the FocusManager not invoke InputMethod onShow function
        UIViewType GetViewType() const override
        {
            return UI_NUMBER_MAX;
        }

        bool OnPressEvent(const PressEvent& event) override
        {
            return UIView::OnPressEvent(event);
        }
    };

    CustomInputMethod() {}
    ~CustomInputMethod() {}

    bool OnClick(UIView& view, const ClickEvent& event) override;
    void OnShow(InputMethodManager::InputMethodParam& param) override;
    void OnHide() override;

private:
    void SetupView(KeyboardType type);
    void TearDownView();
    UILabelButton* SetupButton(const char* title);
    void SetupKeyboard(KeyboardType type);
    void ChangeKeyboard(KeyboardType type);
    FlexLayout* SetupKeyRow(const char* name, int16_t width, int16_t height);
    void DealKeyEvent(UIView& view);

    UIEditTextEx* editView_ = nullptr;
    UILabelButton* inputTypeBtn_ = nullptr;
    UIViewGroup* container_ = nullptr;
    KeyboardType keyboardType_ = KeyboardType::LOW_CASE;
};
} // namespace OHOS
#endif // UI_TEST_CUSTOM_INPUT_METHOD_H