webkit/tools/test_shell/test_geolocation_service.cc


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

// Copyright (c) 2010 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include "webkit/tools/test_shell/test_geolocation_service.h"

#include "third_party/WebKit/WebKit/chromium/public/WebGeolocationServiceBridge.h"

TestGeolocationService::TestGeolocationService()
    : allowed_(false),
      permission_set_(false) {
}

TestGeolocationService::~TestGeolocationService() {
  for (IDMap<WebKit::WebGeolocationServiceBridge>::iterator it(&bridges_map_);
       !it.IsAtEnd(); it.Advance()) {
    it.GetCurrentValue()->onWebGeolocationServiceDestroyed();
  }
}

void TestGeolocationService::SetGeolocationPermission(bool allowed) {
  allowed_ = allowed;
  permission_set_ = true;
  TryToSendPermissions();
}

void TestGeolocationService::requestPermissionForFrame(
    int bridgeId, const WebKit::WebURL& url) {
  DCHECK(bridges_map_.Lookup(bridgeId)) << "Unknown bridge " << bridgeId;
  pending_permissions_.push_back(bridgeId);
  TryToSendPermissions();
}

int TestGeolocationService::attachBridge(
    WebKit::WebGeolocationServiceBridge* bridge) {
  return bridges_map_.Add(bridge);
}

void TestGeolocationService::detachBridge(int bridgeId) {
  bridges_map_.Remove(bridgeId);
  std::vector<int>::iterator i = pending_permissions_.begin();
  while (i != pending_permissions_.end()) {
    if (*i == bridgeId)
      pending_permissions_.erase(i);
    else
      ++i;
  }
}

void TestGeolocationService::TryToSendPermissions() {
  if (permission_set_ && !permission_timer_.IsRunning())
    permission_timer_.Start(base::TimeDelta::FromMilliseconds(0),
                            this, &TestGeolocationService::SendPermission);
}

void TestGeolocationService::SendPermission() {
  DCHECK(permission_set_);
  std::vector<int> pending_permissions;
  pending_permissions.swap(pending_permissions_);
  for (std::vector<int>::const_iterator i = pending_permissions.begin();
      i != pending_permissions.end(); ++i) {
    WebKit::WebGeolocationServiceBridge* bridge =
       bridges_map_.Lookup(*i);
    DCHECK(bridge);
    bridge->setIsAllowed(allowed_);
  }
}