Move WebTestContentIndexProvider to ShellContentIndexProvider
The ContentIndexProvider is used by web tests as well as
content_browsertests.
[email protected]
Bug: 866140
Change-Id: If07c7869c4484a55be8e958ebb1c43321a65d76a
Reviewed-on: https://chromium-review.googlesource.com/c/chromium/src/+/2269513
Reviewed-by: Avi Drissman <[email protected]>
Commit-Queue: danakj <[email protected]>
Cr-Commit-Position: refs/heads/master@{#783245}
diff --git a/content/shell/browser/shell_content_index_provider.h b/content/shell/browser/shell_content_index_provider.h
new file mode 100644
index 0000000..27ebf31
--- /dev/null
+++ b/content/shell/browser/shell_content_index_provider.h
@@ -0,0 +1,56 @@
+// Copyright 2019 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 CONTENT_SHELL_BROWSER_SHELL_CONTENT_INDEX_PROVIDER_H_
+#define CONTENT_SHELL_BROWSER_SHELL_CONTENT_INDEX_PROVIDER_H_
+
+#include <map>
+#include <string>
+#include <vector>
+
+#include "base/macros.h"
+#include "base/memory/weak_ptr.h"
+#include "content/public/browser/content_index_provider.h"
+#include "ui/gfx/geometry/size.h"
+#include "url/origin.h"
+
+namespace content {
+
+// When using ShellContentIndexProvider, IDs need to be globally unique,
+// instead of per Service Worker.
+class ShellContentIndexProvider : public ContentIndexProvider {
+ public:
+ ShellContentIndexProvider();
+ ~ShellContentIndexProvider() override;
+
+ // ContentIndexProvider implementation.
+ std::vector<gfx::Size> GetIconSizes(
+ blink::mojom::ContentCategory category) override;
+ void OnContentAdded(ContentIndexEntry entry) override;
+ void OnContentDeleted(int64_t service_worker_registration_id,
+ const url::Origin& origin,
+ const std::string& description_id) override;
+
+ // Returns the Service Worker Registration ID and the origin of the Content
+ // Index entry registered with |id|. If |id| does not exist, invalid values
+ // are returned, which would cause DB tasks to fail.
+ std::pair<int64_t, url::Origin> GetRegistrationDataFromId(
+ const std::string& id);
+
+ void set_icon_sizes(std::vector<gfx::Size> icon_sizes) {
+ icon_sizes_ = std::move(icon_sizes);
+ }
+
+ private:
+ // Map from |description_id| to <|service_worker_registration_id|, |origin|>.
+ std::map<std::string, std::pair<int64_t, url::Origin>> entries_;
+
+ std::vector<gfx::Size> icon_sizes_;
+
+ DISALLOW_COPY_AND_ASSIGN(ShellContentIndexProvider);
+};
+
+} // namespace content
+
+#endif // CONTENT_SHELL_BROWSER_SHELL_CONTENT_INDEX_PROVIDER_H_
