provide a standard api for calculating pin data for a given public key
This commit is contained in:
parent
b8343b5483
commit
cd793b295d
73
lib/pins.js
73
lib/pins.js
@ -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
|
||||||
|
|
||||||
|
|||||||
55
rpc.js
55
rpc.js
@ -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);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
var ref = {};
|
||||||
|
var lineHandler = Pins.createLineHandler(ref, function (label, data) {
|
||||||
|
Log.error(label, {
|
||||||
|
log: publicKey,
|
||||||
|
data: data,
|
||||||
|
});
|
||||||
|
});
|
||||||
|
|
||||||
// if channels aren't in memory. load them from disk
|
// if channels aren't in memory. load them from disk
|
||||||
var pins = {};
|
Env.pinStore.getMessages(publicKey, lineHandler, function () {
|
||||||
|
|
||||||
var pin = function (channel) {
|
|
||||||
pins[channel] = true;
|
|
||||||
};
|
|
||||||
|
|
||||||
var unpin = function (channel) {
|
|
||||||
// TODO delete?
|
|
||||||
pins[channel] = false;
|
|
||||||
};
|
|
||||||
|
|
||||||
Env.pinStore.getMessages(publicKey, function (msg) {
|
|
||||||
// handle messages...
|
|
||||||
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…
x
Reference in New Issue
Block a user