From 01242a2f0d174b4bf8b51fd5627edced465757e9 Mon Sep 17 00:00:00 2001
From: Shin'ya Ueoka <ueokande@i-beam.org>
Date: Thu, 23 Sep 2021 12:44:49 +0900
Subject: Search a content from frames successfully loaded

---
 .../repositories/ReadyFrameRepository.ts           | 49 ++++++++++++++++++++++
 1 file changed, 49 insertions(+)
 create mode 100644 src/background/repositories/ReadyFrameRepository.ts

(limited to 'src/background/repositories')

diff --git a/src/background/repositories/ReadyFrameRepository.ts b/src/background/repositories/ReadyFrameRepository.ts
new file mode 100644
index 0000000..725f604
--- /dev/null
+++ b/src/background/repositories/ReadyFrameRepository.ts
@@ -0,0 +1,49 @@
+import MemoryStorage from "../infrastructures/MemoryStorage";
+
+const REPOSITORY_KEY = "readyFrameRepository";
+
+type State = { [tabId: number]: number[] };
+
+export default interface ReadyFrameRepository {
+  clearFrameIds(tabId: number): Promise<void>;
+
+  addFrameId(tabId: number, fraemId: number): Promise<void>;
+
+  getFrameIds(tabId: number): Promise<number[] | undefined>;
+}
+
+export class ReadyFrameRepositoryImpl implements ReadyFrameRepository {
+  private cache: MemoryStorage;
+
+  constructor() {
+    this.cache = new MemoryStorage();
+  }
+
+  clearFrameIds(tabId: number): Promise<void> {
+    let state: State | undefined = this.cache.get(REPOSITORY_KEY);
+    if (typeof state === "undefined") {
+      state = {};
+    }
+    delete state[tabId];
+    this.cache.set(REPOSITORY_KEY, state);
+    return Promise.resolve();
+  }
+
+  addFrameId(tabId: number, fraemId: number): Promise<void> {
+    let state: State | undefined = this.cache.get(REPOSITORY_KEY);
+    if (typeof state === "undefined") {
+      state = {};
+    }
+    state[tabId] = (state[tabId] || []).concat(fraemId);
+    this.cache.set(REPOSITORY_KEY, state);
+    return Promise.resolve();
+  }
+
+  getFrameIds(tabId: number): Promise<number[] | undefined> {
+    const state: State | undefined = this.cache.get(REPOSITORY_KEY);
+    if (typeof state === "undefined") {
+      return Promise.resolve(undefined);
+    }
+    return Promise.resolve(state[tabId]);
+  }
+}
-- 
cgit v1.2.3