// Copyright 2020 Toyota Connected North America
// @copyright Copyright (c) 2022 Woven Alpha, Inc.
//
// 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 "text_input.h"

#include <shell/platform/common/json_method_codec.h>

#include <memory>
#include "engine.h"

TextInput::TextInput()
    : channel_(std::make_unique<flutter::MethodChannel<rapidjson::Document>>(
          this,
          kChannelName,
          &flutter::JsonMethodCodec::GetInstance())),
      client_id_(0) {}

void TextInput::SendStateUpdate(const flutter::TextInputModel& model) const {
  std::unique_ptr<std::vector<uint8_t>> result;
  auto args = std::make_unique<rapidjson::Document>(rapidjson::kArrayType);
  auto& allocator = args->GetAllocator();
  args->PushBack(client_id_, allocator);

  const flutter::TextRange selection = model.selection();
  rapidjson::Value editing_state(rapidjson::kObjectType);
  editing_state.AddMember(kComposingBaseKey, -1, allocator);
  editing_state.AddMember(kComposingExtentKey, -1, allocator);
  editing_state.AddMember(kSelectionAffinityKey, kAffinityDownstream,
                          allocator);
  editing_state.AddMember(kSelectionBaseKey, selection.base(), allocator);
  editing_state.AddMember(kSelectionExtentKey, selection.extent(), allocator);
  editing_state.AddMember(kSelectionIsDirectionalKey, false, allocator);
  editing_state.AddMember(
      kTextKey, rapidjson::Value(model.GetText().c_str(), allocator).Move(),
      allocator);
  args->PushBack(editing_state, allocator);

  channel_->InvokeMethod(kUpdateEditingStateMethod, std::move(args));
}

void TextInput::EnterPressed(flutter::TextInputModel* model) const {
  if (input_type_ == kMultilineInputType) {
    model->AddCodePoint('\n');
    SendStateUpdate(*model);
  }
  auto args = std::make_unique<rapidjson::Document>(rapidjson::kArrayType);
  auto& allocator = args->GetAllocator();
  args->PushBack(client_id_, allocator);
  args->PushBack(rapidjson::Value(input_action_.c_str(), allocator).Move(),
                 allocator);

  channel_->InvokeMethod(kPerformActionMethod, std::move(args));
}

void TextInput::SetEngine(const std::shared_ptr<Engine>& engine) {
  if (engine) {
    engine_ = engine;
    engine->SetTextInput(this);
  }
}

