feat: Add password prompt for local activation

open-amt-cloud-toolkit · Sep 5, 2023 · 6cdfa31 · 6cdfa31
1 parent 09fdd03
commit 6cdfa31
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 1 deletion.
diff --git a/internal/flags/deactivate.go b/internal/flags/deactivate.go
@@ -3,10 +3,22 @@ package flags
 import (
 	"fmt"
 	"rpc/pkg/utils"
+
+	log "github.com/sirupsen/logrus"
 )
 
 func (f *Flags) handleDeactivateCommand() int {
 	f.amtDeactivateCommand.BoolVar(&f.Local, "local", false, "Execute command to AMT directly without cloud interaction")
+
+	if len(f.commandLineArgs) == 3 && f.commandLineArgs[2] == "-local" {
+		_, errCode := f.ReadPasswordFromUser()
+		if f.Password == "" {
+			if errCode != 0 {
+				log.Error("Please enter value for AMT Password")
+				return utils.MissingOrIncorrectPassword
+			}
+		}
+	}
 	if len(f.commandLineArgs) == 2 {
 		f.amtDeactivateCommand.PrintDefaults()
 		return utils.IncorrectCommandLineParameters

diff --git a/internal/flags/deactivate_test.go b/internal/flags/deactivate_test.go
@@ -75,12 +75,20 @@ func TestHandleDeactivateCommandWithForce(t *testing.T) {
 }
 
 func TestHandleLocalDeactivation(t *testing.T) {
-	args := []string{"./rpc", "deactivate", "-local"}
+	args := []string{"./rpc", "deactivate", "-local", "password"}
 	flags := NewFlags(args)
 	errCode := flags.ParseFlags()
 	assert.Equal(t, errCode, utils.Success)
 }
 
+func TestParseFlagsDeactivateNoPassword(t *testing.T) {
+	args := []string{"./rpc", "deactivate", "-local"}
+	flags := NewFlags(args)
+	result := flags.ParseFlags()
+	assert.EqualValues(t, result, utils.MissingOrIncorrectPassword)
+	assert.Equal(t, utils.CommandDeactivate, flags.Command)
+}
+
 func TestParseFlagsDeactivate(t *testing.T) {
 	args := []string{"./rpc", "deactivate"}
 	flags := NewFlags(args)