Public Sessions - prompt the user for audioCapture/videoCapture requests

chrome/browser/media/extension_media_access_handler.h

 // Copyright 2015 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.
 
 #ifndef CHROME_BROWSER_MEDIA_EXTENSION_MEDIA_ACCESS_HANDLER_H_
 #define CHROME_BROWSER_MEDIA_EXTENSION_MEDIA_ACCESS_HANDLER_H_
 
+#include "chrome/browser/extensions/extension_install_prompt.h"
 #include "chrome/browser/media/media_access_handler.h"
+#include "extensions/common/extension_id.h"
 
 class MediaStreamCaptureIndicator;
 
 // MediaAccessHandler for extension capturing requests.
 class ExtensionMediaAccessHandler : public MediaAccessHandler {
  public:
   ExtensionMediaAccessHandler();
   ~ExtensionMediaAccessHandler() override;
 
   // MediaAccessHandler implementation.
   bool SupportsStreamType(const content::MediaStreamType type,
                           const extensions::Extension* extension) override;
   bool CheckMediaAccessPermission(
       content::WebContents* web_contents,
       const GURL& security_origin,
       content::MediaStreamType type,
       const extensions::Extension* extension) override;
   void HandleRequest(content::WebContents* web_contents,
                      const content::MediaStreamRequest& request,
                      const content::MediaResponseCallback& callback,
                      const extensions::Extension* extension) override;
+
+ private:
+  // Original HandleRequest function was split in two parts to allow prompting
+  // the user for input, this is the second part.
+  void HandleRequestContinuation(content::WebContents* web_contents,
+                                 const content::MediaStreamRequest& request,
+                                 const content::MediaResponseCallback& callback,
+                                 const extensions::Extension* extension);
+
+  // Function used to resolve user decision regarding allowing audio/video.
+  void ResolvePermissionPrompt(
+      content::WebContents* web_contents,
+      const content::MediaStreamRequest& request,
+      const content::MediaResponseCallback& callback,
+      const extensions::Extension* extension,
+      ExtensionInstallPrompt::Result prompt_result);
+
+  std::unique_ptr<ExtensionInstallPrompt> prompt_;
+
+  // Class used to cache user choice regarding allowing audio/video capture.
+  class UserChoice {
+   public:
+    // Helper function for checking if audio/video is disallowed by user choice.
+    bool Disallowed(content::MediaStreamType type) const;
+    // Helper function which returns true if audio/video wasn't prompted yet.
+    bool NeedsPrompting(content::MediaStreamType type) const;
+    void Set(content::MediaStreamType type, bool allowed);
+
+   private:
+    bool audio_prompted_ = false;
+    bool audio_allowed_ = false;
+    bool video_prompted_ = false;
+    bool video_allowed_ = false;
+  };
+
+  std::map<extensions::ExtensionId, UserChoice> user_choice_cache_;
 };
 
 #endif  // CHROME_BROWSER_MEDIA_EXTENSION_MEDIA_ACCESS_HANDLER_H_