void TextInput::OnPlatformMessage(const FlutterPlatformMessage* message,
                                  void* userdata) {
  std::unique_ptr<std::vector<uint8_t>> result;
  const auto engine = static_cast<Engine*>(userdata);
  const auto text_input = engine->GetTextInput();
  auto& codec = flutter::JsonMethodCodec::GetInstance();
  const auto obj =
      codec.DecodeMethodCall(message->message, message->message_size);
  const auto method = obj->method_name();

  if (method == kShowMethod || method == kHideMethod) {
    // These methods are no-ops.
  } else if (method == kClearClientMethod) {
    text_input->active_model_ = nullptr;
  } else if (method == kSetClientMethod) {
    if (obj->arguments()->IsNull()) {
      auto res = codec.EncodeErrorEnvelope(kBadArgumentError,
                                           "Method invoked without args");
      goto done;
    }
    const rapidjson::Document& args = *(obj->arguments());

    const rapidjson::Value& client_id_json = args[0];
    const rapidjson::Value& client_config = args[1];
    if (client_id_json.IsNull()) {
      auto res = codec.EncodeErrorEnvelope(kBadArgumentError,
                                           "Could not set client, ID is null.");
      goto done;
    }
    if (client_config.IsNull()) {
      auto res = codec.EncodeErrorEnvelope(
          kBadArgumentError, "Could not set client, missing arguments.");
      goto done;
    }
    text_input->client_id_ = client_id_json.GetInt();
    text_input->input_action_ = "";
    const auto input_action_json = client_config.FindMember(kTextInputAction);
    if (input_action_json != client_config.MemberEnd() &&
        input_action_json->value.IsString()) {
      text_input->input_action_ = input_action_json->value.GetString();
    }
    text_input->input_type_ = "";
    const auto input_type_info_json = client_config.FindMember(kTextInputType);
    if (input_type_info_json != client_config.MemberEnd() &&
        input_type_info_json->value.IsObject()) {
      const auto input_type_json =
          input_type_info_json->value.FindMember(kTextInputTypeName);
      if (input_type_json != input_type_info_json->value.MemberEnd() &&
          input_type_json->value.IsString()) {
        text_input->input_type_ = input_type_json->value.GetString();
      }
    }
    text_input->active_model_ = std::make_unique<flutter::TextInputModel>();
  } else if (method == kSetEditingStateMethod) {
    if (!obj->arguments() || obj->arguments()->IsNull()) {
      result = codec.EncodeErrorEnvelope(kBadArgumentError,
                                         "Method invoked without args");
      goto done;
    }
    const rapidjson::Document& args = *(obj->arguments());

    if (text_input->active_model_ == nullptr) {
      result = codec.EncodeErrorEnvelope(
          kInternalConsistencyError,
          "Set editing state has been invoked, but no client is set.");
      goto done;
    }
    const auto text = args.FindMember(kTextKey);
    if (text == args.MemberEnd() || text->value.IsNull()) {
      result = codec.EncodeErrorEnvelope(
          kBadArgumentError,
          "Set editing state has been invoked, but without text.");
      goto done;
    }
    const auto selection_base = args.FindMember(kSelectionBaseKey);
    const auto selection_extent = args.FindMember(kSelectionExtentKey);
    if (selection_base == args.MemberEnd() || selection_base->value.IsNull() ||
        selection_extent == args.MemberEnd() ||
        selection_extent->value.IsNull()) {
      result = codec.EncodeErrorEnvelope(
          kInternalConsistencyError, "Selection base/extent values invalid.");
      goto done;
    }
    // Flutter uses -1/-1 for invalid; translate that to 0/0 for the model.
    int base = selection_base->value.GetInt();
    int extent = selection_extent->value.GetInt();
    if (base == -1 && extent == -1) {
      base = extent = 0;
    }
    text_input->active_model_->SetText(text->value.GetString());
    text_input->active_model_->SetSelection(flutter::TextRange(
        static_cast<size_t>(base), static_cast<size_t>(extent)));
  } else {
    engine->SendPlatformMessageResponse(message->response_handle, nullptr, 0);
    return;
  }
  // All error conditions return early, so if nothing has gone wrong indicate
  // success.
  result = codec.EncodeSuccessEnvelope();

done:
  engine->SendPlatformMessageResponse(message->response_handle, result->data(),
                                      result->size());
}

