@certik/skynet 0.10.6 → 0.10.7

package/labelling.js

@@ -1,90 +1,90 @@
-const fetch = require("node-fetch");
-const { getRecordByKey, updateRecordByKey } = require("./dynamodb");
-const { getEnvironment } = require("./env");
-
-const ADDRESSES_TABLE_NAME = `skynet-${getEnvironment()}-aml-addresses`;
-const SMART_CONTRACTS_TABLE_NAME = `skynet-${getEnvironment()}-aml-smart-contracts`;
-
-async function addTagsToAddress(address, tags, verbose) {
-  let tableName = null;
-  let updateItemTags;
-
-  // Validate tags
-  tags.forEach((tag) => {
-    if (!tag.type || !tag.label) {
-      throw new Error("error: tag must be in format { type, label }");
-    }
-  });
-
-  // check if address is a smart contract
-  let queryResult = await getRecordByKey(SMART_CONTRACTS_TABLE_NAME, { address });
-
-  if (queryResult) {
-    tableName = SMART_CONTRACTS_TABLE_NAME;
-    updateItemTags = queryResult.tags ?? [];
-  } else {
-    // check if address is an EOA
-    queryResult = await getRecordByKey(ADDRESSES_TABLE_NAME, { address });
-
-    if (queryResult) {
-      tableName = ADDRESSES_TABLE_NAME;
-      updateItemTags = queryResult.tags ?? [];
-    } else {
-      // Unknown address type, inquiry endpoint to create a record
-      const url = `https://aml-inquiry.certik-skynet.com/address?address=${address}`;
-      const response = await fetch(url);
-      const json = await response.json();
-      if (!json.type) {
-        throw new Error(`Unknown address type in inquiry response: ${address}`);
-      }
-      const { type } = json;
-      tableName = type === "EOA" ? ADDRESSES_TABLE_NAME : SMART_CONTRACTS_TABLE_NAME;
-      updateItemTags = [];
-    }
-  }
-
-  if (tableName) {
-    // record available, add tag
-    const newTags = tags.filter((tag) => !updateItemTags.includes(tag));
-
-    if (newTags.length + updateItemTags.length === 0) {
-      throw new Error("error: no tags to add and none already in database");
-    }
-
-    if (newTags.length === 0) {
-      return;
-    }
-
-    updateItemTags = [...updateItemTags, ...newTags];
-
-    // keep distinct tags
-    updateItemTags = updateItemTags.filter(
-      (value, index, self) => self.findIndex((t) => t.type === value.type && t.label === value.label) === index
-    );
-
-    await updateRecordByKey(tableName, { address: address }, { tags: updateItemTags }, null, verbose);
-  }
-
-  return { tableName, updateItemTags };
-}
-
-async function queryAddressTags(address) {
-  let updateItemTags = [];
-
-  let queryResult = await getRecordByKey(SMART_CONTRACTS_TABLE_NAME, { address });
-
-  if (queryResult) {
-    updateItemTags = queryResult.tags ?? [];
-  } else {
-    // check if address is an EOA
-    queryResult = await getRecordByKey(ADDRESSES_TABLE_NAME, { address });
-
-    if (queryResult) {
-      updateItemTags = queryResult.tags ?? [];
-    }
-  }
-
-  return updateItemTags;
-}
-
-module.exports = { addTagsToAddress, queryAddressTags };
+const fetch = require("node-fetch");
+const { getRecordByKey, updateRecordByKey } = require("./dynamodb");
+const { getEnvironment } = require("./env");
+
+const ADDRESSES_TABLE_NAME = `skynet-${getEnvironment()}-aml-addresses`;
+const SMART_CONTRACTS_TABLE_NAME = `skynet-${getEnvironment()}-aml-smart-contracts`;
+
+async function addTagsToAddress(address, tags, verbose) {
+  let tableName = null;
+  let updateItemTags;
+
+  // Validate tags
+  tags.forEach((tag) => {
+    if (!tag.type || !tag.label) {
+      throw new Error("error: tag must be in format { type, label }");
+    }
+  });
+
+  // check if address is a smart contract
+  let queryResult = await getRecordByKey(SMART_CONTRACTS_TABLE_NAME, { address });
+
+  if (queryResult) {
+    tableName = SMART_CONTRACTS_TABLE_NAME;
+    updateItemTags = queryResult.tags ?? [];
+  } else {
+    // check if address is an EOA
+    queryResult = await getRecordByKey(ADDRESSES_TABLE_NAME, { address });
+
+    if (queryResult) {
+      tableName = ADDRESSES_TABLE_NAME;
+      updateItemTags = queryResult.tags ?? [];
+    } else {
+      // Unknown address type, inquiry endpoint to create a record
+      const url = `https://aml-inquiry.certik-skynet.com/address?address=${address}`;
+      const response = await fetch(url);
+      const json = await response.json();
+      if (!json.type) {
+        throw new Error(`Unknown address type in inquiry response: ${address}`);
+      }
+      const { type } = json;
+      tableName = type === "EOA" ? ADDRESSES_TABLE_NAME : SMART_CONTRACTS_TABLE_NAME;
+      updateItemTags = [];
+    }
+  }
+
+  if (tableName) {
+    // record available, add tag
+    const newTags = tags.filter((tag) => !updateItemTags.includes(tag));
+
+    if (newTags.length + updateItemTags.length === 0) {
+      throw new Error("error: no tags to add and none already in database");
+    }
+
+    if (newTags.length === 0) {
+      return;
+    }
+
+    updateItemTags = [...updateItemTags, ...newTags];
+
+    // keep distinct tags
+    updateItemTags = updateItemTags.filter(
+      (value, index, self) => self.findIndex((t) => t.type === value.type && t.label === value.label) === index
+    );
+
+    await updateRecordByKey(tableName, { address: address }, { tags: updateItemTags }, null, verbose);
+  }
+
+  return { tableName, updateItemTags };
+}
+
+async function queryAddressTags(address) {
+  let updateItemTags = [];
+
+  let queryResult = await getRecordByKey(SMART_CONTRACTS_TABLE_NAME, { address });
+
+  if (queryResult) {
+    updateItemTags = queryResult.tags ?? [];
+  } else {
+    // check if address is an EOA
+    queryResult = await getRecordByKey(ADDRESSES_TABLE_NAME, { address });
+
+    if (queryResult) {
+      updateItemTags = queryResult.tags ?? [];
+    }
+  }
+
+  return updateItemTags;
+}
+
+module.exports = { addTagsToAddress, queryAddressTags };

