Fix padding

Author: sjmiller609

pkg/table/table.go

@@ -30,15 +30,14 @@ func PrintTableNoPad(data pterm.TableData, hasHeader bool) {
 		return
 	}
 
-	// Pre-compute max width per column for all but the last column
+	// Pre-compute max width per column (including last column for proper alignment)
 	maxColWidths := make([]int, numCols)
 	for _, row := range data {
 		for colIdx := 0; colIdx < numCols && colIdx < len(row); colIdx++ {
-			if colIdx == numCols-1 {
-				continue
-			}
 			for _, line := range strings.Split(row[colIdx], "\n") {
-				if w := utf8.RuneCountInString(line); w > maxColWidths[colIdx] {
+				// Strip color codes for accurate width measurement
+				visibleLine := pterm.RemoveColorFromString(line)
+				if w := utf8.RuneCountInString(visibleLine); w > maxColWidths[colIdx] {
 					maxColWidths[colIdx] = w
 				}
 			}
@@ -50,67 +49,38 @@ func PrintTableNoPad(data pterm.TableData, hasHeader bool) {
 	sepStyled := pterm.ThemeDefault.TableSeparatorStyle.Sprint(sep)
 
 	renderRow := func(row []string, styleHeader bool) {
-		// Build first-line-only for non-last columns; last column is full string
-		firstLineParts := make([]string, 0, numCols)
+		// Build and pad all columns for proper alignment
+		parts := make([]string, 0, numCols)
 		for colIdx := 0; colIdx < numCols; colIdx++ {
 			var cell string
 			if colIdx < len(row) {
 				cell = row[colIdx]
 			}
 
-			if colIdx < numCols-1 {
-				// Only the first line for non-last columns
-				lines := strings.Split(cell, "\n")
-				first := ""
-				if len(lines) > 0 {
-					first = lines[0]
-				}
-				padCount := maxColWidths[colIdx] - utf8.RuneCountInString(first)
-				if padCount < 0 {
-					padCount = 0
-				}
-				firstLineParts = append(firstLineParts, first+strings.Repeat(" ", padCount))
-			} else {
-				// Last column: render the first line now; remaining lines after
-				lines := strings.Split(cell, "\n")
-				if len(lines) > 0 {
-					firstLineParts = append(firstLineParts, lines[0])
-				} else {
-					firstLineParts = append(firstLineParts, "")
-				}
+			// Get first line only
+			lines := strings.Split(cell, "\n")
+			first := ""
+			if len(lines) > 0 {
+				first = lines[0]
 			}
-		}
 
-		line := strings.Join(firstLineParts[:numCols-1], sepStyled)
-		if numCols > 1 {
-			if line != "" {
-				line += sepStyled
+			// Pad to column width (measure visible chars, accounting for color codes)
+			visibleFirst := pterm.RemoveColorFromString(first)
+			padCount := maxColWidths[colIdx] - utf8.RuneCountInString(visibleFirst)
+			if padCount < 0 {
+				padCount = 0
 			}
-			line += firstLineParts[numCols-1]
+			parts = append(parts, first+strings.Repeat(" ", padCount))
 		}
 
+		line := strings.Join(parts, sepStyled)
+
 		if styleHeader {
 			b.WriteString(pterm.ThemeDefault.TableHeaderStyle.Sprint(line))
 		} else {
 			b.WriteString(line)
 		}
 		b.WriteString("\n")
-
-		// Print remaining lines from the last column without alignment padding
-		if numCols > 0 {
-			var lastCell string
-			if len(row) >= numCols {
-				lastCell = row[numCols-1]
-			}
-			lines := strings.Split(lastCell, "\n")
-			if len(lines) > 1 {
-				rest := strings.Join(lines[1:], "\n")
-				if rest != "" {
-					b.WriteString(rest)
-					b.WriteString("\n")
-				}
-			}
-		}
 	}
 
 	for idx, row := range data {