@heroku/skynet 1.4.2 → 1.5.0

package/commands/categories.js

@@ -10,8 +10,8 @@ function * run (context) {
   response = JSON.parse(response)
   cli.table(response, {
     columns: [
-      {key: 'Category'},
-      {key: 'Description'}
+      { key: 'Category' },
+      { key: 'Description' }
     ]
   })
 }

package/commands/deprovision.js

@@ -11,18 +11,18 @@ function readlines (file) {
     let users = []
 
     fs.createReadStream(file).pipe(split())
-    .on('data', function (line) {
-      line = line.trim()
-      if (line) {
-        users.push(line)
-      }
-    })
-    .on('end', function () {
-      resolve(users)
-    })
-    .on('error', function (err) {
-      reject(err)
-    })
+      .on('data', function (line) {
+        line = line.trim()
+        if (line) {
+          users.push(line)
+        }
+      })
+      .on('end', function () {
+        resolve(users)
+      })
+      .on('error', function (err) {
+        reject(err)
+      })
   })
 }
 
@@ -60,11 +60,11 @@ module.exports = {
   help: `Examples:
     $ heroku skynet:deprovision -u foo@bar.com -n "helpful suspend message" -c "spam"`,
   flags: [
-    {name: 'user', char: 'u', description: 'user to deprovision', hasValue: true},
-    {name: 'infile', char: 'i', description: 'file list of users to deprovision', hasValue: true},
-    {name: 'category', char: 'c', description: 'suspension category', hasValue: true, required: true},
-    {name: 'notes', char: 'n', description: 'suspend notes', hasValue: true, required: true},
-    {name: 'bypass', description: 'bypass the whitelist', hasValue: false, required: false}
+    { name: 'user', char: 'u', description: 'user to deprovision', hasValue: true },
+    { name: 'infile', char: 'i', description: 'file list of users to deprovision', hasValue: true },
+    { name: 'category', char: 'c', description: 'suspension category', hasValue: true, required: true },
+    { name: 'notes', char: 'n', description: 'suspend notes', hasValue: true, required: true },
+    { name: 'bypass', description: 'bypass the whitelist', hasValue: false, required: false }
   ],
   run: cli.command(co.wrap(run))
 }

package/commands/suspend/app-owner.js

@@ -11,18 +11,18 @@ function readlines (file) {
     let users = []
 
     fs.createReadStream(file).pipe(split())
-    .on('data', function (line) {
-      line = line.trim()
-      if (line) {
-        users.push(line)
-      }
-    })
-    .on('end', function () {
-      resolve(users)
-    })
-    .on('error', function (err) {
-      reject(err)
-    })
+      .on('data', function (line) {
+        line = line.trim()
+        if (line) {
+          users.push(line)
+        }
+      })
+      .on('end', function () {
+        resolve(users)
+      })
+      .on('error', function (err) {
+        reject(err)
+      })
   })
 }
 
@@ -60,11 +60,11 @@ module.exports = {
   help: `Examples:
     $ heroku skynet:suspend:app-owner -a foobar -n "helpful suspend message" -c "ddos"`,
   flags: [
-    {name: 'app', char: 'a', description: 'app that requires owner suspension', hasValue: true},
-    {name: 'infile', char: 'i', description: 'file of apps that require owner suspension', hasValue: true},
-    {name: 'category', char: 'c', description: 'suspension category', hasValue: true, required: true},
-    {name: 'notes', char: 'n', description: 'suspend notes', hasValue: true, required: true},
-    {name: 'bypass', description: 'bypass the whitelist', hasValue: false, required: false}
+    { name: 'app', char: 'a', description: 'app that requires owner suspension', hasValue: true },
+    { name: 'infile', char: 'i', description: 'file of apps that require owner suspension', hasValue: true },
+    { name: 'category', char: 'c', description: 'suspension category', hasValue: true, required: true },
+    { name: 'notes', char: 'n', description: 'suspend notes', hasValue: true, required: true },
+    { name: 'bypass', description: 'bypass the whitelist', hasValue: false, required: false }
   ],
   run: cli.command(co.wrap(run))
 }

package/commands/suspend/apps.js

@@ -18,8 +18,8 @@ module.exports = {
   help: `Examples:
     $ heroku skynet:suspend:app -a test-app -n "helpful suspend message"`,
   flags: [
-    {name: 'app', char: 'a', description: 'app to suspend', hasValue: true, required: true},
-    {name: 'notes', char: 'n', description: 'suspend notes', hasValue: true, required: true}
+    { name: 'app', char: 'a', description: 'app to suspend', hasValue: true, required: true },
+    { name: 'notes', char: 'n', description: 'suspend notes', hasValue: true, required: true }
   ],
   run: cli.command(co.wrap(run))
 }

