From d436c8e65ce94bcaa1ef03b23200e9ed4292b3b8 Mon Sep 17 00:00:00 2001
From: Kenshin <fxblog@gmail.com>
Date: Tue, 20 Dec 2016 13:53:04 +0800
Subject: [PATCH] Optimize focus controlbar logic.

---
 src/contentscripts.js   | 2 +-
 src/focus/controlbar.js | 8 +++++---
 2 files changed, 6 insertions(+), 4 deletions(-)

diff --git a/src/contentscripts.js b/src/contentscripts.js
index ca83b41d..b49a71fd 100644
--- a/src/contentscripts.js
+++ b/src/contentscripts.js
@@ -61,7 +61,7 @@ function focuseMode() {
     $( "body" ).append( '<div class="ks-simpread-bg"></div>' );
 
     // add control bar
-    fcontrol.Init();
+    fcontrol.Init( ".ks-simpread-bg" );
 
     // click mask remove it
     $( ".ks-simpread-bg" ).on( "click", function( event ) {
diff --git a/src/focus/controlbar.js b/src/focus/controlbar.js
index 7bddf149..df2509e5 100644
--- a/src/focus/controlbar.js
+++ b/src/focus/controlbar.js
@@ -8,9 +8,11 @@ var fcontrol = ( function() {
     /*
         Add focus constrol bar
     */
-    FControl.prototype.Init = function() {
+    FControl.prototype.Init = function( root ) {
+        var me = this;
         $.get( chrome.extension.getURL( "/focus/template.html" ), function( tmpl ) {
-            $( ".ks-simpread-bg" ).append( tmpl );
+            me.constructor.prototype.$root = $(root);
+            $( root ).append( tmpl );
                 $target = $( ".ks-simpread-constrolbar" ).find( "span" );
                 addStyle();
                 addEventHandler();
@@ -49,7 +51,7 @@ var fcontrol = ( function() {
                     break;
                 case "closeicon":
                     console.log("333")
-                    $(".ks-simpread-bg").click();
+                    FControl.prototype.$root.click();
                     break;
             }
         })