@certik/skynet 0.7.17 → 0.8.0

package/s3.js

@@ -84,10 +84,39 @@ async function deleteFile(bucketName, key, verbose = false) {
   }
 }
 
+async function listKeys(bucketname, prefix, continuationToken) {
+  const s3 = getS3();
+
+  const params = {
+    Bucket: bucketname,
+    Prefix: prefix,
+    ContinuationToken: continuationToken,
+  };
+
+  let data = null;
+  try {
+    data = await s3.listObjectsV2(params).promise();
+  } catch (err) {
+    if (err.statusCode && err.statusCode === 400) {
+      return null;
+    }
+
+    throw `unable to list keys with prefix ${prefix}: ${err}`;
+  }
+
+  let res = { keys: data.Contents.map(({ Key }) => Key) };
+  if (data.IsTruncated) {
+    res.continuationToken = data.NextContinuationToken;
+  }
+
+  return res;
+}
+
 module.exports = {
   getS3,
   hasFile,
   readFile,
   writeFile,
-  deleteFile
+  deleteFile,
+  listKeys
 };

package/selector.js

@@ -9,9 +9,9 @@ function getSpaces(num) {
 
 function getSelectorDesc(selector) {
   return Object.keys(selector)
-    .map(name => {
+    .map((name) => {
       return `          --${name}${getSpaces(14 - name.length)}${
-        selector[name].desc
+        selector[name].desc || selector[name].description || ""
       }`;
     })
     .join("\n");
@@ -21,7 +21,7 @@ function getSelectorFlags(selector) {
   return Object.keys(selector).reduce((acc, name) => {
     const flag = {
       type: selector[name].type || "string",
-      isRequired: true
+      isRequired: true,
     };
 
     if (selector[name].default) {
@@ -40,7 +40,7 @@ function getSelectorFlags(selector) {
 function toSelectorString(selectorFlags, delim = ",") {
   return Object.keys(selectorFlags)
     .sort() // deterministic
-    .map(flag => {
+    .map((flag) => {
       return `${flag}=${selectorFlags[flag]}`;
     })
     .join(delim);
@@ -49,5 +49,5 @@ function toSelectorString(selectorFlags, delim = ",") {
 module.exports = {
   getSelectorDesc,
   getSelectorFlags,
-  toSelectorString
+  toSelectorString,
 };

package/slack.js

@@ -1,46 +1,62 @@
-const { WebClient, LogLevel } = require("@slack/web-api");
+const { WebClient } = require("@slack/web-api");
 
-const token = process.env.SLACK_TOKEN;
+function getToken() {
+  return process.env.SKYNET_SLACK_TOKEN;
+}
 
-// WebClient insantiates a client that can call API methods
-const client = new WebClient(token, {
-  logLevel: LogLevel.DEBUG,
-});
+function getClient() {
+  const token = getToken();
 
-async function findConversation(name) {
-  try {
-    // Call the conversations.list method using the built-in WebClient
-    const result = await client.conversations.list({
-      token: token,
-      limit: 1000,
-    });
+  if (!token) {
+    throw new Error("Cannot communicate with slack due to missing slack token: SKYNET_SLACK_TOKEN");
+  }
+
+  const client = new WebClient(token);
+
+  return client;
+}
+
+async function findConversation(client, name) {
+  const { conversations } = client;
+
+  // Call the conversations.list method using the built-in WebClient
+  let result = await conversations.list({
+    limit: 1000,
+  });
 
-    for (const channel of result.channels) {
-      if (channel.name === name) {
-        const conversationId = channel.id;
+  for (const channel of result.channels) {
+    if (channel.name === name) {
+      const conversationId = channel.id;
 
-        console.log("Found conversation ID: " + conversationId);
-        return conversationId;
-      }
+      return conversationId;
     }
-  } catch (error) {
-    console.error(error);
   }
+
+  return null;
 }
 
-async function publishMessage(id, text) {
+async function postMessage(channel, message) {
   try {
-    const result = await client.chat.postMessage({
-      token: token,
-      channel: id,
-      text: text,
-      // You could also use a blocks[] array to send richer content
-    });
+    const client = getClient();
+
+    const conversationId = await findConversation(client, channel);
 
-    console.log(result);
+    let post = {};
+
+    if (typeof message === "string") {
+      post.text = message;
+    } else {
+      post = message;
+    }
+
+    await client.chat.postMessage({
+      channel: conversationId,
+      ...post,
+    });
   } catch (error) {
-    console.error(error);
+    // no blocking
+    console.error("failed to post slack message", error);
   }
 }
 
-module.exports = { publishMessage, findConversation };
+module.exports = { postMessage };

package/token.js

@@ -5,24 +5,26 @@ const BigNumber = require("bignumber.js");
 const TOKEN_PRICE_TABLE_NAME = `skynet-${getEnvironment()}-token-prices`;
 const TOKEN_PRICE_CACHE = {};
 
-async function getTokenPriceAt(tokenAddress, timestamp) {
+async function getTokenPriceAt(tokenAddress, timestamp, useCache = true) {
   // to avoid huge amount of dynamodb query
   // "ceil" timestamp to the closest proximate timestamp
   // 100_000 seconds ~= a bit more than 1 day 86_400 seconds
   // and use cache
   const proximateTimestamp = timestamp + 100_000 - (timestamp % 100_000);
 
-  if (TOKEN_PRICE_CACHE[tokenAddress] && TOKEN_PRICE_CACHE[tokenAddress][proximateTimestamp]) {
+  if (useCache && TOKEN_PRICE_CACHE[tokenAddress] && TOKEN_PRICE_CACHE[tokenAddress][proximateTimestamp]) {
     return TOKEN_PRICE_CACHE[tokenAddress][proximateTimestamp];
   }
 
-  if (!TOKEN_PRICE_CACHE[tokenAddress]) {
-    TOKEN_PRICE_CACHE[tokenAddress] = {};
-  }
-
   const price = await getPriceClosestTo(TOKEN_PRICE_TABLE_NAME, tokenAddress, proximateTimestamp);
 
-  TOKEN_PRICE_CACHE[tokenAddress][proximateTimestamp] = price;
+  if (useCache) {
+    if (!TOKEN_PRICE_CACHE[tokenAddress]) {
+      TOKEN_PRICE_CACHE[tokenAddress] = {};
+    }
+
+    TOKEN_PRICE_CACHE[tokenAddress][proximateTimestamp] = price;
+  }
 
   return price;
 }