1 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
2 // Use of this source code is governed by a BSD-style license that can be
3 // found in the LICENSE file.
4
5 #include "chrome/browser/extensions/api/context_menus/context_menus_api.h"
6
7 #include <string>
8
9 #include "base/strings/string_util.h"
10 #include "base/values.h"
11 #include "chrome/browser/extensions/api/context_menus/context_menus_api_helpers.h"
12 #include "chrome/browser/extensions/menu_manager.h"
13 #include "chrome/browser/profiles/profile.h"
14 #include "chrome/common/extensions/api/context_menus.h"
15 #include "extensions/common/error_utils.h"
16 #include "extensions/common/manifest_handlers/background_info.h"
17 #include "extensions/common/url_pattern_set.h"
18
19 using extensions::ErrorUtils;
20 namespace helpers = extensions::context_menus_api_helpers;
21
22 namespace {
23
24 const char kIdRequiredError[] = "Extensions using event pages must pass an "
25 "id parameter to chrome.contextMenus.create";
26
27 } // namespace
28
29 namespace extensions {
30
31 namespace Create = api::context_menus::Create;
32 namespace Remove = api::context_menus::Remove;
33 namespace Update = api::context_menus::Update;
34
RunSync()35 bool ContextMenusCreateFunction::RunSync() {
36 MenuItem::Id id(GetProfile()->IsOffTheRecord(),
37 MenuItem::ExtensionKey(extension_id()));
38 scoped_ptr<Create::Params> params(Create::Params::Create(*args_));
39 EXTENSION_FUNCTION_VALIDATE(params.get());
40
41 if (params->create_properties.id.get()) {
42 id.string_uid = *params->create_properties.id;
43 } else {
44 if (BackgroundInfo::HasLazyBackgroundPage(GetExtension())) {
45 error_ = kIdRequiredError;
46 return false;
47 }
48
49 // The Generated Id is added by context_menus_custom_bindings.js.
50 base::DictionaryValue* properties = NULL;
51 EXTENSION_FUNCTION_VALIDATE(args_->GetDictionary(0, &properties));
52 EXTENSION_FUNCTION_VALIDATE(
53 properties->GetInteger(helpers::kGeneratedIdKey, &id.uid));
54 }
55
56 return helpers::CreateMenuItem(params->create_properties, GetProfile(),
57 GetExtension(), id, &error_);
58 }
59
RunSync()60 bool ContextMenusUpdateFunction::RunSync() {
61 MenuItem::Id item_id(GetProfile()->IsOffTheRecord(),
62 MenuItem::ExtensionKey(extension_id()));
63 scoped_ptr<Update::Params> params(Update::Params::Create(*args_));
64
65 EXTENSION_FUNCTION_VALIDATE(params.get());
66 if (params->id.as_string)
67 item_id.string_uid = *params->id.as_string;
68 else if (params->id.as_integer)
69 item_id.uid = *params->id.as_integer;
70 else
71 NOTREACHED();
72
73 return helpers::UpdateMenuItem(params->update_properties, GetProfile(),
74 GetExtension(), item_id, &error_);
75 }
76
RunSync()77 bool ContextMenusRemoveFunction::RunSync() {
78 scoped_ptr<Remove::Params> params(Remove::Params::Create(*args_));
79 EXTENSION_FUNCTION_VALIDATE(params.get());
80
81 MenuManager* manager = MenuManager::Get(GetProfile());
82
83 MenuItem::Id id(GetProfile()->IsOffTheRecord(),
84 MenuItem::ExtensionKey(extension_id()));
85 if (params->menu_item_id.as_string)
86 id.string_uid = *params->menu_item_id.as_string;
87 else if (params->menu_item_id.as_integer)
88 id.uid = *params->menu_item_id.as_integer;
89 else
90 NOTREACHED();
91
92 MenuItem* item = manager->GetItemById(id);
93 // Ensure one extension can't remove another's menu items.
94 if (!item || item->extension_id() != extension_id()) {
95 error_ = ErrorUtils::FormatErrorMessage(
96 helpers::kCannotFindItemError, helpers::GetIDString(id));
97 return false;
98 }
99
100 if (!manager->RemoveContextMenuItem(id))
101 return false;
102 manager->WriteToStorage(GetExtension(), id.extension_key);
103 return true;
104 }
105
RunSync()106 bool ContextMenusRemoveAllFunction::RunSync() {
107 MenuManager* manager = MenuManager::Get(GetProfile());
108 manager->RemoveAllContextItems(MenuItem::ExtensionKey(GetExtension()->id()));
109 manager->WriteToStorage(GetExtension(),
110 MenuItem::ExtensionKey(GetExtension()->id()));
111 return true;
112 }
113
114 } // namespace extensions
115