|
|
|
|
@ -1,20 +1,8 @@
|
|
|
|
|
const fs = require('fs');
|
|
|
|
|
const mkdirp = require('mkdirp');
|
|
|
|
|
const path = require('path');
|
|
|
|
|
const jdenticon = require('jdenticon');
|
|
|
|
|
|
|
|
|
|
// Icon config
|
|
|
|
|
jdenticon.config = {
|
|
|
|
|
lightness: {
|
|
|
|
|
color: [0.40, 0.80],
|
|
|
|
|
grayscale: [0.30, 0.90],
|
|
|
|
|
},
|
|
|
|
|
saturation: {
|
|
|
|
|
color: 0.50,
|
|
|
|
|
grayscale: 0.00,
|
|
|
|
|
},
|
|
|
|
|
backColor: '#86444400',
|
|
|
|
|
};
|
|
|
|
|
const Identicon = require('identicon.js');
|
|
|
|
|
const sha224 = require('js-sha512').sha512_224;
|
|
|
|
|
|
|
|
|
|
const { app } = require('electron').remote;
|
|
|
|
|
|
|
|
|
|
@ -22,34 +10,15 @@ const userDataPath = app.getPath('userData');
|
|
|
|
|
const PATH = path.join(userDataPath, 'profileImages');
|
|
|
|
|
mkdirp.sync(PATH);
|
|
|
|
|
|
|
|
|
|
function hashCode(s) {
|
|
|
|
|
let h = 0;
|
|
|
|
|
for(let i = 0; i < s.length; i += 1)
|
|
|
|
|
h = Math.imul(31, h) + s.charCodeAt(i) | 0;
|
|
|
|
|
|
|
|
|
|
return h;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
const hasImage = pubKey => fs.existsSync(getImagePath(pubKey));
|
|
|
|
|
|
|
|
|
|
const getImagePath = pubKey => `${PATH}/${pubKey}.png`;
|
|
|
|
|
const getOrCreateImagePath = pubKey => {
|
|
|
|
|
const imagePath = getImagePath(pubKey);
|
|
|
|
|
|
|
|
|
|
// If the image doesn't exist then create it
|
|
|
|
|
if (!hasImage(pubKey)) {
|
|
|
|
|
/*
|
|
|
|
|
We hash the pubKey and then pass that into jdenticon
|
|
|
|
|
This is because if we pass pubKey directly,
|
|
|
|
|
jdenticon trims the string and then generates a hash
|
|
|
|
|
meaning public keys with the same characters at the beginning
|
|
|
|
|
will get the same images
|
|
|
|
|
*/
|
|
|
|
|
const png = jdenticon.toPng(hashCode(pubKey), 50, 0.12);
|
|
|
|
|
fs.writeFileSync(imagePath, png);
|
|
|
|
|
}
|
|
|
|
|
if (!hasImage(pubKey))
|
|
|
|
|
return generateImage(pubKey);
|
|
|
|
|
|
|
|
|
|
return imagePath;
|
|
|
|
|
return getImagePath(pubKey);
|
|
|
|
|
};
|
|
|
|
|
|
|
|
|
|
const removeImage = pubKey => {
|
|
|
|
|
@ -58,10 +27,38 @@ const removeImage = pubKey => {
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
const removeImagesNotInArray = pubKeyArray => {
|
|
|
|
|
fs.readdirSync(PATH)
|
|
|
|
|
// Get all files that end with png
|
|
|
|
|
.filter(file => file.includes('.png'))
|
|
|
|
|
// Strip the extension
|
|
|
|
|
.map(i => path.basename(i, '.png'))
|
|
|
|
|
// Get any file that is not in the pubKeyArray
|
|
|
|
|
.filter(i => !pubKeyArray.includes(i))
|
|
|
|
|
// Remove them
|
|
|
|
|
.forEach(i => removeImage(i));
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
const generateImage = pubKey => {
|
|
|
|
|
const imagePath = getImagePath(pubKey);
|
|
|
|
|
|
|
|
|
|
/*
|
|
|
|
|
We hash the pubKey and then pass that into Identicon.
|
|
|
|
|
This is to avoid getting the same image
|
|
|
|
|
if 2 public keys start with the same 15 characters.
|
|
|
|
|
*/
|
|
|
|
|
const png = new Identicon(sha224(pubKey), {
|
|
|
|
|
margin: 0.2,
|
|
|
|
|
}).toString();
|
|
|
|
|
fs.writeFileSync(imagePath, png, 'base64');
|
|
|
|
|
return imagePath
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
module.exports = {
|
|
|
|
|
generateImage,
|
|
|
|
|
getOrCreateImagePath,
|
|
|
|
|
getImagePath,
|
|
|
|
|
hasImage,
|
|
|
|
|
removeImage,
|
|
|
|
|
hashCode,
|
|
|
|
|
removeImagesNotInArray,
|
|
|
|
|
};
|
|
|
|
|
|
Mikunj
6 years ago