table: add AlignCenter

Author: RaduBerinde

internal/ascii/table/table.go

@@ -221,6 +221,7 @@ func (l literal[T]) renderValue(tupleIndex int, tuple T) string {
 const (
 	AlignLeft Align = iota
 	AlignRight
+	AlignCenter
 )
 
 type Align uint8
@@ -231,12 +232,15 @@ func pad(cur ascii.Cursor, toWidth int, align Align, s string) ascii.Cursor {
 	if len(s) >= toWidth {
 		return cur.WriteString(s)
 	}
-	startCur := cur
-	if align == AlignRight {
-		cur = cur.Right(toWidth - len(s))
+	padding := 0
+	switch align {
+	case AlignRight:
+		padding = toWidth - len(s)
+	case AlignCenter:
+		padding = (toWidth - len(s)) / 2
 	}
-	cur.WriteString(s)
-	return startCur.Right(toWidth)
+	cur.Right(padding).WriteString(s)
+	return cur.Right(toWidth)
 }
 
 const (

internal/ascii/table/table_test.go

@@ -36,21 +36,28 @@ func TestTable(t *testing.T) {
 		if td.HasArg("horizontally") {
 			opts.Orientation = Horizontally
 		}
+		align := AlignRight
+		if td.HasArg("left") {
+			align = AlignLeft
+		}
+		if td.HasArg("center") {
+			align = AlignCenter
+		}
 		switch td.Cmd {
 		case "cats-autoincrement":
 			def := Define[Cat](
 				AutoIncrement[Cat]("idx", 3, AlignLeft),
 				Div(),
-				String("name", 7, AlignRight, func(c Cat) string { return c.Name }),
+				String("name", 7, align, func(c Cat) string { return c.Name }),
 			)
 			wb.Reset(1)
 			def.Render(wb.At(0, 0), opts, slices.Values(cats))
 			return wb.String()
 		case "cats-nodiv":
 			def := Define[Cat](
 				String("name", 6, AlignLeft, func(c Cat) string { return c.Name }),
-				Int("age", 3, AlignRight, func(c Cat) int { return c.Age }),
-				Int("cuteness", 8, AlignRight, func(c Cat) int { return c.Cuteness }),
+				Int("age", 3, align, func(c Cat) int { return c.Age }),
+				Int("cuteness", 8, align, func(c Cat) int { return c.Cuteness }),
 			)
 			wb.Reset(1)
 			def.Render(wb.At(0, 0), opts, slices.Values(cats))
@@ -63,8 +70,8 @@ func TestTable(t *testing.T) {
 			def := Define[Cat](
 				String("name", 6, AlignLeft, func(c Cat) string { return c.Name }),
 				Div(),
-				Int("age", 3, AlignRight, func(c Cat) int { return c.Age }),
-				Int("c", 1, AlignRight, func(c Cat) int { return c.Cuteness }),
+				Int("age", 3, align, func(c Cat) int { return c.Age }),
+				Int("c", 1, align, func(c Cat) int { return c.Cuteness }),
 			)
 			wb.Reset(1)
 			def.Render(wb.At(0, 0), opts, slices.Values(c))

internal/ascii/table/testdata/table

@@ -11,6 +11,32 @@ Sugar     8       10
 Yaya      5       10
 Yuumi     5       10
 
+cats-nodiv left
+----
+name    age cuteness
+--------------------
+Chicken 5   10
+Heart   4   10
+Mai     2   10
+Poi     15  10
+Pigeon  2   10
+Sugar   8   10
+Yaya    5   10
+Yuumi   5   10
+
+cats-nodiv center
+----
+name    age cuteness
+--------------------
+Chicken  5     10
+Heart    4     10
+Mai      2     10
+Poi     15     10
+Pigeon   2     10
+Sugar    8     10
+Yaya     5     10
+Yuumi    5     10
+
 cats-nodiv horizontally
 ----
     name | Chicken  Heart  Mai  Poi  Pigeon  Sugar  Yaya  Yuumi
@@ -49,6 +75,32 @@ Sugar   |  8000000 10
 Yaya    |  5000000 10
 Yuumi   |  5000000 10
 
+cats-column-too-wide left
+----
+name    | age      c
+--------+------------
+Chicken | 5000000  10
+Heart   | 4000000  10
+Mai     | 2000000  10
+Poi     | 15000000 10
+Pigeon  | 2000000  10
+Sugar   | 8000000  10
+Yaya    | 5000000  10
+Yuumi   | 5000000  10
+
+cats-column-too-wide center
+----
+name    |   age    c
+--------+------------
+Chicken | 5000000  10
+Heart   | 4000000  10
+Mai     | 2000000  10
+Poi     | 15000000 10
+Pigeon  | 2000000  10
+Sugar   | 8000000  10
+Yaya    | 5000000  10
+Yuumi   | 5000000  10
+
 cats-column-too-wide horizontally
 ----
   name | Chicken  Heart    Mai      Poi       Pigeon   Sugar    Yaya     Yuumi