throw tosblock on first detection #103

open-wa · Feb 14, 2020 · c088a76 · c088a76
1 parent e85f7be
commit c088a76
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 4 deletions.
diff --git a/demo/index.ts b/demo/index.ts
@@ -132,13 +132,14 @@ create('session',
     '--disable-offline-load-stale-cache',
     '--disk-cache-size=0'
   ]
-}
+},
+tosBlockGuaranteed
 )
 // create()
 .then(async client => await start(client))
 .catch(e=>{
-  console.log(e);
-  // process.exit();
+  console.log('Error',e.message);
+  process.exit();
 });
 
 //or you can set a 'session id'

diff --git a/src/controllers/initializer.ts b/src/controllers/initializer.ts
@@ -19,6 +19,7 @@ export async function create(sessionId?: string, puppeteerConfigOverride?:any, c
   spinner.start('Initializing whatsapp');
   waPage = await initWhatsapp(sessionId, puppeteerConfigOverride, customUserAgent);
   spinner.succeed();
+  const throwOnError=puppeteerConfigOverride&&puppeteerConfigOverride.throwErrorOnTosBlock==true;
 
   const PAGE_UA =  await waPage.evaluate('navigator.userAgent');
   const BROWSER_VERSION = await waPage.browser().version();
@@ -39,6 +40,7 @@ export async function create(sessionId?: string, puppeteerConfigOverride?:any, c
     waPage = await injectApi(waPage);
     spinner.start('WAPI injected');
   } else {
+    if(throwOnError) throw Error('TOSBLOCK');
     console.log('Possilby TOS_BLOCKed')
   }
 
@@ -49,7 +51,7 @@ export async function create(sessionId?: string, puppeteerConfigOverride?:any, c
   const qrLoop = async () => {
     if(!shouldLoop) return;
     console.log(' ')
-    await retrieveQR(waPage,sessionId,autoRefresh,puppeteerConfigOverride&&puppeteerConfigOverride.throwErrorOnTosBlock==true);
+    await retrieveQR(waPage,sessionId,autoRefresh,throwOnError);
     console.log(' ')
     qrTimeout = timeout((puppeteerConfigOverride?(puppeteerConfigOverride.qrRefreshS || 10):10)*1000);
     await qrTimeout;