Merge pull request #77 from obnoxiousnerd/table_footer

Add `footer` parameter to `out.Table`
dominikbraun · May 31, 2021 · bf934a8 · bf934a8
2 parents 7561157 + f9d1967
commit bf934a8
Show file tree

Hide file tree

Showing 4 changed files with 11 additions and 6 deletions.
diff --git a/cli/get.go b/cli/get.go
@@ -38,7 +38,7 @@ func getProjectCommand(t *core.Timetrace) *cobra.Command {
 				return
 			}
 
-			out.Table([]string{"Key"}, [][]string{{project.Key}})
+			out.Table([]string{"Key"}, [][]string{{project.Key}}, nil)
 		},
 	}
 
@@ -102,5 +102,5 @@ func showRecord(record *core.Record, formatter *core.Formatter) {
 		},
 	}
 
-	out.Table([]string{"Start", "End", "Project", "Billable"}, rows)
+	out.Table([]string{"Start", "End", "Project", "Billable"}, rows, nil)
 }
diff --git a/cli/list.go b/cli/list.go
@@ -52,7 +52,7 @@ func listProjectsCommand(t *core.Timetrace) *cobra.Command {
 				rows[i][2] = allModules
 			}
 
-			out.Table([]string{"#", "Key", "Modules"}, rows)
+			out.Table([]string{"#", "Key", "Modules"}, rows, nil)
 		},
 	}
 
@@ -115,7 +115,7 @@ func listRecordsCommand(t *core.Timetrace) *cobra.Command {
 				rows[i][5] = billable
 			}
 
-			out.Table([]string{"#", "Key", "Project", "Start", "End", "Billable"}, rows)
+			out.Table([]string{"#", "Key", "Project", "Start", "End", "Billable"}, rows, nil)
 		},
 	}
 

diff --git a/cli/status.go b/cli/status.go
@@ -47,7 +47,7 @@ func statusCommand(t *core.Timetrace) *cobra.Command {
 					report.FormatTodayTime(),
 				},
 			}
-			out.Table([]string{"Current project", "Worked since start", "Worked today"}, rows)
+			out.Table([]string{"Current project", "Worked since start", "Worked today"}, rows, nil)
 		},
 	}
 

diff --git a/out/out.go b/out/out.go
@@ -42,11 +42,16 @@ func Err(format string, a ...interface{}) {
 }
 
 // Table renders a table with the given rows to the standard output.
-func Table(header []string, rows [][]string) {
+func Table(header []string, rows [][]string, footer []string) {
 	paddedHeaders := headersWithPadding(header)
 	table := tablewriter.NewWriter(os.Stdout)
 	table.SetHeader(paddedHeaders)
 	setHeaderColor(table, paddedHeaders)
+	// If footer array is not empty, then render footer in table.
+	if len(footer) > 0 {
+		paddedFooters := headersWithPadding(footer)
+		table.SetFooter(paddedFooters)
+	}
 	table.AppendBulk(rows)
 	table.Render()
 }