bookmark_forms.py 12 KB
Newer Older
mathieui's avatar
mathieui committed
1 2 3 4 5
"""
Windows used inthe bookmarkstab
"""
import curses

mathieui's avatar
mathieui committed
6 7
from poezio.windows import base_wins
from poezio.windows.base_wins import Win
mathieui's avatar
mathieui committed
8
from poezio.windows.inputs import Input
mathieui's avatar
mathieui committed
9
from poezio.windows.data_forms import FieldInput, FieldInputMixin
10 11
from poezio.theming import to_curses_attr, get_theme
from poezio.common import safeJID
mathieui's avatar
mathieui committed
12

mathieui's avatar
mathieui committed
13

mathieui's avatar
mathieui committed
14 15 16 17 18
class BookmarkJIDInput(FieldInput, Input):
    def __init__(self, field):
        FieldInput.__init__(self, field)
        Input.__init__(self)
        jid = safeJID(field.jid)
19
        jid.resource = field.nick or None
mathieui's avatar
mathieui committed
20 21 22 23 24 25 26 27 28 29 30 31 32
        self.text = jid.full
        self.pos = len(self.text)
        self.color = get_theme().COLOR_NORMAL_TEXT

    def save(self):
        jid = safeJID(self.get_text())
        self._field.jid = jid.bare
        self._field.name = jid.bare
        self._field.nick = jid.resource

    def get_help_message(self):
        return 'Edit the text'

mathieui's avatar
mathieui committed
33

mathieui's avatar
mathieui committed
34
class BookmarkMethodInput(FieldInputMixin):
mathieui's avatar
mathieui committed
35 36 37 38 39 40 41 42 43 44 45 46
    def __init__(self, field):
        FieldInput.__init__(self, field)
        Win.__init__(self)
        self.options = ('local', 'remote')
        # val_pos is the position of the currently selected option
        self.val_pos = self.options.index(field.method)

    def do_command(self, key):
        if key == 'KEY_LEFT':
            if self.val_pos > 0:
                self.val_pos -= 1
        elif key == 'KEY_RIGHT':
mathieui's avatar
mathieui committed
47
            if self.val_pos < len(self.options) - 1:
mathieui's avatar
mathieui committed
48 49 50 51 52 53 54 55
                self.val_pos += 1
        else:
            return
        self.refresh()

    def refresh(self):
        self._win.erase()
        self._win.attron(to_curses_attr(self.color))
mathieui's avatar
mathieui committed
56
        self.addnstr(0, 0, ' ' * self.width, self.width)
mathieui's avatar
mathieui committed
57 58
        if self.val_pos > 0:
            self.addstr(0, 0, '←')
mathieui's avatar
mathieui committed
59 60
        if self.val_pos < len(self.options) - 1:
            self.addstr(0, self.width - 1, '→')
mathieui's avatar
mathieui committed
61 62
        if self.options:
            option = self.options[self.val_pos]
