From 01c9e7e5c7dad01e01211b882b477b004befa114 Mon Sep 17 00:00:00 2001
From: Sergey <83376337+freak12techno@users.noreply.github.com>
Date: Fri, 11 Oct 2024 22:45:05 +0300
Subject: [PATCH] feat: calculate last round table columns count on start based
 on screen width (#40)

---
 pkg/display/wrapper.go | 9 ++++++---
 1 file changed, 6 insertions(+), 3 deletions(-)

diff --git a/pkg/display/wrapper.go b/pkg/display/wrapper.go
index 80dd9b7..3ddc415 100644
--- 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 {