use async/await

Author: KarateCowboy

src/commands/dynorestart.js

@@ -11,7 +11,7 @@ export default class extends Command {
     DurationFlag
   ]
 
-  async wait (ms: number) { return new Promise(resolve => setTimeout(resolve, ms)) }
+  async sleep (ms: number) { return new Promise(resolve => setTimeout(resolve, ms)) }
 
   duration = new Duration(this)
 
@@ -21,7 +21,7 @@ export default class extends Command {
   }
 
   async main () {
-    await this.wait(5000)
+    await this.sleep(5000)
     console.log('Done restarting dynos')
   }
 }

src/mixins/duration.js

@@ -43,12 +43,13 @@ export default class Duration {
       return magnitude * multiplier
     }
   }
+  async sleep (ms: number) { return new Promise(resolve => setTimeout(resolve, ms)) }
 
   async wait () {
-    return new Promise(resolve => setTimeout(resolve, this.parse(this.cmd.flags.duration)))
-      .then(() => {
-        console.log('operation did not complete in provided duration')
-      })
+
+    let resolved = await this.sleep(this.parse(this.cmd.flags.duration))
+    console.log('operation did not complete in provided duration')
+    return resolved
   }
 
   get duration (): number {