1 files changed, 50 insertions, 2 deletions
diff --git a/chrome/browser/extensions/extension_context_menu_api.h b/chrome/browser/extensions/extension_context_menu_api.h
index 71a0353..c9c3380 100644
--- a/chrome/browser/extensions/extension_context_menu_api.h
+++ b/chrome/browser/extensions/extension_context_menu_api.h
@@ -6,17 +6,65 @@
 #define CHROME_BROWSER_EXTENSIONS_EXTENSION_CONTEXT_MENU_API_H__
 
 #include "chrome/browser/extensions/extension_function.h"
+#include "chrome/browser/extensions/extension_menu_manager.h"
 
-class CreateContextMenuFunction : public SyncExtensionFunction {
+class ExtensionContextMenuFunction : public SyncExtensionFunction {
+ public:
+  ~ExtensionContextMenuFunction() {}
+
+ protected:
+  // Helper function to read and parse a list of menu item contexts.
+  bool ParseContexts(const DictionaryValue& properties,
+                     const wchar_t* key,
+                     ExtensionMenuItem::ContextList* result);
+
+  // Looks in properties for the "type" key, and reads the value in |result|. On
+  // error, returns false and puts an error message into error_. If the key is
+  // not present, |result| is set to |default_value| and the return value is
+  // true.
+  bool ParseType(const DictionaryValue& properties,
+                 const ExtensionMenuItem::Type& default_value,
+                 ExtensionMenuItem::Type* result);
+
+  // Helper to read and parse the "checked" property.
+  bool ParseChecked(ExtensionMenuItem::Type type,
+                    const DictionaryValue& properties,
+                    bool default_value,
+                    bool* checked);
+
+  // If the parentId key was specified in properties, this will try looking up
+  // an ExtensionMenuItem with that id and set it into |result|. Returns false
+  // on error, with an explanation written into error_. Note that if the
+  // parentId key is not in properties, this will return true and leave |result|
+  // unset. Also, it is considered an error if the item found has a type other
+  // than NORMAL.
+  bool GetParent(const DictionaryValue& properties,
+                 const ExtensionMenuManager& manager,
+                 ExtensionMenuItem** result);
+};
+
+class CreateContextMenuFunction : public ExtensionContextMenuFunction {
   ~CreateContextMenuFunction() {}
   virtual bool RunImpl();
   DECLARE_EXTENSION_FUNCTION_NAME("experimental.contextMenu.create")
 };
 
-class RemoveContextMenuFunction : public SyncExtensionFunction {
+class UpdateContextMenuFunction : public ExtensionContextMenuFunction {
+  ~UpdateContextMenuFunction() {}
+  virtual bool RunImpl();
+  DECLARE_EXTENSION_FUNCTION_NAME("experimental.contextMenu.update")
+};
+
+class RemoveContextMenuFunction : public ExtensionContextMenuFunction {
   ~RemoveContextMenuFunction() {}
   virtual bool RunImpl();
   DECLARE_EXTENSION_FUNCTION_NAME("experimental.contextMenu.remove")
 };
 
+class RemoveAllContextMenusFunction : public ExtensionContextMenuFunction {
+  ~RemoveAllContextMenusFunction() {}
+  virtual bool RunImpl();
+  DECLARE_EXTENSION_FUNCTION_NAME("experimental.contextMenu.removeAll")
+};
+
 #endif  // CHROME_BROWSER_EXTENSIONS_EXTENSION_CONTEXT_MENU_API_H__