Add SMS support

Author: CoreyRExtend

package.json

@@ -8,7 +8,7 @@
   ],
   "types": "dist/index.d.ts",
   "bin": {
-    "extaws": "dist/lib/extaws.js"
+    "extaws": "dist/lib/extaws-cli.js"
   },
   "repository": "[email protected]:helloextend/extaws.git",
   "author": "Corey Ryan <[email protected]>",

src/lib/extaws.ts

@@ -108,26 +108,53 @@ export class ExtAws {
           }))
           spinner?.start('Logging in...')
         }
-        do {
-          verify = await this.verifyFactor({ id: factor, stateToken: authResponse.stateToken}, totpPrompt)
-          if (!('status' in verify)) {
-            throw new Error(verify.message)
+        if ( type === 'sms') {
+          do {
+            verify = await this.verifyFactor({ id: factor, stateToken: authResponse.stateToken}, totpPrompt)
+
+            spinner?.stop()
+            ;({ totpPrompt } = await ExtAws.inquire({
+              name: 'totpPrompt',
+              type: 'input',
+              message: 'SMS Code:',
+            }))
+            spinner?.start('Logging in...')
+
+            verify = await this.verifyFactor({ id: factor, stateToken: authResponse.stateToken}, totpPrompt)
+            if (!('status' in verify)) {
+              spinner?.stop()
+              throw new Error(verify.message)
+            }
+
+          } while(verify.status !== 'SUCCESS' && verify.stateToken)
+
+          if (!verify.sessionToken) {
+            throw new Error('No session token in response')
           }
-          if (verify.status !== 'SUCCESS') await sleep(1000)
-          if (counter > 30) {
-            console.log('Timing out after thirty seconds')
-            console.log(`Auth State: ${verify.status}`)
-            process.exit(1)
-          }
-          counter++
-        } while(verify.status !== 'SUCCESS' && verify.stateToken)
+          this.sessionToken = verify.sessionToken
 
-        if (!verify.sessionToken) {
-          throw new Error('No session token in response')
+        } else {
+          do {
+            verify = await this.verifyFactor({ id: factor, stateToken: authResponse.stateToken}, totpPrompt)
+            if (!('status' in verify)) {
+              spinner?.stop()
+              throw new Error(verify.message)
+            }
+            if (verify.status !== 'SUCCESS') await sleep(1000 * counter * counter)
+            if (counter > 30) {
+              console.log('Timing out after thirty seconds')
+              console.log(`Auth State: ${verify.status}`)
+              process.exit(1)
+            }
+            counter++
+          } while(verify.status !== 'SUCCESS' && verify.stateToken)
+
+          if (!verify.sessionToken) {
+            throw new Error('No session token in response')
+          }
+          this.sessionToken = verify.sessionToken
         }
 
-        this.sessionToken = verify.sessionToken
-
       } else {
         throw new Error('Unable to find factors in auth response')
       }
@@ -267,6 +294,8 @@ export class ExtAws {
         duration = inputDuration
       }
 
+      if (duration > 43200) duration = 43200
+
       let awsRegion: string
       if (!inputAwsRegion) {
         const { promptAwsRegion } = await ExtAws.inquire({

src/lib/types.ts

@@ -33,7 +33,7 @@ export interface ExtAwsUserConfig {
   }
 }
 
-export type FactorDevice = 'token:software:totp' | 'push'
+export type FactorDevice = 'token:software:totp' | 'push' | 'sms'
 
 export type FactorSelect = {
     factor: string,