fix: Added the profile name to the mfa token prompt when setting an a…

…ws environment (#12)

README.md

@@ -101,6 +101,7 @@ $ biome save -b my-biome -f my.env
 
 ## Future Plans
 - Have goreleaser create a docker image and publish to ghcr
+- :white_check_mark: Custom MFA token message for AWS profiles
 - :white_check_mark: Export loaded variables to a dotenv file
 - :white_check_mark: Switch to [cobra](https://github.com/spf13/cobra) for the cli
 - :white_check_mark: Allow CLI input to be a secret (for passwords)

src/repos/aws_sts.go

@@ -2,6 +2,7 @@ package repos
 
 import (
 	"context"
+	"fmt"
 	"os"
 
 	"github.com/aws/aws-sdk-go-v2/aws"
@@ -11,6 +12,8 @@ import (
 	"github.com/jeff-roche/biome/src/lib/types"
 )
 
+var awsProfileName string = ""
+
 type AwsStsRepositoryIfc interface {
 	ConfigureSession(profile string) (*types.AwsEnvConfig, error)
 	SetAwsEnvs(*types.AwsEnvConfig)
@@ -67,6 +70,8 @@ func (repo AwsStsRepository) loadProfile(profile string) (config.SharedConfig, e
 }
 
 func (repo AwsStsRepository) loadAwsConfig(profCfg *config.SharedConfig) (aws.Config, error) {
+	awsProfileName = profCfg.Profile
+
 	return config.LoadDefaultConfig(
 		context.TODO(),
 		config.WithSharedConfigProfile(profCfg.Profile),
@@ -75,7 +80,7 @@ func (repo AwsStsRepository) loadAwsConfig(profCfg *config.SharedConfig) (aws.Co
 			func(aro *stscreds.AssumeRoleOptions) {
 				if profCfg.MFASerial != "" {
 					aro.SerialNumber = &profCfg.MFASerial
-					aro.TokenProvider = stscreds.StdinTokenProvider
+					aro.TokenProvider = CustomStdinTokenProvider
 				}
 			},
 		),
@@ -99,3 +104,11 @@ func (repo AwsStsRepository) setupAwsSession(cfg *aws.Config, profile *config.Sh
 
 	return creds, nil
 }
+
+func CustomStdinTokenProvider() (string, error) {
+	var v string
+	fmt.Printf("MFA token for AWS profile '%s': ", awsProfileName)
+	_, err := fmt.Scanln(&v)
+
+	return v, err
+}