/*

 * Copyright (C) 2007 Apple Inc. All rights reserved.

 *

 * Redistribution and use in source and binary forms, with or without

 * modification, are permitted provided that the following conditions

 * are met:

 *

 * 1.  Redistributions of source code must retain the above copyright

 *     notice, this list of conditions and the following disclaimer. 

 * 2.  Redistributions in binary form must reproduce the above copyright

 *     notice, this list of conditions and the following disclaimer in the

 *     documentation and/or other materials provided with the distribution. 

 * 3.  Neither the name of Apple Computer, Inc. ("Apple") nor the names of

 *     its contributors may be used to endorse or promote products derived

 *     from this software without specific prior written permission. 

 *

 * THIS SOFTWARE IS PROVIDED BY APPLE AND ITS CONTRIBUTORS "AS IS" AND ANY

 * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED

 * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE

 * DISCLAIMED. IN NO EVENT SHALL APPLE OR ITS CONTRIBUTORS BE LIABLE FOR ANY

 * DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES

 * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;

 * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND

 * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT

 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF

 * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.

 */

#include "config.h"

#include "InspectorController.h"

#include "CString.h"

#include "CachedResource.h"

#include "DocLoader.h"

#include "Document.h"

#include "DocumentLoader.h"

#include "Element.h"

#include "FloatRect.h"

#include "Frame.h"

#include "FrameLoader.h"

#include "FrameTree.h"

#include "HTMLFrameOwnerElement.h"

#include "InspectorClient.h"

#include "JSRange.h"

#include "Page.h"

#include "Range.h"

#include "ResourceRequest.h"

#include "ResourceResponse.h"

#include "Settings.h"

#include "Shared.h"

#include "SharedBuffer.h"

#include "SystemTime.h"

#include "TextEncoding.h"

#include "TextIterator.h"

#include "kjs_dom.h"

#include "kjs_proxy.h"

#include "kjs_window.h"

#include <JavaScriptCore/APICast.h>

#include <JavaScriptCore/JSLock.h>

#include <JavaScriptCore/JSStringRef.h>

namespace WebCore {

struct ConsoleMessage {

    ConsoleMessage(MessageSource s, MessageLevel l, const String& m, unsigned li, const String& u)

        : source(s)

        , level(l)

        , message(m)

        , line(li)

        , url(u)

    {

    }

    MessageSource source;

    MessageLevel level;

    String message;

    unsigned line;

    String url;

};

struct InspectorResource : public Shared<InspectorResource> {

    // Keep these in sync with WebInspector.Resource.Type

    enum Type {

        Doc,

        Stylesheet,

        Image,

        Script,

        Other

    };

    InspectorResource(long long identifier, DocumentLoader* documentLoader, Frame* frame)

        : identifier(identifier)

        , loader(documentLoader)

        , frame(frame)

        , scriptContext(0)

        , scriptObject(0)

        , expectedContentLength(0)

        , cached(false)

        , finished(false)

        , failed(false)

        , length(0)

        , responseStatusCode(0)

        , startTime(-1.0)

        , responseReceivedTime(-1.0)

        , endTime(-1.0)

    {

    }

    ~InspectorResource()

    {

        setScriptObject(0, 0);

    }

    Type type() const

    {

        if (requestURL == loader->requestURL())

            return Doc;

        FrameLoader* frameLoader = loader->frameLoader();

        if (!frameLoader)

            return Other;

        if (requestURL == frameLoader->iconURL())

            return Image;

        Document* doc = frameLoader->frame()->document();

        if (!doc)

            return Other;

        CachedResource* cachedResource = doc->docLoader()->cachedResource(requestURL.url());

        if (!cachedResource)

            return Other;

        switch (cachedResource->type()) {

            case CachedResource::ImageResource:

                return Image;

            case CachedResource::CSSStyleSheet:

#if ENABLE(XSLT)

            case CachedResource::XSLStyleSheet:

#endif

                return Stylesheet;

            case CachedResource::Script:

                return Script;

            default:

                return Other;

        }

    }

    void setScriptObject(JSContextRef context, JSObjectRef newScriptObject)

    {

        if (scriptContext && scriptObject)

            JSValueUnprotect(scriptContext, scriptObject);

        scriptObject = newScriptObject;

        scriptContext = context;

        ASSERT((context && newScriptObject) || (!context && !newScriptObject));

        if (context && newScriptObject)

            JSValueProtect(context, newScriptObject);

    }

    long long identifier;

    RefPtr<DocumentLoader> loader;

