provide a standard api for calculating pin data for a given public key

pull/1/head
ansuz 6 years ago
parent b8343b5483
commit cd793b295d

@ -2,20 +2,44 @@
var Pins = module.exports; var Pins = module.exports;
/* /* Accepts a reference to an object, and...
takes contents of a pinFile (UTF8 string) either a string describing which log is being processed (backwards compatibility),
and the pin file's name or a function which will log the error with all relevant data
returns an array of of channel ids which are pinned
throw errors on pin logs with invalid pin data
*/ */
Pins.calculateFromLog = function (pinFile, fileName) { var createLineHandler = Pins.createLineHandler = function (ref, errorHandler) {
var pins = {}; var fileName;
pinFile.split('\n').filter((x)=>(x)).map((l) => JSON.parse(l)).forEach((l) => { if (typeof(errorHandler) === 'string') {
fileName = errorHandler;
errorHandler = function (label, data) {
console.error(label, {
log: fileName,
data: data,
});
};
}
// passing the reference to an object allows us to overwrite accumulated pins
// make sure to get ref.pins as the result
// it's a weird API but it's faster than unpinning manually
var pins = ref.pins = {};
return function (line) {
if (!Boolean(line)) { return; }
var l;
try {
l = JSON.parse(line);
} catch (e) {
return void errorHandler('PIN_LINE_PARSE_ERROR', line);
}
if (!Array.isArray(l)) {
return void errorHandler('PIN_LINE_NOT_FORMAT_ERROR', l);
}
switch (l[0]) { switch (l[0]) {
case 'RESET': { case 'RESET': {
pins = {}; pins = ref.pins = {};
if (l[1] && l[1].length) { l[1].forEach((x) => { pins[x] = 1; }); } if (l[1] && l[1].length) { l[1].forEach((x) => { ref.pins[x] = 1; }); }
//jshint -W086 //jshint -W086
// fallthrough // fallthrough
} }
@ -28,16 +52,25 @@ Pins.calculateFromLog = function (pinFile, fileName) {
break; break;
} }
default: default:
// FIXME logging errorHandler("PIN_LINE_UNSUPPORTED_COMMAND", l);
// TODO write to the error log
/* Log.error('CORRUPTED_PIN_LOG', {
line: JSON.stringify(l),
fileName: fileName,
}); */
console.error(new Error (JSON.stringify(l) + ' ' + fileName));
} }
}); };
return Object.keys(pins); };
/*
takes contents of a pinFile (UTF8 string)
and the pin file's name
returns an array of of channel ids which are pinned
throw errors on pin logs with invalid pin data
*/
Pins.calculateFromLog = function (pinFile, fileName) {
var ref = {};
var handler = createLineHandler(ref, fileName);
pinFile.split('\n').forEach(handler);
return Object.keys(ref.pins);
}; };
// TODO refactor to include a streaming version for use in rpc.js as well // TODO refactor to include a streaming version for use in rpc.js as well

@ -16,7 +16,7 @@ const Pinned = require('./scripts/pinned');
const Saferphore = require("saferphore"); const Saferphore = require("saferphore");
const nThen = require("nthen"); const nThen = require("nthen");
const getFolderSize = require("get-folder-size"); const getFolderSize = require("get-folder-size");
const Pins = require("./lib/pins");
var RPC = module.exports; var RPC = module.exports;
@ -237,52 +237,21 @@ var loadUserPins = function (Env, publicKey, cb) {
return cb(session.channels); return cb(session.channels);
} }
// if channels aren't in memory. load them from disk var ref = {};
var pins = {}; var lineHandler = Pins.createLineHandler(ref, function (label, data) {
Log.error(label, {
var pin = function (channel) { log: publicKey,
pins[channel] = true; data: data,
}; });
});
var unpin = function (channel) {
// TODO delete?
pins[channel] = false;
};
Env.pinStore.getMessages(publicKey, function (msg) { // if channels aren't in memory. load them from disk
// handle messages... Env.pinStore.getMessages(publicKey, lineHandler, function () {
var parsed;
try {
parsed = JSON.parse(msg);
session.hasPinned = true;
switch (parsed[0]) {
case 'PIN':
parsed[1].forEach(pin);
break;
case 'UNPIN':
parsed[1].forEach(unpin);
break;
case 'RESET':
// TODO just wipe out the object?
Object.keys(pins).forEach(unpin);
if (parsed[1] && parsed[1].length) {
parsed[1].forEach(pin);
}
break;
default:
Log.warn('INVALID_STORED_MESSAGE', msg);
}
} catch (e) {
Log.warn('STORED_PARSE_ERROR', e);
}
}, function () {
// no more messages // no more messages
// only put this into the cache if it completes // only put this into the cache if it completes
session.channels = pins; session.channels = ref.pins;
cb(pins); cb(ref.pins);
}); });
}; };

Loading…
Cancel
Save