Merge pull request #53 from takker99/take-all-stores

✨ Get both Cursor and Selection at once

Author: takker99

browser/dom/cursor.ts

@@ -5,9 +5,4 @@
 import { takeStores } from "./stores.ts";
 import { Cursor } from "./cursor.d.ts";
 
-export const takeCursor = (): Cursor => {
-  for (const store of takeStores()) {
-    if ("goByAction" in store) return store;
-  }
-  throw Error('#text-input must has a "Cursor" store.');
-};
+export const takeCursor = (): Cursor => takeStores().cursor;

browser/dom/mod.ts

@@ -10,3 +10,4 @@ export * from "./open.ts";
 export * from "./cache.ts";
 export * from "./cursor.ts";
 export * from "./selection.ts";
+export * from "./stores.ts";

browser/dom/selection.ts

@@ -5,9 +5,4 @@
 import { takeStores } from "./stores.ts";
 import { Selection } from "./selection.d.ts";
 
-export const takeSelection = (): Selection => {
-  for (const store of takeStores()) {
-    if ("hasSelection" in store) return store;
-  }
-  throw Error('#text-input must has a "Selection" store.');
-};
+export const takeSelection = (): Selection => takeStores().selection;

browser/dom/stores.ts

@@ -6,7 +6,7 @@ import { textInput } from "./dom.ts";
 import { Cursor } from "./cursor.d.ts";
 import { Selection } from "./selection.d.ts";
 
-export const takeStores = (): (Cursor | Selection)[] => {
+export const takeStores = (): { cursor: Cursor; selection: Selection } => {
   const textarea = textInput();
   if (!textarea) {
     throw Error(`#text-input is not found.`);
@@ -21,9 +21,24 @@ export const takeStores = (): (Cursor | Selection)[] => {
   }
 
   // @ts-ignore DOMを無理矢理objectとして扱っている
-  return (textarea[
+  const stores = (textarea[
     reactKey
-  ] as ReactFiber).return.return.stateNode._stores;
+  ] as ReactFiber).return.return.stateNode._stores as (Cursor | Selection)[];
+
+  const cursor = stores.find((store) =>
+    store.constructor.name === "Cursor"
+  ) as (Cursor | undefined);
+  if (!cursor) {
+    throw Error('#text-input must has a "Cursor" store.');
+  }
+  const selection = stores.find((store) =>
+    store.constructor.name === "Selection"
+  ) as (Selection | undefined);
+  if (!selection) {
+    throw Error('#text-input must has a "Selection" store.');
+  }
+
+  return { cursor, selection };
 };
 
 interface ReactFiber {