// 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/ui/webui/mojo_web_ui_controller.h"

#include "chrome/browser/profiles/profile.h"
#include "chrome/browser/ui/webui/mojo_web_ui_handler.h"
#include "content/public/browser/render_view_host.h"
#include "content/public/browser/web_ui_data_source.h"
#include "content/public/common/bindings_policy.h"
#include "mojo/public/cpp/system/core.h"

MojoWebUIController::MojoWebUIController(content::WebUI* contents)
    : WebUIController(contents),
      mojo_data_source_(NULL) {
}

MojoWebUIController::~MojoWebUIController() {
}

void MojoWebUIController::RenderViewCreated(
    content::RenderViewHost* render_view_host) {
  render_view_host->AllowBindings(content::BINDINGS_POLICY_WEB_UI);

  mojo::MessagePipe pipe;
  ui_handler_ = CreateUIHandler(pipe.handle0.Pass());
  render_view_host->SetWebUIHandle(pipe.handle1.Pass());
}

void MojoWebUIController::AddMojoResourcePath(const std::string& path,
                                              int resource_id) {
  if (!mojo_data_source_) {
    mojo_data_source_ = content::WebUIDataSource::AddMojoDataSource(
        Profile::FromWebUI(web_ui()));
  }
  mojo_data_source_->AddResourcePath(path, resource_id);
}