feat(table): --lazy-quotes and --fields-per-record (#759)

As per [csv.Reader]. closes #345
charmbracelet · Dec 11, 2024 · a8cce1c · a8cce1c
1 parent bf06fce
commit a8cce1c
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 8 deletions.
diff --git a/table/command.go b/table/command.go
@@ -37,6 +37,8 @@ func (o Options) Run() error {
 
 	transformer := unicode.BOMOverride(encoding.Nop.NewDecoder())
 	reader := csv.NewReader(transform.NewReader(input, transformer))
+	reader.LazyQuotes = o.LazyQuotes
+	reader.FieldsPerRecord = o.FieldsPerRecord
 	separatorRunes := []rune(o.Separator)
 	if len(separatorRunes) != 1 {
 		return fmt.Errorf("separator must be single character")

diff --git a/table/options.go b/table/options.go
@@ -8,14 +8,16 @@ import (
 
 // Options is the customization options for the table command.
 type Options struct {
-	Separator string   `short:"s" help:"Row separator" default:","`
-	Columns   []string `short:"c" help:"Column names"`
-	Widths    []int    `short:"w" help:"Column widths"`
-	Height    int      `help:"Table height" default:"0"`
-	Print     bool     `short:"p" help:"static print" default:"false"`
-	File      string   `short:"f" help:"file path" default:""`
-	Border    string   `short:"b" help:"border style" default:"rounded" enum:"rounded,thick,normal,hidden,double,none"`
-	ShowHelp  bool     `help:"Show help keybinds" default:"true" negatable:"" env:"GUM_TABLE_SHOW_HELP"`
+	Separator       string   `short:"s" help:"Row separator" default:","`
+	Columns         []string `short:"c" help:"Column names"`
+	Widths          []int    `short:"w" help:"Column widths"`
+	Height          int      `help:"Table height" default:"0"`
+	Print           bool     `short:"p" help:"static print" default:"false"`
+	File            string   `short:"f" help:"file path" default:""`
+	Border          string   `short:"b" help:"border style" default:"rounded" enum:"rounded,thick,normal,hidden,double,none"`
+	ShowHelp        bool     `help:"Show help keybinds" default:"true" negatable:"" env:"GUM_TABLE_SHOW_HELP"`
+	LazyQuotes      bool     `help:"If LazyQuotes is true, a quote may appear in an unquoted field and a non-doubled quote may appear in a quoted field" default:"false" env:"GUM_TABLE_LAZY_QUOTES"`
+	FieldsPerRecord int      `help:"Sets the number of expected fields per record" default:"0" env:"GUM_TABLE_FIELDS_PER_RECORD"`
 
 	BorderStyle   style.Styles  `embed:"" prefix:"border." envprefix:"GUM_TABLE_BORDER_"`
 	CellStyle     style.Styles  `embed:"" prefix:"cell." envprefix:"GUM_TABLE_CELL_"`