Index: chrome/browser/extensions/api/context_menus/context_menus_api.cc |
=================================================================== |
--- chrome/browser/extensions/api/context_menus/context_menus_api.cc (revision 176053) |
+++ chrome/browser/extensions/api/context_menus/context_menus_api.cc (working copy) |
@@ -1,398 +0,0 @@ |
-// 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. |
- |
-#include "chrome/browser/extensions/api/context_menus/context_menus_api.h" |
- |
-#include <string> |
- |
-#include "base/string_number_conversions.h" |
-#include "base/string_util.h" |
-#include "base/values.h" |
-#include "chrome/browser/extensions/extension_service.h" |
-#include "chrome/browser/extensions/menu_manager.h" |
-#include "chrome/browser/profiles/profile.h" |
-#include "chrome/common/extensions/api/context_menus.h" |
-#include "extensions/common/error_utils.h" |
-#include "extensions/common/url_pattern_set.h" |
- |
-using extensions::ErrorUtils; |
- |
-namespace { |
- |
-const char kGeneratedIdKey[] = "generatedId"; |
- |
-const char kCannotFindItemError[] = "Cannot find menu item with id *"; |
-const char kOnclickDisallowedError[] = "Extensions using event pages cannot " |
- "pass an onclick parameter to chrome.contextMenus.create. Instead, use " |
- "the chrome.contextMenus.onClicked event."; |
-const char kCheckedError[] = |
- "Only items with type \"radio\" or \"checkbox\" can be checked"; |
-const char kDuplicateIDError[] = |
- "Cannot create item with duplicate id *"; |
-const char kIdRequiredError[] = "Extensions using event pages must pass an " |
- "id parameter to chrome.contextMenus.create"; |
-const char kParentsMustBeNormalError[] = |
- "Parent items must have type \"normal\""; |
-const char kTitleNeededError[] = |
- "All menu items except for separators must have a title"; |
-const char kLauncherNotAllowedError[] = |
- "Only packaged apps are allowed to use 'launcher' context"; |
- |
-std::string GetIDString(const extensions::MenuItem::Id& id) { |
- if (id.uid == 0) |
- return id.string_uid; |
- else |
- return base::IntToString(id.uid); |
-} |
- |
-template<typename PropertyWithEnumT> |
-extensions::MenuItem::ContextList GetContexts( |
- const PropertyWithEnumT& property) { |
- extensions::MenuItem::ContextList contexts; |
- for (size_t i = 0; i < property.contexts->size(); ++i) { |
- switch (property.contexts->at(i)) { |
- case PropertyWithEnumT::CONTEXTS_ELEMENT_ALL: |
- contexts.Add(extensions::MenuItem::ALL); |
- break; |
- case PropertyWithEnumT::CONTEXTS_ELEMENT_PAGE: |
- contexts.Add(extensions::MenuItem::PAGE); |
- break; |
- case PropertyWithEnumT::CONTEXTS_ELEMENT_SELECTION: |
- contexts.Add(extensions::MenuItem::SELECTION); |
- break; |
- case PropertyWithEnumT::CONTEXTS_ELEMENT_LINK: |
- contexts.Add(extensions::MenuItem::LINK); |
- break; |
- case PropertyWithEnumT::CONTEXTS_ELEMENT_EDITABLE: |
- contexts.Add(extensions::MenuItem::EDITABLE); |
- break; |
- case PropertyWithEnumT::CONTEXTS_ELEMENT_IMAGE: |
- contexts.Add(extensions::MenuItem::IMAGE); |
- break; |
- case PropertyWithEnumT::CONTEXTS_ELEMENT_VIDEO: |
- contexts.Add(extensions::MenuItem::VIDEO); |
- break; |
- case PropertyWithEnumT::CONTEXTS_ELEMENT_AUDIO: |
- contexts.Add(extensions::MenuItem::AUDIO); |
- break; |
- case PropertyWithEnumT::CONTEXTS_ELEMENT_FRAME: |
- contexts.Add(extensions::MenuItem::FRAME); |
- break; |
- case PropertyWithEnumT::CONTEXTS_ELEMENT_LAUNCHER: |
- contexts.Add(extensions::MenuItem::LAUNCHER); |
- break; |
- case PropertyWithEnumT::CONTEXTS_ELEMENT_NONE: |
- NOTREACHED(); |
- } |
- } |
- return contexts; |
-} |
- |
-template<typename PropertyWithEnumT> |
-extensions::MenuItem::Type GetType(const PropertyWithEnumT& property) { |
- switch (property.type) { |
- case PropertyWithEnumT::TYPE_NONE: |
- case PropertyWithEnumT::TYPE_NORMAL: |
- return extensions::MenuItem::NORMAL; |
- case PropertyWithEnumT::TYPE_CHECKBOX: |
- return extensions::MenuItem::CHECKBOX; |
- case PropertyWithEnumT::TYPE_RADIO: |
- return extensions::MenuItem::RADIO; |
- case PropertyWithEnumT::TYPE_SEPARATOR: |
- return extensions::MenuItem::SEPARATOR; |
- } |
- return extensions::MenuItem::NORMAL; |
-} |
- |
-template<typename PropertyWithEnumT> |
-scoped_ptr<extensions::MenuItem::Id> GetParentId( |
- const PropertyWithEnumT& property, |
- bool is_off_the_record, |
- std::string extension_id) { |
- scoped_ptr<extensions::MenuItem::Id> parent_id( |
- new extensions::MenuItem::Id(is_off_the_record, extension_id)); |
- switch (property.parent_id_type) { |
- case PropertyWithEnumT::PARENT_ID_NONE: |
- return scoped_ptr<extensions::MenuItem::Id>().Pass(); |
- case PropertyWithEnumT::PARENT_ID_INTEGER: |
- parent_id->uid = *property.parent_id_integer; |
- break; |
- case PropertyWithEnumT::PARENT_ID_STRING: |
- parent_id->string_uid = *property.parent_id_string; |
- break; |
- } |
- return parent_id.Pass(); |
-} |
- |
-extensions::MenuItem* GetParent(extensions::MenuItem::Id parent_id, |
- const extensions::MenuManager* menu_manager, |
- std::string* error) { |
- extensions::MenuItem* parent = menu_manager->GetItemById(parent_id); |
- if (!parent) { |
- *error = ErrorUtils::FormatErrorMessage( |
- kCannotFindItemError, GetIDString(parent_id)); |
- return NULL; |
- } |
- if (parent->type() != extensions::MenuItem::NORMAL) { |
- *error = kParentsMustBeNormalError; |
- return NULL; |
- } |
- |
- return parent; |
-} |
- |
-} // namespace |
- |
-namespace extensions { |
- |
-namespace Create = api::context_menus::Create; |
-namespace Remove = api::context_menus::Remove; |
-namespace Update = api::context_menus::Update; |
- |
-bool ContextMenusCreateFunction::RunImpl() { |
- MenuItem::Id id(profile()->IsOffTheRecord(), extension_id()); |
- scoped_ptr<Create::Params> params(Create::Params::Create(*args_)); |
- EXTENSION_FUNCTION_VALIDATE(params.get()); |
- |
- if (params->create_properties.id.get()) { |
- id.string_uid = *params->create_properties.id; |
- } else { |
- if (GetExtension()->has_lazy_background_page()) { |
- error_ = kIdRequiredError; |
- return false; |
- } |
- |
- // The Generated Id is added by context_menus_custom_bindings.js. |
- DictionaryValue* properties = NULL; |
- EXTENSION_FUNCTION_VALIDATE(args_->GetDictionary(0, &properties)); |
- EXTENSION_FUNCTION_VALIDATE(properties->GetInteger(kGeneratedIdKey, |
- &id.uid)); |
- } |
- |
- std::string title; |
- if (params->create_properties.title.get()) |
- title = *params->create_properties.title; |
- |
- MenuManager* menu_manager = profile()->GetExtensionService()->menu_manager(); |
- |
- if (menu_manager->GetItemById(id)) { |
- error_ = ErrorUtils::FormatErrorMessage(kDuplicateIDError, |
- GetIDString(id)); |
- return false; |
- } |
- |
- if (GetExtension()->has_lazy_background_page() && |
- params->create_properties.onclick.get()) { |
- error_ = kOnclickDisallowedError; |
- return false; |
- } |
- |
- MenuItem::ContextList contexts; |
- if (params->create_properties.contexts.get()) |
- contexts = GetContexts(params->create_properties); |
- else |
- contexts.Add(MenuItem::PAGE); |
- |
- if (contexts.Contains(MenuItem::LAUNCHER) && |
- !GetExtension()->is_platform_app()) { |
- error_ = kLauncherNotAllowedError; |
- return false; |
- } |
- |
- MenuItem::Type type = GetType(params->create_properties); |
- |
- if (title.empty() && type != MenuItem::SEPARATOR) { |
- error_ = kTitleNeededError; |
- return false; |
- } |
- |
- bool checked = false; |
- if (params->create_properties.checked.get()) |
- checked = *params->create_properties.checked; |
- |
- bool enabled = true; |
- if (params->create_properties.enabled.get()) |
- enabled = *params->create_properties.enabled; |
- |
- scoped_ptr<MenuItem> item( |
- new MenuItem(id, title, checked, enabled, type, contexts)); |
- |
- if (!item->PopulateURLPatterns( |
- params->create_properties.document_url_patterns.get(), |
- params->create_properties.target_url_patterns.get(), |
- &error_)) { |
- return false; |
- } |
- |
- bool success = true; |
- scoped_ptr<MenuItem::Id> parent_id(GetParentId(params->create_properties, |
- profile()->IsOffTheRecord(), |
- extension_id())); |
- if (parent_id.get()) { |
- MenuItem* parent = GetParent(*parent_id, menu_manager, &error_); |
- if (!parent) |
- return false; |
- success = menu_manager->AddChildItem(parent->id(), item.release()); |
- } else { |
- success = menu_manager->AddContextItem(GetExtension(), item.release()); |
- } |
- |
- if (!success) |
- return false; |
- |
- menu_manager->WriteToStorage(GetExtension()); |
- return true; |
-} |
- |
-bool ContextMenusUpdateFunction::RunImpl() { |
- bool radio_item_updated = false; |
- MenuItem::Id item_id(profile()->IsOffTheRecord(), extension_id()); |
- scoped_ptr<Update::Params> params(Update::Params::Create(*args_)); |
- |
- EXTENSION_FUNCTION_VALIDATE(params.get()); |
- switch (params->id_type) { |
- case Update::Params::ID_STRING: |
- item_id.string_uid = *params->id_string; |
- break; |
- case Update::Params::ID_INTEGER: |
- item_id.uid = *params->id_integer; |
- break; |
- case Update::Params::ID_NONE: |
- NOTREACHED(); |
- } |
- |
- ExtensionService* service = profile()->GetExtensionService(); |
- MenuManager* manager = service->menu_manager(); |
- MenuItem* item = manager->GetItemById(item_id); |
- if (!item || item->extension_id() != extension_id()) { |
- error_ = ErrorUtils::FormatErrorMessage( |
- kCannotFindItemError, GetIDString(item_id)); |
- return false; |
- } |
- |
- // Type. |
- MenuItem::Type type = GetType(params->update_properties); |
- |
- if (type != item->type()) { |
- if (type == MenuItem::RADIO || item->type() == MenuItem::RADIO) |
- radio_item_updated = true; |
- item->set_type(type); |
- } |
- |
- // Title. |
- if (params->update_properties.title.get()) { |
- std::string title(*params->update_properties.title); |
- if (title.empty() && item->type() != MenuItem::SEPARATOR) { |
- error_ = kTitleNeededError; |
- return false; |
- } |
- item->set_title(title); |
- } |
- |
- // Checked state. |
- if (params->update_properties.checked.get()) { |
- bool checked = *params->update_properties.checked; |
- if (checked && |
- item->type() != MenuItem::CHECKBOX && |
- item->type() != MenuItem::RADIO) { |
- error_ = kCheckedError; |
- return false; |
- } |
- if (checked != item->checked()) { |
- if (!item->SetChecked(checked)) { |
- error_ = kCheckedError; |
- return false; |
- } |
- radio_item_updated = true; |
- } |
- } |
- |
- // Enabled. |
- if (params->update_properties.enabled.get()) |
- item->set_enabled(*params->update_properties.enabled); |
- |
- // Contexts. |
- MenuItem::ContextList contexts; |
- if (params->update_properties.contexts.get()) { |
- contexts = GetContexts(params->update_properties); |
- |
- if (contexts.Contains(MenuItem::LAUNCHER) && |
- !GetExtension()->is_platform_app()) { |
- error_ = kLauncherNotAllowedError; |
- return false; |
- } |
- |
- if (contexts != item->contexts()) |
- item->set_contexts(contexts); |
- } |
- |
- // Parent id. |
- MenuItem* parent = NULL; |
- scoped_ptr<MenuItem::Id> parent_id(GetParentId(params->update_properties, |
- profile()->IsOffTheRecord(), |
- extension_id())); |
- if (parent_id.get()) { |
- MenuItem* parent = GetParent(*parent_id, manager, &error_); |
- if (!parent || !manager->ChangeParent(item->id(), &parent->id())) |
- return false; |
- } |
- |
- // URL Patterns. |
- if (!item->PopulateURLPatterns( |
- params->update_properties.document_url_patterns.get(), |
- params->update_properties.target_url_patterns.get(), &error_)) { |
- return false; |
- } |
- |
- // There is no need to call ItemUpdated if ChangeParent is called because |
- // all sanitation is taken care of in ChangeParent. |
- if (!parent && radio_item_updated && !manager->ItemUpdated(item->id())) |
- return false; |
- |
- manager->WriteToStorage(GetExtension()); |
- return true; |
-} |
- |
-bool ContextMenusRemoveFunction::RunImpl() { |
- scoped_ptr<Remove::Params> params(Remove::Params::Create(*args_)); |
- EXTENSION_FUNCTION_VALIDATE(params.get()); |
- |
- ExtensionService* service = profile()->GetExtensionService(); |
- MenuManager* manager = service->menu_manager(); |
- |
- MenuItem::Id id(profile()->IsOffTheRecord(), extension_id()); |
- switch (params->menu_item_id_type) { |
- case Remove::Params::MENU_ITEM_ID_STRING: |
- id.string_uid = *params->menu_item_id_string; |
- break; |
- case Remove::Params::MENU_ITEM_ID_INTEGER: |
- id.uid = *params->menu_item_id_integer; |
- break; |
- case Remove::Params::MENU_ITEM_ID_NONE: |
- NOTREACHED(); |
- } |
- |
- MenuItem* item = manager->GetItemById(id); |
- // Ensure one extension can't remove another's menu items. |
- if (!item || item->extension_id() != extension_id()) { |
- error_ = ErrorUtils::FormatErrorMessage( |
- kCannotFindItemError, GetIDString(id)); |
- return false; |
- } |
- |
- if (!manager->RemoveContextMenuItem(id)) |
- return false; |
- manager->WriteToStorage(GetExtension()); |
- return true; |
-} |
- |
-bool ContextMenusRemoveAllFunction::RunImpl() { |
- ExtensionService* service = profile()->GetExtensionService(); |
- MenuManager* manager = service->menu_manager(); |
- manager->RemoveAllContextItems(GetExtension()->id()); |
- manager->WriteToStorage(GetExtension()); |
- return true; |
-} |
- |
-} // namespace extensions |