feat(context): Treatment of the puppeteer.connect() function

README.md

@@ -154,6 +154,8 @@ venom
     (statusSession) => {
       console.log('Status Session: ', statusSession);
       //return isLogged || notLogged || browserClose || qrReadSuccess || qrReadFail || autocloseCalled
+      //
+A criação de sessão no wss retorna "serverClose" se o servidor não estiver funcionando.
     },
     undefined
   )

src/api/layers/retriever.layer.ts

@@ -54,6 +54,7 @@ MMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMM
 
 */
 import { Page } from 'puppeteer';
+import { async } from 'rxjs';
 import {
   Chat,
   Contact,
@@ -97,7 +98,7 @@ declare module WAPI {
     includeMe: boolean,
     includeNotifications: boolean
   ) => Message[];
-  const getSessionTokenBrowser: () => object;
+  const getSessionTokenBrowser: () => void;
 }
 
 export class RetrieverLayer extends SenderLayer {

src/controllers/auth.ts

@@ -104,6 +104,9 @@ export const isInsideChat = (waPage: puppeteer.Page) => {
 };
 export async function retrieveQR(page: puppeteer.Page) {
   const { code, data } = await decodeQR(page);
+  if (data === null || code === null) {
+    return false;
+  }
   const asciiQR = await asciiQr(code);
   return { code, data, asciiQR };
 }
@@ -120,22 +123,27 @@ async function decodeQR(
   page: puppeteer.Page
 ): Promise<{ code: string; data: string }> {
   await page.waitForSelector('canvas', { timeout: 0 });
+
   await page.addScriptTag({
     path: require.resolve(path.join(__dirname, '../lib/jsQR', 'jsQR.js')),
   });
 
   return await page.evaluate(() => {
-    const canvas = document.querySelector('canvas');
-    const context = canvas.getContext('2d');
-
-    // @ts-ignore
-    const code = jsQR(
-      context.getImageData(0, 0, canvas.width, canvas.height).data,
-      canvas.width,
-      canvas.height
-    );
-
-    return { code: code.data, data: canvas.toDataURL() };
+    const canvas = document.querySelector('canvas') || null;
+    if (canvas !== null) {
+      const context = canvas.getContext('2d') || null;
+      if (context !== null) {
+        // @ts-ignore
+        const code = jsQR(
+          context.getImageData(0, 0, canvas.width, canvas.height).data,
+          canvas.width,
+          canvas.height
+        );
+        return { code: code.data, data: canvas.toDataURL() };
+      }
+    } else {
+      return { code: null, data: null };
+    }
   });
 }
 

src/controllers/browser.ts

@@ -144,7 +144,7 @@ export async function initBrowser(
         browser = e;
       })
       .catch(() => {
-        throw new Error('Error when try to connect');
+        browser = 'connect';
       });
   } else {
     await puppeteer
@@ -159,7 +159,9 @@ export async function initBrowser(
       .then((e) => {
         browser = e;
       })
-      .catch(() => {});
+      .catch(() => {
+        browser = 'launch';
+      });
   }
 
   return browser;