Merge pull request #1 from codeeno/add-spinner

Add spinner

Author: codeeno

lib/aws.js

@@ -1,49 +1,74 @@
 import AWS from "aws-sdk";
 import os from "os";
+import ora from "ora";
 
 export const createAWSClient = (region) => ({
   SSOOIDC: new AWS.SSOOIDC({ region: region }),
   SSO: new AWS.SSO({ region: region }),
 });
 
-export const registerClient = async (AWSClient) =>
-  await AWSClient.SSOOIDC.registerClient({
+export const registerClient = async (AWSClient) => {
+  const spinner = ora("Registering client..").start();
+  const client = await AWSClient.SSOOIDC.registerClient({
     clientName: os.hostname(),
     clientType: "public",
   }).promise();
+  spinner.stop();
+  return client;
+};
 
-export const startDeviceAuthorization = async (clientId, clientSecret, startUrl, AWSClient) =>
-  await AWSClient.SSOOIDC.startDeviceAuthorization({
+export const startDeviceAuthorization = async (clientId, clientSecret, startUrl, AWSClient) => {
+  const spinner = ora("Starting device authorization..").start();
+  const auth = await AWSClient.SSOOIDC.startDeviceAuthorization({
     clientId: clientId,
     clientSecret: clientSecret,
     startUrl: startUrl,
   }).promise();
+  spinner.stop();
+  return auth;
+};
 
-export const createToken = async (clientId, clientSecret, deviceCode, AWSClient) =>
-  await AWSClient.SSOOIDC.createToken({
+export const createToken = async (clientId, clientSecret, deviceCode, AWSClient) => {
+  const spinner = ora("Creating token..").start();
+  const token = await AWSClient.SSOOIDC.createToken({
     clientId: clientId,
     clientSecret: clientSecret,
     deviceCode: deviceCode,
     grantType: "urn:ietf:params:oauth:grant-type:device_code",
   }).promise();
+  spinner.stop();
+  return token;
+};
 
-export const listAccounts = async (accessToken, nextToken, AWSClient) =>
-  await AWSClient.SSO.listAccounts({
+export const listAccounts = async (accessToken, nextToken, AWSClient) => {
+  const spinner = ora("Fetching accounts..").start();
+  const accounts = await AWSClient.SSO.listAccounts({
     maxResults: 500,
     accessToken: accessToken,
     nextToken: nextToken,
   }).promise();
+  spinner.stop();
+  return accounts;
+};
 
-export const listAccountRoles = async (accessToken, accountId, nextToken, AWSClient) =>
-  await AWSClient.SSO.listAccountRoles({
+export const listAccountRoles = async (accessToken, accountId, nextToken, AWSClient) => {
+  const spinner = ora("Fetching roles..").start();
+  const roles = await AWSClient.SSO.listAccountRoles({
     accessToken: accessToken,
     accountId: accountId,
     nextToken: nextToken,
   }).promise();
+  spinner.stop();
+  return roles;
+};
 
-export const getRoleCredentials = async (accessToken, accountId, roleName, AWSClient) =>
-  await AWSClient.SSO.getRoleCredentials({
+export const getRoleCredentials = async (accessToken, accountId, roleName, AWSClient) => {
+  const spinner = ora("Fetching credentials..").start();
+  const credentials = await AWSClient.SSO.getRoleCredentials({
     accessToken: accessToken,
     accountId: accountId,
     roleName: roleName,
   }).promise();
+  spinner.stop();
+  return credentials;
+};

lib/web.js

@@ -1,5 +1,6 @@
 import axios from "axios";
 import os from "os";
+import ora from "ora";
 import { buildError } from "./error.js";
 
 const buildURLQuery = (obj) =>
@@ -12,6 +13,7 @@ export const getSigninUrl = async (accessKeyId, secretAccessKey, sessionToken) =
   const destinationUrl = `https://console.aws.amazon.com/console/home`;
 
   try {
+    const spinner = ora("Fetching sign-in URL..").start();
     const response = await axios.get(federationUrl, {
       params: {
         Action: "getSigninToken",
@@ -30,6 +32,7 @@ export const getSigninUrl = async (accessKeyId, secretAccessKey, sessionToken) =
       SigninToken: encodeURI(response?.data?.SigninToken),
     });
 
+    spinner.stop();
     return `${federationUrl}?${urlParams}`;
   } catch (err) {
     throw buildError("ERR_GET_SIGNIN_URL", `${err}`);