2 files changed, 122 insertions, 6 deletions
diff --git a/src/background/repositories/FindRepository.ts b/src/background/repositories/FindRepository.ts
index 813e065..3492759 100644
--- a/src/background/repositories/FindRepository.ts
+++ b/src/background/repositories/FindRepository.ts
@@ -1,22 +1,68 @@
 import { injectable } from "tsyringe";
 import MemoryStorage from "../infrastructures/MemoryStorage";
 
-const FIND_KEYWORD_KEY = "find-keyword";
+const FIND_GLOBAL_KEYWORD_KEY = "find-global-keyword";
+const FIND_LOCAL_KEYWORD_KEY = "find-local-keyword";
+
+export type FindState = {
+  keyword: string;
+  frameId: number;
+};
+
+export default interface FindRepository {
+  getGlobalKeyword(): Promise<string | undefined>;
+
+  setGlobalKeyword(keyword: string): Promise<void>;
+
+  getLocalState(tabId: number): Promise<undefined | FindState>;
+
+  setLocalState(tabId: number, state: FindState): Promise<void>;
+
+  deleteLocalState(tabId: number): Promise<void>;
+}
 
 @injectable()
-export default class FindRepository {
+export class FindRepositoryImpl implements FindRepository {
   private cache: MemoryStorage;
 
   constructor() {
     this.cache = new MemoryStorage();
   }
 
-  getKeyword(): Promise<string> {
-    return Promise.resolve(this.cache.get(FIND_KEYWORD_KEY));
+  getGlobalKeyword(): Promise<string | undefined> {
+    return Promise.resolve(this.cache.get(FIND_GLOBAL_KEYWORD_KEY));
+  }
+
+  setGlobalKeyword(keyword: string): Promise<void> {
+    this.cache.set(FIND_GLOBAL_KEYWORD_KEY, keyword);
+    return Promise.resolve();
+  }
+
+  getLocalState(tabId: number): Promise<FindState | undefined> {
+    let states = this.cache.get(FIND_LOCAL_KEYWORD_KEY);
+    if (typeof states === "undefined") {
+      states = {};
+    }
+    return Promise.resolve(states[tabId]);
+  }
+
+  setLocalState(tabId: number, state: FindState): Promise<void> {
+    let states = this.cache.get(FIND_LOCAL_KEYWORD_KEY);
+    if (typeof states === "undefined") {
+      states = {};
+    }
+    states[tabId] = state;
+    this.cache.set(FIND_LOCAL_KEYWORD_KEY, states);
+    return Promise.resolve();
   }
 
-  setKeyword(keyword: string): Promise<void> {
-    this.cache.set(FIND_KEYWORD_KEY, keyword);
+  deleteLocalState(tabId: number): Promise<void> {
+    const states = this.cache.get(FIND_LOCAL_KEYWORD_KEY);
+    if (typeof states === "undefined") {
+      return Promise.resolve();
+    }
+    delete states[tabId];
+    this.cache.set(FIND_LOCAL_KEYWORD_KEY, states);
     return Promise.resolve();
   }
 }
diff --git a/src/background/repositories/ReadyFrameRepository.ts b/src/background/repositories/ReadyFrameRepository.ts
new file mode 100644
index 0000000..c993858
--- /dev/null
+++ b/src/background/repositories/ReadyFrameRepository.ts
@@ -0,0 +1,70 @@
+import MemoryStorage from "../infrastructures/MemoryStorage";
+
+const REPOSITORY_KEY = "readyFrameRepository";
+
+type State = { [tabId: number]: { [frameId: number]: number } };
+
+export default interface ReadyFrameRepository {
+  addFrameId(tabId: number, frameId: number): Promise<void>;
+
+  removeFrameId(tabId: number, frameId: number): Promise<void>;
+
+  getFrameIds(tabId: number): Promise<number[] | undefined>;
+}
+
+export class ReadyFrameRepositoryImpl implements ReadyFrameRepository {
+  private cache: MemoryStorage;
+
+  constructor() {
+    this.cache = new MemoryStorage();
+  }
+
+  addFrameId(tabId: number, frameId: number): Promise<void> {
+    let state: State | undefined = this.cache.get(REPOSITORY_KEY);
+    if (typeof state === "undefined") {
+      state = {};
+    }
+    const tab = state[tabId] || {};
+    tab[frameId] = (tab[frameId] || 0) + 1;
+    state[tabId] = tab;
+    this.cache.set(REPOSITORY_KEY, state);
+    return Promise.resolve();
+  }
+
+  removeFrameId(tabId: number, frameId: number): Promise<void> {
+    const state: State | undefined = this.cache.get(REPOSITORY_KEY);
+    if (typeof state === "undefined") {
+      return Promise.resolve();
+    }
+    const ids = state[tabId];
+    if (typeof ids === "undefined") {
+      return Promise.resolve();
+    }
+    const tab = state[tabId] || {};
+    tab[frameId] = (tab[frameId] || 0) - 1;
+    if (tab[frameId] == 0) {
+      delete tab[frameId];
+    }
+    if (Object.keys(tab).length === 0) {
+      delete state[tabId];
+    }
+
+    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);
+    }
+    const tab = state[tabId];
+    if (typeof tab === "undefined") {
+      return Promise.resolve(undefined);
+    }
+    const frameIds = Object.keys(tab)
+      .map((v) => Number(v))
+      .sort();
+    return Promise.resolve(frameIds);
+  }
+}