// 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.

#ifdef UNSAFE_BUFFERS_BUILD
// TODO(crbug.com/40285824): Remove this and convert code to safer constructs.
#pragma allow_unsafe_buffers
#endif

#include "chrome/browser/ui/cocoa/task_manager_mac.h"

#import <Cocoa/Cocoa.h>
#include <Foundation/Foundation.h>
#include <stddef.h>

#include <algorithm>

#include "base/functional/callback.h"
#include "base/strings/pattern.h"
#include "chrome/browser/browser_process.h"
#include "chrome/browser/task_manager/common/task_manager_features.h"
#include "chrome/browser/task_manager/task_manager_browsertest_util.h"
#include "chrome/browser/task_manager/task_manager_tester.h"
#include "chrome/browser/ui/browser_commands.h"
#include "chrome/browser/ui/browser_dialogs.h"
#include "chrome/browser/ui/browser_tabstrip.h"
#include "chrome/browser/ui/tab_contents/tab_contents_iterator.h"
#include "chrome/browser/ui/task_manager/task_manager_columns.h"
#include "chrome/browser/ui/task_manager/task_manager_table_model.h"
#include "chrome/common/pref_names.h"
#include "chrome/test/base/in_process_browser_test.h"
#include "chrome/test/base/ui_test_utils.h"
#include "components/prefs/pref_service.h"
#include "components/prefs/scoped_user_pref_update.h"
#include "components/sessions/content/session_tab_helper.h"
#include "content/public/browser/web_contents.h"
#include "content/public/browser/web_contents_delegate.h"
#include "content/public/test/browser_test.h"
#include "content/public/test/browser_test_utils.h"
#include "content/public/test/no_renderer_crashes_assertion.h"
#include "content/public/test/test_utils.h"
#include "net/dns/mock_host_resolver.h"
#include "net/test/embedded_test_server/embedded_test_server.h"
#include "testing/gtest_mac.h"
#include "ui/base/test/ui_controls.h"
#include "ui/gfx/native_widget_types.h"
#include "url/gurl.h"

namespace task_manager {

using browsertest_util::WaitForTaskManagerRows;

class TaskManagerMacTest : public InProcessBrowserTest {
 public:
  TaskManagerMacTest() {
    feature_list_.InitWithFeatures(
        /*enabled_features=*/{},
        /*disabled_features=*/{features::kTaskManagerDesktopRefresh});
  }
  ~TaskManagerMacTest() override = default;

  TaskManagerMacTest(const TaskManagerMacTest&) = delete;
  TaskManagerMacTest& operator=(const TaskManagerMacTest&) = delete;

  void SetUpOnMainThread() override {
    ASSERT_FALSE(GetTaskManagerMac());
    host_resolver()->AddRule("*", "127.0.0.1");
    ASSERT_TRUE(embedded_test_server()->Start());
  }

  void TearDownOnMainThread() override {
    // Make sure the task manager is closed (if any).
    chrome::HideTaskManager();
    ASSERT_FALSE(GetTaskManagerMac());
  }

  TaskManagerMac* GetTaskManagerMac() const {
    return TaskManagerMac::GetInstanceForTests();
  }

  NSTableView* GetTable() const {
    return GetTaskManagerMac() ? GetTaskManagerMac()
                                     ->CocoaControllerForTests()
                                     .tableViewForTesting
                               : nullptr;
  }

  std::optional<size_t> TableFirstSelectedRow() const {
    int index = GetTable().selectedRowIndexes.firstIndex;
    return (index < 0) ? std::nullopt
                       : std::make_optional(static_cast<size_t>(index));
  }

  void PressKillButton() {
    ASSERT_TRUE(GetTaskManagerMac());
    [GetTaskManagerMac()->CocoaControllerForTests().endProcessButtonForTesting
        performClick:nil];
  }

  void ClearStoredColumnSettings() const {
    PrefService* local_state = g_browser_process->local_state();
    if (!local_state) {
      FAIL();
    }

    local_state->SetDict(prefs::kTaskManagerColumnVisibility,
                         base::Value::Dict());
  }

  void ToggleColumnVisibility(TaskManagerMac* task_manager, int col_id) {
    DCHECK(task_manager);
    task_manager->GetTableModelForTests()->ToggleColumnVisibility(col_id);
  }

