use of generic ascii table package

1 files changed, 8 insertions, 7 deletions

diff --git a/internal/table/render.go b/internal/table/render.go
index edac866..c7ed46c 100644
--- a/internal/table/render.go
+++ b/internal/table/render.go
@@ -25,9 +25,7 @@ func newRender(tab *Table) render {
 		r.lineLen += len + 3
 	}
 
-	separator.WriteString("+\n")
-	r.separator = separator.String()
-
+	r.separator = tab.sprintf("%s+", separator.String()) + "\n"
 	return r
 }
 
@@ -35,23 +33,26 @@ func (r render) String() string {
 	var sb strings.Builder
 
 	sb.WriteString(r.separator)
-	r.writeRow(&sb, r.tab.headers)
+	sb.WriteString(r.rowString(r.tab.headers))
 
 	sb.WriteString(r.separator)
 	for _, row := range r.tab.rows {
-		r.writeRow(&sb, row)
+		sb.WriteString(r.rowString(row))
 	}
 	sb.WriteString(r.separator)
 
 	return sb.String()
 }
 
-func (r render) writeRow(sb *strings.Builder, row []string) {
+func (r render) rowString(row []string) string {
+	var sb strings.Builder
+
 	for i, col := range row {
 		sb.WriteString(fmt.Sprintf("| %s ", col))
 		for j := len(col); j < r.tab.lengths[i]; j++ {
 			sb.WriteString(" ")
 		}
 	}
-	sb.WriteString("|\n")
+
+	return r.tab.sprintf("%s|", sb.String()) + "\n"
 }