mathieui's avatar
mathieui committed
63
            self.addstr(0, self.width // 2 - len(option) // 2, option)
mathieui's avatar
mathieui committed
64 65 66 67 68 69 70 71 72
        self._win.attroff(to_curses_attr(self.color))
        self._refresh()

    def save(self):
        self._field.method = self.options[self.val_pos]

    def get_help_message(self):
        return '←, →: Select a value amongst the others'

mathieui's avatar
mathieui committed
73

mathieui's avatar
mathieui committed
74 75 76 77 78 79 80 81 82 83 84 85
class BookmarkPasswordInput(FieldInput, Input):
    def __init__(self, field):
        FieldInput.__init__(self, field)
        Input.__init__(self)
        self.text = field.password or ''
        self.pos = len(self.text)
        self.color = get_theme().COLOR_NORMAL_TEXT

    def rewrite_text(self):
        self._win.erase()
        if self.color:
            self._win.attron(to_curses_attr(self.color))
mathieui's avatar
mathieui committed
86 87
        self.addstr(
            '*' * len(self.text[self.view_pos:self.view_pos + self.width - 1]))
mathieui's avatar
mathieui committed
88 89
        if self.color:
            (y, x) = self._win.getyx()
mathieui's avatar
mathieui committed
90 91
            size = self.width - x
            self.addnstr(' ' * size, size, to_curses_attr(self.color))
mathieui's avatar
mathieui committed
92 93 94 95 96 97 98 99 100 101 102
        self.addstr(0, self.pos, '')
        if self.color:
            self._win.attroff(to_curses_attr(self.color))
        self._refresh()

    def save(self):
        self._field.password = self.get_text() or None

    def get_help_message(self):
        return 'Edit the secret text'

mathieui's avatar
mathieui committed
103

mathieui's avatar
mathieui committed
104
class BookmarkAutojoinWin(FieldInputMixin):
mathieui's avatar
mathieui committed
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
    def __init__(self, field):
        FieldInput.__init__(self, field)
        Win.__init__(self)
        self.last_key = 'KEY_RIGHT'
        self.value = field.autojoin

    def do_command(self, key):
        if key == 'KEY_LEFT' or key == 'KEY_RIGHT':
            self.value = not self.value
            self.last_key = key
        self.refresh()

    def refresh(self):
        self._win.erase()
        self._win.attron(to_curses_attr(self.color))
        format_string = '←{:^%s}→' % 7
        inp = format_string.format(repr(self.value))
        self.addstr(0, 0, inp)
        if self.last_key == 'KEY_RIGHT':
            self.move(0, 8)
        else:
            self.move(0, 0)
        self._win.attroff(to_curses_attr(self.color))
        self._refresh()

    def save(self):
        self._field.autojoin = self.value

    def get_help_message(self):
        return '← and →: change the value between True and False'


class BookmarksWin(Win):
    def __init__(self, bookmarks, height, width, y, x):
mathieui's avatar
mathieui committed
139
        self._win = base_wins.TAB_WIN.derwin(height, width, y, x)
mathieui's avatar
mathieui committed
140 141 142 143 144 145
        self.scroll_pos = 0
        self._current_input = 0
        self.current_horizontal_input = 0
        self._bookmarks = list(bookmarks)
        self.lines = []
        for bookmark in sorted(self._bookmarks, key=lambda x: x.jid):
mathieui's avatar
mathieui committed
146 147 148
            self.lines.append(
                (BookmarkJIDInput(bookmark), BookmarkPasswordInput(bookmark),
                 BookmarkAutojoinWin(bookmark), BookmarkMethodInput(bookmark)))
mathieui's avatar
mathieui committed
149 150 151 152 153 154 155 156 157

    @property
    def current_input(self):
        return self._current_input

    @current_input.setter
    def current_input(self, value):
        if 0 <= self._current_input < len(self.lines):
            if 0 <= value < len(self.lines):
mathieui's avatar
mathieui committed
158 159 160
                self.lines[self._current_input][
                    self.current_horizontal_input].set_color(
                        get_theme().COLOR_NORMAL_TEXT)
mathieui's avatar
mathieui committed
161 162 163 164 165
                self._current_input = value
        else:
            self._current_input = 0

    def add_bookmark(self, bookmark):
mathieui's avatar
mathieui committed
166 167 168 169 170 171
        self.lines.append(
            (BookmarkJIDInput(bookmark), BookmarkPasswordInput(bookmark),
             BookmarkAutojoinWin(bookmark), BookmarkMethodInput(bookmark)))
        self.lines[
            self.current_input][self.current_horizontal_input].set_color(
                get_theme().COLOR_NORMAL_TEXT)
mathieui's avatar
mathieui committed
172 173
        self.current_horizontal_input = 0
        self.current_input = len(self.lines) - 1
mathieui's avatar
mathieui committed
174
        if self.current_input - self.scroll_pos > self.height - 1:
mathieui's avatar
mathieui committed
175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191
            self.scroll_pos = self.current_input - self.height + 1
        self.refresh()

    def del_current_bookmark(self):
        if self.lines:
            bm = self.lines[self.current_input][0]._field
            to_delete = self.current_input
            self.current_input -= 1
            del self.lines[to_delete]
            if self.scroll_pos:
                self.scroll_pos -= 1
            self.refresh()
            return bm

    def resize(self, height, width, y, x):
        self.height = height
        self.width = width
mathieui's avatar
mathieui committed
192
        self._win = base_wins.TAB_WIN.derwin(height, width, y, x)
mathieui's avatar
mathieui committed
193 194
        # Adjust the scroll position, if resizing made the window too small
        # for the cursor to be visible
mathieui's avatar
mathieui committed
195
        while self.current_input - self.scroll_pos > self.height - 1:
mathieui's avatar
mathieui committed
196 197 198 199 200 201 202
            self.scroll_pos += 1

    def go_to_next_line_input(self):
        if not self.lines:
            return
        if self.current_input == len(self.lines) - 1:
            return
mathieui's avatar
mathieui committed
203 204 205
        self.lines[
            self.current_input][self.current_horizontal_input].set_color(
                get_theme().COLOR_NORMAL_TEXT)
mathieui's avatar
mathieui committed
206 207
        # Adjust the scroll position if the current_input would be outside
        # of the visible area
mathieui's avatar
mathieui committed
208
        if self.current_input + 1 - self.scroll_pos > self.height - 1:
mathieui's avatar
mathieui committed
209 210 211 212 213
            self.current_input += 1
            self.scroll_pos += 1
            self.refresh()
        else:
            self.current_input += 1
mathieui's avatar
mathieui committed
214 215 216
            self.lines[self.current_input][
                self.current_horizontal_input].set_color(
                    get_theme().COLOR_SELECTED_ROW)
mathieui's avatar
mathieui committed
217 218 219 220 221 222

    def go_to_previous_line_input(self):
        if not self.lines:
            return
        if self.current_input == 0:
            return
mathieui's avatar
mathieui committed
223 224 225
        self.lines[
            self.current_input][self.current_horizontal_input].set_color(
                get_theme().COLOR_NORMAL_TEXT)
mathieui's avatar
mathieui committed
226 227 228 229 230 231
        self.current_input -= 1
        # Adjust the scroll position if the current_input would be outside
        # of the visible area
        if self.current_input < self.scroll_pos:
            self.scroll_pos = self.current_input
            self.refresh()
mathieui's avatar
mathieui committed
232 233 234
        self.lines[
            self.current_input][self.current_horizontal_input].set_color(
                get_theme().COLOR_SELECTED_ROW)
mathieui's avatar
mathieui committed
235 236 237 238

    def go_to_next_horizontal_input(self):
        if not self.lines:
            return
mathieui's avatar
mathieui committed
239 240 241
        self.lines[
            self.current_input][self.current_horizontal_input].set_color(
                get_theme().COLOR_NORMAL_TEXT)
mathieui's avatar
mathieui committed
242 243 244
        self.current_horizontal_input += 1
        if self.current_horizontal_input > 3:
            self.current_horizontal_input = 0
mathieui's avatar
mathieui committed
245 246 247
        self.lines[
            self.current_input][self.current_horizontal_input].set_color(
                get_theme().COLOR_SELECTED_ROW)
mathieui's avatar
mathieui committed
248

249 250 251 252 253 254 255
    def go_to_next_page(self):
        if not self.lines:
            return

        if self.current_input == len(self.lines) - 1:
            return

mathieui's avatar
mathieui committed
256 257 258
        self.lines[
            self.current_input][self.current_horizontal_input].set_color(
                get_theme().COLOR_NORMAL_TEXT)
259 260 261 262 263 264 265 266
        inc = min(self.height, len(self.lines) - self.current_input - 1)

        if self.current_input + inc - self.scroll_pos > self.height - 1:
            self.current_input += inc
            self.scroll_pos += inc
            self.refresh()
        else:
            self.current_input += inc
mathieui's avatar
mathieui committed
267 268 269
            self.lines[self.current_input][
                self.current_horizontal_input].set_color(
                    get_theme().COLOR_SELECTED_ROW)
270 271 272 273 274 275 276 277 278
        return True

    def go_to_previous_page(self):
        if not self.lines:
            return

        if self.current_input == 0:
            return

mathieui's avatar
mathieui committed
279 280 281
        self.lines[
            self.current_input][self.current_horizontal_input].set_color(
                get_theme().COLOR_NORMAL_TEXT)
282 283 284 285 286 287 288 289

        dec = min(self.height, self.current_input)
        self.current_input -= dec
        # Adjust the scroll position if the current_input would be outside
        # of the visible area
        if self.current_input < self.scroll_pos:
            self.scroll_pos = self.current_input
            self.refresh()
mathieui's avatar
mathieui committed
290 291 292
        self.lines[
            self.current_input][self.current_horizontal_input].set_color(
                get_theme().COLOR_SELECTED_ROW)
293 294
        return True

mathieui's avatar
mathieui committed
295 296 297 298 299
    def go_to_previous_horizontal_input(self):
        if not self.lines:
            return
        if self.current_horizontal_input == 0:
            return
mathieui's avatar
mathieui committed
300 301 302
        self.lines[
            self.current_input][self.current_horizontal_input].set_color(
                get_theme().COLOR_NORMAL_TEXT)
mathieui's avatar
mathieui committed
303
        self.current_horizontal_input -= 1
mathieui's avatar
mathieui committed
304 305 306
        self.lines[
            self.current_input][self.current_horizontal_input].set_color(
                get_theme().COLOR_SELECTED_ROW)
mathieui's avatar
mathieui committed
307 308 309 310

    def on_input(self, key):
        if not self.lines:
            return
mathieui's avatar
mathieui committed
311 312
        self.lines[self.current_input][
            self.current_horizontal_input].do_command(key)
mathieui's avatar
mathieui committed
313 314 315 316

    def refresh(self):
        # store the cursor status
        self._win.erase()
mathieui's avatar
mathieui committed
317
        y = -self.scroll_pos
mathieui's avatar
mathieui committed
318
        for i in range(len(self.lines)):
mathieui's avatar
mathieui committed
319 320 321 322 323 324
            self.lines[i][0].resize(1, self.width // 3, y + 1, 0)
            self.lines[i][1].resize(1, self.width // 3, y + 1, self.width // 3)
            self.lines[i][2].resize(1, self.width // 6, y + 1,
                                    2 * self.width // 3)
            self.lines[i][3].resize(1, self.width // 6, y + 1,
                                    5 * self.width // 6)
mathieui's avatar
mathieui committed
325 326 327 328 329 330 331 332 333 334
            y += 1
        self._refresh()
        for i, inp in enumerate(self.lines):
            if i < self.scroll_pos:
                continue
            if i >= self.height + self.scroll_pos:
                break
            for j in range(4):
                inp[j].refresh()

mathieui's avatar
mathieui committed
335 336 337 338 339 340
        if self.lines and self.current_input < self.height - 1:
            self.lines[self.current_input][
                self.current_horizontal_input].set_color(
                    get_theme().COLOR_SELECTED_ROW)
            self.lines[self.current_input][
                self.current_horizontal_input].refresh()
mathieui's avatar
mathieui committed
341 342 343 344 345 346 347
        if not self.lines:
            curses.curs_set(0)
        else:
            curses.curs_set(1)

    def refresh_current_input(self):
        if self.lines:
mathieui's avatar
mathieui committed
348 349
            self.lines[self.current_input][
                self.current_horizontal_input].refresh()
mathieui's avatar
mathieui committed
350 351 352 353 354

    def save(self):
        for line in self.lines:
            for item in line:
                item.save()