From f54d875f801aa383c5962b0d308dfccdca4703bd Mon Sep 17 00:00:00 2001
From: sobuj53 <25761237+sobuj53@users.noreply.github.com>
Date: Mon, 5 Apr 2021 12:41:51 +0600
Subject: [PATCH 1/2] Resolved the button clicking issue

This will click the close button in right manner. Thanks to `Albert Einstein`
---
 src/inject/inject.js | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

diff --git a/src/inject/inject.js b/src/inject/inject.js
index 8700464..c90b695 100755
--- a/src/inject/inject.js
+++ b/src/inject/inject.js
@@ -28,10 +28,12 @@ function clickConnect() {
 }
 
 function clickDismiss() {
-	try {
-		document.querySelector('colab-sessions-dialog').shadowRoot.querySelector('.dismiss').click();
+	try {		
+		//colab design changed again
+		document.querySelector("paper-tab").querySelector("paper-icon-button").shadowRoot.getElementById('icon').click();
+		//document.querySelector('colab-sessions-dialog').shadowRoot.querySelector('.dismiss').click();
 		console.log('clicked on dismiss button');
 	} catch (error) {
 		console.log(error);
 	}
-}
\ No newline at end of file
+}

From 2cec4570423da6a2a9fe03f4e48aecd6f91cdc0c Mon Sep 17 00:00:00 2001
From: sobuj53 <25761237+sobuj53@users.noreply.github.com>
Date: Thu, 8 Apr 2021 15:51:09 +0600
Subject: [PATCH 2/2] update clicking method

---
 src/inject/inject.js | 39 +++++++++++++++++++++++++++++++--------
 1 file changed, 31 insertions(+), 8 deletions(-)

diff --git a/src/inject/inject.js b/src/inject/inject.js
index c90b695..1218e96 100755
--- a/src/inject/inject.js
+++ b/src/inject/inject.js
@@ -20,20 +20,43 @@ function clickConnect() {
 		 document.querySelector("#top-toolbar > colab-connect-button").shadowRoot.querySelector("#connect").click();
 		// this also works, if above one doesn't work, comment it and uncomment below one
 		//document.querySelector("colab-connect-button").shadowRoot.getElementById('connect').click();
-		setTimeout(clickDismiss, 2000);
+		setTimeout(clickDismiss, 500);
 		console.log("Keeping Colab Alive!");	
 	} catch (error) {
 		console.log(error);
 	}
 }
 
-function clickDismiss() {
-	try {		
-		//colab design changed again
-		document.querySelector("paper-tab").querySelector("paper-icon-button").shadowRoot.getElementById('icon').click();
-		//document.querySelector('colab-sessions-dialog').shadowRoot.querySelector('.dismiss').click();
-		console.log('clicked on dismiss button');
-	} catch (error) {
+
+function sleep(ms) {
+  return new Promise(resolve => setTimeout(resolve, ms));
+}
+
+async function clickDismiss() {
+	
+	try {	
+		
+		// click manage session button
+		document.querySelector("colab-usage-display").shadowRoot.querySelector("paper-button").click();
+	
+		} catch (error) {
+		console.log(error);
+	}
+	
+		try {
+			// leave from manage session window
+			await sleep(1000);
+			document.querySelector('colab-sessions-dialog').shadowRoot.querySelector('.dismiss').click();
+				} catch (error) {
+		console.log(error);
+	}
+	
+		try {	
+			// click close button
+			await sleep(1000);
+			document.querySelector("paper-tab").querySelector("paper-icon-button").shadowRoot.getElementById('icon').click();
+				} catch (error) {
 		console.log(error);
 	}
+	
 }