// Copyright 2015 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include "base/check_op.h"
#include "chrome/browser/ui/browser_window/public/browser_window_features.h"
#include "chrome/browser/ui/browser_window/public/browser_window_interface.h"
#include "chrome/browser/ui/translate/translate_bubble_model.h"
#include "chrome/browser/ui/translate/translate_bubble_test_utils.h"
#include "chrome/browser/ui/views/translate/translate_bubble_controller.h"
#include "chrome/browser/ui/views/translate/translate_bubble_view.h"
#include "components/translate/core/common/translate_features.h"
#include "ui/events/keycodes/dom/dom_code.h"
#include "ui/views/controls/button/label_button.h"
#include "ui/views/controls/combobox/combobox.h"

namespace translate::test_utils {

TranslateBubbleView* GetTranslateBubble(BrowserWindowInterface* browser) {
  return TranslateBubbleController::From(browser)->GetTranslateBubble();
}

const TranslateBubbleModel* GetCurrentModel(BrowserWindowInterface* browser) {
  DCHECK(browser);
  TranslateBubbleView* view = GetTranslateBubble(browser);

  return view ? view->model() : nullptr;
}

void CloseCurrentBubble(BrowserWindowInterface* browser) {
  DCHECK(browser);
  TranslateBubbleController* controller =
      TranslateBubbleController::From(browser);
  if (controller) {
    controller->CloseBubble();
  }
}

void PressTranslate(BrowserWindowInterface* browser) {
  DCHECK(browser);
  TranslateBubbleView* bubble = GetTranslateBubble(browser);
  DCHECK(bubble);

  bubble->TabSelectedAt(1);
}

void PressRevert(BrowserWindowInterface* browser) {
  DCHECK(browser);
  TranslateBubbleView* bubble = GetTranslateBubble(browser);
  DCHECK(bubble);

  bubble->TabSelectedAt(0);
}

void SelectTargetLanguageByDisplayName(BrowserWindowInterface* browser,
                                       const std::u16string& display_name) {
  DCHECK(browser);

  TranslateBubbleView* bubble = GetTranslateBubble(browser);
  DCHECK(bubble);

  TranslateBubbleModel* model = bubble->model();
  DCHECK(model);

  // Get index of the language with the matching display name.
  int language_index = -1;
  for (int i = 0; i < model->GetNumberOfTargetLanguages(); ++i) {
    const std::u16string& language_name = model->GetTargetLanguageNameAt(i);

    if (language_name == display_name) {
      language_index = i;
      break;
    }
  }
  DCHECK_GE(language_index, 0);

  // Simulate selecting the correct index of the target language.
  if (base::FeatureList::IsEnabled(translate::kTranslateLanguageSearchUI)) {
    bubble->TargetLanguageChangedWithIndex(language_index);
  } else {
    bubble->target_language_combobox_->SetSelectedIndex(language_index);
    bubble->TargetLanguageChanged();
  }
}

}  // namespace translate::test_utils