Added PermissionIDSet to APIPermissions

extensions/common/permissions/media_galleries_permission.cc

 // Copyright 2014 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 "extensions/common/permissions/media_galleries_permission.h"
 
 #include <set>
 #include <string>
 
 #include "base/logging.h"
@@ skipping 92 matching lines @@
     // No other permissions, so reaching this means
     // MediaGalleriesPermissionData is probably out of sync in some way.
     // Fail so developers notice this.
     NOTREACHED();
     return false;
   }
 
   return IsValidPermissionSet(has_read, has_copy_to, has_delete, error);
 }
 
+PermissionIDSet MediaGalleriesPermission::GetPermissions() const {
+  DCHECK(HasMessages());
+  PermissionIDSet result;
+  AddMessagesAndPermissions(&result, NULL);
+  return result;
+}
+
 PermissionMessages MediaGalleriesPermission::GetMessages() const {
   DCHECK(HasMessages());
   PermissionMessages result;
+  AddMessagesAndPermissions(NULL, &result);
+  return result;
+}
 
+void MediaGalleriesPermission::AddMessagesAndPermissions(
+    PermissionIDSet* permission_ids,
+    PermissionMessages* permission_messages) const {
   bool has_all_auto_detected = false;
   bool has_read = false;
   bool has_copy_to = false;
   bool has_delete = false;
 
   for (std::set<MediaGalleriesPermissionData>::const_iterator it =
       data_set_.begin(); it != data_set_.end(); ++it) {
     if (it->permission() == kAllAutoDetectedPermission)
       has_all_auto_detected = true;
     else if (it->permission() == kReadPermission)
       has_read = true;
     else if (it->permission() == kCopyToPermission)
       has_copy_to = true;
     else if (it->permission() == kDeletePermission)
       has_delete = true;
   }
 
   if (!IsValidPermissionSet(has_read, has_copy_to, has_delete, NULL)) {
     NOTREACHED();
-    return result;
+    return;
   }
 
   // If |has_all_auto_detected| is false, then Chrome will prompt the user at
   // runtime when the extension call the getMediaGalleries API.
   if (!has_all_auto_detected)
-    return result;
+    return;
   // No access permission case.
   if (!has_read)
-    return result;
+    return;
 
   // Separate PermissionMessage IDs for read, copyTo, and delete. Otherwise an
   // extension can silently gain new access capabilities.
-  result.push_back(PermissionMessage(
-      PermissionMessage::kMediaGalleriesAllGalleriesRead,
-      l10n_util::GetStringUTF16(
-          IDS_EXTENSION_PROMPT_WARNING_MEDIA_GALLERIES_READ)));
+  if (permission_messages) {
+    permission_messages->push_back(PermissionMessage(
+        PermissionMessage::kMediaGalleriesAllGalleriesRead,
+        l10n_util::GetStringUTF16(
+            IDS_EXTENSION_PROMPT_WARNING_MEDIA_GALLERIES_READ)));
+  }
+  if (permission_ids) {
+    permission_ids->insert(APIPermission::kMediaGalleriesAllGalleriesRead);
+  }
 
   // For copyTo and delete, the proper combined permission message will be
   // derived in ChromePermissionMessageProvider::GetWarningMessages(), such
   // that the user get 1 entry for all media galleries access permissions,
   // rather than several separate entries.
   if (has_copy_to) {
-    result.push_back(PermissionMessage(
-        PermissionMessage::kMediaGalleriesAllGalleriesCopyTo,
-        base::string16()));
+    if (permission_messages) {
+      permission_messages->push_back(PermissionMessage(
+          PermissionMessage::kMediaGalleriesAllGalleriesCopyTo,
+          base::string16()));
+    }
+    if (permission_ids) {
+      permission_ids->insert(APIPermission::kMediaGalleriesAllGalleriesCopyTo);
+    }
   }
   if (has_delete) {
-    result.push_back(PermissionMessage(
-        PermissionMessage::kMediaGalleriesAllGalleriesDelete,
-        base::string16()));
+    if (permission_messages) {
+      permission_messages->push_back(PermissionMessage(
+          PermissionMessage::kMediaGalleriesAllGalleriesDelete,
+          base::string16()));
+    }
+    if (permission_ids) {
+      permission_ids->insert(APIPermission::kMediaGalleriesAllGalleriesDelete);
+    }
   }
-  return result;
+  return;
 }
 
 }  // namespace extensions