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
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
|
// Copyright (c) 2012 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.
#ifndef CHROME_BROWSER_EXTENSIONS_SCRIPT_BADGE_CONTROLLER_H_
#define CHROME_BROWSER_EXTENSIONS_SCRIPT_BADGE_CONTROLLER_H_
#pragma once
#include <map>
#include <set>
#include <string>
#include "base/compiler_specific.h"
#include "base/memory/linked_ptr.h"
#include "base/memory/ref_counted.h"
#include "chrome/browser/extensions/location_bar_controller.h"
#include "chrome/browser/extensions/script_executor.h"
#include "chrome/browser/extensions/script_executor_impl.h"
#include "content/public/browser/notification_observer.h"
#include "content/public/browser/notification_registrar.h"
#include "content/public/browser/web_contents_observer.h"
class ExtensionAction;
class ExtensionService;
class TabContents;
namespace IPC {
class Message;
}
namespace extensions {
class Extension;
// A LocationBarController which displays icons whenever a script is executing
// in a tab. It accomplishes this two different ways:
//
// - For content_script declarations, this receives IPCs from the renderer
// notifying that a content script is running (either on this tab or one of
// its frames), which is recorded.
// - The ScriptExecutor interface is exposed for programmatically executing
// scripts. Successfully executed scripts are recorded.
//
// When extension IDs are recorded a NOTIFICATION_EXTENSION_LOCATION_BAR_UPDATED
// is sent, and those extensions will be returned from GetCurrentActions until
// the next page navigation.
//
// Ref-counted so that it can be safely bound in a base::Bind.
class ScriptBadgeController
: public base::RefCountedThreadSafe<ScriptBadgeController>,
public LocationBarController,
public ScriptExecutor,
public content::WebContentsObserver,
public content::NotificationObserver {
public:
explicit ScriptBadgeController(TabContents* tab_contents);
// LocationBarController implementation.
virtual std::vector<ExtensionAction*> GetCurrentActions() const OVERRIDE;
virtual Action OnClicked(const std::string& extension_id,
int mouse_button) OVERRIDE;
virtual void NotifyChange() OVERRIDE;
// ScriptExecutor implementation.
virtual void ExecuteScript(const std::string& extension_id,
ScriptType script_type,
const std::string& code,
FrameScope frame_scope,
UserScript::RunLocation run_at,
WorldType world_type,
const ExecuteScriptCallback& callback) OVERRIDE;
private:
friend class base::RefCountedThreadSafe<ScriptBadgeController>;
virtual ~ScriptBadgeController();
// Callback for ExecuteScript which if successful and for the current URL
// records that the script is running, then calls the original callback.
void OnExecuteScriptFinished(const std::string& extension_id,
const ExecuteScriptCallback& callback,
bool success,
int32 page_id,
const std::string& error);
// Gets the ExtensionService for |tab_contents_|.
ExtensionService* GetExtensionService();
// Gets the current page ID.
int32 GetPageID();
// content::WebContentsObserver implementation.
virtual void DidNavigateMainFrame(
const content::LoadCommittedDetails& details,
const content::FrameNavigateParams& params) OVERRIDE;
virtual bool OnMessageReceived(const IPC::Message& message) OVERRIDE;
// content::NotificationObserver implementation.
virtual void Observe(int type,
const content::NotificationSource& source,
const content::NotificationDetails& details) OVERRIDE;
// IPC::Message handlers.
void OnContentScriptsExecuting(const std::set<std::string>& extension_ids,
int32 page_id);
// Tries to insert an extension into the relevant collections, and returns
// whether any change was made.
bool InsertExtension(const std::string& extension_id);
// Tries to erase an extension from the relevant collections, and returns
// whether any change was made.
bool EraseExtension(const Extension* extension);
// Delegate ScriptExecutorImpl for running ExecuteScript.
ScriptExecutorImpl script_executor_;
// Our parent TabContents.
TabContents* tab_contents_;
// The current extension actions in the order they appeared.
std::vector<ExtensionAction*> current_actions_;
// The extensions that have called ExecuteScript on the current frame.
std::set<std::string> extensions_executing_scripts_;
// Listen to extension unloaded notifications.
content::NotificationRegistrar registrar_;
DISALLOW_COPY_AND_ASSIGN(ScriptBadgeController);
};
} // namespace extensions
#endif // CHROME_BROWSER_EXTENSIONS_SCRIPT_BADGE_CONTROLLER_H_
|
