Expose visible and total line count methods within viewport

charmbracelet · Nov 4, 2022 · 7f98a14 · 7f98a14
1 parent 0474e12
commit 7f98a14
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/viewport/viewport.go b/viewport/viewport.go
@@ -207,6 +207,16 @@ func (m *Model) LineUp(n int) (lines []string) {
 	return m.visibleLines()
 }
 
+// TotalLineCount returns the total number of lines (both hidden and visible) within the viewport.
+func (m Model) TotalLineCount() int {
+	return len(m.lines)
+}
+
+// VisibleLineCount returns the number of the visible lines within the viewport.
+func (m Model) VisibleLineCount() int {
+	return len(m.visibleLines())
+}
+
 // GotoTop sets the viewport to the top position.
 func (m *Model) GotoTop() (lines []string) {
 	if m.AtTop() {