uBlock/js/whitelist.js

/*******************************************************************************

    µBlock - a Chromium browser extension to block requests.
    Copyright (C) 2014 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/uBlock
*/

/* global chrome, messaging, uDom */

/******************************************************************************/

(function() {

/******************************************************************************/

messaging.start('whitelist.js');

/******************************************************************************/

var cachedWhitelist = '';

// Could make it more fancy if needed. But speed... It's a compromise. 
var reBadHostname = /[\x00-\x08\x0b\x0c\x0e-\x1f\x21-\x2c\x2f\x3a-\x40\x5b-\x60\x7b-\x7e]/;

/******************************************************************************/

var validateHostname = function(s) {
    var hn = punycode.toASCII(s).toLowerCase();
    if ( reBadHostname.test(hn) ) {
        return '';
    }
    return hn;
};

var whitelistFromString = function(s) {
    var whitelist = {};
    var items = s.split(/\s+/);
    var item;
    for ( var i = 0; i < items.length; i++ ) {
        item = validateHostname(items[i]);
        if ( item !== '' ) {
            whitelist[item] = true;
        }
    }
    return whitelist;
};

var stringFromWhitelist = function(whitelist) {
    var s = [];
    var items = Object.keys(whitelist);
    for ( var i = 0; i < items.length; i++ ) {
        s.push(punycode.toUnicode(items[i]));
    }
    return s.sort(function(a, b) { return a.localeCompare(b); }).join('\n');
};

/******************************************************************************/

var badWhitelist = function(s) {
    return reBadHostname.test(s);
};

/******************************************************************************/

var whitelistChanged = function() {
    var s = uDom('#whitelist').val().trim();
    uDom('#whitelistApply').prop(
        'disabled',
        s === cachedWhitelist
    );
    uDom('#whitelist').toggleClass('bad', badWhitelist(s));
};

/******************************************************************************/

function renderWhitelist() {
    var onRead = function(whitelist) {
        cachedWhitelist = stringFromWhitelist(whitelist);
        uDom('#whitelist').val(cachedWhitelist);
    };
    messaging.ask({ what: 'getWhitelist' }, onRead);
};

/******************************************************************************/

var importWhitelistFromFile = function() {
    var input = uDom('<input />').attr({
        type: 'file',
        accept: 'text/plain'
    });
    var fileReaderOnLoadHandler = function() {
        var textarea = uDom('#whitelist');
        textarea.val([textarea.val(), this.result].join('\n').trim());
        whitelistChanged();
    };
    var filePickerOnChangeHandler = function() {
        input.off('change', filePickerOnChangeHandler);
        var file = this.files[0];
        if ( !file ) {
            return;
        }
        if ( file.type.indexOf('text') !== 0 ) {
            return;
        }
        var fr = new FileReader();
        fr.onload = fileReaderOnLoadHandler;
        fr.readAsText(file);
    };
    input.on('change', filePickerOnChangeHandler);
    input.trigger('click');
};

/******************************************************************************/

var exportWhitelistToFile = function() {
    chrome.downloads.download({
        'url': 'data:text/plain,' + encodeURIComponent(uDom('#whitelist').val()),
        'filename': 'my-ublock-whitelist.txt',
        'saveAs': true
    });
};

/******************************************************************************/

var whitelistApplyHandler = function() {
    cachedWhitelist = uDom('#whitelist').val().trim();
    var request = {
        what: 'setWhitelist',
        whitelist: whitelistFromString(cachedWhitelist)
    };
    messaging.tell(request);
    whitelistChanged();
};

/******************************************************************************/

uDom.onLoad(function() {
    uDom('#importWhitelistFromFile').on('click', importWhitelistFromFile);
    uDom('#exportWhitelistToFile').on('click', exportWhitelistToFile);
    uDom('#whitelist').on('input', whitelistChanged);
    uDom('#whitelistApply').on('click', whitelistApplyHandler);

    renderWhitelist();
});

/******************************************************************************/

})();