    RefPtr<Frame> frame;

    KURL requestURL;

    HTTPHeaderMap requestHeaderFields;

    HTTPHeaderMap responseHeaderFields;

    String mimeType;

    String suggestedFilename;

    String textEncodingName;

    JSContextRef scriptContext;

    JSObjectRef scriptObject;

    long long expectedContentLength;

    bool cached;

    bool finished;

    bool failed;

    int length;

    int responseStatusCode;

    double startTime;

    double responseReceivedTime;

    double endTime;

};

static JSValueRef addSourceToFrame(JSContextRef ctx, JSObjectRef /*function*/, JSObjectRef thisObject, size_t argumentCount, const JSValueRef arguments[], JSValueRef* /*exception*/)

{

    JSValueRef undefined = JSValueMakeUndefined(ctx);

    InspectorController* controller = reinterpret_cast<InspectorController*>(JSObjectGetPrivate(thisObject));

    if (argumentCount < 2 || !controller)

        return undefined;

    JSValueRef identifierValue = arguments[0];

    if (!JSValueIsNumber(ctx, identifierValue))

        return undefined;

    unsigned long identifier = static_cast<unsigned long>(JSValueToNumber(ctx, identifierValue, 0));

    RefPtr<InspectorResource> resource = controller->resources().get(identifier);

    ASSERT(resource);

    if (!resource)

        return undefined;

    RefPtr<SharedBuffer> buffer;

    if (resource->requestURL == resource->loader->requestURL())

        buffer = resource->loader->mainResourceData();

    else {

        FrameLoader* frameLoader = resource->loader->frameLoader();

        if (!frameLoader)

            return undefined;

        Document* doc = frameLoader->frame()->document();

        if (!doc)

            return undefined;

        CachedResource* cachedResource = doc->docLoader()->cachedResource(resource->requestURL.url());

        if (!cachedResource)

            return undefined;

        buffer = cachedResource->data();

    }

    if (!buffer)

        return undefined;

    String textEncodingName = resource->loader->overrideEncoding();

    if (!textEncodingName)

        textEncodingName = resource->textEncodingName;

    TextEncoding encoding(textEncodingName);

    if (!encoding.isValid())

        encoding = WindowsLatin1Encoding();

    String sourceString = encoding.decode(buffer->data(), buffer->size());

    Node* node = toNode(toJS(arguments[1]));

    ASSERT(node);

    if (!node)

        return undefined;

    if (!node->attached()) {

        ASSERT_NOT_REACHED();

        return undefined;

    }

    ASSERT(node->isElementNode());

    if (!node->isElementNode())

        return undefined;

    Element* element = static_cast<Element*>(node);

    ASSERT(element->isFrameOwnerElement());

    if (!element->isFrameOwnerElement())

        return undefined;

    HTMLFrameOwnerElement* frameOwner = static_cast<HTMLFrameOwnerElement*>(element);

    ASSERT(frameOwner->contentFrame());

    if (!frameOwner->contentFrame())

        return undefined;

    FrameLoader* loader = frameOwner->contentFrame()->loader();

    loader->setResponseMIMEType(resource->mimeType);

    loader->begin();

    loader->write(sourceString);

    loader->end();

    return undefined;

}

static JSValueRef getResourceDocumentNode(JSContextRef ctx, JSObjectRef /*function*/, JSObjectRef thisObject, size_t argumentCount, const JSValueRef arguments[], JSValueRef* /*exception*/)

{

    JSValueRef undefined = JSValueMakeUndefined(ctx);

    InspectorController* controller = reinterpret_cast<InspectorController*>(JSObjectGetPrivate(thisObject));

    if (!argumentCount || argumentCount > 1 || !controller)

        return undefined;

    JSValueRef identifierValue = arguments[0];

    if (!JSValueIsNumber(ctx, identifierValue))

        return undefined;

    unsigned long identifier = static_cast<unsigned long>(JSValueToNumber(ctx, identifierValue, 0));

    RefPtr<InspectorResource> resource = controller->resources().get(identifier);

    ASSERT(resource);

    if (!resource)

        return undefined;

    FrameLoader* frameLoader = resource->loader->frameLoader();

    if (!frameLoader)

        return undefined;

    Document* document = frameLoader->frame()->document();

    if (!document)

        return undefined;

    if (document->isPluginDocument() || document->isImageDocument())

        return undefined;

    KJS::JSLock lock;

    JSValueRef documentValue = toRef(toJS(toJS(controller->scriptContext()), document));

    return documentValue;

}

static JSValueRef highlightDOMNode(JSContextRef context, JSObjectRef /*function*/, JSObjectRef thisObject, size_t argumentCount, const JSValueRef arguments[], JSValueRef* /*exception*/)

{

    JSValueRef undefined = JSValueMakeUndefined(context);

    InspectorController* controller = reinterpret_cast<InspectorController*>(JSObjectGetPrivate(thisObject));

    if (argumentCount < 1 || !controller)

        return undefined;

    Node* node = toNode(toJS(arguments[0]));

    if (!node)

        return undefined;

    controller->highlight(node);

    return undefined;

}

static JSValueRef hideDOMNodeHighlight(JSContextRef context, JSObjectRef /*function*/, JSObjectRef thisObject, size_t argumentCount, const JSValueRef arguments[], JSValueRef* /*exception*/)

{

    JSValueRef undefined = JSValueMakeUndefined(context);

    InspectorController* controller = reinterpret_cast<InspectorController*>(JSObjectGetPrivate(thisObject));

    if (argumentCount || !controller)

        return undefined;

    controller->hideHighlight();

    return undefined;

}

static JSValueRef loaded(JSContextRef ctx, JSObjectRef /*function*/, JSObjectRef thisObject, size_t /*argumentCount*/, const JSValueRef[] /*arguments[]*/, JSValueRef* /*exception*/)

{

    InspectorController* controller = reinterpret_cast<InspectorController*>(JSObjectGetPrivate(thisObject));

    if (!controller)

        return JSValueMakeUndefined(ctx);

    controller->scriptObjectReady();

    return JSValueMakeUndefined(ctx);

}

static JSValueRef unloading(JSContextRef ctx, JSObjectRef /*function*/, JSObjectRef thisObject, size_t /*argumentCount*/, const JSValueRef[] /*arguments[]*/, JSValueRef* /*exception*/)

{

    InspectorController* controller = reinterpret_cast<InspectorController*>(JSObjectGetPrivate(thisObject));

    if (!controller)

        return JSValueMakeUndefined(ctx);

    controller->windowUnloading();

    return JSValueMakeUndefined(ctx);

}

static JSValueRef attach(JSContextRef ctx, JSObjectRef /*function*/, JSObjectRef thisObject, size_t /*argumentCount*/, const JSValueRef[] /*arguments[]*/, JSValueRef* /*exception*/)

{

    InspectorController* controller = reinterpret_cast<InspectorController*>(JSObjectGetPrivate(thisObject));

    if (!controller)

        return JSValueMakeUndefined(ctx);

    controller->attachWindow();

    return JSValueMakeUndefined(ctx);

}

static JSValueRef detach(JSContextRef ctx, JSObjectRef /*function*/, JSObjectRef thisObject, size_t /*argumentCount*/, const JSValueRef[] /*arguments[]*/, JSValueRef* /*exception*/)

{

    InspectorController* controller = reinterpret_cast<InspectorController*>(JSObjectGetPrivate(thisObject));

    if (!controller)

        return JSValueMakeUndefined(ctx);

    controller->detachWindow();

    return JSValueMakeUndefined(ctx);

}

static JSValueRef log(JSContextRef ctx, JSObjectRef /*function*/, JSObjectRef /*thisObject*/, size_t argumentCount, const JSValueRef arguments[], JSValueRef* /*exception*/)

{

    if (argumentCount < 1 || !JSValueIsString(ctx, arguments[0]))

        return JSValueMakeUndefined(ctx);

#ifndef NDEBUG

    JSStringRef string = JSValueToStringCopy(ctx, arguments[0], 0);

    String message(JSStringGetCharactersPtr(string), JSStringGetLength(string));

    JSStringRelease(string);

    fprintf(stderr, "%s\n", message.latin1().data());

#endif

    return JSValueMakeUndefined(ctx);

}

static JSValueRef search(JSContextRef ctx, JSObjectRef /*function*/, JSObjectRef thisObject, size_t argumentCount, const JSValueRef arguments[], JSValueRef* /*exception*/)

{

    InspectorController* controller = reinterpret_cast<InspectorController*>(JSObjectGetPrivate(thisObject));

    if (!controller)

        return JSValueMakeUndefined(ctx);

    if (argumentCount < 2 || !JSValueIsString(ctx, arguments[1]))

        return JSValueMakeUndefined(ctx);

    Node* node = toNode(toJS(arguments[0]));

    if (!node)

        return JSValueMakeUndefined(ctx);

    JSStringRef string = JSValueToStringCopy(ctx, arguments[1], 0);

    String target(JSStringGetCharactersPtr(string), JSStringGetLength(string));

    JSStringRelease(string);

    JSObjectRef globalObject = JSContextGetGlobalObject(ctx);

    JSStringRef constructorString = JSStringCreateWithUTF8CString("Array");

    JSObjectRef arrayConstructor = JSValueToObject(ctx, JSObjectGetProperty(ctx, globalObject, constructorString, 0), 0);

    JSStringRelease(constructorString);

    JSObjectRef array = JSObjectCallAsConstructor(ctx, arrayConstructor, 0, 0, 0);

    JSStringRef pushString = JSStringCreateWithUTF8CString("push");

    JSValueRef pushValue = JSObjectGetProperty(ctx, array, pushString, 0);

    JSStringRelease(pushString);

    JSObjectRef push = JSValueToObject(ctx, pushValue, 0);

    RefPtr<Range> searchRange(rangeOfContents(node));

    int exception = 0;

    do {

        RefPtr<Range> resultRange(findPlainText(searchRange.get(), target, true, false));

        if (resultRange->collapsed(exception))

            break;

        // A non-collapsed result range can in some funky whitespace cases still not

        // advance the range's start position (4509328). Break to avoid infinite loop.

        VisiblePosition newStart = endVisiblePosition(resultRange.get(), DOWNSTREAM);

        if (newStart == startVisiblePosition(searchRange.get(), DOWNSTREAM))

            break;

        KJS::JSLock lock;

        JSValueRef arg0 = toRef(toJS(toJS(ctx), resultRange.get()));

        JSObjectCallAsFunction(ctx, push, array, 1, &arg0, 0);

        setStart(searchRange.get(), newStart);

    } while (true);

    return array;

}

static JSValueRef inspectedWindow(JSContextRef ctx, JSObjectRef /*function*/, JSObjectRef thisObject, size_t /*argumentCount*/, const JSValueRef[] /*arguments[]*/, JSValueRef* /*exception*/)

{

    InspectorController* controller = reinterpret_cast<InspectorController*>(JSObjectGetPrivate(thisObject));

    if (!controller)

        return JSValueMakeUndefined(ctx);

    return toRef(KJS::Window::retrieve(controller->inspectedPage()->mainFrame()));

}

InspectorController::InspectorController(Page* page, InspectorClient* client)

