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

#include "chrome/browser/ui/webui/ash/login/oobe_display_chooser.h"

#include <stdint.h>

#include <algorithm>

#include "ash/display/cros_display_config.h"
#include "base/check_deref.h"
#include "base/functional/bind.h"
#include "content/public/browser/browser_task_traits.h"
#include "content/public/browser/browser_thread.h"
#include "ui/display/display.h"
#include "ui/display/screen.h"
#include "ui/events/devices/touchscreen_device.h"

using content::BrowserThread;

namespace ash {

namespace {

bool TouchSupportAvailable(const display::Display& display) {
  return display.touch_support() == display::Display::TouchSupport::AVAILABLE;
}

// TODO(felixe): More context at crbug.com/40528488
const uint16_t kDeviceIds[] = {0x0457, 0x266e, 0x222a};

// Returns true if `vendor_id` is a valid vendor id that may be made the primary
// display.
bool IsAllowListedVendorId(uint16_t vendor_id) {
  return std::ranges::contains(kDeviceIds, vendor_id);
}

}  // namespace

OobeDisplayChooser::OobeDisplayChooser(
    ash::CrosDisplayConfig* cros_display_config)
    : cros_display_config_(CHECK_DEREF(cros_display_config)) {}

OobeDisplayChooser::~OobeDisplayChooser() = default;

void OobeDisplayChooser::TryToPlaceUiOnTouchDisplay() {
  DCHECK_CURRENTLY_ON(BrowserThread::UI);

  // Don't (potentially) queue a second task to run MoveToTouchDisplay if one
  // already is queued.
  if (weak_ptr_factory_.HasWeakPtrs()) {
    return;
  }

  display::Display primary_display =
      display::Screen::Get()->GetPrimaryDisplay();

  if (primary_display.is_valid() && !TouchSupportAvailable(primary_display)) {
    content::GetUIThreadTaskRunner({})->PostTask(
        FROM_HERE, base::BindOnce(&OobeDisplayChooser::MaybeMoveToTouchDisplay,
                                  weak_ptr_factory_.GetWeakPtr()));
  }
}

void OobeDisplayChooser::MaybeMoveToTouchDisplay() {
  ui::DeviceDataManager* device_data_manager =
      ui::DeviceDataManager::GetInstance();
  if (device_data_manager->AreDeviceListsComplete() &&
      device_data_manager->AreTouchscreenTargetDisplaysValid()) {
    MoveToTouchDisplay();
  } else if (!scoped_observation_.IsObserving()) {
    scoped_observation_.Observe(device_data_manager);
  } else {
    DCHECK(scoped_observation_.IsObservingSource(device_data_manager));
  }
}

void OobeDisplayChooser::MoveToTouchDisplay() {
  DCHECK_CURRENTLY_ON(BrowserThread::UI);

  scoped_observation_.Reset();

  const ui::DeviceDataManager* device_data_manager =
      ui::DeviceDataManager::GetInstance();
  for (const ui::TouchscreenDevice& device :
       device_data_manager->GetTouchscreenDevices()) {
    if (IsAllowListedVendorId(device.vendor_id) &&
        device.target_display_id != display::kInvalidDisplayId) {
      DisplayConfigProperties config_properties;
      config_properties.set_primary = true;
      cros_display_config_->SetDisplayProperties(device.target_display_id,
                                                 config_properties,
                                                 DisplayConfigSource::kUser);
      break;
    }
  }
}

void OobeDisplayChooser::OnTouchDeviceAssociationChanged() {
  MaybeMoveToTouchDisplay();
}

void OobeDisplayChooser::OnInputDeviceConfigurationChanged(
    uint8_t input_device_types) {
  if (input_device_types & ui::InputDeviceEventObserver::kTouchscreen) {
    MaybeMoveToTouchDisplay();
  }
}

void OobeDisplayChooser::OnDeviceListsComplete() {
  MaybeMoveToTouchDisplay();
}

}  // namespace ash