PS - Showing permission prompt for activeTab

chrome/browser/chromeos/extensions/public_session_permission_helper.cc

 // Copyright 2017 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/chromeos/extensions/public_session_permission_helper.h"
 
 #include <algorithm>
 #include <map>
 #include <memory>
 #include <utility>
@@ skipping 25 matching lines @@
 
 // This class is the internal implementation of HandlePermissionRequest(). It
 // contains the actual prompt showing and resolving logic, and it caches the
 // user choices.
 class PublicSessionPermissionHelper {
  public:
   PublicSessionPermissionHelper();
   PublicSessionPermissionHelper(PublicSessionPermissionHelper&& other);
   ~PublicSessionPermissionHelper();
 
-  void HandlePermissionRequestImpl(const Extension& extension,
+  bool HandlePermissionRequestImpl(const Extension& extension,
                                    const PermissionIDSet& requested_permissions,
                                    content::WebContents* web_contents,
                                    const RequestResolvedCallback& callback,
                                    const PromptFactory& prompt_factory);
 
+  bool PermissionAllowedImpl(APIPermission::ID permission);
+
  private:
   void ResolvePermissionPrompt(const ExtensionInstallPrompt* prompt,
                                const PermissionIDSet& unprompted_permissions,
                                ExtensionInstallPrompt::Result prompt_result);
 
   PermissionIDSet FilterAllowedPermissions(const PermissionIDSet& permissions);
 
   struct RequestCallback {
     RequestCallback(const RequestResolvedCallback& callback,
                     const PermissionIDSet& permission_list);
@@ skipping 13 matching lines @@
   DISALLOW_COPY_AND_ASSIGN(PublicSessionPermissionHelper);
 };
 
 PublicSessionPermissionHelper::PublicSessionPermissionHelper() {}
 
 PublicSessionPermissionHelper::PublicSessionPermissionHelper(
     PublicSessionPermissionHelper&& other) = default;
 
 PublicSessionPermissionHelper::~PublicSessionPermissionHelper() {}
 
-void PublicSessionPermissionHelper::HandlePermissionRequestImpl(
+bool PublicSessionPermissionHelper::HandlePermissionRequestImpl(
     const Extension& extension,
     const PermissionIDSet& requested_permissions,
     content::WebContents* web_contents,
     const RequestResolvedCallback& callback,
     const PromptFactory& prompt_factory) {
   CHECK(web_contents);
 
   PermissionIDSet unresolved_permissions = PermissionIDSet::Difference(
       requested_permissions, allowed_permission_set_);
   unresolved_permissions = PermissionIDSet::Difference(
       unresolved_permissions, denied_permission_set_);
   if (unresolved_permissions.empty()) {
     // All requested permissions are already resolved.
-    callback.Run(FilterAllowedPermissions(requested_permissions));
-    return;
+    if (!callback.is_null())
+      callback.Run(FilterAllowedPermissions(requested_permissions));
+    return true;
   }
 
   // Since not all permissions are resolved yet, queue the callback to be called
   // when all of them are resolved.
-  callbacks_.push_back(RequestCallback(callback, requested_permissions));
+  if (!callback.is_null())
+    callbacks_.push_back(RequestCallback(callback, requested_permissions));
 
   PermissionIDSet unprompted_permissions = PermissionIDSet::Difference(
       unresolved_permissions, prompted_permission_set_);
   if (unprompted_permissions.empty()) {
     // Some permissions aren't resolved yet, but they are currently being
     // prompted for, so no need to show a prompt.
-    return;
+    return false;
   }
 
   // Some permissions need prompting, setup the prompt and show it.
   APIPermissionSet new_apis;
   for (const auto& permission : unprompted_permissions) {
     prompted_permission_set_.insert(permission.id());
     new_apis.insert(permission.id());
   }
   auto permission_set = base::MakeUnique<PermissionSet>(
       new_apis, ManifestPermissionSet(), URLPatternSet(), URLPatternSet());
   auto prompt = prompt_factory.Run(web_contents);
   // This Unretained is safe because the lifetime of this object is until
   // process exit.
   prompt->ShowDialog(
       base::Bind(&PublicSessionPermissionHelper::ResolvePermissionPrompt,
                  base::Unretained(this), prompt.get(),
                  std::move(unprompted_permissions)),
       &extension,
       nullptr,  // Use the extension icon.
       base::MakeUnique<ExtensionInstallPrompt::Prompt>(
           ExtensionInstallPrompt::PERMISSIONS_PROMPT),
       std::move(permission_set),
       ExtensionInstallPrompt::GetDefaultShowDialogCallback());
   prompts_.insert(std::move(prompt));
+
+  return false;
+}
+
+bool PublicSessionPermissionHelper::PermissionAllowedImpl(
+    APIPermission::ID permission) {
+  return allowed_permission_set_.ContainsID(permission);
 }
 
 void PublicSessionPermissionHelper::ResolvePermissionPrompt(
     const ExtensionInstallPrompt* prompt,
     const PermissionIDSet& unprompted_permissions,
     ExtensionInstallPrompt::Result prompt_result) {
   PermissionIDSet& add_to_set =
       prompt_result == ExtensionInstallPrompt::Result::ACCEPTED ?
           allowed_permission_set_ : denied_permission_set_;
   for (const auto& permission : unprompted_permissions) {
@@ skipping 49 matching lines @@
 PublicSessionPermissionHelper::RequestCallback::RequestCallback(
     const RequestCallback& other) = default;
 
 PublicSessionPermissionHelper::RequestCallback::~RequestCallback() {}
 
 base::LazyInstance<std::map<ExtensionId, PublicSessionPermissionHelper>>::Leaky
     g_helpers = LAZY_INSTANCE_INITIALIZER;
 
 }  // namespace
 
-void HandlePermissionRequest(const Extension& extension,
+bool HandlePermissionRequest(const Extension& extension,
                              const PermissionIDSet& requested_permissions,
                              content::WebContents* web_contents,
                              const RequestResolvedCallback& callback,
                              const PromptFactory& prompt_factory) {
   DCHECK_CURRENTLY_ON(content::BrowserThread::UI);
   const PromptFactory& factory = prompt_factory.is_null()
                                      ? base::Bind(&CreateExtensionInstallPrompt)
                                      : prompt_factory;
   return g_helpers.Get()[extension.id()].HandlePermissionRequestImpl(
       extension, requested_permissions, web_contents, callback, factory);
 }
 
+bool PermissionAllowed(const Extension* extension,
+                       APIPermission::ID permission) {
+  DCHECK_CURRENTLY_ON(content::BrowserThread::UI);
+  return g_helpers.Get()[extension->id()].PermissionAllowedImpl(permission);
+}
+
 void ResetPermissionsForTesting() {
   // Clear out the std::map between tests. Just setting g_helpers to
   // LAZY_INSTANCE_INITIALIZER again causes a memory leak (because of the
   // LazyInstance::Leaky).
   g_helpers.Get().clear();
 }
 
 }  // namespace permission_helper
 }  // namespace extensions