void TextInput::keyboard_handle_key(void* data,
                                    xkb_keysym_t keysym,
                                    uint32_t state) {
  const auto* text_input = static_cast<TextInput*>(data);

  if (text_input->active_model_ == nullptr) {
    return;
  }

  if (state == WL_KEYBOARD_KEY_STATE_PRESSED) {
    switch (keysym) {
      case XKB_KEY_Left:
      case XKB_KEY_KP_Left:
        if (text_input->active_model_->MoveCursorBack()) {
          text_input->SendStateUpdate(*text_input->active_model_);
        }
        break;
      case XKB_KEY_Right:
      case XKB_KEY_KP_Right:
        if (text_input->active_model_->MoveCursorForward()) {
          text_input->SendStateUpdate(*text_input->active_model_);
        }
        break;
      case XKB_KEY_End:
      case XKB_KEY_KP_End:
        text_input->active_model_->MoveCursorToEnd();
        text_input->SendStateUpdate(*text_input->active_model_);
        break;
      case XKB_KEY_Home:
      case XKB_KEY_KP_Home:
        text_input->active_model_->MoveCursorToBeginning();
        text_input->SendStateUpdate(*text_input->active_model_);
        break;
      case XKB_KEY_BackSpace:
        if (text_input->active_model_->Backspace()) {
          text_input->SendStateUpdate(*text_input->active_model_);
        }
        break;
      case XKB_KEY_Delete:
      case XKB_KEY_KP_Delete:
        if (text_input->active_model_->Delete()) {
          text_input->SendStateUpdate(*text_input->active_model_);
        }
        break;
      case XKB_KEY_ISO_Enter:
      case XKB_KEY_KP_Enter:
        text_input->EnterPressed(text_input->active_model_.get());
        break;
      case XKB_KEY_Shift_L:
      case XKB_KEY_Shift_R:
      case XKB_KEY_Control_L:
      case XKB_KEY_Control_R:
      case XKB_KEY_Caps_Lock:
      case XKB_KEY_Shift_Lock:
      case XKB_KEY_Meta_L:
      case XKB_KEY_Meta_R:
      case XKB_KEY_Alt_L:
      case XKB_KEY_Alt_R:
      case XKB_KEY_Super_L:
      case XKB_KEY_Super_R:
      case XKB_KEY_Hyper_L:
      case XKB_KEY_Hyper_R:
      case XKB_KEY_Tab:
      case XKB_KEY_Linefeed:
      case XKB_KEY_Clear:
      case XKB_KEY_Return:
      case XKB_KEY_Pause:
      case XKB_KEY_Scroll_Lock:
      case XKB_KEY_Sys_Req:
      case XKB_KEY_Escape:
      case XKB_KEY_Up:
      case XKB_KEY_Down:
      case XKB_KEY_Page_Up:
      case XKB_KEY_Page_Down:
      case XKB_KEY_Begin:
      case XKB_KEY_Print:
      case XKB_KEY_Insert:
      case XKB_KEY_Menu:
      case XKB_KEY_Num_Lock:
      case XKB_KEY_KP_Up:
      case XKB_KEY_KP_Down:
      case XKB_KEY_KP_Page_Up:
      case XKB_KEY_KP_Page_Down:
      case XKB_KEY_KP_Begin:
      case XKB_KEY_KP_Insert:
      case XKB_KEY_KP_Tab:
      case XKB_KEY_F1:
      case XKB_KEY_F2:
      case XKB_KEY_F3:
      case XKB_KEY_F4:
      case XKB_KEY_F5:
      case XKB_KEY_F6:
      case XKB_KEY_F7:
      case XKB_KEY_F8:
      case XKB_KEY_F9:
      case XKB_KEY_F10:
      case XKB_KEY_F11:
      case XKB_KEY_F12:
      case XKB_KEY_F13:
      case XKB_KEY_F14:
      case XKB_KEY_F15:
      case XKB_KEY_F16:
      case XKB_KEY_F17:
      case XKB_KEY_F18:
      case XKB_KEY_F19:
      case XKB_KEY_F20:
      case XKB_KEY_F21:
      case XKB_KEY_F22:
      case XKB_KEY_F23:
      case XKB_KEY_F24:
        break;
      default:
        text_input->active_model_->AddCodePoint(keysym);
        text_input->SendStateUpdate(*(text_input->active_model_));
        break;
    }
  }
}

#if ENABLE_PLUGIN_KEY_EVENT
std::shared_ptr<DelegateHandleKey> TextInput::GetDelegate(
    void* data,
    FlutterKeyEventType type,
    uint32_t xkb_scancode,
    xkb_keysym_t keysym) {
  if (type == kFlutterKeyEventTypeUp) {
    // ignore
    return nullptr;
  }
  if (data == nullptr) {
    // ignore
    return nullptr;
  }

  return std::make_shared<DelegateHandleKey>(data, TextInput::handle_key_hook,
                                             type, xkb_scancode, keysym);
}

void TextInput::handle_key_hook(void* data,
                                FlutterKeyEventType type,
                                uint32_t xkb_scancode,
                                xkb_keysym_t keysym) {
  auto* text_input = static_cast<TextInput*>(data);

  if ((type == kFlutterKeyEventTypeRepeat &&
       xkb_scancode != XKB_KEY_NoSymbol) ||
      (type == kFlutterKeyEventTypeDown)) {
    TextInput::keyboard_handle_key(text_input, keysym,
                                   WL_KEYBOARD_KEY_STATE_PRESSED);
  }
}
#endif

void TextInput::Send(const std::string& channel,
                     const uint8_t* message,
                     const size_t message_size,
                     const flutter::BinaryReply reply) const {
  if (kSuccess !=
      engine_->SendPlatformMessage(channel.c_str(), message, message_size)) {
    spdlog::error("Failed to send platform message");
  }
  last_reply_handler_ = reply;
}

void TextInput::SetMessageHandler(const std::string& channel,
                                  const flutter::BinaryMessageHandler handler) {
  last_message_handler_channel_ = channel;
  last_message_handler_ = handler;
}