OLD | NEW |
1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. |
2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 #ifndef CHROME_TEST_LIVE_SYNC_LIVE_SYNC_EXTENSION_HELPER_H_ | 5 #ifndef CHROME_TEST_LIVE_SYNC_LIVE_SYNC_EXTENSION_HELPER_H_ |
6 #define CHROME_TEST_LIVE_SYNC_LIVE_SYNC_EXTENSION_HELPER_H_ | 6 #define CHROME_TEST_LIVE_SYNC_LIVE_SYNC_EXTENSION_HELPER_H_ |
7 #pragma once | 7 #pragma once |
8 | 8 |
9 #include <map> | 9 #include <map> |
10 #include <string> | 10 #include <string> |
11 | 11 |
12 #include "base/basictypes.h" | 12 #include "base/basictypes.h" |
13 #include "base/compiler_specific.h" | 13 #include "base/compiler_specific.h" |
14 #include "base/memory/ref_counted.h" | 14 #include "base/memory/ref_counted.h" |
15 #include "chrome/common/extensions/extension.h" | 15 #include "chrome/common/extensions/extension.h" |
16 | 16 |
17 class Extension; | 17 class Extension; |
18 class LiveSyncTest; | 18 class LiveSyncTest; |
19 class Profile; | 19 class Profile; |
20 | 20 |
21 class LiveSyncExtensionHelper { | 21 class LiveSyncExtensionHelper { |
22 public: | 22 public: |
23 enum ExtensionState { DISABLED, PENDING, ENABLED }; | |
24 | |
25 typedef std::map<std::string, ExtensionState> ExtensionStateMap; | |
26 | |
27 LiveSyncExtensionHelper(); | 23 LiveSyncExtensionHelper(); |
28 ~LiveSyncExtensionHelper(); | 24 ~LiveSyncExtensionHelper(); |
29 | 25 |
30 // Returns a generated extension ID for the given name. | 26 // Returns a generated extension ID for the given name. |
31 static std::string NameToId(const std::string& name); | 27 static std::string NameToId(const std::string& name); |
32 | 28 |
33 // Initializes the profiles in |test| and registers them with | 29 // Initializes the profiles in |test| and registers them with |
34 // internal data structures. | 30 // internal data structures. |
35 void Setup(LiveSyncTest* test); | 31 void Setup(LiveSyncTest* test); |
36 | 32 |
37 // Installs the extension with the given name to |profile|. | 33 // Installs the extension with the given name to |profile|. |
38 void InstallExtension( | 34 void InstallExtension( |
39 Profile* profile, const std::string& name, Extension::Type type); | 35 Profile* profile, const std::string& name, Extension::Type type); |
40 | 36 |
41 // Uninstalls the extension with the given name from |profile|. | 37 // Uninstalls the extension with the given name from |profile|. |
42 void UninstallExtension(Profile* profile, const std::string& name); | 38 void UninstallExtension(Profile* profile, const std::string& name); |
43 | 39 |
| 40 // Enables the extension with the given name on |profile|. |
| 41 void EnableExtension(Profile* profile, const std::string& name); |
| 42 |
| 43 // Disables the extension with the given name on |profile|. |
| 44 void DisableExtension(Profile* profile, const std::string& name); |
| 45 |
| 46 // Enables the extension with the given name to run in incognito mode |
| 47 void IncognitoEnableExtension(Profile* profile, const std::string& name); |
| 48 |
| 49 // Disables the extension with the given name from running in incognito mode |
| 50 void IncognitoDisableExtension(Profile* profile, const std::string& name); |
| 51 |
44 // Returns true iff the extension with the given id is pending | 52 // Returns true iff the extension with the given id is pending |
45 // install in |profile|. | 53 // install in |profile|. |
46 bool IsExtensionPendingInstallForSync( | 54 bool IsExtensionPendingInstallForSync( |
47 Profile* profile, const std::string& id) const; | 55 Profile* profile, const std::string& id) const; |
48 | 56 |
49 // Installs all extensions pending sync in |profile| of the given | 57 // Installs all extensions pending sync in |profile| of the given |
50 // type. | 58 // type. |
51 void InstallExtensionsPendingForSync(Profile* profile, Extension::Type type); | 59 void InstallExtensionsPendingForSync(Profile* profile, Extension::Type type); |
52 | 60 |
53 // Returns a map from |profile|'s installed extensions to their | 61 // Returns true iff |profile1| and |profile2| have the same extensions and |
54 // state. | 62 // they are all in the same state. |
55 ExtensionStateMap GetExtensionStates(Profile* profile) const; | 63 static bool ExtensionStatesMatch(Profile* profile1, Profile* profile2); |
56 | 64 |
57 private: | 65 private: |
| 66 struct ExtensionState { |
| 67 enum EnabledState { DISABLED, PENDING, ENABLED }; |
| 68 |
| 69 ExtensionState(); |
| 70 ~ExtensionState(); |
| 71 bool Equals(const ExtensionState &other) const; |
| 72 |
| 73 EnabledState enabled_state; |
| 74 bool incognito_enabled; |
| 75 }; |
| 76 |
| 77 typedef std::map<std::string, ExtensionState> ExtensionStateMap; |
58 typedef std::map<std::string, scoped_refptr<Extension> > ExtensionNameMap; | 78 typedef std::map<std::string, scoped_refptr<Extension> > ExtensionNameMap; |
59 typedef std::map<Profile*, ExtensionNameMap> ProfileExtensionNameMap; | 79 typedef std::map<Profile*, ExtensionNameMap> ProfileExtensionNameMap; |
60 typedef std::map<std::string, std::string> StringMap; | 80 typedef std::map<std::string, std::string> StringMap; |
61 | 81 |
| 82 // Returns a map from |profile|'s installed extensions to their state. |
| 83 static ExtensionStateMap GetExtensionStates(Profile* profile); |
| 84 |
62 // Initializes extensions for |profile| and creates an entry in | 85 // Initializes extensions for |profile| and creates an entry in |
63 // |profile_extensions_| for it. | 86 // |profile_extensions_| for it. |
64 void SetupProfile(Profile* profile); | 87 void SetupProfile(Profile* profile); |
65 | 88 |
66 // Returns an extension for the given name in |profile|. type and | 89 // Returns an extension for the given name in |profile|. type and |
67 // index. Two extensions with the name but different profiles will | 90 // index. Two extensions with the name but different profiles will |
68 // have the same id. | 91 // have the same id. |
69 scoped_refptr<Extension> GetExtension( | 92 scoped_refptr<Extension> GetExtension( |
70 Profile* profile, const std::string& name, | 93 Profile* profile, const std::string& name, |
71 Extension::Type type) WARN_UNUSED_RESULT; | 94 Extension::Type type) WARN_UNUSED_RESULT; |
72 | 95 |
73 ProfileExtensionNameMap profile_extensions_; | 96 ProfileExtensionNameMap profile_extensions_; |
74 StringMap id_to_name_; | 97 StringMap id_to_name_; |
75 | 98 |
76 DISALLOW_COPY_AND_ASSIGN(LiveSyncExtensionHelper); | 99 DISALLOW_COPY_AND_ASSIGN(LiveSyncExtensionHelper); |
77 }; | 100 }; |
78 | 101 |
79 #endif // CHROME_TEST_LIVE_SYNC_LIVE_SYNC_EXTENSION_HELPER_H_ | 102 #endif // CHROME_TEST_LIVE_SYNC_LIVE_SYNC_EXTENSION_HELPER_H_ |
OLD | NEW |