From e83f34c885f67b02f5c81470e610595b97016f25 Mon Sep 17 00:00:00 2001
From: 502647092 <admin@yumc.pw>
Date: Tue, 15 May 2018 17:34:39 +0800
Subject: [PATCH] =?UTF-8?q?=E6=9B=B4=E6=96=B0=20'WebIDE/enhance.user.js'?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 WebIDE/enhance.user.js | 20 ++++++++++++--------
 1 file changed, 12 insertions(+), 8 deletions(-)

diff --git a/WebIDE/enhance.user.js b/WebIDE/enhance.user.js
index e2bb369..71d2ad3 100644
--- a/WebIDE/enhance.user.js
+++ b/WebIDE/enhance.user.js
@@ -11,10 +11,13 @@
 
 (function() {
     'use strict';
-    setTimeout(function (){
-        window.onbeforeunload = function () { return false; }
-        console.log('已阻止 Ctrl + W 关闭页面...')
-    }, 1000)
+    var id = setInterval(function (){
+        if (document.getElementsByClassName('ide-container')[0]) {
+            window.onbeforeunload = function () { return false; }
+            console.log('已阻止 Ctrl + W 关闭页面...')
+            clearInterval(id);
+        }
+    }, 300)
     var dqKeys = function (keys, isOrder, sucFuc, cancelFuc) {
         return new dqKeys.fn.init(keys, isOrder, sucFuc, cancelFuc);
     }
@@ -97,11 +100,12 @@
                 document.addEventListener(ev, fn, bool);
             }
         }
-
     }
     dqKeys.fn.init.prototype = dqKeys.fn;
-    dqKeys({'alt':18,'w':87}, true, function(){
-    },function(){
-        document.getElementsByClassName('tab-label active')[0].childNodes[2].childNodes[0].click();
+    dqKeys({'alt':18,'w':87}, true, function(){},function(){
+      	var activeTab = document.getElementsByClassName('tab-label active')[0];
+        if (activeTab) {
+            activeTab.childNodes[2].childNodes[0].click();
+        }
     }).listenkeys();
 })();