1 files changed, 31 insertions, 0 deletions
diff --git a/src/background/repositories/CompletionsRepository.js b/src/background/repositories/CompletionsRepository.js
new file mode 100644
index 0000000..1318d36
--- /dev/null
+++ b/src/background/repositories/CompletionsRepository.js
@@ -0,0 +1,31 @@
+export default class CompletionsRepository {
+  async queryBookmarks(keywords) {
+    let items = await browser.bookmarks.search({ query: keywords });
+    return items.filter((item) => {
+      let url = undefined;
+      try {
+        url = new URL(item.url);
+      } catch (e) {
+        return false;
+      }
+      return item.type === 'bookmark' && url.protocol !== 'place:';
+    });
+  }
+
+  queryHistories(keywords) {
+    return browser.history.search({
+      text: keywords,
+      startTime: 0,
+    });
+  }
+
+  async queryTabs(keywords, excludePinned) {
+    let tabs = await browser.tabs.query({ currentWindow: true });
+    return tabs.filter((t) => {
+      return t.url.toLowerCase().includes(keywords.toLowerCase()) ||
+        t.title && t.title.toLowerCase().includes(keywords.toLowerCase());
+    }).filter((t) => {
+      return !(excludePinned && t.pinned);
+    });
+  }
+}