mirror of
https://github.com/gorhill/uBlock.git
synced 2024-11-17 16:02:33 +01:00
458 lines
15 KiB
JavaScript
458 lines
15 KiB
JavaScript
/*******************************************************************************
|
|
|
|
uBlock Origin - a browser extension to block requests.
|
|
Copyright (C) 2014-present Raymond Hill
|
|
|
|
This program is free software: you can redistribute it and/or modify
|
|
it under the terms of the GNU General Public License as published by
|
|
the Free Software Foundation, either version 3 of the License, or
|
|
(at your option) any later version.
|
|
|
|
This program is distributed in the hope that it will be useful,
|
|
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
GNU General Public License for more details.
|
|
|
|
You should have received a copy of the GNU General Public License
|
|
along with this program. If not, see {http://www.gnu.org/licenses/}.
|
|
|
|
Home: https://github.com/gorhill/uMatrix
|
|
*/
|
|
|
|
/* global diff_match_patch, CodeMirror, uDom, uBlockDashboard */
|
|
|
|
'use strict';
|
|
|
|
/******************************************************************************/
|
|
|
|
(function() {
|
|
|
|
/******************************************************************************/
|
|
|
|
var messaging = vAPI.messaging;
|
|
|
|
var mergeView = new CodeMirror.MergeView(
|
|
document.querySelector('.codeMirrorMergeContainer'),
|
|
{
|
|
allowEditingOriginals: true,
|
|
connect: 'align',
|
|
inputStyle: 'contenteditable',
|
|
lineNumbers: true,
|
|
lineWrapping: false,
|
|
origLeft: '',
|
|
revertButtons: true,
|
|
value: ''
|
|
}
|
|
);
|
|
mergeView.editor().setOption('styleActiveLine', true);
|
|
mergeView.editor().setOption('lineNumbers', false);
|
|
mergeView.leftOriginal().setOption('readOnly', 'nocursor');
|
|
|
|
uBlockDashboard.patchCodeMirrorEditor(mergeView.editor());
|
|
|
|
var unfilteredRules = {
|
|
orig: { doc: mergeView.leftOriginal(), rules: [] },
|
|
edit: { doc: mergeView.editor(), rules: [] }
|
|
};
|
|
|
|
var cleanEditToken = 0;
|
|
var cleanEditText = '';
|
|
|
|
var differ;
|
|
|
|
/******************************************************************************/
|
|
|
|
// Borrowed from...
|
|
// https://github.com/codemirror/CodeMirror/blob/3e1bb5fff682f8f6cbfaef0e56c61d62403d4798/addon/search/search.js#L22
|
|
// ... and modified as needed.
|
|
|
|
var updateOverlay = (function() {
|
|
let reFilter;
|
|
let mode = {
|
|
token: function(stream) {
|
|
if ( reFilter !== undefined ) {
|
|
reFilter.lastIndex = stream.pos;
|
|
let match = reFilter.exec(stream.string);
|
|
if ( match !== null ) {
|
|
if ( match.index === stream.pos ) {
|
|
stream.pos += match[0].length || 1;
|
|
return 'searching';
|
|
}
|
|
stream.pos = match.index;
|
|
return;
|
|
}
|
|
}
|
|
stream.skipToEnd();
|
|
}
|
|
};
|
|
return function(filter) {
|
|
reFilter = typeof filter === 'string' && filter !== '' ?
|
|
new RegExp(filter.replace(/[.*+?^${}()|[\]\\]/g, '\\$&'), 'gi') :
|
|
undefined;
|
|
return mode;
|
|
};
|
|
})();
|
|
|
|
/******************************************************************************/
|
|
|
|
// Incrementally update text in a CodeMirror editor for best user experience:
|
|
// - Scroll position preserved
|
|
// - Minimum amount of text updated
|
|
|
|
var rulesToDoc = function(clearHistory) {
|
|
for ( let key in unfilteredRules ) {
|
|
if ( unfilteredRules.hasOwnProperty(key) === false ) { continue; }
|
|
let doc = unfilteredRules[key].doc;
|
|
let rules = filterRules(key);
|
|
if (
|
|
doc.lineCount() === 1 && doc.getValue() === '' ||
|
|
rules.length === 0
|
|
) {
|
|
doc.setValue(rules.length !== 0 ? rules.join('\n') : '');
|
|
continue;
|
|
}
|
|
if ( differ === undefined ) { differ = new diff_match_patch(); }
|
|
let beforeText = doc.getValue();
|
|
let afterText = rules.join('\n');
|
|
let diffs = differ.diff_main(beforeText, afterText);
|
|
doc.startOperation();
|
|
let i = diffs.length,
|
|
iedit = beforeText.length;
|
|
while ( i-- ) {
|
|
let diff = diffs[i];
|
|
if ( diff[0] === 0 ) {
|
|
iedit -= diff[1].length;
|
|
continue;
|
|
}
|
|
let end = doc.posFromIndex(iedit);
|
|
if ( diff[0] === 1 ) {
|
|
doc.replaceRange(diff[1], end, end);
|
|
continue;
|
|
}
|
|
/* diff[0] === -1 */
|
|
iedit -= diff[1].length;
|
|
let beg = doc.posFromIndex(iedit);
|
|
doc.replaceRange('', beg, end);
|
|
}
|
|
doc.endOperation();
|
|
}
|
|
cleanEditText = mergeView.editor().getValue().trim();
|
|
cleanEditToken = mergeView.editor().changeGeneration();
|
|
if ( clearHistory ) {
|
|
mergeView.editor().clearHistory();
|
|
}
|
|
};
|
|
|
|
/******************************************************************************/
|
|
|
|
var filterRules = function(key) {
|
|
let rules = unfilteredRules[key].rules;
|
|
let filter = uDom.nodeFromSelector('#ruleFilter input').value;
|
|
if ( filter !== '' ) {
|
|
rules = rules.slice();
|
|
let i = rules.length;
|
|
while ( i-- ) {
|
|
if ( rules[i].indexOf(filter) === -1 ) {
|
|
rules.splice(i, 1);
|
|
}
|
|
}
|
|
}
|
|
return rules;
|
|
};
|
|
|
|
/******************************************************************************/
|
|
|
|
var renderRules = (function() {
|
|
let firstVisit = true;
|
|
let reIsSwitchRule = /^[a-z-]+: /;
|
|
|
|
// Switches always listed at the top.
|
|
let customSort = (a, b) => {
|
|
let aIsSwitch = reIsSwitchRule.test(a);
|
|
if ( reIsSwitchRule.test(b) === aIsSwitch ) {
|
|
return a.localeCompare(b);
|
|
}
|
|
return aIsSwitch ? -1 : 1;
|
|
};
|
|
|
|
return function(details) {
|
|
details.permanentRules.sort(customSort);
|
|
details.sessionRules.sort(customSort);
|
|
unfilteredRules.orig.rules = details.permanentRules;
|
|
unfilteredRules.edit.rules = details.sessionRules;
|
|
rulesToDoc(firstVisit);
|
|
if ( firstVisit ) {
|
|
firstVisit = false;
|
|
mergeView.editor().execCommand('goNextDiff');
|
|
}
|
|
onTextChanged(true);
|
|
};
|
|
})();
|
|
|
|
/******************************************************************************/
|
|
|
|
var applyDiff = function(permanent, toAdd, toRemove) {
|
|
messaging.send(
|
|
'dashboard',
|
|
{
|
|
what: 'modifyRuleset',
|
|
permanent: permanent,
|
|
toAdd: toAdd,
|
|
toRemove: toRemove
|
|
},
|
|
renderRules
|
|
);
|
|
};
|
|
|
|
/******************************************************************************/
|
|
|
|
// CodeMirror quirk: sometimes fromStart.ch and/or toStart.ch is undefined.
|
|
// When this happens, use 0.
|
|
|
|
mergeView.options.revertChunk = function(
|
|
mv,
|
|
from, fromStart, fromEnd,
|
|
to, toStart, toEnd
|
|
) {
|
|
// https://github.com/gorhill/uBlock/issues/3611
|
|
if ( document.body.getAttribute('dir') === 'rtl' ) {
|
|
let tmp = from; from = to; to = tmp;
|
|
tmp = fromStart; fromStart = toStart; toStart = tmp;
|
|
tmp = fromEnd; fromEnd = toEnd; toEnd = tmp;
|
|
}
|
|
if ( typeof fromStart.ch !== 'number' ) { fromStart.ch = 0; }
|
|
if ( fromEnd.ch !== 0 ) { fromEnd.line += 1; }
|
|
let toAdd = from.getRange(
|
|
{ line: fromStart.line, ch: 0 },
|
|
{ line: fromEnd.line, ch: 0 }
|
|
);
|
|
if ( typeof toStart.ch !== 'number' ) { toStart.ch = 0; }
|
|
if ( toEnd.ch !== 0 ) { toEnd.line += 1; }
|
|
let toRemove = to.getRange(
|
|
{ line: toStart.line, ch: 0 },
|
|
{ line: toEnd.line, ch: 0 }
|
|
);
|
|
applyDiff(from === mv.editor(), toAdd, toRemove);
|
|
};
|
|
|
|
/******************************************************************************/
|
|
|
|
function handleImportFilePicker() {
|
|
let fileReaderOnLoadHandler = function() {
|
|
if ( typeof this.result !== 'string' || this.result === '' ) { return; }
|
|
// https://github.com/chrisaljoudi/uBlock/issues/757
|
|
// Support RequestPolicy rule syntax
|
|
let result = this.result;
|
|
let matches = /\[origins-to-destinations\]([^\[]+)/.exec(result);
|
|
if ( matches && matches.length === 2 ) {
|
|
result = matches[1].trim()
|
|
.replace(/\|/g, ' ')
|
|
.replace(/\n/g, ' * noop\n');
|
|
}
|
|
applyDiff(false, result, '');
|
|
};
|
|
let file = this.files[0];
|
|
if ( file === undefined || file.name === '' ) { return; }
|
|
if ( file.type.indexOf('text') !== 0 ) { return; }
|
|
let fr = new FileReader();
|
|
fr.onload = fileReaderOnLoadHandler;
|
|
fr.readAsText(file);
|
|
}
|
|
|
|
/******************************************************************************/
|
|
|
|
var startImportFilePicker = function() {
|
|
let input = document.getElementById('importFilePicker');
|
|
// Reset to empty string, this will ensure an change event is properly
|
|
// triggered if the user pick a file, even if it is the same as the last
|
|
// one picked.
|
|
input.value = '';
|
|
input.click();
|
|
};
|
|
|
|
/******************************************************************************/
|
|
|
|
function exportUserRulesToFile() {
|
|
let filename = vAPI.i18n('rulesDefaultFileName')
|
|
.replace('{{datetime}}', uBlockDashboard.dateNowToSensibleString())
|
|
.replace(/ +/g, '_');
|
|
vAPI.download({
|
|
url: 'data:text/plain,' + encodeURIComponent(
|
|
mergeView.leftOriginal().getValue().trim() + '\n'
|
|
),
|
|
filename: filename,
|
|
saveAs: true
|
|
});
|
|
}
|
|
|
|
/******************************************************************************/
|
|
|
|
var onFilterChanged = (function() {
|
|
let timer,
|
|
overlay = null,
|
|
last = '';
|
|
|
|
let process = function() {
|
|
timer = undefined;
|
|
if ( mergeView.editor().isClean(cleanEditToken) === false ) { return; }
|
|
let filter = uDom.nodeFromSelector('#ruleFilter input').value;
|
|
if ( filter === last ) { return; }
|
|
last = filter;
|
|
if ( overlay !== null ) {
|
|
mergeView.leftOriginal().removeOverlay(overlay);
|
|
mergeView.editor().removeOverlay(overlay);
|
|
overlay = null;
|
|
}
|
|
if ( filter !== '' ) {
|
|
overlay = updateOverlay(filter);
|
|
mergeView.leftOriginal().addOverlay(overlay);
|
|
mergeView.editor().addOverlay(overlay);
|
|
}
|
|
rulesToDoc(true);
|
|
};
|
|
|
|
return function() {
|
|
if ( timer !== undefined ) { clearTimeout(timer); }
|
|
timer = vAPI.setTimeout(process, 773);
|
|
};
|
|
})();
|
|
|
|
/******************************************************************************/
|
|
|
|
var onTextChanged = (function() {
|
|
let timer;
|
|
|
|
let process = function(now) {
|
|
timer = undefined;
|
|
let isClean = mergeView.editor().isClean(cleanEditToken);
|
|
let diff = document.getElementById('diff');
|
|
if (
|
|
now &&
|
|
isClean === false &&
|
|
mergeView.editor().getValue().trim() === cleanEditText
|
|
) {
|
|
cleanEditToken = mergeView.editor().changeGeneration();
|
|
isClean = true;
|
|
}
|
|
diff.classList.toggle('editing', isClean === false);
|
|
diff.classList.toggle('dirty', mergeView.leftChunks().length !== 0);
|
|
document.getElementById('editSaveButton').classList.toggle(
|
|
'disabled',
|
|
isClean
|
|
);
|
|
let input = document.querySelector('#ruleFilter input');
|
|
if ( isClean ) {
|
|
input.removeAttribute('disabled');
|
|
CodeMirror.commands.save = undefined;
|
|
} else {
|
|
input.setAttribute('disabled', '');
|
|
CodeMirror.commands.save = editSaveHandler;
|
|
}
|
|
};
|
|
|
|
return function(now) {
|
|
if ( timer !== undefined ) { clearTimeout(timer); }
|
|
timer = now ? process(now) : vAPI.setTimeout(process, 57);
|
|
};
|
|
})();
|
|
|
|
/******************************************************************************/
|
|
|
|
var revertAllHandler = function() {
|
|
let toAdd = [], toRemove = [];
|
|
let left = mergeView.leftOriginal(),
|
|
edit = mergeView.editor();
|
|
for ( let chunk of mergeView.leftChunks() ) {
|
|
let addedLines = left.getRange(
|
|
{ line: chunk.origFrom, ch: 0 },
|
|
{ line: chunk.origTo, ch: 0 }
|
|
);
|
|
let removedLines = edit.getRange(
|
|
{ line: chunk.editFrom, ch: 0 },
|
|
{ line: chunk.editTo, ch: 0 }
|
|
);
|
|
toAdd.push(addedLines.trim());
|
|
toRemove.push(removedLines.trim());
|
|
}
|
|
applyDiff(false, toAdd.join('\n'), toRemove.join('\n'));
|
|
};
|
|
|
|
/******************************************************************************/
|
|
|
|
var commitAllHandler = function() {
|
|
let toAdd = [], toRemove = [];
|
|
let left = mergeView.leftOriginal(),
|
|
edit = mergeView.editor();
|
|
for ( let chunk of mergeView.leftChunks() ) {
|
|
let addedLines = edit.getRange(
|
|
{ line: chunk.editFrom, ch: 0 },
|
|
{ line: chunk.editTo, ch: 0 }
|
|
);
|
|
let removedLines = left.getRange(
|
|
{ line: chunk.origFrom, ch: 0 },
|
|
{ line: chunk.origTo, ch: 0 }
|
|
);
|
|
toAdd.push(addedLines.trim());
|
|
toRemove.push(removedLines.trim());
|
|
}
|
|
applyDiff(true, toAdd.join('\n'), toRemove.join('\n'));
|
|
};
|
|
|
|
/******************************************************************************/
|
|
|
|
var editSaveHandler = function() {
|
|
let editor = mergeView.editor();
|
|
let editText = editor.getValue().trim();
|
|
if ( editText === cleanEditText ) {
|
|
onTextChanged(true);
|
|
return;
|
|
}
|
|
if ( differ === undefined ) { differ = new diff_match_patch(); }
|
|
let toAdd = [], toRemove = [];
|
|
let diffs = differ.diff_main(cleanEditText, editText);
|
|
for ( let diff of diffs ) {
|
|
if ( diff[0] === 1 ) {
|
|
toAdd.push(diff[1]);
|
|
} else if ( diff[0] === -1 ) {
|
|
toRemove.push(diff[1]);
|
|
}
|
|
}
|
|
applyDiff(false, toAdd.join(''), toRemove.join(''));
|
|
};
|
|
|
|
/******************************************************************************/
|
|
|
|
self.cloud.onPush = function() {
|
|
return mergeView.leftOriginal().getValue().trim();
|
|
};
|
|
|
|
self.cloud.onPull = function(data, append) {
|
|
if ( typeof data !== 'string' ) { return; }
|
|
applyDiff(
|
|
false,
|
|
data,
|
|
append ? '' : mergeView.editor().getValue().trim()
|
|
);
|
|
};
|
|
|
|
/******************************************************************************/
|
|
|
|
messaging.send('dashboard', { what: 'getRules' }, renderRules);
|
|
|
|
// Handle user interaction
|
|
uDom('#importButton').on('click', startImportFilePicker);
|
|
uDom('#importFilePicker').on('change', handleImportFilePicker);
|
|
uDom('#exportButton').on('click', exportUserRulesToFile);
|
|
uDom('#revertButton').on('click', revertAllHandler);
|
|
uDom('#commitButton').on('click', commitAllHandler);
|
|
uDom('#editSaveButton').on('click', editSaveHandler);
|
|
uDom('#ruleFilter input').on('input', onFilterChanged);
|
|
|
|
// https://groups.google.com/forum/#!topic/codemirror/UQkTrt078Vs
|
|
mergeView.editor().on('updateDiff', function() { onTextChanged(); });
|
|
|
|
/******************************************************************************/
|
|
|
|
})();
|
|
|