(#51) Find hooks

Author: s1hofmann

lib/screen.class.spec.ts

@@ -45,6 +45,22 @@ describe("Screen.", () => {
     expect(visionAdapterMock.findOnScreenRegion).toHaveBeenCalledWith(matchRequest);
   });
 
+  it("should call registered hook before resolve", async () => {
+    const matchResult = new MatchResult(0.99, searchRegion);
+    VisionAdapter.prototype.findOnScreenRegion = jest.fn(() => {
+      return Promise.resolve(matchResult);
+    });
+    const visionAdapterMock = new VisionAdapter();
+
+    const SUT = new Screen(visionAdapterMock);
+    const testCallback = jest.fn(() => Promise.resolve());
+    const imagePath = "test/path/to/image.png";
+    SUT.on(imagePath, testCallback);
+    await SUT.find(imagePath);
+    expect(testCallback).toBeCalledTimes(1);
+    expect(testCallback).toBeCalledWith(matchResult);
+  });
+
   it("should reject with insufficient confidence.", async () => {
     const matchResult = new MatchResult(0.8, searchRegion);
 

lib/screen.class.ts

@@ -5,16 +5,21 @@ import { FileType } from "./file-type.enum";
 import { generateOutputPath } from "./generate-output-path.function";
 import { LocationParameters } from "./locationparameters.class";
 import { MatchRequest } from "./match-request.class";
+import { MatchResult } from "./match-result.class";
 import { Region } from "./region.class";
 import { timeout } from "./util/poll-action.function";
 
+export type FindHookCallback = (target: MatchResult) => Promise<void>;
+
 export class Screen {
   public config = {
     confidence: 0.99,
     resourceDirectory: cwd(),
   };
 
-  constructor(private vision: VisionAdapter) {
+  constructor(
+    private vision: VisionAdapter,
+    private findHooks: Map<string, FindHookCallback> = new Map<string, FindHookCallback>()) {
   }
 
   public width() {
@@ -34,7 +39,6 @@ export class Screen {
       (params && params.searchRegion) || await this.vision.screenSize();
 
     const fullPathToNeedle = normalize(join(this.config.resourceDirectory, pathToNeedle));
-    // console.log(`Full path to needle: ${fullPathToNeedle}`);
 
     const screenImage = await this.vision.grabScreen();
 
@@ -49,6 +53,10 @@ export class Screen {
       try {
         const matchResult = await this.vision.findOnScreenRegion(matchRequest);
         if (matchResult.confidence >= minMatch) {
+          const possibleHook = this.findHooks.get(pathToNeedle);
+          if (possibleHook) {
+            await possibleHook(matchResult);
+          }
           resolve(matchResult.location);
         } else {
           reject(
@@ -73,6 +81,10 @@ export class Screen {
     return timeout(500, timeoutMs, () => this.find(pathToNeedle, params));
   }
 
+  public on(pathToNeedle: string, callback: FindHookCallback) {
+    this.findHooks.set(pathToNeedle, callback);
+  }
+
   public async capture(
     fileName: string,
     fileFormat: FileType = FileType.PNG,