From 4bd120e4389ed73c4b58f97cffbf765b064c86c9 Mon Sep 17 00:00:00 2001
From: Zachary Huff <zach.huff.386@gmail.com>
Date: Fri, 26 Jul 2024 23:00:57 -0400
Subject: [PATCH] Add password prompt to sprofile utils

---
 cli/sprofile/utils.go | 87 ++++++++++++++++++++++++++++++++++++++++++-
 1 file changed, 86 insertions(+), 1 deletion(-)

diff --git a/cli/sprofile/utils.go b/cli/sprofile/utils.go
index 480fffc12..3228fc33c 100644
--- a/cli/sprofile/utils.go
+++ b/cli/sprofile/utils.go
@@ -13,11 +13,14 @@ import (
 	"strings"
 	"time"
 
+	"github.com/dropbox/godropbox/container/set"
 	"github.com/dropbox/godropbox/errors"
 	"github.com/pritunl/pritunl-client-electron/cli/errortypes"
 	"github.com/pritunl/pritunl-client-electron/cli/profile"
 	"github.com/pritunl/pritunl-client-electron/cli/service"
+	"github.com/pritunl/pritunl-client-electron/cli/terminal"
 	"github.com/pritunl/pritunl-client-electron/cli/utils"
+	"github.com/spf13/cobra"
 )
 
 var (
@@ -277,7 +280,82 @@ func GetAll() (sprfls []*Sprofile, err error) {
 	return
 }
 
-func Start(sprflId, mode, password string) (err error) {
+func PasswordPrompt(sprfl *Sprofile) (pass string, err error) {
+	passModes := set.NewSet()
+
+	passModesStr := strings.Split(sprfl.PasswordMode, "_")
+	for _, passMode := range passModesStr {
+		passModes.Add(passMode)
+	}
+
+	if passModes.Contains("pin") {
+		part := terminal.ReadPassword("Pin")
+		if part == "" {
+			cobra.CheckErr("cmd: Pin is empty")
+		}
+		pass += part
+	}
+
+	if passModes.Contains("duo") {
+		part := terminal.ReadPassword("Duo Passcode")
+		if part == "" {
+			cobra.CheckErr("cmd: Duo Passcode is empty")
+		}
+		pass += part
+	}
+
+	if passModes.Contains("onelogin") {
+		part := terminal.ReadPassword("OneLogin Passcode")
+		if part == "" {
+			cobra.CheckErr("cmd: OneLogin Passcode is empty")
+		}
+		pass += part
+	}
+
+	if passModes.Contains("okta") {
+		part := terminal.ReadPassword("Okta Passcode")
+		if part == "" {
+			cobra.CheckErr("cmd: Okta Passcode is empty")
+		}
+		pass += part
+	}
+
+	if passModes.Contains("otp") {
+		part := terminal.ReadPassword("Authenticator Passcode")
+		if part == "" {
+			cobra.CheckErr("cmd: Authenticator Passcode is empty")
+		}
+		pass += part
+	}
+
+	if passModes.Contains("yubikey") {
+		part := terminal.ReadPassword("YubiKey")
+		if part == "" {
+			cobra.CheckErr("cmd: YubiKey is empty")
+		}
+		pass += part
+	}
+
+	if passModes.Contains("yubikey") {
+		part := terminal.ReadPassword("YubiKey")
+		if part == "" {
+			cobra.CheckErr("cmd: YubiKey is empty")
+		}
+		pass += part
+	}
+
+	if pass == "" {
+		part := terminal.ReadPassword("Password")
+		if part == "" {
+			cobra.CheckErr("cmd: Password is empty")
+		}
+		pass += part
+	}
+
+	return
+}
+
+func Start(sprflId, mode, password string, passwordPrompt bool) (err error) {
 	sprfl, err := Match(sprflId)
 	if err != nil {
 		return
@@ -302,6 +380,13 @@ func Start(sprflId, mode, password string) (err error) {
 
 	reqUrl := service.GetAddress() + "/profile"
 
+	if passwordPrompt {
+		password, err = PasswordPrompt(sprfl)
+		if err != nil {
+			return
+		}
+	}
+
 	authKey, err := service.GetAuthKey()
 	if err != nil {
 		return