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
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
|
import curses
import re
from typing import Optional, Tuple, Union
from epy_reader.models import Direction, InlineStyle, Key, NoUpdate
from epy_reader.settings import DoubleSpreadPadding
class InfiniBoard:
"""
Wrapper for curses screen to render infinite texts.
The idea is instead of pre render all the text before reading,
this will only renders part of text on demand by which available
page on screen.
And what this does is only drawing text/string on curses screen
without .clear() or .refresh() to optimize performance.
"""
def __init__(
self,
screen,
text: Tuple[str, ...],
textwidth: int = 80,
default_style: Tuple[InlineStyle, ...] = tuple(),
spread: int = 1,
):
self.screen = screen
self.screen_rows, self.screen_cols = self.screen.getmaxyx()
self.textwidth = textwidth
self.x = ((self.screen_cols - self.textwidth) // 2) + 1
self.text = text
self.total_lines = len(text)
self.default_style: Tuple[InlineStyle, ...] = default_style
self.temporary_style: Tuple[InlineStyle, ...] = ()
self.spread = spread
if self.spread == 2:
self.x = DoubleSpreadPadding.LEFT.value
self.x_alt = (
DoubleSpreadPadding.LEFT.value + self.textwidth + DoubleSpreadPadding.MIDDLE.value
)
def feed_temporary_style(self, styles: Optional[Tuple[InlineStyle, ...]] = None) -> None:
"""Reset styling if `styles` is None"""
self.temporary_style = styles if styles else ()
def render_styles(
self, row: int, styles: Tuple[InlineStyle, ...] = (), bottom_padding: int = 0
) -> None:
for i in styles:
if i.row in range(row, row + self.screen_rows - bottom_padding):
self.chgat(row, i.row, i.col, i.n_letters, self.screen.getbkgd() | i.attr)
if self.spread == 2 and i.row in range(
row + self.screen_rows - bottom_padding,
row + 2 * (self.screen_rows - bottom_padding),
):
self.chgat(
row,
i.row - (self.screen_rows - bottom_padding),
-self.x + self.x_alt + i.col,
i.n_letters,
self.screen.getbkgd() | i.attr,
)
def getch(self) -> Union[NoUpdate, Key]:
input = self.screen.getch()
if input == -1:
return NoUpdate()
return Key(input)
def getbkgd(self):
return self.screen.getbkgd()
def chgat(self, row: int, y: int, x: int, n: int, attr: int) -> None:
self.screen.chgat(y - row, self.x + x, n, attr)
def write(self, row: int, bottom_padding: int = 0) -> None:
for n_row in range(min(self.screen_rows - bottom_padding, self.total_lines - row)):
text_line = self.text[row + n_row]
# NOTE: A bug with python itself: https://bugs.python.org/issue8243
# It's stated in python docs:
# > Attempting to write to the lower right corner of a window, subwindow,
# > or pad will cause an exception to be raised after the character is printed.
# https://github.com/python/cpython/commit/ef5ce884a41c8553a7eff66ebace908c1dcc1f89#diff-cb5622768373b8c93cc8eee30dfb041108783bb419d9eaf205501989cea0049fR691-R692
#
# Since the exception is raised "after the character is printed"
# then it seems to be safe to catch it.
try:
self.screen.addstr(n_row, self.x, text_line)
except curses.error:
pass
if (
self.spread == 2
and row + self.screen_rows - bottom_padding + n_row < self.total_lines
):
text_line = self.text[row + self.screen_rows - bottom_padding + n_row]
# TODO: clean this up
if re.search("\\[IMG:[0-9]+\\]", text_line):
self.screen.addstr(
n_row, self.x_alt, text_line.center(self.textwidth), curses.A_BOLD
)
else:
self.screen.addstr(n_row, self.x_alt, text_line)
self.render_styles(row, self.default_style, bottom_padding)
self.render_styles(row, self.temporary_style, bottom_padding)
# self.screen.refresh()
def write_n(
self,
row: int,
n: int = 1,
direction: Direction = Direction.FORWARD,
bottom_padding: int = 0,
) -> None:
assert n > 0
for n_row in range(min(self.screen_rows - bottom_padding, self.total_lines - row)):
text_line = self.text[row + n_row]
if direction == Direction.FORWARD:
# self.screen.addnstr(n_row, self.x + self.textwidth - n, self.text[row+n_row], n)
# `+ " " * (self.textwidth - len(self.text[row + n_row]))` is workaround to
# to prevent curses trace because not calling screen.clear()
self.screen.addnstr(
n_row,
self.x + self.textwidth - n,
text_line + " " * (self.textwidth - len(text_line)),
n,
)
if (
self.spread == 2
and row + self.screen_rows - bottom_padding + n_row < self.total_lines
):
text_line_alt = self.text[row + n_row + self.screen_rows - bottom_padding]
self.screen.addnstr(
n_row,
self.x_alt + self.textwidth - n,
text_line_alt + " " * (self.textwidth - len(text_line_alt)),
n,
)
else:
if text_line[self.textwidth - n :]:
self.screen.addnstr(n_row, self.x, text_line[self.textwidth - n :], n)
if (
self.spread == 2
and row + self.screen_rows - bottom_padding + n_row < self.total_lines
):
text_line_alt = self.text[row + n_row + self.screen_rows - bottom_padding]
self.screen.addnstr(
n_row,
self.x_alt,
text_line_alt[self.textwidth - n :],
n,
)
|
