mirror of
https://github.com/awatertrevi/infisical.git
synced 2026-09-22 13:39:35 +00:00
114 lines
3.8 KiB
Go
114 lines
3.8 KiB
Go
/*
|
|
Copyright (c) 2023 Infisical Inc.
|
|
*/
|
|
package cmd
|
|
|
|
import (
|
|
"encoding/base64"
|
|
"fmt"
|
|
"strings"
|
|
|
|
"github.com/Infisical/infisical-merge/packages/util"
|
|
"github.com/posthog/posthog-go"
|
|
"github.com/rs/zerolog/log"
|
|
"github.com/spf13/cobra"
|
|
)
|
|
|
|
type VaultBackendType struct {
|
|
Name string
|
|
Description string
|
|
}
|
|
|
|
var AvailableVaults = []VaultBackendType{
|
|
{
|
|
Name: "auto",
|
|
Description: "automatically select the system keyring",
|
|
},
|
|
{
|
|
Name: "file",
|
|
Description: "encrypted file vault",
|
|
},
|
|
}
|
|
|
|
var vaultSetCmd = &cobra.Command{
|
|
Example: `infisical vault set file`,
|
|
Use: "set [file|auto]",
|
|
Short: "Used to configure the vault backends",
|
|
DisableFlagsInUseLine: true,
|
|
Args: cobra.MinimumNArgs(1),
|
|
Run: func(cmd *cobra.Command, args []string) {
|
|
wantedVaultTypeName := args[0]
|
|
currentVaultBackend, err := util.GetCurrentVaultBackend()
|
|
if err != nil {
|
|
log.Error().Msgf("Unable to set vault to [%s] because of [err=%s]", wantedVaultTypeName, err)
|
|
return
|
|
}
|
|
|
|
if wantedVaultTypeName == string(currentVaultBackend) {
|
|
log.Error().Msgf("You are already on vault backend [%s]", currentVaultBackend)
|
|
return
|
|
}
|
|
|
|
if wantedVaultTypeName == util.VAULT_BACKEND_AUTO_MODE || wantedVaultTypeName == util.VAULT_BACKEND_FILE_MODE {
|
|
configFile, err := util.GetConfigFile()
|
|
if err != nil {
|
|
log.Error().Msgf("Unable to set vault to [%s] because of [err=%s]", wantedVaultTypeName, err)
|
|
return
|
|
}
|
|
|
|
configFile.VaultBackendType = wantedVaultTypeName
|
|
configFile.LoggedInUserEmail = ""
|
|
configFile.VaultBackendPassphrase = base64.StdEncoding.EncodeToString([]byte(util.GenerateRandomString(10)))
|
|
|
|
err = util.WriteConfigFile(&configFile)
|
|
if err != nil {
|
|
log.Error().Msgf("Unable to set vault to [%s] because an error occurred when saving the config file [err=%s]", wantedVaultTypeName, err)
|
|
return
|
|
}
|
|
|
|
fmt.Printf("\nSuccessfully, switched vault backend from [%s] to [%s]. Please login in again to store your login details in the new vault with [infisical login]\n", currentVaultBackend, wantedVaultTypeName)
|
|
|
|
Telemetry.CaptureEvent("cli-command:vault set", posthog.NewProperties().Set("currentVault", currentVaultBackend).Set("wantedVault", wantedVaultTypeName).Set("version", util.CLI_VERSION))
|
|
} else {
|
|
var availableVaultsNames []string
|
|
for _, vault := range AvailableVaults {
|
|
availableVaultsNames = append(availableVaultsNames, vault.Name)
|
|
}
|
|
log.Error().Msgf("The requested vault type [%s] is not available on this system. Only the following vault backends are available for you system: %s", wantedVaultTypeName, strings.Join(availableVaultsNames, ", "))
|
|
}
|
|
},
|
|
}
|
|
|
|
// runCmd represents the run command
|
|
var vaultCmd = &cobra.Command{
|
|
Use: "vault",
|
|
Short: "Used to manage where your Infisical login token is saved on your machine",
|
|
DisableFlagsInUseLine: true,
|
|
Args: cobra.NoArgs,
|
|
Run: func(cmd *cobra.Command, args []string) {
|
|
printAvailableVaultBackends()
|
|
},
|
|
}
|
|
|
|
func printAvailableVaultBackends() {
|
|
fmt.Printf("Vaults are used to securely store your login details locally. Available vaults:")
|
|
for _, vaultType := range AvailableVaults {
|
|
fmt.Printf("\n- %s (%s)", vaultType.Name, vaultType.Description)
|
|
}
|
|
|
|
currentVaultBackend, err := util.GetCurrentVaultBackend()
|
|
if err != nil {
|
|
log.Error().Msgf("printAvailableVaultBackends: unable to print the available vault backend because of error [err=%s]", err)
|
|
}
|
|
|
|
Telemetry.CaptureEvent("cli-command:vault", posthog.NewProperties().Set("currentVault", currentVaultBackend).Set("version", util.CLI_VERSION))
|
|
|
|
fmt.Printf("\n\nYou are currently using [%s] vault to store your login credentials\n", string(currentVaultBackend))
|
|
}
|
|
|
|
func init() {
|
|
vaultCmd.AddCommand(vaultSetCmd)
|
|
|
|
rootCmd.AddCommand(vaultCmd)
|
|
}
|