table/border.go at master · vanillaiice/table · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
package table

// AddOuterBorderRows adds outer borders around rows.
func AddOuterBorderRows(borderStyle string, rows ...*Row) {
	if len(rows) == 1 {
		for i, c := range rows[0].cells {
			if i == 0 {
				c.Opts.Style.Border.Left = borderStyle
			} else if i == len(rows[0].cells)-1 {
				c.Opts.Style.Border.Right = borderStyle
			}
			c.Opts.Style.Border.Top = borderStyle
			c.Opts.Style.Border.Bottom = borderStyle
		}
	} else {
		for i, row := range rows {
			if i == 0 {
				for j, cell := range row.cells {
					if j == 0 {
						cell.Opts.Style.Border.Left = borderStyle
					} else if j == len(row.cells)-1 {
						cell.Opts.Style.Border.Right = borderStyle
					}
					cell.Opts.Style.Border.Top = borderStyle
				}
			} else if i == len(rows)-1 {
				for j, cell := range row.cells {
					if j == 0 {
						cell.Opts.Style.Border.Left = borderStyle
					} else if j == len(row.cells)-1 {
						cell.Opts.Style.Border.Right = borderStyle
					}
					cell.Opts.Style.Border.Bottom = borderStyle
				}
			} else {
				for j, cell := range row.cells {
					if j == 0 {
						cell.Opts.Style.Border.Left = borderStyle
					} else if j == len(row.cells)-1 {
						cell.Opts.Style.Border.Right = borderStyle
					}
				}
			}
		}
	}
}

// AddInnerBorderRows adds inner borders for rows.
func AddInnerBorderRows(borderStyle string, rows ...*Row) {
	if len(rows) == 1 {
		for i, c := range rows[0].cells {
			if i == 0 || i == len(rows[0].cells)-1 {
				continue
			}
			c.Opts.Style.Border.Right = borderStyle
			c.Opts.Style.Border.Left = borderStyle
		}
	} else {
		for i, row := range rows {
			for j, c := range row.cells {
				if i == len(rows)-1 {
					if j != 0 && j != len(row.cells)-1 {
						c.Opts.Style.Border.Left = borderStyle
						c.Opts.Style.Border.Right = borderStyle
					}
					c.Opts.Style.Border.Top = borderStyle
				} else {
					if j != 0 && j != len(row.cells)-1 {
						c.Opts.Style.Border.Left = borderStyle
						c.Opts.Style.Border.Right = borderStyle
					}
					c.Opts.Style.Border.Bottom = borderStyle
				}
			}
		}
	}
}

// AddOuterAndInnerBorderRow adds outer and inner borders for rows.
func AddOuterAndInnerBorderRow(borderStyle string, rows ...*Row) {
	for _, r := range rows {
		for _, c := range r.cells {
			c.Opts.Style.SetBorders(borderStyle)
		}
	}
}