// Copyright 2013 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/android/webapps/single_tab_mode_tab_helper.h"

#include "base/lazy_instance.h"
#include "chrome/browser/android/tab_android.h"
#include "content/public/browser/browser_thread.h"
#include "content/public/browser/render_process_host.h"
#include "content/public/browser/render_view_host.h"
#include "content/public/browser/web_contents.h"

DEFINE_WEB_CONTENTS_USER_DATA_KEY(SingleTabModeTabHelper);

namespace {

typedef std::pair<int32, int32> RenderWidgetHostID;
typedef std::set<RenderWidgetHostID> SingleTabIDSet;
base::LazyInstance<SingleTabIDSet> g_blocked_ids = LAZY_INSTANCE_INITIALIZER;

void AddPairOnIOThread(int32 process_id, int32 routing_id) {
  DCHECK(content::BrowserThread::CurrentlyOn(content::BrowserThread::IO));
  RenderWidgetHostID single_tab_pair(process_id, routing_id);
  g_blocked_ids.Get().insert(single_tab_pair);
}

void RemovePairOnIOThread(int32 process_id, int32 routing_id) {
  DCHECK(content::BrowserThread::CurrentlyOn(content::BrowserThread::IO));
  RenderWidgetHostID single_tab_pair(process_id, routing_id);
  SingleTabIDSet::iterator itr = g_blocked_ids.Get().find(single_tab_pair);
  DCHECK(itr != g_blocked_ids.Get().end());
  g_blocked_ids.Get().erase(itr);
}

void AddPair(content::RenderViewHost* render_view_host) {
  if (!render_view_host)
    return;

  int32 process_id = render_view_host->GetProcess()->GetID();
  int32 routing_id = render_view_host->GetRoutingID();
  content::BrowserThread::PostTask(
      content::BrowserThread::IO, FROM_HERE,
      base::Bind(&AddPairOnIOThread, process_id, routing_id));
}

void RemovePair(content::RenderViewHost* render_view_host) {
  if (!render_view_host)
    return;

  int32 process_id = render_view_host->GetProcess()->GetID();
  int32 routing_id = render_view_host->GetRoutingID();
  content::BrowserThread::PostTask(
      content::BrowserThread::IO, FROM_HERE,
      base::Bind(&RemovePairOnIOThread, process_id, routing_id));
}

}  // namespace

SingleTabModeTabHelper::SingleTabModeTabHelper(
    content::WebContents* web_contents)
    : content::WebContentsObserver(web_contents),
      block_all_new_windows_(false) {
}

SingleTabModeTabHelper::~SingleTabModeTabHelper() {
}

void SingleTabModeTabHelper::RenderViewCreated(
    content::RenderViewHost* render_view_host) {
  if (!block_all_new_windows_)
    return;
  AddPair(render_view_host);
}

void SingleTabModeTabHelper::RenderViewDeleted(
    content::RenderViewHost* render_view_host) {
  if (!block_all_new_windows_)
    return;
  RemovePair(render_view_host);
}

void SingleTabModeTabHelper::PermanentlyBlockAllNewWindows() {
  block_all_new_windows_ = true;
  AddPair(web_contents()->GetRenderViewHost());
}

bool SingleTabModeTabHelper::IsRegistered(int32 process_id,
                                          int32 routing_id) {
  DCHECK(content::BrowserThread::CurrentlyOn(content::BrowserThread::IO));
  RenderWidgetHostID single_tab_pair(process_id, routing_id);
  SingleTabIDSet::iterator itr = g_blocked_ids.Get().find(single_tab_pair);
  return itr != g_blocked_ids.Get().end();
}

void SingleTabModeTabHelper::HandleOpenUrl(const BlockedWindowParams &params) {
  TabAndroid* tab = TabAndroid::FromWebContents(web_contents());
  if (!tab)
    return;

  chrome::NavigateParams nav_params =
      params.CreateNavigateParams(web_contents());
  tab->HandlePopupNavigation(&nav_params);
}