Use factory pattern to determine which scanner to use (#129)

Author: schang1146

src/common/messages/message-handler.ts

@@ -34,7 +34,12 @@ const notifyHandler: MessageHandler<'notify'> = async (payload, _context) => {
 const pageInfoHandler: MessageHandler<'pageInfo'> = (payload, context) => {
     return new Promise((resolve, reject) => {
         console.log('Page Info Received, triggering page load handler:', payload);
-        context.main.onPageLoaded(payload.domain, payload.url).then(resolve).catch(reject);
+        try {
+            context.main.onPageLoaded(payload.domain, payload.url);
+            resolve();
+        } catch (e) {
+            reject(e as Error);
+        }
     });
 };
 

src/common/services/content-scanner.ts

@@ -0,0 +1,18 @@
+import { IScanParameters } from '@/common/services/content-scanner.types';
+import { UrlScanner } from '@/scanners/url-scanner';
+import AmazonScanner from '@/scanners/amazon/amazon-scanner';
+
+const amazonProductPageRegex = /^https?:\/\/(www\.)?amazon\.[a-z.]+\/.+\/dp\/[A-Z0-9]{10}(?:[/?].*)?$/i;
+
+export class ScannerFactory {
+    getScanner(parameters: IScanParameters) {
+        if (amazonProductPageRegex.test(parameters.url)) {
+            return new AmazonScanner(parameters);
+        }
+
+        return new UrlScanner({
+            url: new URL(parameters.url),
+            pagesDb: parameters.pagesDb,
+        });
+    }
+}