package/log.js

@@ -1,29 +1,29 @@
-function getLine(params) {
-  let line = "";
-
-  // Convert to string and filter out newline to tabs (AWS Athena)
-  for (let i = 0, l = params.length; i < l; i++) {
-    // Certain objects don't get converted
-    // Note using JSON.stringfy may be too slow for large objects
-    line += `${params[i]} `.replace(/\n/gm, "\t");
-  }
-
-  return line.trim();
-}
-
-function timestamp() {
-  return new Date().toISOString();
-}
-
-const inline = {
-  log: function (...args) {
-    console.log(`${timestamp()} ${getLine(args)}`);
-  },
-  error: function (...args) {
-    console.error(`${timestamp()} ${getLine(args)}`);
-  },
-};
-
-module.exports = {
-  inline,
-};
+function getLine(params) {
+  let line = "";
+
+  // Convert to string and filter out newline to tabs (AWS Athena)
+  for (let i = 0, l = params.length; i < l; i++) {
+    // Certain objects don't get converted
+    // Note using JSON.stringfy may be too slow for large objects
+    line += `${params[i]} `.replace(/\n/gm, "\t");
+  }
+
+  return line.trim();
+}
+
+function timestamp() {
+  return new Date().toISOString();
+}
+
+const inline = {
+  log: function (...args) {
+    console.log(`${timestamp()} ${getLine(args)}`);
+  },
+  error: function (...args) {
+    console.error(`${timestamp()} ${getLine(args)}`);
+  },
+};
+
+module.exports = {
+  inline,
+};

package/metric.js

@@ -1,65 +1,65 @@
-const { getDocClient } = require("./dynamodb");
-
-/* Assume table has name/timestamp/value fields */
-
-async function getMetricAt(tableName, name, timestamp) {
-  const docClient = getDocClient();
-
-  const query = await docClient
-    .query({
-      TableName: tableName,
-      KeyConditionExpression: "#name = :name and #timestamp <= :timestamp",
-      ExpressionAttributeNames: {
-        "#name": "name",
-        "#timestamp": "timestamp"
-      },
-      ExpressionAttributeValues: {
-        ":name": name,
-        ":timestamp": timestamp
-      },
-      Limit: 1,
-      ScanIndexForward: false
-    })
-    .promise();
-
-  if (query.Count > 0) {
-    const { value } = query.Items[0];
-
-    return value;
-  } else {
-    return null;
-  }
-}
-
-async function getMetricPreviousValue(tableName, name, timestamp) {
-  const docClient = getDocClient();
-
-  const query = await docClient
-    .query({
-      TableName: tableName,
-      KeyConditionExpression: "#name = :name and #timestamp < :timestamp",
-      ExpressionAttributeNames: {
-        "#timestamp": "timestamp",
-        "#name": "name"
-      },
-      ExpressionAttributeValues: {
-        ":timestamp": timestamp,
-        ":name": name
-      },
-      Limit: 1,
-      ScanIndexForward: false
-    })
-    .promise();
-
-  if (query.Count === 0) {
-    return 0;
-  } else {
-    const { value } = query.Items[0];
-    return value;
-  }
-}
-
-module.exports = {
-  getMetricAt,
-  getMetricPreviousValue
-}
+const { getDocClient } = require("./dynamodb");
+
+/* Assume table has name/timestamp/value fields */
+
+async function getMetricAt(tableName, name, timestamp) {
+  const docClient = getDocClient();
+
+  const query = await docClient
+    .query({
+      TableName: tableName,
+      KeyConditionExpression: "#name = :name and #timestamp <= :timestamp",
+      ExpressionAttributeNames: {
+        "#name": "name",
+        "#timestamp": "timestamp"
+      },
+      ExpressionAttributeValues: {
+        ":name": name,
+        ":timestamp": timestamp
+      },
+      Limit: 1,
+      ScanIndexForward: false
+    })
+    .promise();
+
+  if (query.Count > 0) {
+    const { value } = query.Items[0];
+
+    return value;
+  } else {
+    return null;
+  }
+}
+
+async function getMetricPreviousValue(tableName, name, timestamp) {
+  const docClient = getDocClient();
+
+  const query = await docClient
+    .query({
+      TableName: tableName,
+      KeyConditionExpression: "#name = :name and #timestamp < :timestamp",
+      ExpressionAttributeNames: {
+        "#timestamp": "timestamp",
+        "#name": "name"
+      },
+      ExpressionAttributeValues: {
+        ":timestamp": timestamp,
+        ":name": name
+      },
+      Limit: 1,
+      ScanIndexForward: false
+    })
+    .promise();
+
+  if (query.Count === 0) {
+    return 0;
+  } else {
+    const { value } = query.Items[0];
+    return value;
+  }
+}
+
+module.exports = {
+  getMetricAt,
+  getMetricPreviousValue
+}