1 files changed, 145 insertions, 0 deletions
diff --git a/WebKit/android/wds/Command.cpp b/WebKit/android/wds/Command.cpp
new file mode 100644
index 0000000..132b8e5
--- /dev/null
+++ b/WebKit/android/wds/Command.cpp
@@ -0,0 +1,145 @@
+/*
+ * Copyright 2008, The Android Open Source Project
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+#define LOG_TAG "wds"
+#include "config.h"
+
+#include "AndroidLog.h"
+#include "CString.h"
+#include "Command.h"
+#include "Connection.h"
+#include "DebugServer.h"
+#include "Frame.h"
+#include "RenderTreeAsText.h"
+#include "RenderView.h"
+#include "WebViewCore.h"
+#include <utils/Log.h>
+
+#if ENABLE(WDS)
+
+using namespace WebCore;
+
+namespace android {
+
+namespace WDS {
+
+//------------------------------------------------------------------------------
+// Actual commands -- XXX should be moved somewhere else
+//------------------------------------------------------------------------------
+static bool callDumpRenderTree(const Frame* frame, const Connection* conn) {
+    CString str = externalRepresentation(frame->contentRenderer()).latin1();
+    conn->write(str.data(), str.length());
+    return true;
+}
+
+static bool callDumpDomTree(const Frame* frame, const Connection* conn) {
+    WebViewCore::getWebViewCore(frame->view())->dumpDomTree(true);
+
+    FILE* f = fopen(DOM_TREE_LOG_FILE, "r");
+    if (!f) {
+        conn->write("Dom tree written to logcat\n");
+    } else {
+        char buf[512];
+        while (true) {
+            int nread = fread(buf, 1, sizeof(buf), f);
+            if (nread <= 0)
+                break;
+            conn->write(buf, nread);
+        }
+        fclose(f);
+    }
+    return true;
+}
+
+class WebCoreHandler : public Handler {
+public:
+    virtual void post(TargetThreadFunction func, void* v) const {
+        callOnMainThread(func, v);
+    }
+};
+static WebCoreHandler s_webcoreHandler;
+
+//------------------------------------------------------------------------------
+// End command section
+//------------------------------------------------------------------------------
+
+class InternalCommand : public Command {
+public:
+    InternalCommand(const Command* comm, const Frame* frame,
+            const Connection* connection)
+        : Command(*comm)
+        , m_frame(frame)
+        , m_connection(connection) {}
+    virtual ~InternalCommand() { delete m_connection; }
+
+    void doCommand() const {
+        LOGD("Executing command '%s' (%s)", m_name, m_description);
+        if (!m_dispatch(m_frame, m_connection))
+            // XXX: Have useful failure messages
+            m_connection->write("EPIC FAIL!\n", 11);
+    }
+
+private:
+    const Frame* m_frame;
+    const Connection* m_connection;
+};
+
+static void commandDispatcher(void* v) {
+    InternalCommand* c = static_cast<InternalCommand*>(v);
+    c->doCommand();
+    delete c;
+}
+
+void Command::dispatch() {
+    m_handler.post(commandDispatcher, this);
+}
+
+Vector<const Command*>* Command::s_commands;
+
+void Command::Init() {
+    // Do not initialize twice.
+    if (s_commands)
+        return;
+    // XXX: Move this somewhere else.
+    s_commands = new Vector<const Command*>();
+    s_commands->append(new Command("DDOM", "Dump Dom Tree",
+                callDumpDomTree, s_webcoreHandler));
+    s_commands->append(new Command("DDRT", "Dump Render Tree",
+                callDumpRenderTree, s_webcoreHandler));
+}
+
+Command* Command::Find(const Connection* conn) {
+    char buf[COMMAND_LENGTH];
+    if (conn->read(buf, sizeof(buf)) != COMMAND_LENGTH)
+        return NULL;
+
+    // Linear search of commands. TODO: binary search when more commands are
+    // added.
+    Vector<const Command*>::const_iterator i = s_commands->begin();
+    Vector<const Command*>::const_iterator end = s_commands->end();
+    while (i != end) {
+        if (strncmp(buf, (*i)->name(), sizeof(buf)) == 0)
+            return new InternalCommand(*i, server()->getFrame(0), conn);
+        i++;
+    }
+    return NULL;
+}
+
+} // end namespace WDS
+
+} // end namespace android
+
+#endif