From 2aea73861b682ecbc2d236e4390646cc8fce1d0f Mon Sep 17 00:00:00 2001
From: Akhil Mohan <akhilmhdh@gmail.com>
Date: Thu, 18 Apr 2024 13:09:09 +0530
Subject: [PATCH] feat(cli): added template feature to cli export command

---
 cli/packages/cmd/export.go   | 32 ++++++++++++++++++++++++++++++++
 docs/cli/commands/export.mdx | 19 +++++++++++++++++++
 2 files changed, 51 insertions(+)

diff --git a/cli/packages/cmd/export.go b/cli/packages/cmd/export.go
index 4e08e5fa8..951d538ba 100644
--- a/cli/packages/cmd/export.go
+++ b/cli/packages/cmd/export.go
@@ -7,6 +7,7 @@ import (
 	"encoding/csv"
 	"encoding/json"
 	"fmt"
+	"os"
 	"strings"
 
 	"github.com/Infisical/infisical-merge/packages/models"
@@ -59,6 +60,11 @@ var exportCmd = &cobra.Command{
 			util.HandleError(err)
 		}
 
+		templatePath, err := cmd.Flags().GetString("template")
+		if err != nil {
+			util.HandleError(err)
+		}
+
 		secretOverriding, err := cmd.Flags().GetBool("secret-overriding")
 		if err != nil {
 			util.HandleError(err, "Unable to parse flag")
@@ -93,6 +99,31 @@ var exportCmd = &cobra.Command{
 			request.UniversalAuthAccessToken = token.Token
 		}
 
+		if templatePath != "" {
+			sigChan := make(chan os.Signal, 1)
+			dynamicSecretLeases := NewDynamicSecretLeaseManager(sigChan)
+			newEtag := ""
+
+			accessToken := ""
+			if token != nil {
+				accessToken = token.Token
+			} else {
+				log.Debug().Msg("GetAllEnvironmentVariables: Trying to fetch secrets using logged in details")
+				loggedInUserDetails, err := util.GetCurrentLoggedInUserDetails()
+				if err != nil {
+					util.HandleError(err)
+				}
+				accessToken = loggedInUserDetails.UserCredentials.JTWToken
+			}
+
+			processedTemplate, err := ProcessTemplate(1, templatePath, nil, accessToken, "", &newEtag, dynamicSecretLeases)
+			if err != nil {
+				util.HandleError(err)
+			}
+			fmt.Print(processedTemplate.String())
+			return
+		}
+
 		secrets, err := util.GetAllEnvironmentVariables(request, "")
 		if err != nil {
 			util.HandleError(err, "Unable to fetch secrets")
@@ -140,6 +171,7 @@ func init() {
 	exportCmd.Flags().StringP("tags", "t", "", "filter secrets by tag slugs")
 	exportCmd.Flags().String("projectId", "", "manually set the projectId to fetch secrets from")
 	exportCmd.Flags().String("path", "/", "get secrets within a folder path")
+	exportCmd.Flags().String("template", "", "The path to the template file to output secrets")
 }
 
 // Format according to the format flag
diff --git a/docs/cli/commands/export.mdx b/docs/cli/commands/export.mdx
index 49446e734..246c674cd 100644
--- a/docs/cli/commands/export.mdx
+++ b/docs/cli/commands/export.mdx
@@ -33,6 +33,9 @@ Export environment variables from the platform into a file format.
 
   # Export variables to a YAML file
   infisical export --format=yaml > secrets.yaml
+
+  # Export variables from a template
+  infisical export --template=<path to template>
   ```
 
   ### Environment variables
@@ -57,6 +60,22 @@ Export environment variables from the platform into a file format.
   </Accordion>
 
   ### flags 
+  <Accordion title="--template">
+    The --template flag specifies the path to the template file used for rendering secrets. In template mode, you can omit the other flags since the settings are already configured within the template.
+
+    ```bash
+    # Example
+    infisical export --template="/path/to/template"
+    ```
+
+		```text my-dot-env-secret-template
+		{{- with secret "<projectId>" "<environment slug>" "/" }}
+		{{- range . }}
+		{{ .Key }}={{ .Value }}
+		{{- end }}
+		{{- end }}
+		```
+  </Accordion>
   <Accordion title="--env">
     Used to set the environment that secrets are pulled from.