1 files changed, 35 insertions, 0 deletions
diff --git a/src/background/repositories/MarkRepository.ts b/src/background/repositories/MarkRepository.ts
new file mode 100644
index 0000000..69c85f6
--- /dev/null
+++ b/src/background/repositories/MarkRepository.ts
@@ -0,0 +1,35 @@
+import MemoryStorage from '../infrastructures/MemoryStorage';
+import GlobalMark from '../domains/GlobalMark';
+
+const MARK_KEY = 'mark';
+
+export default class MarkRepository {
+  private cache: MemoryStorage;
+
+  constructor() {
+    this.cache = new MemoryStorage();
+  }
+
+  getMark(key: string): Promise<GlobalMark | undefined> {
+    let marks = this.getOrEmptyMarks();
+    let data = marks[key];
+    if (!data) {
+      return Promise.resolve(undefined);
+    }
+    let mark = { tabId: data.tabId, url: data.url, x: data.x, y: data.y };
+    return Promise.resolve(mark);
+  }
+
+  setMark(key: string, mark: GlobalMark): Promise<any> {
+    let marks = this.getOrEmptyMarks();
+    marks[key] = { tabId: mark.tabId, url: mark.url, x: mark.x, y: mark.y };
+    this.cache.set(MARK_KEY, marks);
+
+    return Promise.resolve();
+  }
+
+  getOrEmptyMarks() {
+    return this.cache.get(MARK_KEY) || {};
+  }
+}
+