@abtnode/util 1.7.23 → 1.7.24

package/lib/user-avatar.js

@@ -0,0 +1,69 @@
+const path = require('path');
+const fs = require('fs-extra');
+
+const { USER_AVATAR_DIR, USER_AVATAR_URL_PREFIX } = require('@abtnode/constant');
+
+const md5 = require('./md5');
+
+const getAvatarFile = (dataDir, fileName) =>
+  path.join(dataDir, USER_AVATAR_DIR, fileName.substring(0, 2), fileName.substring(2));
+
+/**
+ * save base64 avatar data to storage and return url
+ */
+const extractUserAvatar = async (avatar, { dataDir } = {}) => {
+  if (!avatar) {
+    return avatar;
+  }
+
+  const regex = /^data:image\/(\w+);base64,/;
+  const match = regex.exec(avatar);
+
+  if (!match) {
+    return avatar;
+  }
+
+  const fileData = Buffer.from(avatar.replace(regex, ''), 'base64');
+  const fileName = `${md5(fileData)}.${match[1]}`;
+
+  const avatarFile = getAvatarFile(dataDir, fileName);
+  await fs.outputFile(avatarFile, fileData);
+
+  const url = `${USER_AVATAR_URL_PREFIX}/${fileName}`;
+
+  return url;
+};
+
+/**
+ * generate base64 avatar data to from file by url
+ */
+const parseUserAvatar = async (avatar, { dataDir } = {}) => {
+  if (!avatar) {
+    return avatar;
+  }
+
+  if (!avatar.startsWith(USER_AVATAR_URL_PREFIX)) {
+    return avatar;
+  }
+
+  const fileName = avatar.split('/').slice(-1)[0];
+
+  const avatarFile = getAvatarFile(dataDir, fileName);
+
+  if (!fs.existsSync(avatarFile)) {
+    return avatar;
+  }
+
+  const buf = await fs.promises.readFile(avatarFile);
+  const extname = path.extname(fileName).substring(1);
+
+  const data = `data:image/${extname};base64,${buf.toString('base64')}`;
+
+  return data;
+};
+
+module.exports = {
+  extractUserAvatar,
+  parseUserAvatar,
+  getAvatarFile,
+};

package/package.json

@@ -3,7 +3,7 @@
   "publishConfig": {
     "access": "public"
   },
-  "version": "1.7.23",
+  "version": "1.7.24",
   "description": "ArcBlock's JavaScript utility",
   "main": "lib/index.js",
   "files": [
@@ -18,6 +18,7 @@
   "author": "polunzh <polunzh@gmail.com> (http://github.com/polunzh)",
   "license": "MIT",
   "dependencies": {
+    "@abtnode/constant": "1.7.24",
     "@ocap/mcrypto": "^1.16.15",
     "@ocap/util": "^1.16.15",
     "@ocap/wallet": "^1.16.15",
@@ -54,5 +55,5 @@
     "fs-extra": "^10.0.1",
     "jest": "^27.4.5"
   },
-  "gitHead": "aa13838b7597bce11851d0b902575f861aa1ab34"
+  "gitHead": "904178e4dd0cd14c2a0a8b9131708c7ec68351e1"
 }