// Copyright (c) 2009 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 "config.h"

#include <string>

#include "Document.h"
#include "EventListener.h"
#include "InspectorController.h"
#include "Node.h"
#include "Page.h"
#include "PlatformString.h"
#undef LOG

#include "base/values.h"
#include "webkit/glue/devtools/dom_agent_impl.h"
#include "webkit/glue/devtools/net_agent_impl.h"
#include "webkit/glue/glue_util.h"
#include "webkit/glue/webdevtoolsagent_delegate.h"
#include "webkit/glue/webdevtoolsagent_impl.h"
#include "webkit/glue/webview_impl.h"

using WebCore::Document;
using WebCore::InspectorController;
using WebCore::Node;
using WebCore::Page;
using WebCore::String;

WebDevToolsAgentImpl::WebDevToolsAgentImpl(
    WebViewImpl* web_view_impl,
    WebDevToolsAgentDelegate* delegate)
    : delegate_(delegate),
      web_view_impl_(web_view_impl),
      document_(NULL) {
  dom_agent_delegate_stub_.reset(new DomAgentDelegateStub(this));
  net_agent_delegate_stub_.reset(new NetAgentDelegateStub(this));
  tools_agent_delegate_stub_.reset(new ToolsAgentDelegateStub(this));
}

WebDevToolsAgentImpl::~WebDevToolsAgentImpl() {
}

void WebDevToolsAgentImpl::SetDomAgentEnabled(bool enabled) {
  if (enabled && !dom_agent_impl_.get()) {
    dom_agent_impl_.reset(new DomAgentImpl(dom_agent_delegate_stub_.get()));
    if (document_)
      dom_agent_impl_->SetDocument(document_);
  } else if (!enabled && dom_agent_impl_.get()) {
    dom_agent_impl_.reset(NULL);
  }
}

void WebDevToolsAgentImpl::SetNetAgentEnabled(bool enabled) {
  if (enabled && !net_agent_impl_.get()) {
    net_agent_impl_.reset(new NetAgentImpl(net_agent_delegate_stub_.get()));
    if (document_)
      net_agent_impl_->SetDocument(document_);
  } else if (!enabled && net_agent_impl_.get()) {
    net_agent_impl_.reset(NULL);
  }
}

void WebDevToolsAgentImpl::SetMainFrameDocumentReady(bool ready) {
  if (ready) {
    Page* page = web_view_impl_->page();
    document_ = page->mainFrame()->document();
  } else {
    document_ = NULL;
  }
  if (dom_agent_impl_.get())
    dom_agent_impl_->SetDocument(document_);
  if (net_agent_impl_.get())
    net_agent_impl_->SetDocument(document_);
}

void WebDevToolsAgentImpl::HighlightDOMNode(int node_id) {
  if (!dom_agent_impl_.get())
    return;
  Node* node = dom_agent_impl_->GetNodeForId(node_id);
  if (!node)
    return;
  Page* page = web_view_impl_->page();
  page->inspectorController()->highlight(node);
}

void WebDevToolsAgentImpl::HideDOMNodeHighlight() {
  Page* page = web_view_impl_->page();
  page->inspectorController()->hideHighlight();
}

void WebDevToolsAgentImpl::Inspect(Node* node) {
  if (!dom_agent_impl_.get())
    return;

  int node_id = dom_agent_impl_->GetPathToNode(node);
  tools_agent_delegate_stub_->UpdateFocusedNode(node_id);
}

void WebDevToolsAgentImpl::DispatchMessageFromClient(
    const std::string& raw_msg) {
  if (dom_agent_impl_.get() &&
      dom_agent_dispatch_.Dispatch(dom_agent_impl_.get(), raw_msg))
    return;
  if (net_agent_impl_.get() &&
      net_agent_dispatch_.Dispatch(net_agent_impl_.get(), raw_msg))
    return;
  tools_agent_dispatch_.Dispatch(this, raw_msg);
}

void WebDevToolsAgentImpl::SendRpcMessage(const std::string& raw_msg) {
  delegate_->SendMessageToClient(raw_msg);
}