ash/mus/window_manager.cc

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

#include "ash/mus/window_manager.h"

#include <stdint.h>

#include <utility>

#include "ash/common/shell_window_ids.h"
#include "ash/mus/bridge/wm_lookup_mus.h"
#include "ash/mus/bridge/wm_shell_mus.h"
#include "ash/mus/bridge/wm_window_mus.h"
#include "ash/mus/non_client_frame_controller.h"
#include "ash/mus/property_util.h"
#include "ash/mus/root_window_controller.h"
#include "ash/mus/root_windows_observer.h"
#include "ash/mus/shadow_controller.h"
#include "ash/mus/window_manager_application.h"
#include "ash/public/interfaces/container.mojom.h"
#include "components/mus/common/event_matcher_util.h"
#include "components/mus/common/types.h"
#include "components/mus/public/cpp/property_type_converters.h"
#include "components/mus/public/cpp/window.h"
#include "components/mus/public/cpp/window_property.h"
#include "components/mus/public/cpp/window_tree_client.h"
#include "components/mus/public/interfaces/mus_constants.mojom.h"
#include "components/mus/public/interfaces/window_manager.mojom.h"
#include "ui/events/mojo/event.mojom.h"
#include "ui/views/mus/screen_mus.h"

namespace ash {
namespace mus {

const uint32_t kWindowSwitchAccelerator = 1;

void AssertTrue(bool success) {
  DCHECK(success);
}

WindowManager::WindowManager(WindowManagerApplication* window_manager_app,
                             shell::Connector* connector)
    : window_manager_app_(window_manager_app), connector_(connector) {}

WindowManager::~WindowManager() {
  // NOTE: |window_tree_client_| may already be null.
  delete window_tree_client_;
}

void WindowManager::Init(::mus::WindowTreeClient* window_tree_client) {
  DCHECK(!window_tree_client_);
  window_tree_client_ = window_tree_client;

  shadow_controller_.reset(new ShadowController(window_tree_client));

  AddAccelerators();

  // The insets are roughly what is needed by CustomFrameView. The expectation
  // is at some point we'll write our own NonClientFrameView and get the insets
  // from it.
  ::mus::mojom::FrameDecorationValuesPtr frame_decoration_values =
      ::mus::mojom::FrameDecorationValues::New();
  const gfx::Insets client_area_insets =
      NonClientFrameController::GetPreferredClientAreaInsets();
  frame_decoration_values->normal_client_area_insets = client_area_insets;
  frame_decoration_values->maximized_client_area_insets = client_area_insets;
  frame_decoration_values->max_title_bar_button_width =
      NonClientFrameController::GetMaxTitleBarButtonWidth();
  window_manager_client_->SetFrameDecorationValues(
      std::move(frame_decoration_values));

  shell_.reset(new WmShellMus(window_tree_client_));
  lookup_.reset(new WmLookupMus);
}

void WindowManager::SetScreenLocked(bool is_locked) {
  // TODO: screen locked state needs to be persisted for newly added displays.
  for (auto& root_window_controller : root_window_controllers_) {
    WmWindowMus* non_lock_screen_containers_container =
        root_window_controller->GetWindowByShellWindowId(
            kShellWindowId_NonLockScreenContainersContainer);
    non_lock_screen_containers_container->mus_window()->SetVisible(!is_locked);
  }
}

::mus::Window* WindowManager::NewTopLevelWindow(
    std::map<std::string, std::vector<uint8_t>>* properties) {
  // TODO(sky): need to maintain active as well as allowing specifying display.
  RootWindowController* root_window_controller =
      root_window_controllers_.begin()->get();
  return root_window_controller->NewTopLevelWindow(properties);
}

std::set<RootWindowController*> WindowManager::GetRootWindowControllers() {
  std::set<RootWindowController*> result;
  for (auto& root_window_controller : root_window_controllers_)
    result.insert(root_window_controller.get());
  return result;
}

void WindowManager::AddRootWindowsObserver(RootWindowsObserver* observer) {
  root_windows_observers_.AddObserver(observer);
}

void WindowManager::RemoveRootWindowsObserver(RootWindowsObserver* observer) {
  root_windows_observers_.RemoveObserver(observer);
}

void WindowManager::AddAccelerators() {
  // TODO(sky): this is broke for multi-display case. Need to fix mus to
  // deal correctly.
  window_manager_client_->AddAccelerator(
      kWindowSwitchAccelerator,
      ::mus::CreateKeyMatcher(ui::mojom::KeyboardCode::TAB,
                              ui::mojom::kEventFlagControlDown),
      base::Bind(&AssertTrue));
}

RootWindowController* WindowManager::CreateRootWindowController(
    ::mus::Window* window,
    const display::Display& display) {
  // TODO(sky): there is timing issues with using ScreenMus.
  if (!screen_) {
    std::unique_ptr<views::ScreenMus> screen(new views::ScreenMus(nullptr));
    screen->Init(connector_);
    screen_ = std::move(screen);
  }

  std::unique_ptr<RootWindowController> root_window_controller_ptr(
      new RootWindowController(this, window, display));
  RootWindowController* root_window_controller =
      root_window_controller_ptr.get();
  root_window_controllers_.insert(std::move(root_window_controller_ptr));
  window->AddObserver(this);

  FOR_EACH_OBSERVER(RootWindowsObserver, root_windows_observers_,
                    OnRootWindowControllerAdded(root_window_controller));
  return root_window_controller;
}

void WindowManager::OnWindowDestroying(::mus::Window* window) {
  for (auto it = root_window_controllers_.begin();
       it != root_window_controllers_.end(); ++it) {
    if ((*it)->root() == window) {
      FOR_EACH_OBSERVER(RootWindowsObserver, root_windows_observers_,
                        OnWillDestroyRootWindowController((*it).get()));
      return;
    }
  }
  NOTREACHED();
}

void WindowManager::OnWindowDestroyed(::mus::Window* window) {
  window->RemoveObserver(this);
  for (auto it = root_window_controllers_.begin();
       it != root_window_controllers_.end(); ++it) {
    if ((*it)->root() == window) {
      root_window_controllers_.erase(it);
      return;
    }
  }
  NOTREACHED();
}

void WindowManager::OnEmbed(::mus::Window* root) {
  // WindowManager should never see this, instead OnWmNewDisplay() is called.
  NOTREACHED();
}

void WindowManager::OnWindowTreeClientDestroyed(
    ::mus::WindowTreeClient* client) {
  // Destroying the roots should result in removal from
  // |root_window_controllers_|.
  DCHECK(root_window_controllers_.empty());

  lookup_.reset();
  shell_.reset();
  shadow_controller_.reset();

  window_tree_client_ = nullptr;
  // TODO(sky): this should likely shutdown.
}

void WindowManager::OnEventObserved(const ui::Event& event,
                                    ::mus::Window* target) {
  // Does not use EventObservers.
}

void WindowManager::SetWindowManagerClient(::mus::WindowManagerClient* client) {
  window_manager_client_ = client;
}

bool WindowManager::OnWmSetBounds(::mus::Window* window, gfx::Rect* bounds) {
  // TODO(sky): this indirectly sets bounds, which is against what
  // OnWmSetBounds() recommends doing. Remove that restriction, or fix this.
  WmWindowMus::Get(window)->SetBounds(*bounds);
  *bounds = window->bounds();
  return true;
}

bool WindowManager::OnWmSetProperty(
    ::mus::Window* window,
    const std::string& name,
    std::unique_ptr<std::vector<uint8_t>>* new_data) {
  // TODO(sky): constrain this to set of keys we know about, and allowed
  // values.
  return name == ::mus::mojom::WindowManager::kShowState_Property ||
         name == ::mus::mojom::WindowManager::kPreferredSize_Property ||
         name == ::mus::mojom::WindowManager::kResizeBehavior_Property ||
         name == ::mus::mojom::WindowManager::kWindowAppIcon_Property ||
         name == ::mus::mojom::WindowManager::kWindowTitle_Property;
}

::mus::Window* WindowManager::OnWmCreateTopLevelWindow(
    std::map<std::string, std::vector<uint8_t>>* properties) {
  return NewTopLevelWindow(properties);
}

void WindowManager::OnWmClientJankinessChanged(
    const std::set<::mus::Window*>& client_windows,
    bool janky) {
  for (auto window : client_windows)
    SetWindowIsJanky(window, janky);
}

void WindowManager::OnWmNewDisplay(::mus::Window* window,
                                   const display::Display& display) {
  CreateRootWindowController(window, display);
}

void WindowManager::OnAccelerator(uint32_t id, const ui::Event& event) {
  switch (id) {
    case kWindowSwitchAccelerator:
      window_manager_client()->ActivateNextWindow();
      break;
    default:
      window_manager_app_->OnAccelerator(id, event);
      break;
  }
}

}  // namespace mus
}  // namespace ash