package/commands/suspend/user.js

@@ -11,18 +11,18 @@ function readlines (file) {
     let users = []
 
     fs.createReadStream(file).pipe(split())
-    .on('data', function (line) {
-      line = line.trim()
-      if (line) {
-        users.push(line)
-      }
-    })
-    .on('end', function () {
-      resolve(users)
-    })
-    .on('error', function (err) {
-      reject(err)
-    })
+      .on('data', function (line) {
+        line = line.trim()
+        if (line) {
+          users.push(line)
+        }
+      })
+      .on('end', function () {
+        resolve(users)
+      })
+      .on('error', function (err) {
+        reject(err)
+      })
   })
 }
 
@@ -34,6 +34,7 @@ function * run (context) {
   let notes = context.flags.notes
   let category = context.flags.category
   let force = context.flags.bypass
+  let notify = !context.flags['no-notify']
 
   if (user && file) {
     throw new Error('Either --user USER or --infile must be passed, but not both')
@@ -47,7 +48,7 @@ function * run (context) {
       throw new Error('Required flag: --user USER or --infile FILE')
     }
 
-    let response = yield cli.action(`suspending ${cli.color.cyan(user)}`, skynet.suspendUser(user, notes, category, force))
+    let response = yield cli.action(`suspending ${cli.color.cyan(user)}`, skynet.suspendUser(user, notes, category, notify, force))
     response = JSON.parse(response)
     cli.log(`${response.status}. ${response.message}`)
   }
@@ -60,11 +61,12 @@ module.exports = {
   help: `Examples:
     $ heroku sudo skynet:suspend:user -u foo@bar.com -n "helpful suspend message" -c "ddos"`,
   flags: [
-    {name: 'user', char: 'u', description: 'user to suspend', hasValue: true},
-    {name: 'infile', char: 'i', description: 'file of users to suspend', hasValue: true},
-    {name: 'category', char: 'c', description: 'suspension category', hasValue: true, required: true},
-    {name: 'notes', char: 'n', description: 'suspend notes', hasValue: true, required: true},
-    {name: 'bypass', description: 'bypass the whitelist', hasValue: false, required: false}
+    { name: 'user', char: 'u', description: 'user to suspend', hasValue: true },
+    { name: 'infile', char: 'i', description: 'file of users to suspend', hasValue: true },
+    { name: 'category', char: 'c', description: 'suspension category', hasValue: true, required: true },
+    { name: 'notes', char: 'n', description: 'suspend notes', hasValue: true, required: true },
+    { name: 'bypass', description: 'bypass the whitelist', hasValue: false, required: false },
+    { name: 'no-notify', description: 'skip user suspension email notification', hasValue: false, required: false }
   ],
   run: cli.command(co.wrap(run))
 }

package/commands/unsuspend/apps.js

@@ -18,7 +18,7 @@ module.exports = {
   help: `Examples:
     $ heroku skynet:unsuspend:app -a test-app`,
   flags: [
-    {name: 'app', char: 'a', description: 'app to unsuspend', hasValue: true, required: true}
+    { name: 'app', char: 'a', description: 'app to unsuspend', hasValue: true, required: true }
   ],
   run: cli.command(co.wrap(run))
 }

package/commands/unsuspend/user.js

@@ -20,7 +20,7 @@ module.exports = {
   help: `Examples:
     $ heroku skynet:unsuspend:user -u foo@bar.com`,
   flags: [
-    {name: 'user', char: 'u', description: 'user to unsuspend', hasValue: true}
+    { name: 'user', char: 'u', description: 'user to unsuspend', hasValue: true }
   ],
   run: cli.command(co.wrap(run))
 }

package/commands/userpass/add.js

@@ -14,8 +14,8 @@ module.exports = {
   help: `Examples:
     $ heroku skynet:userpass:add -u foo@bar.com -f cant-install-abused-addons`,
   flags: [
-    {name: 'user', char: 'u', description: 'user to apply userpass flag to', hasValue: true, required: true},
-    {name: 'flag', char: 'f', description: 'flag to add to the given user', hasValue: true, required: true}
+    { name: 'user', char: 'u', description: 'user to apply userpass flag to', hasValue: true, required: true },
+    { name: 'flag', char: 'f', description: 'flag to add to the given user', hasValue: true, required: true }
   ],
   run: cli.command(co.wrap(run))
 }

package/commands/userpass/remove.js

@@ -14,8 +14,8 @@ module.exports = {
   help: `Examples:
     $ heroku skynet:userpass:remove -u foo@bar.com -f cant-install-abused-addons`,
   flags: [
-    {name: 'user', char: 'u', description: 'user to remove user_pass from', hasValue: true, required: true},
-    {name: 'flag', char: 'f', description: 'flag to remove from given user', hasValue: true, required: true}
+    { name: 'user', char: 'u', description: 'user to remove user_pass from', hasValue: true, required: true },
+    { name: 'flag', char: 'f', description: 'flag to remove from given user', hasValue: true, required: true }
   ],
   run: cli.command(co.wrap(run))
 }

package/commands/whitelist/add.js

@@ -0,0 +1,38 @@
+let cli = require('heroku-cli-util')
+let co = require('co')
+let SkynetAPI = require('../../lib/skynet')
+
+function * run (context) {
+  const skynet = new SkynetAPI(context.auth.password)
+  yield cli.action(
+    `adding ${cli.color.cyan(
+      context.flags.value
+    )} to whitelist with ${cli.color.cyan(context.flags.notes)}`,
+    skynet.addWhitelist(context.flags.value, context.flags.notes)
+  )
+}
+
+module.exports = {
+  topic: 'skynet',
+  command: 'whitelist:add',
+  description: 'adds an app or user to the whitelist',
+  help: `Examples:
+    $ heroku skynet:whitelist:add -v foo@bar.com -n Additional info`,
+  flags: [
+    {
+      name: 'value',
+      char: 'v',
+      description: 'app name or user email to whitelist',
+      hasValue: true,
+      required: true
+    },
+    {
+      name: 'notes',
+      char: 'n',
+      description: 'additional information for the whitelist entry',
+      hasValue: true,
+      required: false
+    }
+  ],
+  run: cli.command(co.wrap(run))
+}

package/commands/whitelist/remove.js

@@ -0,0 +1,29 @@
+let cli = require('heroku-cli-util')
+let co = require('co')
+let SkynetAPI = require('../../lib/skynet')
+
+function * run (context) {
+  const skynet = new SkynetAPI(context.auth.password)
+  yield cli.action(
+    `removing ${cli.color.cyan(context.flags.value)} from whitelist`,
+    skynet.removeWhitelist(context.flags.value)
+  )
+}
+
+module.exports = {
+  topic: 'skynet',
+  command: 'whitelist:remove',
+  description: 'Remove user or app from the whitelist',
+  help: `Examples:
+    $ heroku skynet:whitelist:remove -v foo@bar.com`,
+  flags: [
+    {
+      name: 'value',
+      char: 'v',
+      description: 'app name or user email to remove from whitelist',
+      hasValue: true,
+      required: true
+    }
+  ],
+  run: cli.command(co.wrap(run))
+}

package/index.js

@@ -13,5 +13,7 @@ exports.commands = [
   require('./commands/unsuspend/apps.js'),
   require('./commands/unsuspend/user.js'),
   require('./commands/deprovision.js'),
-  require('./commands/categories.js')
+  require('./commands/categories.js'),
+  require('./commands/whitelist/add.js'),
+  require('./commands/whitelist/remove.js')
 ]

package/lib/skynet.js

@@ -21,7 +21,7 @@ module.exports = class SkynetAPI {
 
     options.json = false
 
-    return cli.got(SKYNET_BASE_URL + url, options).then((res) => res.body)
+    return cli.got(SKYNET_BASE_URL + url, options).then(res => res.body)
   }
 
   categories () {
@@ -30,6 +30,24 @@ module.exports = class SkynetAPI {
     })
   }
 
