1/*
2 * Copyright (C) 2014 Google Inc. All rights reserved.
3 *
4 * Redistribution and use in source and binary forms, with or without
5 * modification, are permitted provided that the following conditions are
6 * met:
7 *
8 *     * Redistributions of source code must retain the above copyright
9 * notice, this list of conditions and the following disclaimer.
10 *     * Redistributions in binary form must reproduce the above
11 * copyright notice, this list of conditions and the following disclaimer
12 * in the documentation and/or other materials provided with the
13 * distribution.
14 *     * Neither the name of Google Inc. nor the names of its
15 * contributors may be used to endorse or promote products derived from
16 * this software without specific prior written permission.
17 *
18 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
19 * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
20 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
21 * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
22 * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
23 * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
24 * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
25 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
26 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
27 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
28 * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
29 */
30
31#include "config.h"
32#include "InternalsGeolocation.h"
33
34#include "core/dom/Document.h"
35#include "core/page/Page.h"
36#include "core/testing/Internals.h"
37#include "modules/geolocation/GeolocationController.h"
38#include "modules/geolocation/GeolocationError.h"
39#include "modules/geolocation/GeolocationPosition.h"
40#include "modules/geolocation/testing/GeolocationClientMock.h"
41#include "wtf/CurrentTime.h"
42
43namespace WebCore {
44
45void InternalsGeolocation::setGeolocationClientMock(Internals&, Document* document)
46{
47    ASSERT(document && document->frame());
48    GeolocationClientMock* client = new GeolocationClientMock();
49
50    for (Frame* childFrame = document->page()->mainFrame(); childFrame; childFrame = childFrame->tree().traverseNext()) {
51        if (childFrame->isLocalFrame())
52            GeolocationController::from(toLocalFrame(childFrame))->setClientForTest(client);
53    }
54}
55
56void InternalsGeolocation::setGeolocationPosition(Internals&, Document* document, double latitude, double longitude, double accuracy)
57{
58    ASSERT(document && document->frame());
59    GeolocationClientMock* client = geolocationClient(document);
60    if (!client)
61        return;
62    client->setPosition(GeolocationPosition::create(currentTime(), latitude, longitude, accuracy));
63}
64
65void InternalsGeolocation::setGeolocationPositionUnavailableError(Internals&, Document* document, const String& message)
66{
67    ASSERT(document && document->frame());
68    GeolocationClientMock* client = geolocationClient(document);
69    if (!client)
70        return;
71    client->setPositionUnavailableError(message);
72}
73
74void InternalsGeolocation::setGeolocationPermission(Internals&, Document* document, bool allowed)
75{
76    ASSERT(document && document->frame());
77    GeolocationClientMock* client = geolocationClient(document);
78    if (!client)
79        return;
80    client->setPermission(allowed);
81}
82
83int InternalsGeolocation::numberOfPendingGeolocationPermissionRequests(Internals&, Document* document)
84{
85    ASSERT(document && document->frame());
86    GeolocationClientMock* client = geolocationClient(document);
87    if (!client)
88        return -1;
89    return client->numberOfPendingPermissionRequests();
90}
91
92GeolocationClientMock* InternalsGeolocation::geolocationClient(Document* document)
93{
94    GeolocationController* controller = GeolocationController::from(document->frame());
95    if (!controller->hasClientForTest())
96        return 0;
97    return static_cast<GeolocationClientMock*>(controller->client());
98}
99
100} // namespace WebCore
101