  // Looks up a tab based on its tab ID.
  content::WebContents* FindWebContentsByTabId(SessionID tab_id) {
    auto& all_tabs = AllTabContentses();
    auto it = std::ranges::find(all_tabs, tab_id,
                                &sessions::SessionTabHelper::IdForTab);

    return (it == all_tabs.end()) ? nullptr : *it;
  }

  // Returns the current TaskManagerTableModel index for a particular tab. Don't
  // cache this value, since it can change whenever the message loop runs.
  std::optional<size_t> FindRowForTab(content::WebContents* tab) {
    SessionID tab_id = sessions::SessionTabHelper::IdForTab(tab);
    std::unique_ptr<TaskManagerTester> tester =
        TaskManagerTester::Create(base::RepeatingClosure());
    for (size_t i = 0; i < tester->GetRowCount(); ++i) {
      if (tester->GetTabId(i) == tab_id) {
        return i;
      }
    }
    return std::nullopt;
  }

 private:
  base::test::ScopedFeatureList feature_list_;
};

// Tests that all defined columns have a corresponding string IDs for keying
// into the user preferences dictionary.
IN_PROC_BROWSER_TEST_F(TaskManagerMacTest, AllColumnsHaveStringIds) {
  for (size_t i = 0; i < kColumnsSize; ++i) {
    EXPECT_NE("", GetColumnIdAsString(kColumns[i].id));
  }
}

// In the case of no settings stored in the user preferences local store, test
// that the task manager table starts with the default columns visibility as
// stored in |kColumns|.
IN_PROC_BROWSER_TEST_F(TaskManagerMacTest, TableStartsWithDefaultColumns) {
  ASSERT_NO_FATAL_FAILURE(ClearStoredColumnSettings());

  chrome::ShowTaskManager(browser());
  NSTableView* table = GetTable();
  ASSERT_TRUE(table);

  EXPECT_EQ(0u, table.sortDescriptors.count);
  NSArray* tableColumns = table.tableColumns;
  for (size_t i = 0; i < kColumnsSize; ++i) {
    EXPECT_EQ(kColumns[i].id, [[tableColumns[i] identifier] intValue]);
    EXPECT_EQ(kColumns[i].default_visibility, ![tableColumns[i] isHidden]);
  }
}

// Tests that changing columns visibility and sort order will be stored upon
// closing the task manager view and restored when re-opened.
IN_PROC_BROWSER_TEST_F(TaskManagerMacTest, ColumnsSettingsAreRestored) {
  ASSERT_NO_FATAL_FAILURE(ClearStoredColumnSettings());

  chrome::ShowTaskManager(browser());
  TaskManagerMac* task_manager = GetTaskManagerMac();
  ASSERT_TRUE(task_manager);
  NSTableView* table = GetTable();
  ASSERT_TRUE(table);

  // Toggle the visibility of all columns.
  EXPECT_EQ(0u, table.sortDescriptors.count);
  NSArray* tableColumns = [table tableColumns];
  for (size_t i = 0; i < kColumnsSize; ++i) {
    EXPECT_EQ(kColumns[i].id, [[tableColumns[i] identifier] intValue]);
    EXPECT_EQ(kColumns[i].default_visibility, ![tableColumns[i] isHidden]);
    ToggleColumnVisibility(task_manager, kColumns[i].id);
  }

  // Sort by the first visible and initially ascending sortable column. It would
  // be nice to fake a click with -performClick: but that doesn't work (see
  // http://www.cocoabuilder.com/archive/cocoa/177610-programmatically-click-column-header-in-nstableview.html).
  bool is_sorted = false;
  for (NSTableColumn* column in tableColumns) {
    if (column.hidden) {
      continue;
    }
    if (column.sortDescriptorPrototype.ascending) {
      // Toggle the sort for a descending sort.
      NSSortDescriptor* newSortDescriptor =
          column.sortDescriptorPrototype.reversedSortDescriptor;
      table.sortDescriptors = @[ newSortDescriptor ];
      is_sorted = true;
      break;
    }
  }

  NSArray* expectedSortDescriptors = table.sortDescriptors;
  EXPECT_EQ(is_sorted, expectedSortDescriptors.count > 0);

  // Close the task manager view and re-open. Expect the inverse of the default
  // visibility, and the last sort order.
  chrome::HideTaskManager();
  ASSERT_FALSE(GetTaskManagerMac());
  chrome::ShowTaskManager(browser());
  task_manager = GetTaskManagerMac();
  ASSERT_TRUE(task_manager);
  table = GetTable();
  ASSERT_TRUE(table);

  if (is_sorted) {
    EXPECT_NSEQ(expectedSortDescriptors, table.sortDescriptors);
  }
}

// Tests that pressing Enter/return kills a process.
IN_PROC_BROWSER_TEST_F(TaskManagerMacTest, PressingEnterKillsProcess) {
  ASSERT_NO_FATAL_FAILURE(ClearStoredColumnSettings());
  ui_controls::EnableUIControls();

  // Set up a running tab.
  ASSERT_TRUE(ui_test_utils::NavigateToURL(
      browser(), embedded_test_server()->GetURL("a.com", "/title2.html")));

  chrome::ShowTaskManager(browser());

  TaskManagerMac* task_manager = GetTaskManagerMac();

  ASSERT_TRUE(task_manager);

  NSTableView* table = GetTable();

  ASSERT_TRUE(table);

  TaskManagerWindowController* window_controller =
      task_manager->CocoaControllerForTests();
  std::unique_ptr<TaskManagerTester> tester =
      TaskManagerTester::Create(base::RepeatingClosure());

  auto pattern = browsertest_util::MatchTab("Title *");

  // Wait for the title of the tab to appear in the task manager.
  size_t rows = 1;
  ASSERT_NO_FATAL_FAILURE(WaitForTaskManagerRows(rows, pattern));

  std::u16string title = u"";

  for (size_t i = 0; i < tester->GetRowCount(); ++i) {
    // Press down to select the first/next row.
    ui_controls::SendKeyPress(gfx::NativeWindow(window_controller.window),
                              ui::VKEY_DOWN, false, false, false, false);

    ASSERT_TRUE(TableFirstSelectedRow().has_value());

    std::u16string currentTitle =
        tester->GetRowTitle(TableFirstSelectedRow().value());

    // Found the tab, stop pressing down.
    if (base::MatchPattern(currentTitle, pattern)) {
      title = currentTitle;
      break;
    }
  }

  // Found the tab.
  EXPECT_NE(u"", title);

  {
    content::ScopedAllowRendererCrashes scoped_allow_renderer_crashes;

    // Press Enter/return to simulate a click on the end process button.
    ui_controls::SendKeyPress(gfx::NativeWindow(window_controller.window),
                              ui::VKEY_RETURN, false, false, false, false);

    // The rows for the tab should disappear.
    size_t no_rows = 0;
    ASSERT_NO_FATAL_FAILURE(WaitForTaskManagerRows(no_rows, pattern));
  }
}

// TODO(crbug.com/40261286): Re-enable when fixed.
IN_PROC_BROWSER_TEST_F(TaskManagerMacTest, DISABLED_SelectionConsistency) {
  ASSERT_NO_FATAL_FAILURE(ClearStoredColumnSettings());

  chrome::ShowTaskManager(browser());

  // Set up a total of three tabs in different processes.
  ASSERT_TRUE(ui_test_utils::NavigateToURL(
      browser(), embedded_test_server()->GetURL("a.com", "/title2.html")));
  ui_test_utils::NavigateToURLWithDisposition(
      browser(), embedded_test_server()->GetURL("b.com", "/title2.html"),
      WindowOpenDisposition::NEW_FOREGROUND_TAB,
      ui_test_utils::BROWSER_TEST_WAIT_FOR_LOAD_STOP);
  ui_test_utils::NavigateToURLWithDisposition(
      browser(), embedded_test_server()->GetURL("c.com", "/title2.html"),
      WindowOpenDisposition::NEW_FOREGROUND_TAB,
      ui_test_utils::BROWSER_TEST_WAIT_FOR_LOAD_STOP);

  // Wait for their titles to appear in the TaskManager. There should be three
  // rows.
  auto pattern = browsertest_util::MatchTab("Title *");
  size_t rows = 3;
  ASSERT_NO_FATAL_FAILURE(WaitForTaskManagerRows(rows, pattern));

  // Find the three tabs we set up, in TaskManager model order. Because we have
  // not sorted the table yet, this should also be their UI display order.
  std::unique_ptr<TaskManagerTester> tester =
      TaskManagerTester::Create(base::RepeatingClosure());
  std::vector<content::WebContents*> tabs;
  for (size_t i = 0; i < tester->GetRowCount(); ++i) {
    // Filter based on our title.
    if (!base::MatchPattern(tester->GetRowTitle(i), pattern)) {
      continue;
    }
    content::WebContents* tab = FindWebContentsByTabId(tester->GetTabId(i));
    EXPECT_NE(nullptr, tab);
    tabs.push_back(tab);
  }
  EXPECT_EQ(3U, tabs.size());

  // Select the middle row, and store its tab id.
  [GetTable()
          selectRowIndexes:[NSIndexSet
                               indexSetWithIndex:FindRowForTab(tabs[1]).value()]
      byExtendingSelection:NO];
  EXPECT_EQ(TableFirstSelectedRow(), FindRowForTab(tabs[1]));
  EXPECT_EQ(1, GetTable().numberOfSelectedRows);

  // Add 3 rows above the selection. The selected tab should not change.
  for (int i = 0; i < 3; ++i) {
    ASSERT_TRUE(content::ExecJs(tabs[0], "window.open('title3.html');"));
    EXPECT_EQ(TableFirstSelectedRow(), FindRowForTab(tabs[1]));
  }
  ASSERT_NO_FATAL_FAILURE(WaitForTaskManagerRows((rows += 3), pattern));
  EXPECT_EQ(TableFirstSelectedRow(), FindRowForTab(tabs[1]));
  EXPECT_EQ(1, GetTable().numberOfSelectedRows);

  // Add 2 rows below the selection. The selected tab should not change.
  for (int i = 0; i < 2; ++i) {
    ASSERT_TRUE(content::ExecJs(tabs[2], "window.open('title3.html');"));
    EXPECT_EQ(TableFirstSelectedRow(), FindRowForTab(tabs[1]));
  }
  ASSERT_NO_FATAL_FAILURE(WaitForTaskManagerRows((rows += 2), pattern));
  EXPECT_EQ(TableFirstSelectedRow(), FindRowForTab(tabs[1]));
  EXPECT_EQ(1, GetTable().numberOfSelectedRows);

  // Add a new row in the same process as the selection. The selected tab should
  // not change.
  ASSERT_TRUE(content::ExecJs(tabs[1], "window.open('title3.html');"));
  EXPECT_EQ(TableFirstSelectedRow(), FindRowForTab(tabs[1]));
  ASSERT_NO_FATAL_FAILURE(WaitForTaskManagerRows((rows += 1), pattern));
  EXPECT_EQ(TableFirstSelectedRow(), FindRowForTab(tabs[1]));
  EXPECT_EQ(2, GetTable().numberOfSelectedRows);

  {
    content::ScopedAllowRendererCrashes scoped_allow_renderer_crashes;

    // Press the button, which kills the process of the selected row.
    PressKillButton();

    // Two rows should disappear.
    ASSERT_NO_FATAL_FAILURE(WaitForTaskManagerRows((rows -= 2), pattern));
  }

  // No row should now be selected.
  ASSERT_FALSE(TableFirstSelectedRow().has_value());

  // Now select tabs[2].
  [GetTable()
          selectRowIndexes:[NSIndexSet
                               indexSetWithIndex:FindRowForTab(tabs[2]).value()]
      byExtendingSelection:NO];

  // Focus and reload one of the sad tabs. It should reappear in the TM. The
  // other sad tab should not reappear.
  tabs[1]->GetDelegate()->ActivateContents(tabs[1]);
  chrome::Reload(browser(), WindowOpenDisposition::CURRENT_TAB);
  ASSERT_NO_FATAL_FAILURE(WaitForTaskManagerRows((rows += 1), pattern));

  // tabs[2] should still be selected.
  EXPECT_EQ(TableFirstSelectedRow(), FindRowForTab(tabs[2]));

  // Close tabs[0]. The selection should not change.
  chrome::CloseWebContents(browser(), tabs[0], false);
  ASSERT_NO_FATAL_FAILURE(WaitForTaskManagerRows((rows -= 1), pattern));
  EXPECT_EQ(TableFirstSelectedRow(), FindRowForTab(tabs[2]));
}

// TODO(crbug.com/40261286): Re-enable when fixed.
IN_PROC_BROWSER_TEST_F(TaskManagerMacTest, DISABLED_NavigateSelection) {
  ASSERT_NO_FATAL_FAILURE(ClearStoredColumnSettings());
  ui_controls::EnableUIControls();
  chrome::ShowTaskManager(browser());

  // Set up two tabs in different processes.
  ASSERT_TRUE(ui_test_utils::NavigateToURL(
      browser(), embedded_test_server()->GetURL("a.com", "/title2.html")));
  ui_test_utils::NavigateToURLWithDisposition(
      browser(), embedded_test_server()->GetURL("b.com", "/title2.html"),
      WindowOpenDisposition::NEW_FOREGROUND_TAB,
      ui_test_utils::BROWSER_TEST_WAIT_FOR_LOAD_STOP);

  // Wait for their titles to appear in the TaskManager.
  auto pattern = browsertest_util::MatchTab("Title *");
  size_t rows = 2;
  ASSERT_NO_FATAL_FAILURE(WaitForTaskManagerRows(rows, pattern));

  // Find the two tabs we set up, in TaskManager model order. Because we have
  // not sorted the table yet, this should also be their UI display order.
  std::unique_ptr<TaskManagerTester> tester =
      TaskManagerTester::Create(base::RepeatingClosure());
  std::vector<content::WebContents*> tabs;
  for (size_t i = 0; i < tester->GetRowCount(); ++i) {
    // Filter based on our title.
    if (!base::MatchPattern(tester->GetRowTitle(i), pattern)) {
      continue;
    }
    content::WebContents* tab = FindWebContentsByTabId(tester->GetTabId(i));
    EXPECT_NE(nullptr, tab);
    tabs.push_back(tab);
  }
  EXPECT_EQ(2U, tabs.size());

  // Select the first row, and store its tab id.
  [GetTable()
          selectRowIndexes:[NSIndexSet
                               indexSetWithIndex:FindRowForTab(tabs[0]).value()]
      byExtendingSelection:NO];
  EXPECT_EQ(TableFirstSelectedRow(), FindRowForTab(tabs[0]));
  EXPECT_EQ(1, GetTable().numberOfSelectedRows);

  // Add two new tasks to the same process as a.com
  int num_group_tasks = 1;
  for (int i = 0; i < 2; i++) {
    ASSERT_TRUE(content::ExecJs(tabs[0], "window.open('title3.html');"));
    EXPECT_EQ(TableFirstSelectedRow(), FindRowForTab(tabs[0]));
    ASSERT_NO_FATAL_FAILURE(WaitForTaskManagerRows((rows += 1), pattern));
    EXPECT_EQ(TableFirstSelectedRow(), FindRowForTab(tabs[0]));
    num_group_tasks++;
    EXPECT_EQ(num_group_tasks, GetTable().numberOfSelectedRows);
  }

  std::optional<size_t> selected_row = TableFirstSelectedRow();
  ASSERT_TRUE(selected_row.has_value());
  size_t expected_selected_row = selected_row.value();
  TaskManagerWindowController* window_controller =
      GetTaskManagerMac()->CocoaControllerForTests();

  // Navigate off of the grouped tasks into a different process task
  ui_controls::SendKeyPress(gfx::NativeWindow(window_controller.window),
                            ui::VKEY_DOWN, false, false, false, false);
  expected_selected_row = expected_selected_row + 3;
  EXPECT_EQ(expected_selected_row, TableFirstSelectedRow().value());
  EXPECT_EQ(1, GetTable().numberOfSelectedRows);

  // Navigate into the three grouped tasks
  ui_controls::SendKeyPress(gfx::NativeWindow(window_controller.window),
                            ui::VKEY_UP, false, false, false, false);
  expected_selected_row -= num_group_tasks;
  EXPECT_EQ(expected_selected_row, TableFirstSelectedRow().value());
  EXPECT_EQ(num_group_tasks, GetTable().numberOfSelectedRows);

  // Navigate off of grouped tasks
  ui_controls::SendKeyPress(gfx::NativeWindow(window_controller.window),
                            ui::VKEY_UP, false, false, false, false);
  expected_selected_row--;
  EXPECT_EQ(expected_selected_row, TableFirstSelectedRow().value());
  EXPECT_EQ(1, GetTable().numberOfSelectedRows);

  // Navigate back into the three grouped tasks
  ui_controls::SendKeyPress(gfx::NativeWindow(window_controller.window),
                            ui::VKEY_DOWN, false, false, false, false);
  expected_selected_row++;
  EXPECT_EQ(expected_selected_row, TableFirstSelectedRow().value());
  EXPECT_EQ(num_group_tasks, GetTable().numberOfSelectedRows);
}
}  // namespace task_manager