uBlock/js/asset-updater.js
2014-07-25 16:12:20 -04:00

227 lines
7 KiB
JavaScript

/*******************************************************************************
µ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, µBlock */
/******************************************************************************/
// Asset update manager
µBlock.assetUpdater = (function() {
/******************************************************************************/
var getUpdateList = function(callback) {
var localChecksumsText = '';
var remoteChecksumsText = '';
var compareChecksums = function() {
var parseChecksumsText = function(text) {
var result = {};
var lines = text.split(/\n+/);
var i = lines.length;
var fields;
while ( i-- ) {
fields = lines[i].trim().split(/\s+/);
if ( fields.length !== 2 ) {
continue;
}
result[fields[1]] = fields[0];
}
return result;
};
if ( remoteChecksumsText === 'Error' || localChecksumsText === 'Error' ) {
remoteChecksumsText = localChecksumsText = '';
}
var localAssetChecksums = parseChecksumsText(localChecksumsText);
var remoteAssetChecksums = parseChecksumsText(remoteChecksumsText);
var toUpdate = {};
var path;
for ( path in remoteAssetChecksums ) {
if ( !remoteAssetChecksums.hasOwnProperty(path) ) {
continue;
}
if ( localAssetChecksums[path] === undefined ) {
toUpdate[path] = {
status: 'Added',
remoteChecksum: remoteAssetChecksums[path],
localChecksum: ''
};
continue;
}
if ( localAssetChecksums[path] === remoteAssetChecksums[path] ) {
toUpdate[path] = {
status: 'Unchanged',
remoteChecksum: remoteAssetChecksums[path],
localChecksum: localAssetChecksums[path]
};
continue;
}
toUpdate[path] = {
status: 'Changed',
remoteChecksum: remoteAssetChecksums[path],
localChecksum: localAssetChecksums[path]
};
}
for ( path in localAssetChecksums ) {
if ( !localAssetChecksums.hasOwnProperty(path) ) {
continue;
}
if ( remoteAssetChecksums[path] === undefined ) {
toUpdate[path] = {
status: 'Removed',
remoteChecksum: '',
localChecksum: localAssetChecksums[path]
};
}
}
callback({ 'list': toUpdate });
};
var validateChecksums = function(details) {
if ( details.error || details.content === '' ) {
return 'Error';
}
if ( /^(?:[0-9a-f]{32}\s+\S+(\s+|$))+/.test(details.content) ) {
return details.content;
}
return 'Error';
};
var onLocalChecksumsLoaded = function(details) {
localChecksumsText = validateChecksums(details);
if ( remoteChecksumsText !== '' ) {
compareChecksums();
}
};
var onRemoteChecksumsLoaded = function(details) {
remoteChecksumsText = validateChecksums(details);
if ( localChecksumsText !== '' ) {
compareChecksums();
}
};
µBlock.assets.getRepo('assets/checksums.txt', onRemoteChecksumsLoaded);
µBlock.assets.get('assets/checksums.txt', onLocalChecksumsLoaded);
};
/******************************************************************************/
// If `list` is null, it will be fetched internally.
var update = function(list, callback) {
var assetChangedCount = 0;
var assetProcessedCount;
var updatedAssetChecksums = [];
var onCompleted = function() {
var details = {
what: 'allLocalAssetsUpdated',
changedCount: assetChangedCount
};
callback(details);
µBlock.messaging.announce(details);
};
var doCountdown = function() {
assetProcessedCount -= 1;
if ( assetProcessedCount > 0 ) {
return;
}
µBlock.assets.put(
'assets/checksums.txt',
updatedAssetChecksums.join('\n'),
onCompleted
);
chrome.storage.local.set({ 'assetsUpdateTimestamp': Date.now() });
};
var assetUpdated = function(details) {
var path = details.path;
var entry = list[path];
if ( details.error ) {
updatedAssetChecksums.push(entry.localChecksum + ' ' + path);
} else {
updatedAssetChecksums.push(entry.remoteChecksum + ' ' + path);
assetChangedCount += 1;
}
doCountdown();
};
var processList = function() {
assetProcessedCount = Object.keys(list).length;
if ( assetProcessedCount === 0 ) {
onCompleted();
return;
}
var entry;
var details = { path: '', md5: '' };
for ( var path in list ) {
if ( list.hasOwnProperty(path) === false ) {
continue;
}
entry = list[path];
if ( entry.status === 'Added' || entry.status === 'Changed' ) {
details.path = path;
details.md5 = entry.remoteChecksum;
µBlock.assets.update(details, assetUpdated);
continue;
}
if ( entry.status === 'Unchanged' ) {
updatedAssetChecksums.push(entry.localChecksum + ' ' + path);
}
doCountdown();
}
};
var listLoaded = function(details) {
list = details.list;
processList();
};
// Purge obsolete external assets
µBlock.assets.purge(/^https?:\/\/.+$/, Date.now() - µBlock.updateAssetsEvery);
if ( list ) {
processList();
} else {
getUpdateList(listLoaded);
}
};
/******************************************************************************/
// Export API
return {
'getList': getUpdateList,
'update': update
};
/******************************************************************************/
})();
/******************************************************************************/