// Copyright (c) 2014 The Chromium Embedded Framework 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 "browser_process_handler.h"
#include "client_handler.h"

#include "jni_util.h"
#include "print_handler.h"
#include "util.h"

#include "include/cef_base.h"

// instantiate static values
std::set<CefMessageRouterConfig, cmpCfg> BrowserProcessHandler::router_cfg_;
base::Lock BrowserProcessHandler::router_cfg_lock_;

BrowserProcessHandler::BrowserProcessHandler(jobject app_handler) {
  app_handler_ = app_handler;
}

BrowserProcessHandler::~BrowserProcessHandler() {
  base::AutoLock lock_scope(router_cfg_lock_);
  router_cfg_.clear();
}

void BrowserProcessHandler::OnContextInitialized() {
  if (!app_handler_)
    return;

  BEGIN_ENV(env)
  JNI_CALL_VOID_METHOD(env, app_handler_, "onContextInitialized", "()V");
  END_ENV(env)
}

void BrowserProcessHandler::OnRenderProcessThreadCreated(
    CefRefPtr<CefListValue> extra_info) {
  int idx=0;
  static std::set<CefMessageRouterConfig, cmpCfg>::iterator iter;

  // Delegate creation of the renderer-side router for query handling.
  base::AutoLock lock_scope(router_cfg_lock_);
  for (iter = router_cfg_.begin(); iter != router_cfg_.end(); ++iter) {
    CefRefPtr<CefDictionaryValue> dict = CefDictionaryValue::Create();
    dict->SetString("js_query_function", iter->js_query_function);
    dict->SetString("js_cancel_function", iter->js_cancel_function);
    extra_info->SetDictionary(idx, dict);
    idx++;
  }
}

CefRefPtr<CefPrintHandler> BrowserProcessHandler::GetPrintHandler() {
  CefRefPtr<CefPrintHandler> result = NULL;
  BEGIN_ENV(env)
  jobject handler = NULL;
  JNI_CALL_METHOD(env, app_handler_, "getPrintHandler",
      "()Lorg/cef/handler/CefPrintHandler;", Object, handler);

  if (handler != NULL) {
    result = GetCefFromJNIObject<CefPrintHandler>(env, handler,
        "CefPrintHandler");
    if (!result.get()) {
      result = new PrintHandler(env, handler);
      SetCefForJNIObject(env, handler, result.get(), "CefPrintHandler");
    }
  }
  END_ENV(env)
  return result;
}

void BrowserProcessHandler::AddMessageRouterConfig(
    const CefMessageRouterConfig& cfg) {
  base::AutoLock lock_scope(router_cfg_lock_);
  router_cfg_.insert(cfg);
}

void BrowserProcessHandler::RemoveMessageRouterConfig(
    const CefMessageRouterConfig& cfg) {
  base::AutoLock lock_scope(router_cfg_lock_);
  router_cfg_.erase(cfg);
}