@abtnode/util 1.7.26 → 1.8.1

package/lib/axios.js

@@ -1,6 +1,6 @@
 // set proxy to false to fix bug with HTTPS_PROXY environment
 // https://github.com/axios/axios/issues/3384
-const axios = require('axios');
+const { default: axios } = require('axios');
 
 axios.defaults.proxy = false;
 

package/lib/download-file.js

@@ -7,18 +7,56 @@ const axios = require('./axios');
 const CANCEL = '__cancel__';
 
 /**
- * download file
- * @param {string} url
- * @param {string} dest directory
- * @param {Number} timeout unit: ms
- * @param {CancelCtrl} cancelCtrl
+ *
+ *
+ * @param {Stream} stream
+ * @returns {Promise<string>}
  */
-const downloadFile = async (url, dest, { timeout = 600 * 1000, cancelCtrl } = {}) => {
+async function getErrorMessageFromStream(stream) {
+  const str = await streamToString(stream);
+
+  try {
+    const json = JSON.parse(str);
+    return json.error;
+  } catch (error) {
+    return str;
+  }
+}
+
+/**
+ *
+ *
+ * @param {*} stream
+ * @return {Promise<string>}
+ */
+async function streamToString(stream) {
+  const chunks = [];
+
+  for await (const chunk of stream) {
+    chunks.push(Buffer.from(chunk));
+  }
+
+  return Buffer.concat(chunks).toString('utf-8');
+}
+
+/**
+ *
+ *
+ * @param {*} url
+ * @param {*} dest directory
+ * @param {{
+ *  timeout: number; // unit: ms
+ *  cancelCtrl: any;
+ * }} [{ timeout = 600 * 1000, cancelCtrl, data }={}]
+ * @param {{}} context
+ * @return {*}
+ */
+const downloadFile = async (url, dest, { timeout = 600 * 1000, cancelCtrl } = {}, context = {}) => {
   const CONNECTION_TIMEOUT = 20 * 1000;
   const source = CancelToken.source();
 
   try {
-    const result = await tryWithTimeout(async () => {
+    return await tryWithTimeout(async () => {
       const timer = setTimeout(() => {
         source.cancel(`Connection timeout: ${Math.ceil(CONNECTION_TIMEOUT / 1000)}s`);
       }, CONNECTION_TIMEOUT);
@@ -39,6 +77,7 @@ const downloadFile = async (url, dest, { timeout = 600 * 1000, cancelCtrl } = {}
 
       const response = await axios({
         url,
+        headers: context?.headers,
         method: 'GET',
         responseType: 'stream',
         cancelToken: source.token,
@@ -61,8 +100,6 @@ const downloadFile = async (url, dest, { timeout = 600 * 1000, cancelCtrl } = {}
 
       return dest;
     }, timeout);
-
-    return result;
   } catch (err) {
     if (fs.existsSync(dest)) {
       fs.unlinkSync(dest);
@@ -76,6 +113,11 @@ const downloadFile = async (url, dest, { timeout = 600 * 1000, cancelCtrl } = {}
     }
 
     source.cancel();
+
+    if (err?.response?.data) {
+      throw new Error(await getErrorMessageFromStream(err.response.data));
+    }
+
     throw err;
   }
 };

package/lib/public-to-store.js

@@ -0,0 +1,81 @@
+const { sign } = require('@arcblock/jwt');
+const { BlockletSource } = require('@blocklet/meta/lib/constants');
+const { WHO_CAN_ACCESS } = require('@abtnode/constant');
+const logger = require('@abtnode/logger')('@abtnode/util:public-to-store');
+
+const getWallet = require('./get-app-wallet');
+const axios = require('./axios');
+
+const getAppToken = (blocklet) =>
+  sign(blocklet.environmentObj?.BLOCKLET_APP_ID, blocklet.environmentObj?.BLOCKLET_APP_SK);
+
+const getAppId = (blocklet) => blocklet.environmentObj?.BLOCKLET_APP_ID;
+
+const getAppSK = (blocklet) => blocklet.environmentObj?.BLOCKLET_APP_SK;
+
+const getBlockletDid = (blocklet) => blocklet.meta?.bundleDid;
+
+const getAppUrl = (blocklet) => blocklet.environmentObj?.BLOCKLET_APP_URL;
+
+/**
+ * verify manages the permissions of blocklet public instance
+ * @param {*} blocklet
+ * @returns bool
+ */
+const verifyPublicInstance = (blocklet) => {
+  const { whoCanAccess = WHO_CAN_ACCESS.ALL } = blocklet.settings;
+  return blocklet.source === BlockletSource.registry && whoCanAccess === WHO_CAN_ACCESS.ALL;
+};
+
+/**
+ *
+ * @param {*} blocklet
+ * @param {*} {userDid publicToStore}
+ */
+async function handleInstanceInStore(blocklet, { userDid = null, publicToStore = true } = {}) {
+  if (!blocklet) {
+    throw new Error('blocklet argument is required');
+  }
+
+  const ownerDid = userDid || blocklet.settings?.owner?.did;
+  if (!verifyPublicInstance(blocklet)) {
+    throw new Error('no permission to set publicInstance ');
+  }
+
+  if (!ownerDid) {
+    return false;
+  }
+
+  const appToken = getAppToken(blocklet);
+  const blockletDid = getBlockletDid(blocklet);
+  const appId = getAppId(blocklet);
+  const appSK = getAppSK(blocklet);
+  const wallet = getWallet(appSK);
+  const body = {
+    appToken,
+    appId,
+    appPK: wallet.publicKey,
+    ownerDid,
+    appUrl: getAppUrl(blocklet),
+    blockletDid,
+  };
+
+  const api = axios.create({ baseURL: blocklet.deployedFrom, timeout: 1000 * 10 });
+
+  if (!publicToStore) {
+    try {
+      await api.delete(`/api/blocklet-instances/${appId}`, { data: body });
+    } catch (error) {
+      logger.error('failed to delete blocklet instance', { error });
+    }
+  } else {
+    try {
+      await api.put(`/api/blocklet-instances/${appId}`, body);
+    } catch (error) {
+      logger.error('failed to update blocklet instance', { error });
+    }
+  }
+  return true;
+}
+
+module.exports = handleInstanceInStore;

package/lib/sleep.js

@@ -1 +1,2 @@
+// eslint-disable-next-line no-promise-executor-return
 module.exports = (timeout = 0) => new Promise((resolve) => setTimeout(resolve, timeout));

package/package.json

@@ -3,7 +3,7 @@
   "publishConfig": {
     "access": "public"
   },
-  "version": "1.7.26",
+  "version": "1.8.1",
   "description": "ArcBlock's JavaScript utility",
   "main": "lib/index.js",
   "files": [
@@ -18,7 +18,10 @@
   "author": "polunzh <polunzh@gmail.com> (http://github.com/polunzh)",
   "license": "MIT",
   "dependencies": {
-    "@abtnode/constant": "1.7.26",
+    "@abtnode/constant": "1.8.1",
+    "@abtnode/logger": "1.8.1",
+    "@arcblock/jwt": "^1.17.0",
+    "@blocklet/meta": "1.8.1",
     "@ocap/mcrypto": "1.17.0",
     "@ocap/util": "1.17.0",
     "@ocap/wallet": "1.17.0",
@@ -30,6 +33,7 @@
     "find-up": "^5.0.0",
     "flush-write-stream": "^2.0.0",
     "folder-walker": "^3.2.0",
+    "form-data": "^4.0.0",
     "hasha": "^5.2.0",
     "internal-ip": "^6.1.0",
     "is-docker": "^2.1.1",
@@ -55,5 +59,5 @@
     "fs-extra": "^10.0.1",
     "jest": "^27.4.5"
   },
-  "gitHead": "b7ef9b4ddb18f7a0c3898177fe06d9cefe966566"
+  "gitHead": "c970b8a386bebd7fe6dbc8b8eedf8bd8328b4bb5"
 }