Skip to content

Allow browser plugin to be compatible with the Firefox Nightly 61.0a1. #47

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
TanShun wants to merge 1 commit into
base: master
Choose a base branch
from
Open

Allow browser plugin to be compatible with the Firefox Nightly 61.0a1. #47

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
296 changes: 126 additions & 170 deletions chrome/background.js
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,147 +1,118 @@
/**
* github: https://github.com/luofei614/SocketLog
* @author luofei614<weibo.com/luofei614>
*/
var websocket=null;
var websocket_timeout=0;
function ws_init()
{
if(websocket)
{
*/
var websocket = null;
var websocket_timeout = 0;

function ws_init() {
if (websocket) {
//避免重复监听
websocket.onclose=function(){};//onclose 函数置空，防止重复链接
websocket.close();
websocket.onclose = function() {}; //onclose 函数置空，防止重复链接
websocket.close();
}
var address=localStorage.getItem('address');
var client_id=localStorage.getItem('client_id');
if(!address)
{
address="ws://localhost:1229";
var address = localStorage.getItem('address');
var client_id = localStorage.getItem('client_id');
if (!address) {
address = "ws://localhost:1229";
}
if(client_id)
{
if (client_id) {
//client_id作为地址
address+='/'+client_id;
address += '/' + client_id;
}
websocket=new WebSocket(address);

websocket.onerror=function(msg)
{
websocket = new WebSocket(address);
websocket.onerror = function(msg) {
clearTimeout(websocket_timeout);
websocket_timeout=setTimeout(ws_init,2000);
localStorage.setItem('status','error');
websocket_timeout = setTimeout(ws_init, 2000);
localStorage.setItem('status', 'error');
disable_icon();
};

websocket.onclose=function()
{
setTimeout(function(){
websocket.onclose = function() {
setTimeout(function() {
clearTimeout(websocket_timeout);
websocket_timeout=setTimeout(ws_init,2000);
},1000);
localStorage.setItem('status','close');
websocket_timeout = setTimeout(ws_init, 2000);
}, 1000);
localStorage.setItem('status', 'close');
disable_icon();
}

websocket.onopen=function()
{
localStorage.setItem('status','open');
enable_icon();
websocket.onopen = function() {
localStorage.setItem('status', 'open');
enable_icon();
}


websocket.onmessage=function(event){

var check_error=function()
{
if(event.data.indexOf('SocketLog error handler')!='-1')
{
var opt = {
type: "basic",
title: "注意",
message: "有异常报错，请注意查看console 控制台中的日志",
iconUrl: "logo.png"
websocket.onmessage = function(event) {
var check_error = function() {
if (event.data.indexOf('SocketLog error handler') != '-1') {
var opt = {
type: "basic",
title: "注意",
message: "有异常报错，请注意查看console 控制台中的日志",
iconUrl: "logo.png"
};
chrome.notifications.create('',opt,function(id){
setTimeout(function(){
chrome.notifications.clear(id,function(){});
},3000);
chrome.notifications.create('', opt, function(id) {
setTimeout(function() {
chrome.notifications.clear(id, function() {});
}, 3000);
});

}


if(event.data.indexOf('[NO WHERE]')!='-1')
{
if (event.data.indexOf('[NO WHERE]') != '-1') {
var opt = {
type: "basic",
title: "注意",
message: "存在没有WHERE语句的操作sql语句",
iconUrl: "logo.png"
type: "basic",
title: "注意",
message: "存在没有WHERE语句的操作sql语句",
iconUrl: "logo.png"
};
chrome.notifications.create('',opt,function(id){
setTimeout(function(){
chrome.notifications.clear(id,function(){});
},3000);
chrome.notifications.create('', opt, function(id) {
setTimeout(function() {
chrome.notifications.clear(id, function() {});
}, 3000);
});
}

};

try
{
var data=JSON.parse(event.data);
}
catch(e)
{
if(0==event.data.indexOf('close:')){
websocket.onclose=function(){};//onclose 函数置空，防止重复链接
alert('此client_id不允许连接服务');
}else{
alert('日志格式错误，'+event.data);
}
return ;
try {
var data = JSON.parse(event.data);
} catch (e) {
if (0 == event.data.indexOf('close:')) {
websocket.onclose = function() {}; //onclose 函数置空，防止重复链接
alert('此client_id不允许连接服务');
} else {
alert('日志格式错误，' + event.data);
}
return;
}
var client_id=localStorage.getItem('client_id');
var client_id = localStorage.getItem('client_id');
//判断是否有强制日志
if(client_id && data.force_client_id==client_id)
{
//将强制日志输出到当前的tab页
chrome.tabs.query(
{currentWindow: true, active: true},
function(tabArray) {
if (tabArray && tabArray[0])
{
//延迟保证日志每次都能记录
setTimeout(function(){
check_error();
chrome.tabs.sendMessage(tabArray[0].id,data.logs);
},100);
}
if (client_id && data.force_client_id == client_id) {
//将强制日志输出到当前的tab页
chrome.tabs.query({
currentWindow: true,
active: true
}, function(tabArray) {
if (tabArray && tabArray[0]) {
//延迟保证日志每次都能记录
setTimeout(function() {
check_error();
chrome.tabs.sendMessage(tabArray[0].id, data.logs);
}, 100);
}
);
return ;
});
return;
}

if((client_id && data.client_id!=client_id) || !data.tabid)
{
//不是当前用户的日志不显示。
return ;
if ((client_id && data.client_id != client_id) || !data.tabid) {
//不是当前用户的日志不显示。
return;
}
//延迟保证日志每次都能记录
setTimeout(function(){
setTimeout(function() {
check_error();
chrome.tabs.sendMessage(parseInt(data.tabid),data.logs);
},100);
chrome.tabs.sendMessage(parseInt(data.tabid), data.logs);
}, 100);
};
}

function ws_restart()
{
function ws_restart() {
ws_init();
}


function enable_icon() {
chrome.browserAction.setIcon({
path: "logo.png"
Expand All @@ -153,72 +124,57 @@ function disable_icon() {
path: "logo_disabled.png"
});
}



ws_init();


function url_exp(url)
{
var splatParam = /\*/g;
var escapeRegExp = /[-[\]{}()+?.,\\^$#\s]/g;
url = url.replace(escapeRegExp, '\\$&')
.replace(splatParam, '(.*?)');
function url_exp(url) {
var splatParam = /\*/g;
var escapeRegExp = /[-[\]{}()+?.,\\^$#\s]/g;
url = url.replace(escapeRegExp, '\\$&').replace(splatParam, '(.*?)');
return new RegExp(url, 'i');
}

chrome.webRequest.onBeforeSendHeaders.addListener(
function(details) {

var header="tabid="+details.tabId;

var client_id=localStorage.getItem('client_id');
if(!client_id)
{
client_id='';
}


header+="&client_id="+client_id;

var special_domain=localStorage.getItem('special_domain');
if(!special_domain)
{
special_domain='*.sinaapp.com';
localStorage.setItem('special_domain',special_domain);
}
var exp=url_exp(special_domain);
if (exp.test(details.url))
{
//如果是特殊环境域名
for (var i = 0; i < details.requestHeaders.length; ++i)
{
if (details.requestHeaders[i].name === 'User-Agent')
{
//将参数放在User-agent中，兼容SAE的情况
details.requestHeaders[i].value+=" SocketLog("+header+")";
break;
}
}
chrome.webRequest.onBeforeSendHeaders.addListener(function(details) {
var header = "tabid=" + details.tabId;
var client_id = localStorage.getItem('client_id');
if (!client_id) {
client_id = '';
}
header += "&client_id=" + client_id;
var special_domain = localStorage.getItem('special_domain');
if (!special_domain) {
special_domain = '*.sinaapp.com';
localStorage.setItem('special_domain', special_domain);
}
var exp = url_exp(special_domain);
if (exp.test(details.url)) {
//如果是特殊环境域名
for (var i = 0; i < details.requestHeaders.length; ++i) {
if (details.requestHeaders[i].name === 'User-Agent') {
//将参数放在User-agent中，兼容SAE的情况
details.requestHeaders[i].value += " SocketLog(" + header + ")";
break;
}
}
else
{
details.requestHeaders.push({name:'SocketLog',value:" SocketLog("+header+")"});
}

return {requestHeaders: details.requestHeaders};
},
{urls: ["<all_urls>"]},
["blocking", "requestHeaders"]);

chrome.webRequest.onCompleted.addListener(function(details){
var online_domain=localStorage.getItem('online_domain');
if(online_domain){
var exp=url_exp(online_domain);
if(exp.test(details.url))
{
chrome.tabs.sendMessage(details.tabId,'online_evn');
} else {
details.requestHeaders.push({
name: 'SocketLog',
value: " SocketLog(" + header + ")"
});
}
return {
requestHeaders: details.requestHeaders
};
}, {
urls: ["<all_urls>"]
}, ["blocking", "requestHeaders"]);
chrome.webRequest.onCompleted.addListener(function(details) {
var online_domain = localStorage.getItem('online_domain');
if (online_domain) {
var exp = url_exp(online_domain);
if (exp.test(details.url)) {
chrome.tabs.sendMessage(details.tabId, 'online_evn');
}
}
},{urls: ["<all_urls>"],types:['main_frame','sub_frame']});
}, {
urls: ["<all_urls>"],
types: ['main_frame', 'sub_frame']
});
Loading