    : m_inspectedPage(page)

    , m_client(client)

    , m_page(0)

    , m_scriptObject(0)

    , m_controllerScriptObject(0)

    , m_scriptContext(0)

    , m_windowVisible(false)

    , m_nextIdentifier(-2)

{

    ASSERT_ARG(page, page);

    ASSERT_ARG(client, client);

}

InspectorController::~InspectorController()

{

    if (m_scriptContext) {

        JSObjectRef global = JSContextGetGlobalObject(m_scriptContext);

        JSStringRef controllerProperty = JSStringCreateWithUTF8CString("InspectorController");

        JSObjectRef controller = JSValueToObject(m_scriptContext, JSObjectGetProperty(m_scriptContext, global, controllerProperty, 0), 0);

        JSStringRelease(controllerProperty);

        JSObjectSetPrivate(controller, 0);

    }

    m_client->closeWindow();

    m_client->inspectorDestroyed();

    if (m_page)

        m_page->setParentInspectorController(0);

    deleteAllValues(m_frameResources);

    deleteAllValues(m_consoleMessages);

}

bool InspectorController::enabled() const

{

    return m_inspectedPage->settings()->developerExtrasEnabled();

}

void InspectorController::inspect(Node* node)

{

    if (!node || !enabled())

        return;

    if (!m_page) {

        m_page = m_client->createPage();

        if (!m_page)

            return;

        m_page->setParentInspectorController(this);

    }

    if (node->nodeType() != Node::ELEMENT_NODE && node->nodeType() != Node::DOCUMENT_NODE)

        node = node->parentNode();

    m_nodeToFocus = node;

    if (!m_scriptObject)

        return;

    if (windowVisible())

        focusNode();

    else

        m_client->showWindow();

}

void InspectorController::focusNode()

{

    if (!enabled())

        return;

    ASSERT(m_scriptContext);

    ASSERT(m_scriptObject);

    ASSERT(m_nodeToFocus);

    JSValueRef arg0;

    {

        KJS::JSLock lock;

        arg0 = toRef(toJS(toJS(m_scriptContext), m_nodeToFocus.get()));

    }

    m_nodeToFocus = 0;