central logging and 2v service token

cli/packages/api/api.go

@@ -1,14 +1,13 @@
-package http
+package api
 
 import (
 	"fmt"
 
 	"github.com/Infisical/infisical-merge/packages/config"
-	"github.com/Infisical/infisical-merge/packages/models"
 	"github.com/go-resty/resty/v2"
 )
 
-func CallBatchModifySecretsByWorkspaceAndEnv(httpClient *resty.Client, request models.BatchModifySecretsByWorkspaceAndEnvRequest) error {
+func CallBatchModifySecretsByWorkspaceAndEnv(httpClient *resty.Client, request BatchModifySecretsByWorkspaceAndEnvRequest) error {
 	endpoint := fmt.Sprintf("%v/v2/secret/batch-modify/workspace/%v/environment/%v", config.INFISICAL_URL, request.WorkspaceId, request.EnvironmentName)
 	response, err := httpClient.
 		R().
@@ -26,7 +25,7 @@ func CallBatchModifySecretsByWorkspaceAndEnv(httpClient *resty.Client, request m
 	return nil
 }
 
-func CallBatchCreateSecretsByWorkspaceAndEnv(httpClient *resty.Client, request models.BatchCreateSecretsByWorkspaceAndEnvRequest) error {
+func CallBatchCreateSecretsByWorkspaceAndEnv(httpClient *resty.Client, request BatchCreateSecretsByWorkspaceAndEnvRequest) error {
 	endpoint := fmt.Sprintf("%v/v2/secret/batch-create/workspace/%v/environment/%v", config.INFISICAL_URL, request.WorkspaceId, request.EnvironmentName)
 	response, err := httpClient.
 		R().
@@ -44,7 +43,7 @@ func CallBatchCreateSecretsByWorkspaceAndEnv(httpClient *resty.Client, request m
 	return nil
 }
 
-func CallBatchDeleteSecretsByWorkspaceAndEnv(httpClient *resty.Client, request models.BatchDeleteSecretsBySecretIdsRequest) error {
+func CallBatchDeleteSecretsByWorkspaceAndEnv(httpClient *resty.Client, request BatchDeleteSecretsBySecretIdsRequest) error {
 	endpoint := fmt.Sprintf("%v/v2/secret/batch/workspace/%v/environment/%v", config.INFISICAL_URL, request.WorkspaceId, request.EnvironmentName)
 	response, err := httpClient.
 		R().
@@ -62,45 +61,45 @@ func CallBatchDeleteSecretsByWorkspaceAndEnv(httpClient *resty.Client, request m
 	return nil
 }
 
-func CallGetEncryptedWorkspaceKey(httpClient *resty.Client, request models.GetEncryptedWorkspaceKeyRequest) (models.GetEncryptedWorkspaceKeyResponse, error) {
+func CallGetEncryptedWorkspaceKey(httpClient *resty.Client, request GetEncryptedWorkspaceKeyRequest) (GetEncryptedWorkspaceKeyResponse, error) {
 	endpoint := fmt.Sprintf("%v/v2/workspace/%v/encrypted-key", config.INFISICAL_URL, request.WorkspaceId)
-	var result models.GetEncryptedWorkspaceKeyResponse
+	var result GetEncryptedWorkspaceKeyResponse
 	response, err := httpClient.
 		R().
 		SetResult(&result).
 		Get(endpoint)
 
 	if err != nil {
-		return models.GetEncryptedWorkspaceKeyResponse{}, fmt.Errorf("CallGetEncryptedWorkspaceKey: Unable to complete api request [err=%s]", err)
+		return GetEncryptedWorkspaceKeyResponse{}, fmt.Errorf("CallGetEncryptedWorkspaceKey: Unable to complete api request [err=%s]", err)
 	}
 
 	if response.StatusCode() > 299 {
-		return models.GetEncryptedWorkspaceKeyResponse{}, fmt.Errorf("CallGetEncryptedWorkspaceKey: Unsuccessful response: [response=%s]", response)
+		return GetEncryptedWorkspaceKeyResponse{}, fmt.Errorf("CallGetEncryptedWorkspaceKey: Unsuccessful response: [response=%s]", response)
 	}
 
 	return result, nil
 }
 
-func CallGetServiceTokenDetailsV2(httpClient *resty.Client) (models.GetServiceTokenDetailsResponse, error) {
-	var tokenDetailsResponse models.GetServiceTokenDetailsResponse
+func CallGetServiceTokenDetailsV2(httpClient *resty.Client) (GetServiceTokenDetailsResponse, error) {
+	var tokenDetailsResponse GetServiceTokenDetailsResponse
 	response, err := httpClient.
 		R().
 		SetResult(&tokenDetailsResponse).
 		Get(fmt.Sprintf("%v/v2/service-token", config.INFISICAL_URL))
 
 	if err != nil {
-		return models.GetServiceTokenDetailsResponse{}, fmt.Errorf("CallGetServiceTokenDetails: Unable to complete api request [err=%s]", err)
+		return GetServiceTokenDetailsResponse{}, fmt.Errorf("CallGetServiceTokenDetails: Unable to complete api request [err=%s]", err)
 	}
 
 	if response.StatusCode() > 299 {
-		return models.GetServiceTokenDetailsResponse{}, fmt.Errorf("CallGetServiceTokenDetails: Unsuccessful response: [response=%s]", response)
+		return GetServiceTokenDetailsResponse{}, fmt.Errorf("CallGetServiceTokenDetails: Unsuccessful response: [response=%s]", response)
 	}
 
 	return tokenDetailsResponse, nil
 }
 
-func CallGetSecretsV2(httpClient *resty.Client, request models.GetEncryptedSecretsV2Request) (models.GetEncryptedSecretsV2Response, error) {
-	var secretsResponse models.GetEncryptedSecretsV2Response
+func CallGetSecretsV2(httpClient *resty.Client, request GetEncryptedSecretsV2Request) (GetEncryptedSecretsV2Response, error) {
+	var secretsResponse GetEncryptedSecretsV2Response
 	response, err := httpClient.
 		R().
 		SetResult(&secretsResponse).
@@ -108,12 +107,30 @@ func CallGetSecretsV2(httpClient *resty.Client, request models.GetEncryptedSecre
 		Get(fmt.Sprintf("%v/v2/secret/workspace/%v", config.INFISICAL_URL, request.WorkspaceId))
 
 	if err != nil {
-		return models.GetEncryptedSecretsV2Response{}, fmt.Errorf("CallGetSecretsV2: Unable to complete api request [err=%s]", err)
+		return GetEncryptedSecretsV2Response{}, fmt.Errorf("CallGetSecretsV2: Unable to complete api request [err=%s]", err)
 	}
 
 	if response.StatusCode() > 299 {
-		return models.GetEncryptedSecretsV2Response{}, fmt.Errorf("CallGetSecretsV2: Unsuccessful response: [response=%s]", response)
+		return GetEncryptedSecretsV2Response{}, fmt.Errorf("CallGetSecretsV2: Unsuccessful response: [response=%s]", response)
 	}
 
 	return secretsResponse, nil
 }
+
+func CallGetAllWorkSpacesUserBelongsTo(httpClient *resty.Client) (GetWorkSpacesResponse, error) {
+	var workSpacesResponse GetWorkSpacesResponse
+	response, err := httpClient.
+		R().
+		SetResult(&workSpacesResponse).
+		Get(fmt.Sprintf("%v/v1/workspace", config.INFISICAL_URL))
+
+	if err != nil {
+		return GetWorkSpacesResponse{}, err
+	}
+
+	if response.StatusCode() > 299 {
+		return GetWorkSpacesResponse{}, fmt.Errorf("CallGetAllWorkSpacesUserBelongsTo: Unsuccessful response:  [response=%v]", response)
+	}
+
+	return workSpacesResponse, nil
+}

cli/packages/api/model.go

@@ -1,4 +1,4 @@
-package models
+package api
 
 import "time"
 
@@ -119,14 +119,13 @@ type PullSecretsByInfisicalTokenResponse struct {
 }
 
 type GetWorkSpacesResponse struct {
-	Workspaces []Workspace `json:"workspaces"`
-}
-type Workspace struct {
-	ID           string `json:"_id"`
-	Name         string `json:"name"`
-	Plan         string `json:"plan,omitempty"`
-	V            int    `json:"__v"`
-	Organization string `json:"organization,omitempty"`
+	Workspaces []struct {
+		ID           string `json:"_id"`
+		Name         string `json:"name"`
+		Plan         string `json:"plan,omitempty"`
+		V            int    `json:"__v"`
+		Organization string `json:"organization,omitempty"`
+	} `json:"workspaces"`
 }
 
 type Secret struct {

cli/packages/cmd/export.go

@@ -29,58 +29,46 @@ var exportCmd = &cobra.Command{
 	DisableFlagsInUseLine: true,
 	Example:               "infisical export --env=prod --format=json > secrets.json",
 	Args:                  cobra.NoArgs,
-	PreRun:                toggleDebug,
+	PreRun: func(cmd *cobra.Command, args []string) {
+		toggleDebug(cmd, args)
+		util.RequireLogin()
+		util.RequireLocalWorkspaceFile()
+	},
 	Run: func(cmd *cobra.Command, args []string) {
 		envName, err := cmd.Flags().GetString("env")
 		if err != nil {
-			log.Errorln("Unable to parse the environment flag")
-			log.Debugln(err)
-			return
+			util.HandleError(err)
 		}
 
 		shouldExpandSecrets, err := cmd.Flags().GetBool("expand")
 		if err != nil {
-			log.Errorln("Unable to parse the substitute flag")
-			log.Debugln(err)
-			return
-		}
-
-		projectId, err := cmd.Flags().GetString("projectId")
-		if err != nil {
-			log.Errorln("Unable to parse the project id flag")
-			log.Debugln(err)
-			return
+			util.HandleError(err)
 		}
 
 		format, err := cmd.Flags().GetString("format")
 		if err != nil {
-			log.Errorln("Unable to parse the format flag")
-			log.Debugln(err)
-			return
+			util.HandleError(err)
 		}
 
-		envsFromApi, err := util.GetAllEnvironmentVariables(projectId, envName)
+		secrets, err := util.GetAllEnvironmentVariables(envName)
 		if err != nil {
-			log.Errorln("Something went wrong when pulling secrets using your Infisical token. Double check the token, project id or environment name (dev, prod, ect.)")
-			log.Debugln(err)
-			return
+			util.HandleError(err, "Unable to fetch secrets")
 		}
 
 		var output string
 		if shouldExpandSecrets {
-			substitutions := util.SubstituteSecrets(envsFromApi)
+			substitutions := util.SubstituteSecrets(secrets)
 			output, err = formatEnvs(substitutions, format)
 			if err != nil {
-				log.Errorln(err)
-				return
+				util.HandleError(err)
 			}
 		} else {
-			output, err = formatEnvs(envsFromApi, format)
+			output, err = formatEnvs(secrets, format)
 			if err != nil {
-				log.Errorln(err)
-				return
+				util.HandleError(err)
 			}
 		}
+
 		fmt.Print(output)
 	},
 }
@@ -88,7 +76,6 @@ var exportCmd = &cobra.Command{
 func init() {
 	rootCmd.AddCommand(exportCmd)
 	exportCmd.Flags().StringP("env", "e", "dev", "Set the environment (dev, prod, etc.) from which your secrets should be pulled from")
-	exportCmd.Flags().String("projectId", "", "The project ID from which your secrets should be pulled from")
 	exportCmd.Flags().Bool("expand", true, "Parse shell parameter expansions in your secrets")
 	exportCmd.Flags().StringP("format", "f", "dotenv", "Set the format of the output file (dotenv, json, csv)")
 }

cli/packages/cmd/init.go

@@ -5,10 +5,13 @@ package cmd
 
 import (
 	"encoding/json"
+	"fmt"
 	"os"
 
+	"github.com/Infisical/infisical-merge/packages/api"
 	"github.com/Infisical/infisical-merge/packages/models"
 	"github.com/Infisical/infisical-merge/packages/util"
+	"github.com/go-resty/resty/v2"
 	"github.com/manifoldco/promptui"
 	log "github.com/sirupsen/logrus"
 	"github.com/spf13/cobra"
@@ -21,21 +24,10 @@ var initCmd = &cobra.Command{
 	DisableFlagsInUseLine: true,
 	Example:               "infisical init",
 	Args:                  cobra.ExactArgs(0),
-	PreRun:                toggleDebug,
+	PreRun: func(cmd *cobra.Command, args []string) {
+		util.RequireLogin()
+	},
 	Run: func(cmd *cobra.Command, args []string) {
-		// check if user is logged
-		hasUserLoggedInbefore, loggedInUserEmail, err := util.IsUserLoggedIn()
-		if err != nil {
-			log.Info("Unexpected issue occurred while checking login status. To see more details, add flag --debug")
-			log.Debugln(err)
-			return
-		}
-
-		if !hasUserLoggedInbefore {
-			log.Infoln("No logged in user. To login, please run command [infisical login]")
-			return
-		}
-
 		if util.WorkspaceConfigFileExistsInCurrentPath() {
 			shouldOverride, err := shouldOverrideWorkspacePrompt()
 			if err != nil {
@@ -49,23 +41,22 @@ var initCmd = &cobra.Command{
 			}
 		}
 
-		userCreds, err := util.GetUserCredsFromKeyRing(loggedInUserEmail)
+		userCreds, err := util.GetCurrentLoggedInUserDetails()
 		if err != nil {
-			log.Infoln("Unable to get user creds from key ring")
-			log.Debug(err)
-			return
+			util.HandleError(err, "Unable to get your login details")
 		}
 
-		workspaces, err := util.GetWorkSpacesFromAPI(userCreds)
+		httpClient := resty.New()
+		httpClient.SetAuthToken(userCreds.UserCredentials.JTWToken)
+		workspaceResponse, err := api.CallGetAllWorkSpacesUserBelongsTo(httpClient)
 		if err != nil {
-			log.Errorln("Unable to pull your projects. To see more logs add the --debug flag to this command")
-			log.Debugln("Unable to get your projects because:", err)
-			return
+			util.HandleError(err, "Unable to pull projects that belong to you")
 		}
 
+		workspaces := workspaceResponse.Workspaces
 		if len(workspaces) == 0 {
-			log.Infoln("You don't have any projects created in Infisical. You must first create a project at https://infisical.com")
-			return
+			message := fmt.Sprintf("You don't have any projects created in Infisical. You must first create a project at %s", util.INFISICAL_TOKEN_NAME)
+			util.PrintMessageAndExit(message)
 		}
 
 		var workspaceNames []string
@@ -81,16 +72,12 @@ var initCmd = &cobra.Command{
 
 		index, _, err := prompt.Run()
 		if err != nil {
-			log.Errorln("Unable to parse your response")
-			log.Debug(err)
-			return
+			util.HandleError(err)
 		}
 
 		err = writeWorkspaceFile(workspaces[index])
 		if err != nil {
-			log.Errorln("Something went wrong when creating your workspace file")
-			log.Debug("Error while writing your workspace file:", err)
-			return
+			util.HandleError(err)
 		}
 	},
 }

cli/packages/cmd/login.go

@@ -11,6 +11,7 @@ import (
 	"fmt"
 	"regexp"
 
+	"github.com/Infisical/infisical-merge/packages/api"
 	"github.com/Infisical/infisical-merge/packages/config"
 	"github.com/Infisical/infisical-merge/packages/crypto"
 	"github.com/Infisical/infisical-merge/packages/models"
@@ -29,18 +30,15 @@ var loginCmd = &cobra.Command{
 	DisableFlagsInUseLine: true,
 	PreRun:                toggleDebug,
 	Run: func(cmd *cobra.Command, args []string) {
-		hasUserLoggedInbefore, currentLoggedInUserEmail, err := util.IsUserLoggedIn()
-
+		currentLoggedInUserDetails, err := util.GetCurrentLoggedInUserDetails()
 		if err != nil {
-			log.Debugln("Unable to get current logged in user.", err)
+			util.HandleError(err)
 		}
 
-		if hasUserLoggedInbefore {
-			shouldOverride, err := shouldOverrideLoginPrompt(currentLoggedInUserEmail)
+		if currentLoggedInUserDetails.IsUserLoggedIn {
+			shouldOverride, err := shouldOverrideLoginPrompt(currentLoggedInUserDetails.UserCredentials.Email)
 			if err != nil {
-				log.Errorln("Unable to parse your answer")
-				log.Debug(err)
-				return
+				util.HandleError(err)
 			}
 
 			if !shouldOverride {
@@ -50,14 +48,12 @@ var loginCmd = &cobra.Command{
 
 		email, password, err := askForLoginCredentials()
 		if err != nil {
-			log.Errorln("Unable to parse email and password for authentication")
-			log.Debugln(err)
-			return
+			util.HandleError(err, "Unable to parse email and password for authentication")
 		}
 
 		userCredentials, err := getFreshUserCredentials(email, password)
 		if err != nil {
-			log.Errorln("Unable to authenticate with the provided credentials, please try again")
+			log.Infoln("Unable to authenticate with the provided credentials, please try again")
 			log.Debugln(err)
 			return
 		}
@@ -65,14 +61,12 @@ var loginCmd = &cobra.Command{
 		encryptedPrivateKey, _ := base64.StdEncoding.DecodeString(userCredentials.EncryptedPrivateKey)
 		tag, err := base64.StdEncoding.DecodeString(userCredentials.Tag)
 		if err != nil {
-			log.Errorln("Unable to decode the auth tag")
-			log.Debugln(err)
+			util.HandleError(err)
 		}
 
 		IV, err := base64.StdEncoding.DecodeString(userCredentials.IV)
 		if err != nil {
-			log.Errorln("Unable to decode the IV/Nonce")
-			log.Debugln(err)
+			util.HandleError(err)
 		}
 
 		paddedPassword := fmt.Sprintf("%032s", password)
@@ -80,9 +74,7 @@ var loginCmd = &cobra.Command{
 
 		decryptedPrivateKey, err := crypto.DecryptSymmetric(key, encryptedPrivateKey, tag, IV)
 		if err != nil || len(decryptedPrivateKey) == 0 {
-			log.Errorln("There was an issue decrypting your keys")
-			log.Debugln(err)
-			return
+			util.HandleError(err)
 		}
 
 		userCredentialsToBeStored := &models.UserCredentials{
@@ -102,9 +94,7 @@ var loginCmd = &cobra.Command{
 
 		err = util.WriteInitalConfig(userCredentialsToBeStored)
 		if err != nil {
-			log.Errorln("Unable to write write to Infisical Config file. Please try again")
-			log.Debugln(err)
-			return
+			util.HandleError(err, "Unable to write write to Infisical Config file. Please try again")
 		}
 
 		log.Infoln("Nice! You are loggin as:", email)
@@ -158,7 +148,7 @@ func askForLoginCredentials() (email string, password string, err error) {
 	return userEmail, userPassword, nil
 }
 
-func getFreshUserCredentials(email string, password string) (*models.LoginTwoResponse, error) {
+func getFreshUserCredentials(email string, password string) (*api.LoginTwoResponse, error) {
 	log.Debugln("getFreshUserCredentials:", "email", email, "password", password)
 	httpClient := resty.New()
 	httpClient.SetRetryCount(5)
@@ -169,12 +159,12 @@ func getFreshUserCredentials(email string, password string) (*models.LoginTwoRes
 	srpA := hex.EncodeToString(srpClient.ComputeA())
 
 	// ** Login one
-	loginOneRequest := models.LoginOneRequest{
+	loginOneRequest := api.LoginOneRequest{
 		Email:           email,
 		ClientPublicKey: srpA,
 	}
 
-	var loginOneResponseResult models.LoginOneResponse
+	var loginOneResponseResult api.LoginOneResponse
 
 	loginOneResponse, err := httpClient.
 		R().
@@ -206,12 +196,12 @@ func getFreshUserCredentials(email string, password string) (*models.LoginTwoRes
 
 	srpM1 := srpClient.ComputeM1()
 
-	LoginTwoRequest := models.LoginTwoRequest{
+	LoginTwoRequest := api.LoginTwoRequest{
 		Email:       email,
 		ClientProof: hex.EncodeToString(srpM1),
 	}
 
-	var loginTwoResponseResult models.LoginTwoResponse
+	var loginTwoResponseResult api.LoginTwoResponse
 	loginTwoResponse, err := httpClient.
 		R().
 		SetBody(LoginTwoRequest).

cli/packages/cmd/root.go

@@ -30,7 +30,7 @@ func Execute() {
 func init() {
 	rootCmd.Flags().BoolP("toggle", "t", false, "Help message for toggle")
 	rootCmd.PersistentFlags().BoolVarP(&debugLogging, "debug", "d", false, "Enable verbose logging")
-	rootCmd.PersistentFlags().StringVar(&config.INFISICAL_URL, "domain", "http://localhost:8080/api", "Point the CLI to your own backend")
+	rootCmd.PersistentFlags().StringVar(&config.INFISICAL_URL, "domain", "https://app.infisical.com/api", "Point the CLI to your own backend")
 	// rootCmd.PersistentPreRun = func(cmd *cobra.Command, args []string) {
 	// }
 }

cli/packages/cmd/run.go

@@ -55,36 +55,26 @@ var runCmd = &cobra.Command{
 	Run: func(cmd *cobra.Command, args []string) {
 		envName, err := cmd.Flags().GetString("env")
 		if err != nil {
-			log.Errorln("Unable to parse the environment flag")
-			log.Debugln(err)
-			return
+			util.HandleError(err, "Unable to parse flag")
+		}
+
+		if !util.IsSecretEnvironmentValid(envName) {
+			util.PrintMessageAndExit("Invalid environment name passed. Environment names can only be prod, dev, test or staging")
 		}
 
 		secretOverriding, err := cmd.Flags().GetBool("secret-overriding")
 		if err != nil {
-			log.Errorln("Unable to parse the secret-overriding flag")
-			log.Debugln(err)
-			return
+			util.HandleError(err, "Unable to parse flag")
 		}
 
 		shouldExpandSecrets, err := cmd.Flags().GetBool("expand")
 		if err != nil {
-			log.Errorln("Unable to parse the substitute flag")
-			log.Debugln(err)
-			return
+			util.HandleError(err, "Unable to parse flag")
 		}
 
-		projectId, err := cmd.Flags().GetString("projectId")
+		secrets, err := util.GetAllEnvironmentVariables(envName)
 		if err != nil {
-			log.Errorln("Unable to parse the project id flag")
-			log.Debugln(err)
-			return
-		}
-
-		secrets, err := util.GetAllEnvironmentVariables(projectId, envName)
-		if err != nil {
-			log.Debugln(err)
-			return
+			util.HandleError(err, "Could not fetch secrets", "If you are using a service token to fetch secrets, please ensure it is valid")
 		}
 
 		if shouldExpandSecrets {
@@ -97,29 +87,26 @@ var runCmd = &cobra.Command{
 
 		if cmd.Flags().Changed("command") {
 			command := cmd.Flag("command").Value.String()
+
 			err = executeMultipleCommandWithEnvs(command, secrets)
 			if err != nil {
-				log.Errorf("Something went wrong when executing your command [error=%s]", err)
-				return
+				util.HandleError(err, "Unable to execute your chained command")
 			}
+
 		} else {
 			err = executeSingleCommandWithEnvs(args, secrets)
 			if err != nil {
-				log.Errorf("Something went wrong when executing your command [error=%s]", err)
-				return
+				util.HandleError(err, "Unable to execute your single command")
 			}
-			return
 		}
-
 	},
 }
 
 func init() {
 	rootCmd.AddCommand(runCmd)
 	runCmd.Flags().StringP("env", "e", "dev", "Set the environment (dev, prod, etc.) from which your secrets should be pulled from")
-	runCmd.Flags().String("projectId", "", "The project ID from which your secrets should be pulled from")
 	runCmd.Flags().Bool("expand", true, "Parse shell parameter expansions in your secrets")
-	runCmd.Flags().Bool("secret-overriding", true, "Prioritizes personal secrets with the same name over shared secrets")
+	runCmd.Flags().Bool("secret-overriding", true, "Prioritizes personal secrets, if any, with the same name over shared secrets")
 	runCmd.Flags().StringP("command", "c", "", "chained commands to execute (e.g. \"npm install && npm run dev; echo ...\")")
 }
 
@@ -130,6 +117,7 @@ func executeSingleCommandWithEnvs(args []string, secrets []models.SingleEnvironm
 	numberOfSecretsInjected := fmt.Sprintf("\u2713 Injected %v Infisical secrets into your application process successfully", len(secrets))
 	log.Infof("\x1b[%dm%s\x1b[0m", 32, numberOfSecretsInjected)
 	log.Debugf("executing command: %s %s \n", command, strings.Join(argsForCommand, " "))
+	log.Debugf("Secrets injected: %v", secrets)
 
 	cmd := exec.Command(command, argsForCommand...)
 	cmd.Stdin = os.Stdin
@@ -157,6 +145,7 @@ func executeMultipleCommandWithEnvs(fullCommand string, secrets []models.SingleE
 	numberOfSecretsInjected := fmt.Sprintf("\u2713 Injected %v Infisical secrets into your application process successfully", len(secrets))
 	log.Infof("\x1b[%dm%s\x1b[0m", 32, numberOfSecretsInjected)
 	log.Debugf("executing command: %s %s %s \n", shell[0], shell[1], fullCommand)
+	log.Debugf("Secrets injected: %v", secrets)
 
 	return execCmd(cmd)
 }

cli/packages/cmd/secrets.go

@@ -11,8 +11,8 @@ import (
 
 	"crypto/sha256"
 
+	"github.com/Infisical/infisical-merge/packages/api"
 	"github.com/Infisical/infisical-merge/packages/crypto"
-	"github.com/Infisical/infisical-merge/packages/http"
 	"github.com/Infisical/infisical-merge/packages/models"
 	"github.com/Infisical/infisical-merge/packages/util"
 	"github.com/Infisical/infisical-merge/packages/visualize"
@@ -31,35 +31,23 @@ var secretsCmd = &cobra.Command{
 	Run: func(cmd *cobra.Command, args []string) {
 		environmentName, err := cmd.Flags().GetString("env")
 		if err != nil {
-			log.Errorln("Unable to parse the environment name flag")
-			log.Debugln(err)
-			return
+			util.HandleError(err)
 		}
 
 		shouldExpandSecrets, err := cmd.Flags().GetBool("expand")
 		if err != nil {
-			log.Errorln("Unable to parse the substitute flag")
-			log.Debugln(err)
-			return
+			util.HandleError(err)
 		}
 
-		workspaceFileExists := util.WorkspaceConfigFileExistsInCurrentPath()
-		if !workspaceFileExists {
-			log.Error("You have not yet connected to an Infisical Project. Please run [infisical init]")
-			return
+		secrets, err := util.GetAllEnvironmentVariables(environmentName)
+		if err != nil {
+			util.HandleError(err)
 		}
 
-		secrets, err := util.GetAllEnvironmentVariables("", environmentName)
-
 		if shouldExpandSecrets {
 			secrets = util.SubstituteSecrets(secrets)
 		}
 
-		if err != nil {
-			log.Debugln(err)
-			return
-		}
-
 		visualize.PrintAllSecretDetails(secrets)
 	},
 }
@@ -82,70 +70,36 @@ var secretsSetCmd = &cobra.Command{
 	PreRun:                toggleDebug,
 	Args:                  cobra.MinimumNArgs(1),
 	Run: func(cmd *cobra.Command, args []string) {
-		// secretType, err := cmd.Flags().GetString("type")
-		// if err != nil {
-		// 	log.Errorln("Unable to parse the secret type flag")
-		// 	log.Debugln(err)
-		// 	return
-		// }
-
-		// if !util.IsSecretTypeValid(secretType) {
-		// 	log.Errorf("secret type can only be `personal` or `shared`. You have entered [%v]", secretType)
-		// 	return
-		// }
-
 		environmentName, err := cmd.Flags().GetString("env")
 		if err != nil {
-			log.Errorln("Unable to parse the environment name flag")
-			log.Debugln(err)
-			return
+			util.HandleError(err, "Unable to parse flag")
 		}
 
 		if !util.IsSecretEnvironmentValid(environmentName) {
-			log.Errorln("You have entered a invalid environment name. Environment names can only be prod, dev, test or staging")
-			return
-		}
-
-		workspaceFileExists := util.WorkspaceConfigFileExistsInCurrentPath()
-		if !workspaceFileExists {
-			log.Error("You have not yet connected to an Infisical Project. Please run [infisical init]")
-			return
+			util.PrintMessageAndExit("You have entered a invalid environment name", "Environment names can only be prod, dev, test or staging")
 		}
 
 		workspaceFile, err := util.GetWorkSpaceFromFile()
 		if err != nil {
-			log.Error(err)
-			return
+			util.HandleError(err, "Unable to get your local config details")
 		}
 
 		loggedInUserDetails, err := util.GetCurrentLoggedInUserDetails()
 		if err != nil {
-			log.Error(err)
-			return
-		}
-
-		if !loggedInUserDetails.IsUserLoggedIn {
-			log.Error("You are not logged in yet. Please run [infisical login] then try again")
-			return
-		}
-
-		if loggedInUserDetails.IsUserLoggedIn && loggedInUserDetails.LoginExpired {
-			log.Error("Your login has expired. Please run [infisical login] then try again")
-			return
+			util.HandleError(err, "Unable to authenticate")
 		}
 
 		httpClient := resty.New().
 			SetAuthToken(loggedInUserDetails.UserCredentials.JTWToken).
 			SetHeader("Accept", "application/json")
 
-		request := models.GetEncryptedWorkspaceKeyRequest{
+		request := api.GetEncryptedWorkspaceKeyRequest{
 			WorkspaceId: workspaceFile.WorkspaceId,
 		}
 
-		workspaceKeyResponse, err := http.CallGetEncryptedWorkspaceKey(httpClient, request)
+		workspaceKeyResponse, err := api.CallGetEncryptedWorkspaceKey(httpClient, request)
 		if err != nil {
-			log.Errorf("unable to get your encrypted workspace key. [err=%v]", err)
-			return
+			util.HandleError(err, "unable to get your encrypted workspace key")
 		}
 
 		encryptedWorkspaceKey, _ := base64.StdEncoding.DecodeString(workspaceKeyResponse.EncryptedKey)
@@ -157,10 +111,9 @@ var secretsSetCmd = &cobra.Command{
 		plainTextEncryptionKey := crypto.DecryptAsymmetric(encryptedWorkspaceKey, encryptedWorkspaceKeyNonce, encryptedWorkspaceKeySenderPublicKey, currentUsersPrivateKey)
 
 		// pull current secrets
-		secrets, err := util.GetAllEnvironmentVariables("", environmentName)
+		secrets, err := util.GetAllEnvironmentVariables(environmentName)
 		if err != nil {
-			log.Error("unable to retrieve secrets. Run with -d to see full logs")
-			log.Debug(err)
+			util.HandleError(err, "unable to retrieve secrets")
 		}
 
 		type SecretSetOperation struct {
@@ -169,8 +122,8 @@ var secretsSetCmd = &cobra.Command{
 			SecretOperation string
 		}
 
-		secretsToCreate := []models.Secret{}
-		secretsToModify := []models.Secret{}
+		secretsToCreate := []api.Secret{}
+		secretsToModify := []api.Secret{}
 		secretOperations := []SecretSetOperation{}
 
 		secretByKey := getSecretsByKeys(secrets)
@@ -178,13 +131,11 @@ var secretsSetCmd = &cobra.Command{
 		for _, arg := range args {
 			splitKeyValueFromArg := strings.SplitN(arg, "=", 2)
 			if splitKeyValueFromArg[0] == "" || splitKeyValueFromArg[1] == "" {
-				log.Error("ensure that each secret has a none empty key and value. Modify the input and try again")
-				return
+				util.PrintMessageAndExit("ensure that each secret has a none empty key and value. Modify the input and try again")
 			}
 
 			if unicode.IsNumber(rune(splitKeyValueFromArg[0][0])) {
-				log.Error("keys of secrets cannot start with a number. Modify the key name(s) and try again")
-				return
+				util.PrintMessageAndExit("keys of secrets cannot start with a number. Modify the key name(s) and try again")
 			}
 
 			// Key and value from argument
@@ -194,18 +145,18 @@ var secretsSetCmd = &cobra.Command{
 			hashedKey := fmt.Sprintf("%x", sha256.Sum256([]byte(key)))
 			encryptedKey, err := crypto.EncryptSymmetric([]byte(key), []byte(plainTextEncryptionKey))
 			if err != nil {
-				log.Errorf("unable to encrypt your secrets [err=%v]", err)
+				util.HandleError(err, "unable to encrypt your secrets")
 			}
 
 			hashedValue := fmt.Sprintf("%x", sha256.Sum256([]byte(value)))
 			encryptedValue, err := crypto.EncryptSymmetric([]byte(value), []byte(plainTextEncryptionKey))
 			if err != nil {
-				log.Errorf("unable to encrypt your secrets [err=%v]", err)
+				util.HandleError(err, "unable to encrypt your secrets")
 			}
 
 			if existingSecret, ok := secretByKey[key]; ok {
 				// case: secret exists in project so it needs to be modified
-				encryptedSecretDetails := models.Secret{
+				encryptedSecretDetails := api.Secret{
 					ID:                    existingSecret.ID,
 					SecretValueCiphertext: base64.StdEncoding.EncodeToString(encryptedValue.CipherText),
 					SecretValueIV:         base64.StdEncoding.EncodeToString(encryptedValue.Nonce),
@@ -232,7 +183,7 @@ var secretsSetCmd = &cobra.Command{
 
 			} else {
 				// case: secret doesn't exist in project so it needs to be created
-				encryptedSecretDetails := models.Secret{
+				encryptedSecretDetails := api.Secret{
 					SecretKeyCiphertext:   base64.StdEncoding.EncodeToString(encryptedKey.CipherText),
 					SecretKeyIV:           base64.StdEncoding.EncodeToString(encryptedKey.Nonce),
 					SecretKeyTag:          base64.StdEncoding.EncodeToString(encryptedKey.AuthTag),
@@ -253,29 +204,29 @@ var secretsSetCmd = &cobra.Command{
 		}
 
 		if len(secretsToCreate) > 0 {
-			batchCreateRequest := models.BatchCreateSecretsByWorkspaceAndEnvRequest{
+			batchCreateRequest := api.BatchCreateSecretsByWorkspaceAndEnvRequest{
 				WorkspaceId:     workspaceFile.WorkspaceId,
 				EnvironmentName: environmentName,
 				Secrets:         secretsToCreate,
 			}
 
-			err = http.CallBatchCreateSecretsByWorkspaceAndEnv(httpClient, batchCreateRequest)
+			err = api.CallBatchCreateSecretsByWorkspaceAndEnv(httpClient, batchCreateRequest)
 			if err != nil {
-				log.Errorf("Unable to process new secret creations because %v", err)
+				util.HandleError(err, "Unable to process new secret creations")
 				return
 			}
 		}
 
 		if len(secretsToModify) > 0 {
-			batchModifyRequest := models.BatchModifySecretsByWorkspaceAndEnvRequest{
+			batchModifyRequest := api.BatchModifySecretsByWorkspaceAndEnvRequest{
 				WorkspaceId:     workspaceFile.WorkspaceId,
 				EnvironmentName: environmentName,
 				Secrets:         secretsToModify,
 			}
 
-			err = http.CallBatchModifySecretsByWorkspaceAndEnv(httpClient, batchModifyRequest)
+			err = api.CallBatchModifySecretsByWorkspaceAndEnv(httpClient, batchModifyRequest)
 			if err != nil {
-				log.Errorf("Unable to process the modifications to your secrets because %v", err)
+				util.HandleError(err, "Unable to process the modifications to your secrets")
 				return
 			}
 		}
@@ -308,36 +259,17 @@ var secretsDeleteCmd = &cobra.Command{
 
 		loggedInUserDetails, err := util.GetCurrentLoggedInUserDetails()
 		if err != nil {
-			log.Error(err)
-			return
-		}
-
-		if !loggedInUserDetails.IsUserLoggedIn {
-			log.Error("You are not logged in yet. Please run [infisical login] then try again")
-			return
-		}
-
-		if loggedInUserDetails.IsUserLoggedIn && loggedInUserDetails.LoginExpired {
-			log.Error("Your login has expired. Please run [infisical login] then try again")
-			return
-		}
-
-		workspaceFileExists := util.WorkspaceConfigFileExistsInCurrentPath()
-		if !workspaceFileExists {
-			log.Error("You have not yet connected to an Infisical Project. Please run [infisical init]")
-			return
+			util.HandleError(err, "Unable to authenticate")
 		}
 
 		workspaceFile, err := util.GetWorkSpaceFromFile()
 		if err != nil {
-			log.Error(err)
-			return
+			util.HandleError(err, "Unable to get local project details")
 		}
 
-		secrets, err := util.GetAllEnvironmentVariables("", environmentName)
+		secrets, err := util.GetAllEnvironmentVariables(environmentName)
 		if err != nil {
-			log.Error("Unable to retrieve secrets. Run with -d to see full logs")
-			log.Debug(err)
+			util.HandleError(err, "Unable to fetch secrets")
 		}
 
 		secretByKey := getSecretsByKeys(secrets)
@@ -353,11 +285,11 @@ var secretsDeleteCmd = &cobra.Command{
 		}
 
 		if len(invalidSecretNamesThatDoNotExist) != 0 {
-			log.Errorf("secret name(s) [%v] does not exist in your project. To see which secrets exist run [infisical secrets]", strings.Join(invalidSecretNamesThatDoNotExist, ", "))
-			return
+			message := fmt.Sprintf("secret name(s) [%v] does not exist in your project. To see which secrets exist run [infisical secrets]", strings.Join(invalidSecretNamesThatDoNotExist, ", "))
+			util.PrintMessageAndExit(message)
 		}
 
-		request := models.BatchDeleteSecretsBySecretIdsRequest{
+		request := api.BatchDeleteSecretsBySecretIdsRequest{
 			WorkspaceId:     workspaceFile.WorkspaceId,
 			EnvironmentName: environmentName,
 			SecretIds:       validSecretIdsToDelete,
@@ -367,45 +299,43 @@ var secretsDeleteCmd = &cobra.Command{
 			SetAuthToken(loggedInUserDetails.UserCredentials.JTWToken).
 			SetHeader("Accept", "application/json")
 
-		err = http.CallBatchDeleteSecretsByWorkspaceAndEnv(httpClient, request)
+		err = api.CallBatchDeleteSecretsByWorkspaceAndEnv(httpClient, request)
 		if err != nil {
-			log.Errorf("Unable to complete your request because %v", err)
-			return
+			util.HandleError(err, "Unable to complete your batch delete request")
 		}
 
-		log.Infof("secret name(s) [%v] have been deleted from your project", strings.Join(args, ", "))
+		fmt.Printf("secret name(s) [%v] have been deleted from your project \n", strings.Join(args, ", "))
 
 	},
 }
 
 func init() {
 	secretsCmd.AddCommand(secretsGetCmd)
-	// secretsSetCmd.Flags().String("type", "shared", "Used to set the type for secrets")
 	secretsCmd.AddCommand(secretsSetCmd)
 	secretsCmd.AddCommand(secretsDeleteCmd)
 	secretsCmd.PersistentFlags().String("env", "dev", "Used to define the environment name on which actions should be taken on")
 	secretsCmd.Flags().Bool("expand", true, "Parse shell parameter expansions in your secrets")
+	secretsCmd.PersistentPreRun = func(cmd *cobra.Command, args []string) {
+		util.RequireLogin()
+		util.RequireLocalWorkspaceFile()
+	}
 	rootCmd.AddCommand(secretsCmd)
 }
 
 func getSecretsByNames(cmd *cobra.Command, args []string) {
 	environmentName, err := cmd.Flags().GetString("env")
 	if err != nil {
-		log.Errorln("Unable to parse the environment name flag")
-		log.Debugln(err)
-		return
+		util.HandleError(err, "Unable to parse flag")
 	}
 
 	workspaceFileExists := util.WorkspaceConfigFileExistsInCurrentPath()
 	if !workspaceFileExists {
-		log.Error("You have not yet connected to an Infisical Project. Please run [infisical init]")
-		return
+		util.HandleError(err, "Unable to parse flag")
 	}
 
-	secrets, err := util.GetAllEnvironmentVariables("", environmentName)
+	secrets, err := util.GetAllEnvironmentVariables(environmentName)
 	if err != nil {
-		log.Error("Unable to retrieve secrets. Run with -d to see full logs")
-		log.Debug(err)
+		util.HandleError(err, "To fetch all secrets")
 	}
 
 	requestedSecrets := []models.SingleEnvironmentVariable{}

cli/packages/models/cli.go

@@ -21,6 +21,14 @@ type SingleEnvironmentVariable struct {
 	ID    string `json:"_id"`
 }
 
+type Workspace struct {
+	ID           string `json:"_id"`
+	Name         string `json:"name"`
+	Plan         string `json:"plan,omitempty"`
+	V            int    `json:"__v"`
+	Organization string `json:"organization,omitempty"`
+}
+
 type WorkspaceConfigFile struct {
 	WorkspaceId string `json:"workspaceId"`
 }

cli/packages/util/common.go

@@ -5,15 +5,6 @@ import (
 	"os"
 )
 
-const (
-	CONFIG_FILE_NAME                     = "infisical-config.json"
-	CONFIG_FOLDER_NAME                   = ".infisical"
-	INFISICAL_WORKSPACE_CONFIG_FILE_NAME = ".infisical.json"
-	INFISICAL_TOKEN_NAME                 = "INFISICAL_TOKEN"
-	SECRET_TYPE_PERSONAL                 = "personal"
-	SECRET_TYPE_SHARED                   = "shared"
-)
-
 func GetHomeDir() (string, error) {
 	directory, err := os.UserHomeDir()
 	return directory, err
@@ -23,7 +14,7 @@ func GetHomeDir() (string, error) {
 func WriteToFile(fileName string, dataToWrite []byte, filePerm os.FileMode) error {
 	err := os.WriteFile(fileName, dataToWrite, filePerm)
 	if err != nil {
-		return fmt.Errorf("Unable to wrote to file", err)
+		return fmt.Errorf("unable to wrote to file [err=%v]", err)
 	}
 
 	return nil

cli/packages/util/constants.go

@@ -0,0 +1,13 @@
+package util
+
+const (
+	CONFIG_FILE_NAME                     = "infisical-config.json"
+	CONFIG_FOLDER_NAME                   = ".infisical"
+	INFISICAL_WORKSPACE_CONFIG_FILE_NAME = ".infisical.json"
+	INFISICAL_TOKEN_NAME                 = "INFISICAL_TOKEN"
+	SECRET_TYPE_PERSONAL                 = "personal"
+	SECRET_TYPE_SHARED                   = "shared"
+	KEYRING_SERVICE_NAME                 = "infisical"
+	PERSONAL_SECRET_TYPE_NAME            = "personal"
+	SHARED_SECRET_TYPE_NAME              = "shared"
+)

cli/packages/util/credentials.go

@@ -8,18 +8,14 @@ import (
 	"github.com/Infisical/infisical-merge/packages/config"
 	"github.com/Infisical/infisical-merge/packages/models"
 	"github.com/go-resty/resty/v2"
-	log "github.com/sirupsen/logrus"
 )
 
-const SERVICE_NAME = "infisical"
-
 type LoggedInUserDetails struct {
 	IsUserLoggedIn  bool
 	LoginExpired    bool
 	UserCredentials models.UserCredentials
 }
 
-// To do: what happens if the user doesn't have a keyring in their system?
 func StoreUserCredsInKeyRing(userCred *models.UserCredentials) error {
 	userCredMarshalled, err := json.Marshal(userCred)
 	if err != nil {
@@ -70,46 +66,6 @@ func GetUserCredsFromKeyRing(userEmail string) (credentials models.UserCredentia
 	return userCredentials, err
 }
 
-func IsUserLoggedIn() (hasUserLoggedIn bool, theUsersEmail string, err error) {
-	if ConfigFileExists() {
-		configFile, err := GetConfigFile()
-		if err != nil {
-			return false, "", fmt.Errorf("IsUserLoggedIn: unable to get logged in user from config file [err=%s]", err)
-		}
-
-		if configFile.LoggedInUserEmail == "" {
-			return false, "", nil
-		}
-
-		userCreds, err := GetUserCredsFromKeyRing(configFile.LoggedInUserEmail)
-		if err != nil {
-			return false, "", err
-		}
-
-		// check to to see if the JWT is still valid
-		httpClient := resty.New().
-			SetAuthToken(userCreds.JTWToken).
-			SetHeader("Accept", "application/json")
-
-		response, err := httpClient.
-			R().
-			Post(fmt.Sprintf("%v/v1/auth/checkAuth", config.INFISICAL_URL))
-
-		if err != nil {
-			return false, "", err
-		}
-
-		if response.StatusCode() > 299 {
-			log.Infoln("Login expired, please login again.")
-			return false, "", fmt.Errorf("GetUserCredsFromKeyRing: Login expired, please login again.")
-		}
-
-		return true, configFile.LoggedInUserEmail, nil
-	} else {
-		return false, "", nil
-	}
-}
-
 func GetCurrentLoggedInUserDetails() (LoggedInUserDetails, error) {
 	if ConfigFileExists() {
 		configFile, err := GetConfigFile()

cli/packages/util/errors.go

@@ -0,0 +1,38 @@
+package util
+
+import (
+	"fmt"
+	"os"
+
+	"github.com/fatih/color"
+)
+
+func HandleError(err error, messages ...string) {
+	PrintErrorAndExit(1, err, messages...)
+}
+
+func PrintErrorAndExit(exitCode int, err error, messages ...string) {
+	printError(err)
+
+	if len(messages) > 0 {
+		for _, message := range messages {
+			fmt.Println(message)
+		}
+	}
+
+	os.Exit(exitCode)
+}
+
+func PrintMessageAndExit(messages ...string) {
+	if len(messages) > 0 {
+		for _, message := range messages {
+			fmt.Println(message)
+		}
+	}
+
+	os.Exit(1)
+}
+
+func printError(e error) {
+	color.Red("Hmm, we ran into an error: %v", e)
+}

cli/packages/util/secrets.go

@@ -7,187 +7,15 @@ import (
 	"regexp"
 	"strings"
 
-	"github.com/Infisical/infisical-merge/packages/config"
+	"github.com/Infisical/infisical-merge/packages/api"
 	"github.com/Infisical/infisical-merge/packages/crypto"
-	"github.com/Infisical/infisical-merge/packages/http"
 	"github.com/Infisical/infisical-merge/packages/models"
+	log "github.com/sirupsen/logrus"
 
 	"github.com/go-resty/resty/v2"
-	log "github.com/sirupsen/logrus"
 )
 
-const PERSONAL_SECRET_TYPE_NAME = "personal"
-const SHARED_SECRET_TYPE_NAME = "shared"
-
-func getSecretsByWorkspaceIdAndEnvName(httpClient resty.Client, envName string, workspace models.WorkspaceConfigFile, userCreds models.UserCredentials) (listOfSecrets []models.SingleEnvironmentVariable, err error) {
-	var pullSecretsRequestResponse models.PullSecretsResponse
-	response, err := httpClient.
-		R().
-		SetQueryParam("environment", envName).
-		SetQueryParam("channel", "cli").
-		SetResult(&pullSecretsRequestResponse).
-		Get(fmt.Sprintf("%v/v1/secret/%v", config.INFISICAL_URL, workspace.WorkspaceId)) // need to change workspace id
-
-	if err != nil {
-		return nil, err
-	}
-
-	if response.StatusCode() > 299 {
-		return nil, fmt.Errorf(response.Status())
-	}
-
-	// Get workspace key
-	workspaceKey, err := base64.StdEncoding.DecodeString(pullSecretsRequestResponse.Key.EncryptedKey)
-	if err != nil {
-		return nil, err
-	}
-
-	nonce, err := base64.StdEncoding.DecodeString(pullSecretsRequestResponse.Key.Nonce)
-	if err != nil {
-		return nil, err
-	}
-
-	senderPublicKey, err := base64.StdEncoding.DecodeString(pullSecretsRequestResponse.Key.Sender.PublicKey)
-	if err != nil {
-		return nil, err
-	}
-
-	currentUsersPrivateKey, err := base64.StdEncoding.DecodeString(userCreds.PrivateKey)
-	if err != nil {
-		return nil, err
-	}
-
-	// log.Debugln("workspaceKey", workspaceKey, "nonce", nonce, "senderPublicKey", senderPublicKey, "currentUsersPrivateKey", currentUsersPrivateKey)
-	workspaceKeyInBytes := crypto.DecryptAsymmetric(workspaceKey, nonce, senderPublicKey, currentUsersPrivateKey)
-	var listOfEnv []models.SingleEnvironmentVariable
-
-	for _, secret := range pullSecretsRequestResponse.Secrets {
-		key_iv, _ := base64.StdEncoding.DecodeString(secret.SecretKeyIV)
-		key_tag, _ := base64.StdEncoding.DecodeString(secret.SecretKeyTag)
-		key_ciphertext, _ := base64.StdEncoding.DecodeString(secret.SecretKeyCiphertext)
-
-		plainTextKey, err := crypto.DecryptSymmetric(workspaceKeyInBytes, key_ciphertext, key_tag, key_iv)
-		if err != nil {
-			return nil, err
-		}
-
-		value_iv, _ := base64.StdEncoding.DecodeString(secret.SecretValueIV)
-		value_tag, _ := base64.StdEncoding.DecodeString(secret.SecretValueTag)
-		value_ciphertext, _ := base64.StdEncoding.DecodeString(secret.SecretValueCiphertext)
-
-		plainTextValue, err := crypto.DecryptSymmetric(workspaceKeyInBytes, value_ciphertext, value_tag, value_iv)
-		if err != nil {
-			return nil, err
-		}
-
-		env := models.SingleEnvironmentVariable{
-			Key:   string(plainTextKey),
-			Value: string(plainTextValue),
-			Type:  string(secret.Type),
-			ID:    secret.ID,
-		}
-
-		listOfEnv = append(listOfEnv, env)
-	}
-
-	return listOfEnv, nil
-}
-
-func GetSecretsFromAPIUsingCurrentLoggedInUser(envName string, userCreds models.UserCredentials) ([]models.SingleEnvironmentVariable, error) {
-	log.Debugln("GetSecretsFromAPIUsingCurrentLoggedInUser", "envName", envName, "userCreds", userCreds)
-	// check if user has configured a workspace
-	workspaces, err := GetAllWorkSpaceConfigsStartingFromCurrentPath()
-	if err != nil {
-		return nil, fmt.Errorf("Unable to read workspace file(s):", err)
-	}
-
-	// create http client
-	httpClient := resty.New().
-		SetAuthToken(userCreds.JTWToken).
-		SetHeader("Accept", "application/json")
-
-	secrets := []models.SingleEnvironmentVariable{}
-	for _, workspace := range workspaces {
-		secretsFromAPI, err := getSecretsByWorkspaceIdAndEnvName(*httpClient, envName, workspace, userCreds)
-		if err != nil {
-			return nil, fmt.Errorf("GetSecretsFromAPIUsingCurrentLoggedInUser: Unable to get secrets by workspace id and env name")
-		}
-
-		secrets = append(secrets, secretsFromAPI...)
-	}
-
-	return secrets, nil
-}
-
-// func GetPlainTextSecretsViaJWT(JTWToken string, privateKey string, workspaceId string, environment string) ([]models.SingleEnvironmentVariable, error) {
-
-// 		// get key first
-
-// 		// Get workspace key
-// 		workspaceKey, err := base64.StdEncoding.DecodeString(pullSecretsRequestResponse.Key.EncryptedKey)
-// 		if err != nil {
-// 			return nil, err
-// 		}
-
-// 		nonce, err := base64.StdEncoding.DecodeString(pullSecretsRequestResponse.Key.Nonce)
-// 		if err != nil {
-// 			return nil, err
-// 		}
-
-// 		senderPublicKey, err := base64.StdEncoding.DecodeString(pullSecretsRequestResponse.Key.Sender.PublicKey)
-// 		if err != nil {
-// 			return nil, err
-// 		}
-
-// 		currentUsersPrivateKey, err := base64.StdEncoding.DecodeString(userCreds.PrivateKey)
-// 		if err != nil {
-// 			return nil, err
-// 		}
-
-// 		// log.Debugln("workspaceKey", workspaceKey, "nonce", nonce, "senderPublicKey", senderPublicKey, "currentUsersPrivateKey", currentUsersPrivateKey)
-// 		workspaceKeyInBytes := crypto.DecryptAsymmetric(workspaceKey, nonce, senderPublicKey, currentUsersPrivateKey)
-
-// 	httpClient := resty.New()
-// 	httpClient.SetAuthToken(JTWToken).
-// 		SetHeader("Accept", "application/json")
-
-// 	serviceTokenDetails, err := http.CallGetServiceTokenDetailsV2(httpClient)
-// 	if err != nil {
-// 		return nil, fmt.Errorf("unable to get service token details. [err=%v]", err)
-// 	}
-
-// 	encryptedSecrets, err := http.CallGetSecretsV2(httpClient, models.GetEncryptedSecretsV2Request{
-// 		WorkspaceId:     workspaceId,
-// 		EnvironmentName: environment,
-// 	})
-
-// 	if err != nil {
-// 		return nil, err
-// 	}
-
-// 	// workspace key
-
-// 	decodedSymmetricEncryptionDetails, err := GetBase64DecodedSymmetricEncryptionDetails(serviceTokenParts[3], serviceTokenDetails.EncryptedKey, serviceTokenDetails.Iv, serviceTokenDetails.Tag)
-// 	if err != nil {
-// 		return nil, fmt.Errorf("unable to decode symmetric encryption details [err=%v]", err)
-// 	}
-
-// 	plainTextWorkspaceKey, err := crypto.DecryptSymmetric(decodedSymmetricEncryptionDetails.Key, decodedSymmetricEncryptionDetails.Cipher, decodedSymmetricEncryptionDetails.Tag, decodedSymmetricEncryptionDetails.IV)
-// 	if err != nil {
-// 		return nil, fmt.Errorf("unable to decrypt the required workspace key")
-// 	}
-
-// 	plainTextSecrets, err := GetPlainTextSecrets(plainTextWorkspaceKey, encryptedSecrets)
-// 	if err != nil {
-// 		return nil, fmt.Errorf("unable to decrypt your secrets [err=%v]", err)
-// 	}
-
-// 	return plainTextSecrets, nil
-// }
-
 func GetPlainTextSecretsViaServiceToken(fullServiceToken string) ([]models.SingleEnvironmentVariable, error) {
-	// encryotion key, http,
-
 	serviceTokenParts := strings.SplitN(fullServiceToken, ".", 4)
 	if len(serviceTokenParts) < 4 {
 		return nil, fmt.Errorf("invalid service token entered. Please double check your service token and try again")
@@ -199,12 +27,12 @@ func GetPlainTextSecretsViaServiceToken(fullServiceToken string) ([]models.Singl
 	httpClient.SetAuthToken(serviceToken).
 		SetHeader("Accept", "application/json")
 
-	serviceTokenDetails, err := http.CallGetServiceTokenDetailsV2(httpClient)
+	serviceTokenDetails, err := api.CallGetServiceTokenDetailsV2(httpClient)
 	if err != nil {
 		return nil, fmt.Errorf("unable to get service token details. [err=%v]", err)
 	}
 
-	encryptedSecrets, err := http.CallGetSecretsV2(httpClient, models.GetEncryptedSecretsV2Request{
+	encryptedSecrets, err := api.CallGetSecretsV2(httpClient, api.GetEncryptedSecretsV2Request{
 		WorkspaceId:     serviceTokenDetails.Workspace,
 		EnvironmentName: serviceTokenDetails.Environment,
 	})
@@ -231,75 +59,68 @@ func GetPlainTextSecretsViaServiceToken(fullServiceToken string) ([]models.Singl
 	return plainTextSecrets, nil
 }
 
-func GetAllEnvironmentVariables(projectId string, envName string) ([]models.SingleEnvironmentVariable, error) {
-	infisicalToken := os.Getenv(INFISICAL_TOKEN_NAME)
-
-	if infisicalToken == "" {
-		hasUserLoggedInbefore, loggedInUserEmail, err := IsUserLoggedIn()
-		if err != nil {
-			log.Info("Unexpected issue occurred while checking login status. To see more details, add flag --debug")
-			log.Debugln(err)
-			return nil, err
-		}
-
-		if !hasUserLoggedInbefore {
-			log.Infoln("No logged in user. To login, please run command [infisical login]")
-			return nil, fmt.Errorf("user not logged in")
-		}
-
-		userCreds, err := GetUserCredsFromKeyRing(loggedInUserEmail)
-		if err != nil {
-			log.Infoln("Unable to get user creds from key ring")
-			log.Debug(err)
-			return nil, err
-		}
-
-		// TODO: Should be based on flag. I.e only get all workspaces if desired, otherwise only get the one in the current root of project
-		workspaceConfigs, err := GetAllWorkSpaceConfigsStartingFromCurrentPath()
-		if err != nil {
-			return nil, fmt.Errorf("unable to check if you have a %s file in your current directory", INFISICAL_WORKSPACE_CONFIG_FILE_NAME)
-		}
-
-		if len(workspaceConfigs) == 0 {
-			log.Infoln("Your local project is not connected to a Infisical project yet. Run command [infisical init]")
-			return nil, fmt.Errorf("project not initialized")
-		}
-
-		envsFromApi, err := GetSecretsFromAPIUsingCurrentLoggedInUser(envName, userCreds)
-		if err != nil {
-			log.Errorln("Something went wrong when pulling secrets using your logged in credentials. If the issue persists, double check your project id/try logging in again.")
-			log.Debugln(err)
-			return nil, err
-		}
-
-		return envsFromApi, nil
-
-	} else {
-		return GetPlainTextSecretsViaServiceToken(infisicalToken)
-	}
-}
-
-func GetWorkSpacesFromAPI(userCreds models.UserCredentials) (workspaces []models.Workspace, err error) {
-	// create http client
-	httpClient := resty.New().
-		SetAuthToken(userCreds.JTWToken).
+func GetPlainTextSecretsViaJTW(JTWToken string, receiversPrivateKey string, workspaceId string, environmentName string) ([]models.SingleEnvironmentVariable, error) {
+	httpClient := resty.New()
+	httpClient.SetAuthToken(JTWToken).
 		SetHeader("Accept", "application/json")
 
-	var getWorkSpacesResponse models.GetWorkSpacesResponse
-	response, err := httpClient.
-		R().
-		SetResult(&getWorkSpacesResponse).
-		Get(fmt.Sprintf("%v/v1/workspace", config.INFISICAL_URL))
+	request := api.GetEncryptedWorkspaceKeyRequest{
+		WorkspaceId: workspaceId,
+	}
+
+	workspaceKeyResponse, err := api.CallGetEncryptedWorkspaceKey(httpClient, request)
+	if err != nil {
+		return nil, fmt.Errorf("unable to get your encrypted workspace key. [err=%v]", err)
+	}
+
+	encryptedWorkspaceKey, _ := base64.StdEncoding.DecodeString(workspaceKeyResponse.EncryptedKey)
+	encryptedWorkspaceKeySenderPublicKey, _ := base64.StdEncoding.DecodeString(workspaceKeyResponse.Sender.PublicKey)
+	encryptedWorkspaceKeyNonce, _ := base64.StdEncoding.DecodeString(workspaceKeyResponse.Nonce)
+	currentUsersPrivateKey, _ := base64.StdEncoding.DecodeString(receiversPrivateKey)
+	plainTextWorkspaceKey := crypto.DecryptAsymmetric(encryptedWorkspaceKey, encryptedWorkspaceKeyNonce, encryptedWorkspaceKeySenderPublicKey, currentUsersPrivateKey)
+
+	encryptedSecrets, err := api.CallGetSecretsV2(httpClient, api.GetEncryptedSecretsV2Request{
+		WorkspaceId:     workspaceId,
+		EnvironmentName: environmentName,
+	})
 
 	if err != nil {
 		return nil, err
 	}
 
-	if response.StatusCode() > 299 {
-		return nil, fmt.Errorf("ops, unsuccessful response code. [response=%v]", response)
+	plainTextSecrets, err := GetPlainTextSecrets(plainTextWorkspaceKey, encryptedSecrets)
+	if err != nil {
+		return nil, fmt.Errorf("unable to decrypt your secrets [err=%v]", err)
 	}
 
-	return getWorkSpacesResponse.Workspaces, nil
+	return plainTextSecrets, nil
+}
+
+func GetAllEnvironmentVariables(envName string) ([]models.SingleEnvironmentVariable, error) {
+	infisicalToken := os.Getenv(INFISICAL_TOKEN_NAME)
+
+	if infisicalToken == "" {
+		RequireLocalWorkspaceFile()
+		RequireLogin()
+		log.Debug("Trying to fetch secrets using logged in details")
+
+		loggedInUserDetails, err := GetCurrentLoggedInUserDetails()
+		if err != nil {
+			return nil, err
+		}
+
+		workspaceFile, err := GetWorkSpaceFromFile()
+		if err != nil {
+			return nil, err
+		}
+
+		secrets, err := GetPlainTextSecretsViaJTW(loggedInUserDetails.UserCredentials.JTWToken, loggedInUserDetails.UserCredentials.PrivateKey, workspaceFile.WorkspaceId, envName)
+		return secrets, err
+
+	} else {
+		log.Debug("Trying to fetch secrets using service token")
+		return GetPlainTextSecretsViaServiceToken(infisicalToken)
+	}
 }
 
 func getExpandedEnvVariable(secrets []models.SingleEnvironmentVariable, variableWeAreLookingFor string, hashMapOfCompleteVariables map[string]string, hashMapOfSelfRefs map[string]string) string {
@@ -373,6 +194,8 @@ func SubstituteSecrets(secrets []models.SingleEnvironmentVariable) []models.Sing
 	return expandedSecrets
 }
 
+//
+
 // if two secrets with the same name are found, the one that has type `personal` will be in the returned list
 func OverrideWithPersonalSecrets(secrets []models.SingleEnvironmentVariable) []models.SingleEnvironmentVariable {
 	personalSecret := make(map[string]models.SingleEnvironmentVariable)
@@ -381,37 +204,27 @@ func OverrideWithPersonalSecrets(secrets []models.SingleEnvironmentVariable) []m
 
 	for _, secret := range secrets {
 		if secret.Type == PERSONAL_SECRET_TYPE_NAME {
-			personalSecret[secret.Key] = models.SingleEnvironmentVariable{
-				Key:   secret.Key,
-				Value: secret.Value,
-				Type:  secret.Type,
-			}
+			personalSecret[secret.Key] = secret
 		}
 
 		if secret.Type == SHARED_SECRET_TYPE_NAME {
-			sharedSecret[secret.Key] = models.SingleEnvironmentVariable{
-				Key:   secret.Key,
-				Value: secret.Value,
-				Type:  secret.Type,
-			}
+			sharedSecret[secret.Key] = secret
 		}
 	}
 
-	for _, secret := range secrets {
+	for _, secret := range sharedSecret {
 		personalValue, personalExists := personalSecret[secret.Key]
-		sharedValue, sharedExists := sharedSecret[secret.Key]
-
-		if personalExists && sharedExists || personalExists && !sharedExists {
+		if personalExists {
 			secretsToReturn = append(secretsToReturn, personalValue)
 		} else {
-			secretsToReturn = append(secretsToReturn, sharedValue)
+			secretsToReturn = append(secretsToReturn, secret)
 		}
 	}
 
 	return secretsToReturn
 }
 
-func GetPlainTextSecrets(key []byte, encryptedSecrets models.GetEncryptedSecretsV2Response) ([]models.SingleEnvironmentVariable, error) {
+func GetPlainTextSecrets(key []byte, encryptedSecrets api.GetEncryptedSecretsV2Response) ([]models.SingleEnvironmentVariable, error) {
 	plainTextSecrets := []models.SingleEnvironmentVariable{}
 	for _, secret := range encryptedSecrets {
 		// Decrypt key
@@ -468,28 +281,3 @@ func GetPlainTextSecrets(key []byte, encryptedSecrets models.GetEncryptedSecrets
 
 	return plainTextSecrets, nil
 }
-
-// func GetWorkspaceKeyViaEnvelopeEncryption(JTWToken string, workspaceId string, privateKey string) {
-// 	httpClient := resty.New().
-// 		SetAuthToken(JTWToken).
-// 		SetHeader("Accept", "application/json")
-
-// 	request := models.GetEncryptedWorkspaceKeyRequest{
-// 		WorkspaceId: workspaceId,
-// 	}
-
-// 	workspaceKeyResponse, err := http.CallGetEncryptedWorkspaceKey(httpClient, request)
-// 	if err != nil {
-// 		log.Errorf("unable to get your encrypted workspace key. [err=%v]", err)
-// 		return
-// 	}
-
-// 	encryptedWorkspaceKey, _ := base64.StdEncoding.DecodeString(workspaceKeyResponse.EncryptedKey)
-// 	encryptedWorkspaceKeySenderPublicKey, _ := base64.StdEncoding.DecodeString(workspaceKeyResponse.Sender.PublicKey)
-// 	encryptedWorkspaceKeyNonce, _ := base64.StdEncoding.DecodeString(workspaceKeyResponse.Nonce)
-// 	currentUsersPrivateKey, _ := base64.StdEncoding.DecodeString(loggedInUserDetails.UserCredentials.PrivateKey)
-
-// 	// decrypt workspace key
-// 	plainTextEncryptionKey := crypto.DecryptAsymmetric(encryptedWorkspaceKey, encryptedWorkspaceKeyNonce, encryptedWorkspaceKeySenderPublicKey, currentUsersPrivateKey)
-
-// }

cli/packages/util/vault.go

@@ -29,13 +29,13 @@ func GetKeyRing() (keyring.Keyring, error) {
 
 	keyringInstanceConfig := keyring.Config{
 		FilePasswordFunc:               fileKeyringPassphrasePrompt,
-		ServiceName:                    SERVICE_NAME,
-		LibSecretCollectionName:        SERVICE_NAME,
-		KWalletAppID:                   SERVICE_NAME,
-		KWalletFolder:                  SERVICE_NAME,
+		ServiceName:                    KEYRING_SERVICE_NAME,
+		LibSecretCollectionName:        KEYRING_SERVICE_NAME,
+		KWalletAppID:                   KEYRING_SERVICE_NAME,
+		KWalletFolder:                  KEYRING_SERVICE_NAME,
 		KeychainTrustApplication:       true,
-		WinCredPrefix:                  SERVICE_NAME,
-		FileDir:                        fmt.Sprintf("~/%s-file-vault", SERVICE_NAME),
+		WinCredPrefix:                  KEYRING_SERVICE_NAME,
+		FileDir:                        fmt.Sprintf("~/%s-file-vault", KEYRING_SERVICE_NAME),
 		KeychainAccessibleWhenUnlocked: true,
 	}