/*

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

 * Copyright (C) 2008 Matt Lilek <webkit@mattlilek.com>

 * Copyright (C) 2010 Google 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:

 *

 *     * Redistributions of source code must retain the above copyright

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

 *     * 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.

 *     * Neither the name of Google Inc. 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 THE COPYRIGHT HOLDERS AND 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 THE COPYRIGHT

 * OWNER OR 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 "JSInjectedScriptHost.h"

#if ENABLE(INSPECTOR)

#include "Console.h"

#include "JSMainThreadExecState.h"

#if ENABLE(DATABASE)

#include "Database.h"

#include "JSDatabase.h"

#endif

#include "ExceptionCode.h"

#include "Frame.h"

#include "FrameLoader.h"

#include "InjectedScript.h"

#include "InjectedScriptHost.h"

#include "InspectorController.h"

#include "InspectorResource.h"

#include "JSDOMWindow.h"

#include "JSDOMWindowCustom.h"

#include "JSNode.h"

#include "JSRange.h"

#include "Node.h"

#include "Page.h"

#if ENABLE(DOM_STORAGE)

#include "SerializedScriptValue.h"

#include "Storage.h"

#include "JSStorage.h"

#endif

#include "TextIterator.h"

#include "VisiblePosition.h"

#include <parser/SourceCode.h>

#include <runtime/JSArray.h>

#include <runtime/JSLock.h>

#include <wtf/RefPtr.h>

#include <wtf/Vector.h>

#if ENABLE(JAVASCRIPT_DEBUGGER)

#include "JavaScriptCallFrame.h"

#include "JSJavaScriptCallFrame.h"

#include "ScriptDebugServer.h"

#endif

using namespace JSC;

namespace WebCore {

ScriptObject InjectedScriptHost::createInjectedScript(const String& source, ScriptState* scriptState, long id)

{

    SourceCode sourceCode = makeSource(stringToUString(source));

    JSLock lock(SilenceAssertionsOnly);

    JSDOMGlobalObject* globalObject = static_cast<JSDOMGlobalObject*>(scriptState->lexicalGlobalObject());

    JSValue globalThisValue = scriptState->globalThisValue();

    Completion comp = JSMainThreadExecState::evaluate(scriptState, globalObject->globalScopeChain(), sourceCode, globalThisValue);

    if (comp.complType() != JSC::Normal && comp.complType() != JSC::ReturnValue)

        return ScriptObject();

    JSValue functionValue = comp.value();

    CallData callData;

    CallType callType = getCallData(functionValue, callData);

    if (callType == CallTypeNone)

        return ScriptObject();

    MarkedArgumentBuffer args;

    args.append(toJS(scriptState, globalObject, this));

    args.append(globalThisValue);

    args.append(jsNumber(scriptState, id));

    args.append(jsString(scriptState, String("JSC")));

    JSValue result = JSC::call(scriptState, functionValue, callType, callData, globalThisValue, args);

    if (result.isObject())

        return ScriptObject(scriptState, result.getObject());

    return ScriptObject();

}

#if ENABLE(DATABASE)

JSValue JSInjectedScriptHost::databaseForId(ExecState* exec)

{

    if (exec->argumentCount() < 1)

        return jsUndefined();

    InspectorController* ic = impl()->inspectorController();

    if (!ic)

        return jsUndefined();

    Database* database = impl()->databaseForId(exec->argument(0).toInt32(exec));

    if (!database)

        return jsUndefined();

    return toJS(exec, database);

}

#endif

#if ENABLE(JAVASCRIPT_DEBUGGER)

JSValue JSInjectedScriptHost::currentCallFrame(ExecState* exec)

{

    JavaScriptCallFrame* callFrame = ScriptDebugServer::shared().currentCallFrame();

    if (!callFrame || !callFrame->isValid())

        return jsUndefined();

    JSLock lock(SilenceAssertionsOnly);

    return toJS(exec, callFrame);

}

#endif

JSValue JSInjectedScriptHost::nodeForId(ExecState* exec)

{

    if (exec->argumentCount() < 1)

        return jsUndefined();

    Node* node = impl()->nodeForId(exec->argument(0).toInt32(exec));

    if (!node)

        return jsUndefined();

    InspectorController* ic = impl()->inspectorController();

    if (!ic)

        return jsUndefined();

    JSLock lock(SilenceAssertionsOnly);

    return toJS(exec, node);

}

JSValue JSInjectedScriptHost::pushNodePathToFrontend(ExecState* exec)

{

    if (exec->argumentCount() < 3)

        return jsUndefined();

    Node* node = toNode(exec->argument(0));

    if (!node)

        return jsUndefined();

    bool withChildren = exec->argument(1).toBoolean(exec);

    bool selectInUI = exec->argument(2).toBoolean(exec);

    return jsNumber(exec, impl()->pushNodePathToFrontend(node, withChildren, selectInUI));

}

#if ENABLE(DATABASE)

JSValue JSInjectedScriptHost::selectDatabase(ExecState* exec)

{

    if (exec->argumentCount() < 1)

        return jsUndefined();

    Database* database = toDatabase(exec->argument(0));

    if (database)

        impl()->selectDatabase(database);

    return jsUndefined();

}

#endif

#if ENABLE(DOM_STORAGE)

JSValue JSInjectedScriptHost::selectDOMStorage(ExecState* exec)

{

    if (exec->argumentCount() < 1)

        return jsUndefined();

    InspectorController* ic = impl()->inspectorController();

    if (!ic)

        return jsUndefined();

    Storage* storage = toStorage(exec->argument(0));

    if (storage)

        impl()->selectDOMStorage(storage);

    return jsUndefined();

}

#endif

JSValue JSInjectedScriptHost::reportDidDispatchOnInjectedScript(ExecState* exec)

{

    if (exec->argumentCount() < 3)

        return jsUndefined();

    if (!exec->argument(0).isInt32())

        return jsUndefined();

    int callId = exec->argument(0).asInt32();

    RefPtr<SerializedScriptValue> result(SerializedScriptValue::create(exec, exec->argument(1)));

    bool isException;

    if (!exec->argument(2).getBoolean(isException))

        return jsUndefined();

    impl()->reportDidDispatchOnInjectedScript(callId, result.get(), isException);

    return jsUndefined();

}

InjectedScript InjectedScriptHost::injectedScriptFor(ScriptState* scriptState)

{

    JSLock lock(SilenceAssertionsOnly);

    JSDOMGlobalObject* globalObject = static_cast<JSDOMGlobalObject*>(scriptState->lexicalGlobalObject());

    JSObject* injectedScript = globalObject->injectedScript();

    if (injectedScript)

        return InjectedScript(ScriptObject(scriptState, injectedScript));

    ASSERT(!m_injectedScriptSource.isEmpty()); 

    pair<long, ScriptObject> injectedScriptObject = injectScript(m_injectedScriptSource, scriptState);

    globalObject->setInjectedScript(injectedScriptObject.second.jsObject());

    InjectedScript result(injectedScriptObject.second);

    m_idToInjectedScript.set(injectedScriptObject.first, result);

    return result;

}

bool InjectedScriptHost::canAccessInspectedWindow(ScriptState* scriptState)

{

    JSLock lock(SilenceAssertionsOnly);

    JSDOMWindow* inspectedWindow = toJSDOMWindow(scriptState->lexicalGlobalObject());

    if (!inspectedWindow)

        return false;

    return inspectedWindow->allowsAccessFromNoErrorMessage(scriptState);

}

} // namespace WebCore

#endif // ENABLE(INSPECTOR)