+  addWhitelist (appOrUserEmail, notes) {
+    var body = {
+      value: appOrUserEmail,
+      notes: notes
+    }
+
+    return this.request(`/whitelist`, {
+      method: 'POST',
+      body: qs.stringify(body)
+    })
+  }
+
+  removeWhitelist (appOrUserEmail) {
+    return this.request(`/whitelist/${appOrUserEmail}`, {
+      method: 'DELETE'
+    })
+  }
+
   removeUserpass (user, flag) {
     var body = {
       user: user,
@@ -88,13 +106,15 @@ module.exports = class SkynetAPI {
       method: 'DELETE'
     })
   }
-  suspendUser (user, notes, category, force = false) {
+
+  suspendUser (user, notes, category, notify, force = false) {
     var body = {
       value: user,
       reason: notes,
       method: 'skynet-cli',
       category: category,
-      force: force
+      force: force,
+      notify: notify
     }
 
     return this.request(`/suspend/user`, {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@heroku/skynet",
-  "version": "1.4.2",
+  "version": "1.5.0",
   "description": "use Skynet from Heroku CLI",
   "main": "index.js",
   "author": "Bob Argenbright @byt3smith",
@@ -25,7 +25,7 @@
     "mocha": "^3.4.2",
     "mockdate": "^2.0.1",
     "nock": "^9.0.13",
-    "standard": "^10.0.2",
+    "standard": "^12.0.1",
     "unexpected": "^10.29.0"
   },
   "scripts": {