Merge pull request #1264 from ueokande/fix-establish-connection-issue

Search a content from frames successfully loaded

1 files changed, 45 insertions, 48 deletions

diff --git a/src/background/operators/impls/FindNextOperator.ts b/src/background/operators/impls/FindNextOperator.ts
index 241f71d..99f1759 100644
--- a/src/background/operators/impls/FindNextOperator.ts
+++ b/src/background/operators/impls/FindNextOperator.ts
@@ -3,7 +3,7 @@ import TabPresenter from "../../presenters/TabPresenter";
 import FindRepository from "../../repositories/FindRepository";
 import FindClient from "../../clients/FindClient";
 import ConsoleClient from "../../infrastructures/ConsoleClient";
-import FramePresenter from "../../presenters/FramePresenter";
+import ReadyFrameRepository from "../../repositories/ReadyFrameRepository";
 
 export default class FindNextOperator implements Operator {
   constructor(
@@ -11,7 +11,7 @@ export default class FindNextOperator implements Operator {
     private readonly findRepository: FindRepository,
     private readonly findClient: FindClient,
     private readonly consoleClient: ConsoleClient,
-    private readonly framePresenter: FramePresenter
+    private readonly frameRepository: ReadyFrameRepository
   ) {}
 
   async run(): Promise<void> {
@@ -21,67 +21,64 @@ export default class FindNextOperator implements Operator {
       return;
     }
 
+    const frameIds = await this.frameRepository.getFrameIds(tabId);
+    if (typeof frameIds === "undefined") {
+      // No frames are ready
+      return;
+    }
+
     const state = await this.findRepository.getLocalState(tabId);
     if (state) {
-      // Start to find the keyword from the current frame which last found on,
-      // and concat it to end of frame ids to perform a wrap-search
-      //
-      //                ,- keyword should be in this frame
-      //                |
-      // [100, 101, 0, 100]
-      //   |
-      //   `- continue from frame id 100
-      //
-      const targetFrameIds = state.frameIds
-        .slice(state.framePos)
-        .concat(
-          state.frameIds.slice(0, state.framePos),
-          state.frameIds[state.framePos]
-        );
+      const framePos = frameIds.indexOf(state.frameId);
+      if (framePos !== -1) {
+        // Start to find the keyword from the current frame which last found on,
+        // and concat it to end of frame ids to perform a wrap-search
+        //
+        //                ,- keyword should be in this frame
+        //                |
+        // [100, 101, 0, 100]
+        //   |
+        //   `- continue from frame id 100
+        //
+        const targetFrameIds = frameIds
+          .slice(framePos)
+          .concat(frameIds.slice(0, framePos), frameIds[framePos]);
 
-      for (let i = 0; i < targetFrameIds.length; ++i) {
-        const found = await this.findClient.findNext(
+        for (const frameId of targetFrameIds) {
+          const found = await this.findClient.findNext(
+            tabId,
+            frameId,
+            state.keyword
+          );
+          if (found) {
+            this.findRepository.setLocalState(tabId, {
+              keyword: state.keyword,
+              frameId,
+            });
+            return;
+          }
+          this.findClient.clearSelection(tabId, frameId);
+        }
+
+        // The keyword is gone.
+        this.consoleClient.showError(
           tabId,
-          targetFrameIds[i],
-          state.keyword
+          "Pattern not found: " + state.keyword
         );
-        if (found) {
-          this.findRepository.setLocalState(tabId, {
-            ...state,
-            framePos: (i + state.framePos) % state.frameIds.length, // save current frame position or first
-          });
-          return;
-        }
-        this.findClient.clearSelection(tabId, targetFrameIds[i]);
+        return;
       }
-
-      // The keyword is gone.
-      this.consoleClient.showError(
-        tabId,
-        "Pattern not found: " + state.keyword
-      );
-      return;
     }
 
     const keyword = await this.findRepository.getGlobalKeyword();
     if (keyword) {
-      const frameIds = await this.framePresenter.getAllFrameIds(tabId);
       for (const frameId of frameIds) {
         await this.findClient.clearSelection(tabId, frameId);
       }
 
-      for (let framePos = 0; framePos < frameIds.length; ++framePos) {
-        const found = await this.findClient.findNext(
-          tabId,
-          frameIds[framePos],
-          keyword
-        );
+      for (const frameId of frameIds) {
+        const found = await this.findClient.findNext(tabId, frameId, keyword);
         if (found) {
-          await this.findRepository.setLocalState(tabId, {
-            frameIds,
-            framePos,
-            keyword,
-          });
+          await this.findRepository.setLocalState(tabId, { frameId, keyword });
           await this.consoleClient.showInfo(tabId, "Pattern found: " + keyword);
           return;
         }