Add ref count to service workers for extension API.

extensions/browser/extension_util.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/browser/extension_util.h"
 
+#include "content/public/browser/browser_context.h"
+#include "content/public/browser/site_instance.h"
 #include "extensions/browser/extension_prefs.h"
 #include "extensions/browser/extension_registry.h"
 #include "extensions/common/manifest_handlers/app_isolation_info.h"
 #include "extensions/common/manifest_handlers/incognito_info.h"
 
 namespace extensions {
 namespace util {
 
+namespace {
+
+GURL GetSiteForExtensionId(const std::string& extension_id,

[Devlin, 2016/10/05 17:02:25]

nit: if this is only used here, inline it.

[lazyboy, 2016/10/06 01:02:06]

Done.

+                           content::BrowserContext* context) {
+  return content::SiteInstance::GetSiteForURL(
+      context, Extension::GetBaseURLFromExtensionId(extension_id));
+}
+
+}  // namespace
+
 bool HasIsolatedStorage(const ExtensionInfo& info) {
   if (!info.extension_manifest.get())
     return false;
 
   std::string error;
   scoped_refptr<const Extension> extension(Extension::Create(
       info.extension_path,
       info.extension_location,
       *info.extension_manifest,
       Extension::NO_FLAGS,
@@ skipping 11 matching lines @@
 
   return extension && AppIsolationInfo::HasIsolatedStorage(extension);
 }
 
 bool CanBeIncognitoEnabled(const Extension* extension) {
   return IncognitoInfo::IsIncognitoAllowed(extension) &&
          (!extension->is_platform_app() ||
           extension->location() == Manifest::COMPONENT);
 }
 
+content::StoragePartition* GetStoragePartitionForExtensionId(
+    const std::string& extension_id,
+    content::BrowserContext* browser_context) {
+  GURL site_url = GetSiteForExtensionId(extension_id, browser_context);
+  content::StoragePartition* storage_partition =
+      content::BrowserContext::GetStoragePartitionForSite(browser_context,
+                                                          site_url);
+  DCHECK(storage_partition);

[Devlin, 2016/10/05 17:02:25]

seems like this should be relaxed if the extension e.g. isn't enabled.

[lazyboy, 2016/10/06 01:02:06]

Done.

+  return storage_partition;
+}
+
 }  // namespace util
 }  // namespace extensions