fix: Ability to suppress columns from output

dg.go

@@ -16,6 +16,7 @@ import (
 	"github.com/codingconcepts/dg/internal/pkg/model"
 	"github.com/codingconcepts/dg/internal/pkg/source"
 	"github.com/codingconcepts/dg/internal/pkg/ui"
+	"github.com/samber/lo"
 )
 
 var (
@@ -64,6 +65,8 @@ func main() {
 		log.Fatalf("error generating tables: %v", err)
 	}
 
+	removeSuppressedColumns(c, tt, files)
+
 	if err := writeFiles(*outputDir, files, tt); err != nil {
 		log.Fatalf("error writing csv files: %v", err)
 	}
@@ -216,6 +219,33 @@ func generateTable(t model.Table, files map[string]model.CSVFile, tt ui.TimerFun
 	return nil
 }
 
+func removeSuppressedColumns(c model.Config, tt ui.TimerFunc, files map[string]model.CSVFile) {
+	for _, table := range c.Tables {
+		for _, column := range table.Columns {
+			if !column.Suppress {
+				continue
+			}
+
+			file := files[table.Name]
+
+			// Remove suppressed column from header.
+			var headerIndex int
+			file.Header = lo.Reject(file.Header, func(v string, i int) bool {
+				if v == column.Name {
+					headerIndex = i
+					return true
+				}
+				return false
+			})
+
+			// Remove suppressed column from lines.
+			file.Lines = append(file.Lines[:headerIndex], file.Lines[headerIndex+1:]...)
+
+			files[table.Name] = file
+		}
+	}
+}
+
 func writeFiles(outputDir string, cfs map[string]model.CSVFile, tt ui.TimerFunc) error {
 	defer tt(time.Now(), "wrote all csvs")
 

examples/supress_column/config.yaml

@@ -0,0 +1,16 @@
+tables:
+
+  - name: example
+    count: 1
+    columns:
+
+      - name: one
+        type: gen
+        processor:
+          value: ${uint8}
+
+      - name: two
+        suppress: true
+        type: gen
+        processor:
+          value: ${uint64}

internal/pkg/model/config.go

@@ -26,6 +26,7 @@ type Table struct {
 type Column struct {
 	Name      string     `yaml:"name"`
 	Type      string     `yaml:"type"`
+	Suppress  bool       `yaml:"suppress"`
 	Generator RawMessage `yaml:"processor"`
 }