add print --export option

jippi · Apr 4, 2024 · f532915 · f532915
1 parent d18463b
commit f532915
Show file tree

Hide file tree

Showing 5 changed files with 20 additions and 0 deletions.
diff --git a/cmd/print/print.go b/cmd/print/print.go
@@ -20,6 +20,7 @@ func NewCommand() *cobra.Command {
 	}
 
 	cmd.Flags().Bool("pretty", false, "implies --color --comments --blank-lines --group-banners")
+	cmd.Flags().Bool("export", false, "prefix all key/value pairs with [export] statement")
 
 	cmd.Flags().String("key-prefix", "", "Filter by key prefix")
 	cmd.Flags().String("group", "", "Filter by group name")
@@ -99,5 +100,9 @@ func setup(cmd *cobra.Command) (*ast.Document, *render.Settings, error) {
 		settings.Apply(render.WithFormattedOutput(true))
 	}
 
+	if boolFlag("export") {
+		settings.Apply(render.WithExport(true))
+	}
+
 	return doc, settings, allErrors
 }
diff --git a/pkg/render/output_colorized.go b/pkg/render/output_colorized.go
@@ -33,6 +33,10 @@ func (ColorizedOutput) Assignment(ctx context.Context, assignment *ast.Assignmen
 		printer.Danger().Print("#")
 	}
 
+	if settings.export {
+		printer.Dark().Print("export ")
+	}
+
 	val := assignment.Literal
 
 	if settings.InterpolatedValues {

diff --git a/pkg/render/output_plain.go b/pkg/render/output_plain.go
@@ -29,6 +29,10 @@ func (PlainOutput) Assignment(ctx context.Context, assignment *ast.Assignment, s
 		buf.WriteString("#")
 	}
 
+	if settings.export {
+		buf.WriteString("export ")
+	}
+
 	val := assignment.Literal
 
 	if settings.InterpolatedValues {

diff --git a/pkg/render/settings.go b/pkg/render/settings.go
@@ -19,6 +19,7 @@ type Settings struct {
 	showComments       bool
 	ShowGroupBanners   bool
 	formatOutput       bool
+	export             bool
 	InterpolatedValues bool
 	outputter          Output
 }

diff --git a/pkg/render/settings_options.go b/pkg/render/settings_options.go
@@ -59,6 +59,12 @@ func WithFormattedOutput(boolean bool) SettingsOption {
 	}
 }
 
+func WithExport(boolean bool) SettingsOption {
+	return func(s *Settings) {
+		s.export = boolean
+	}
+}
+
 func WithInterpolation(b bool) SettingsOption {
 	return func(s *Settings) {
 		s.InterpolatedValues = b