feat: calculate last round table columns count on start based on scre…

…en width (#40)
QuokkaStake · Oct 11, 2024 · 01c9e7e · 01c9e7e
1 parent e763d05
commit 01c9e7e
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/pkg/display/wrapper.go b/pkg/display/wrapper.go
@@ -196,9 +196,12 @@ func (w *Wrapper) Start() {
 	_, _ = fmt.Fprint(w.ConsensusInfoTextView, "Loading...")
 	_, _ = fmt.Fprint(w.ProgressTextView, "Loading...")
 
-	w.App.SetBeforeDrawFunc(func(screen tcell.Screen) bool {
-		screen.Clear()
-		return false
+	w.App.SetAfterDrawFunc(func(screen tcell.Screen) {
+		_, _, width, _ := w.LastRoundTable.GetInnerRect()
+		columns := width / 50
+		w.LastRoundTableData.SetColumnsCount(columns)
+
+		w.App.SetAfterDrawFunc(nil)
 	})
 
 	if err := w.App.Run(); err != nil {