func FormatTable(table *gauge.Table) string {
columnToWidthMap := make(map[int]int)
for i, header := range table.Headers {
//table.get(header) returns a list of cells in that particular column
cells := table.Get(header)
columnToWidthMap[i] = findLongestCellWidth(cells, len(header))
}
var tableStringBuffer bytes.Buffer
tableStringBuffer.WriteString(fmt.Sprintf("%s|", getRepeatedChars(" ", tableLeftSpacing)))
for i, header := range table.Headers {
width := columnToWidthMap[i]
tableStringBuffer.WriteString(fmt.Sprintf("%s|", addPaddingToCell(header, width)))
}
tableStringBuffer.WriteString("\n")
tableStringBuffer.WriteString(fmt.Sprintf("%s|", getRepeatedChars(" ", tableLeftSpacing)))
for i := range table.Headers {
width := columnToWidthMap[i]
cell := getRepeatedChars("-", width)
tableStringBuffer.WriteString(fmt.Sprintf("%s|", addPaddingToCell(cell, width)))
}
tableStringBuffer.WriteString("\n")
for _, row := range table.Rows() {
tableStringBuffer.WriteString(fmt.Sprintf("%s|", getRepeatedChars(" ", tableLeftSpacing)))
for i, cell := range row {
width := columnToWidthMap[i]
tableStringBuffer.WriteString(fmt.Sprintf("%s|", addPaddingToCell(cell, width)))
}
tableStringBuffer.WriteString("\n")
}
return string(tableStringBuffer.Bytes())
}
func (resolver *ParamResolver) createProtoStepTable(table *gauge.Table, dataTableLookup *gauge.ArgLookup) *gauge_messages.ProtoTable {
protoTable := new(gauge_messages.ProtoTable)
protoTable.Headers = &gauge_messages.ProtoTableRow{Cells: table.Headers}
tableRows := make([]*gauge_messages.ProtoTableRow, 0)
if len(table.Columns) == 0 {
protoTable.Rows = tableRows
return protoTable
}
for i := 0; i < len(table.Columns[0]); i++ {
row := make([]string, 0)
for _, header := range table.Headers {
tableCell := table.Get(header)[i]
value := tableCell.Value
if tableCell.CellType == gauge.Dynamic {
//if concept has a table with dynamic cell, fetch from datatable
value = dataTableLookup.GetArg(tableCell.Value).Value
}
row = append(row, value)
}
tableRows = append(tableRows, &gauge_messages.ProtoTableRow{Cells: row})
}
protoTable.Rows = tableRows
return protoTable
}