java-cef/java/tests/detailed/handler/MessageRouterHandlerEx.java at 612992f37cf5b8f70993e88df0f2e96130e32fae · chromiumembedded/java-cef · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
// Copyright (c) 2014 The Chromium Embedded Framework Authors. All rights
// reserved. Use of this source code is governed by a BSD-style license that
// can be found in the LICENSE file.

package tests.detailed.handler;

import org.cef.CefClient;
import org.cef.browser.CefBrowser;
import org.cef.browser.CefFrame;
import org.cef.browser.CefMessageRouter;
import org.cef.browser.CefMessageRouter.CefMessageRouterConfig;
import org.cef.callback.CefQueryCallback;
import org.cef.handler.CefMessageRouterHandlerAdapter;

import java.nio.ByteBuffer;

public class MessageRouterHandlerEx extends CefMessageRouterHandlerAdapter {
    private final CefClient client_;
    private final CefMessageRouterConfig config_ =
            new CefMessageRouterConfig("myQuery", "myQueryAbort");
    private CefMessageRouter router_ = null;
    private boolean binary_direct_ = false;

    public MessageRouterHandlerEx(final CefClient client) {
        client_ = client;
    }

    @Override
    public boolean onQuery(CefBrowser browser, CefFrame frame, long query_id, String request,
            boolean persistent, CefQueryCallback callback) {
        if (request.startsWith("hasExtension")) {
            if (router_ != null)
                callback.success("");
            else
                callback.failure(0, "");
        } else if (request.startsWith("enableExt")) {
            if (router_ != null) {
                callback.failure(-1, "Already enabled");
            } else {
                router_ = CefMessageRouter.create(config_, new JavaVersionMessageRouter());
                client_.addMessageRouter(router_);
                callback.success("");
            }
        } else if (request.startsWith("disableExt")) {
            if (router_ == null) {
                callback.failure(-2, "Already disabled");
            } else {
                client_.removeMessageRouter(router_);
                router_.dispose();
                router_ = null;
                callback.success("");
            }
        } else if (request.startsWith("doPersistent")) {
            if (persistent) {
                callback.success("Hello,");
                callback.success("World!");
                callback.failure(0, "Finished");
            } else {
                callback.failure(-1, "Request not marked as persistent");
            }
        } else {
            // not handled
            return false;
        }
        return true;
    }

    @Override
    public boolean onQuery(CefBrowser browser, CefFrame frame, long query_id, ByteBuffer request,
            boolean persistent, CefQueryCallback callback) {
        if (request != null) {
            int size = request.capacity();
            boolean direct = (binary_direct_ = !binary_direct_);
            ByteBuffer response = direct ? ByteBuffer.allocateDirect(size)
                                         : ByteBuffer.allocate(size);
            // reverse bytes
            for (int i = 0; i < size; i++) {
                byte b = request.get(i);
                response.put(size - i - 1, b);
            }
            callback.success(response);
            if (persistent) {
                callback.failure(0, direct ? "Finished (direct)"
                                           : "Finished (array)");
            }
        } else {
            // not handled
            return false;
        }
        return true;
    }

    private class JavaVersionMessageRouter extends CefMessageRouterHandlerAdapter {
        @Override
        public boolean onQuery(CefBrowser browser, CefFrame frame, long query_id, String request,
                boolean persistent, CefQueryCallback callback) {
            if (request.startsWith("jcefJava")) {
                callback.success(System.getProperty("java.version"));
                return true;
            }
            return false;
        };
    }
}