// Copyright 2014 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 "chrome/browser/chromeos/events/event_rewriter_controller.h"

#include "ash/shell.h"
#include "ui/aura/env.h"
#include "ui/aura/window_tree_host.h"
#include "ui/events/event_source.h"

namespace chromeos {

EventRewriterController::EventRewriterController() : initialized_(false) {
  // Add the controller as an observer for new root windows.
  aura::Env::GetInstance()->AddObserver(this);
}

EventRewriterController::~EventRewriterController() {
  aura::Env::GetInstance()->RemoveObserver(this);
  // Remove the rewriters from every root window EventSource and destroy them.
  for (EventRewriters::iterator rewriter_iter = rewriters_.begin();
       rewriter_iter != rewriters_.end();
       ++rewriter_iter) {
    aura::Window::Windows windows = ash::Shell::GetAllRootWindows();
    for (aura::Window::Windows::iterator window_iter = windows.begin();
         window_iter != windows.end();
         ++window_iter) {
      (*window_iter)->GetHost()->GetEventSource()->RemoveEventRewriter(
          *rewriter_iter);
    }
  }
  rewriters_.clear();
}

void EventRewriterController::AddEventRewriter(
    scoped_ptr<ui::EventRewriter> rewriter) {
  DCHECK(!initialized_);
  rewriters_.push_back(rewriter.release());
}

void EventRewriterController::Init() {
  DCHECK(!initialized_);
  initialized_ = true;
  // Add the rewriters to each existing root window EventSource.
  aura::Window::Windows windows = ash::Shell::GetAllRootWindows();
  for (aura::Window::Windows::iterator it = windows.begin();
       it != windows.end();
       ++it) {
    AddToEventSource((*it)->GetHost()->GetEventSource());
  }
}

void EventRewriterController::OnHostInitialized(aura::WindowTreeHost* host) {
  if (initialized_)
    AddToEventSource(host->GetEventSource());
}

void EventRewriterController::AddToEventSource(ui::EventSource* source) {
  DCHECK(source);
  for (EventRewriters::iterator it = rewriters_.begin(); it != rewriters_.end();
       ++it) {
    source->AddEventRewriter(*it);
  }
}

}  // namespace chromeos