#include "config.h"
#include "WebWorkerBase.h"
#include "GenericWorkerTask.h"
#include "MessagePortChannel.h"
#include "PlatformMessagePortChannel.h"
#include "WebDataSourceImpl.h"
#include "WebFrameClient.h"
#include "WebFrameImpl.h"
#include "WebMessagePortChannel.h"
#include "WebView.h"
#include "WebWorkerClient.h"
#include "WorkerThread.h"
#include <wtf/MainThread.h>
using namespace WebCore;
namespace WebKit {
#if ENABLE(WORKERS)
static void initializeWebKitStaticValues()
{
static bool initialized = false;
if (!initialized) {
initialized = true;
RefPtr<SecurityOrigin> origin =
SecurityOrigin::create(KURL(ParsedURLString, "http://localhost"));
origin.release();
}
}
WebWorkerBase::WebWorkerBase()
: m_webView(0)
, m_askedToTerminate(false)
{
initializeWebKitStaticValues();
}
WebWorkerBase::~WebWorkerBase()
{
ASSERT(m_webView);
WebFrameImpl* webFrame = static_cast<WebFrameImpl*>(m_webView->mainFrame());
if (webFrame)
webFrame->setClient(0);
m_webView->close();
}
void WebWorkerBase::stopWorkerThread()
{
if (m_askedToTerminate)
return;
m_askedToTerminate = true;
if (m_workerThread)
m_workerThread->stop();
}
void WebWorkerBase::initializeLoader(const WebURL& url)
{
ASSERT(!m_webView);
m_webView = WebView::create(0);
m_webView->initializeMainFrame(this);
WebFrameImpl* webFrame = static_cast<WebFrameImpl*>(m_webView->mainFrame());
CString content("");
int len = static_cast<int>(content.length());
RefPtr<SharedBuffer> buf(SharedBuffer::create(content.data(), len));
SubstituteData substData(buf, String("text/html"), String("UTF-8"), KURL());
webFrame->frame()->loader()->load(ResourceRequest(url), substData, false);
m_loadingDocument = webFrame->frame()->document();
}
void WebWorkerBase::dispatchTaskToMainThread(PassOwnPtr<ScriptExecutionContext::Task> task)
{
return callOnMainThread(invokeTaskMethod, task.release());
}
void WebWorkerBase::invokeTaskMethod(void* param)
{
ScriptExecutionContext::Task* task =
static_cast<ScriptExecutionContext::Task*>(param);
task->performTask(0);
delete task;
}
void WebWorkerBase::didCreateDataSource(WebFrame*, WebDataSource* ds)
{
static_cast<WebDataSourceImpl*>(ds)->setDeferMainResourceDataLoad(false);
}
WebApplicationCacheHost* WebWorkerBase::createApplicationCacheHost(WebFrame*, WebApplicationCacheHostClient* appcacheHostClient)
{
if (commonClient())
return commonClient()->createApplicationCacheHost(appcacheHostClient);
return 0;
}
void WebWorkerBase::postMessageToWorkerObject(PassRefPtr<SerializedScriptValue> message,
PassOwnPtr<MessagePortChannelArray> channels)
{
dispatchTaskToMainThread(createCallbackTask(&postMessageTask, this,
message->toWireString(), channels));
}
void WebWorkerBase::postMessageTask(ScriptExecutionContext* context,
WebWorkerBase* thisPtr,
String message,
PassOwnPtr<MessagePortChannelArray> channels)
{
if (!thisPtr->client())
return;
WebMessagePortChannelArray webChannels(channels.get() ? channels->size() : 0);
for (size_t i = 0; i < webChannels.size(); ++i) {
webChannels[i] = (*channels)[i]->channel()->webChannelRelease();
webChannels[i]->setClient(0);
}
thisPtr->client()->postMessageToWorkerObject(message, webChannels);
}
void WebWorkerBase::postExceptionToWorkerObject(const String& errorMessage,
int lineNumber,
const String& sourceURL)
{
dispatchTaskToMainThread(createCallbackTask(&postExceptionTask, this,
errorMessage, lineNumber,
sourceURL));
}
void WebWorkerBase::postExceptionTask(ScriptExecutionContext* context,
WebWorkerBase* thisPtr,
const String& errorMessage,
int lineNumber, const String& sourceURL)
{
if (!thisPtr->commonClient())
return;
thisPtr->commonClient()->postExceptionToWorkerObject(errorMessage,
lineNumber,
sourceURL);
}
void WebWorkerBase::postConsoleMessageToWorkerObject(MessageSource source,
MessageType type,
MessageLevel level,
const String& message,
int lineNumber,
const String& sourceURL)
{
dispatchTaskToMainThread(createCallbackTask(&postConsoleMessageTask, this,
source, type, level,
message, lineNumber, sourceURL));
}
void WebWorkerBase::postConsoleMessageTask(ScriptExecutionContext* context,
WebWorkerBase* thisPtr,
int source,
int type, int level,
const String& message,
int lineNumber,
const String& sourceURL)
{
if (!thisPtr->commonClient())
return;
thisPtr->commonClient()->postConsoleMessageToWorkerObject(source,
type, level, message,
lineNumber, sourceURL);
}
void WebWorkerBase::confirmMessageFromWorkerObject(bool hasPendingActivity)
{
dispatchTaskToMainThread(createCallbackTask(&confirmMessageTask, this,
hasPendingActivity));
}
void WebWorkerBase::confirmMessageTask(ScriptExecutionContext* context,
WebWorkerBase* thisPtr,
bool hasPendingActivity)
{
if (!thisPtr->client())
return;
thisPtr->client()->confirmMessageFromWorkerObject(hasPendingActivity);
}
void WebWorkerBase::reportPendingActivity(bool hasPendingActivity)
{
dispatchTaskToMainThread(createCallbackTask(&reportPendingActivityTask,
this, hasPendingActivity));
}
void WebWorkerBase::reportPendingActivityTask(ScriptExecutionContext* context,
WebWorkerBase* thisPtr,
bool hasPendingActivity)
{
if (!thisPtr->client())
return;
thisPtr->client()->reportPendingActivity(hasPendingActivity);
}
void WebWorkerBase::workerContextClosed()
{
dispatchTaskToMainThread(createCallbackTask(&workerContextClosedTask,
this));
}
void WebWorkerBase::workerContextClosedTask(ScriptExecutionContext* context,
WebWorkerBase* thisPtr)
{
if (thisPtr->commonClient())
thisPtr->commonClient()->workerContextClosed();
thisPtr->stopWorkerThread();
}
void WebWorkerBase::workerContextDestroyed()
{
dispatchTaskToMainThread(createCallbackTask(&workerContextDestroyedTask,
this));
}
void WebWorkerBase::workerContextDestroyedTask(ScriptExecutionContext* context,
WebWorkerBase* thisPtr)
{
if (thisPtr->commonClient())
thisPtr->commonClient()->workerContextDestroyed();
delete thisPtr;
}
void WebWorkerBase::postTaskToLoader(PassOwnPtr<ScriptExecutionContext::Task> task)
{
ASSERT(m_loadingDocument->isDocument());
m_loadingDocument->postTask(task);
}
void WebWorkerBase::postTaskForModeToWorkerContext(
PassOwnPtr<ScriptExecutionContext::Task> task, const String& mode)
{
m_workerThread->runLoop().postTaskForMode(task, mode);
}
#endif // ENABLE(WORKERS)
}