Added another example program: bed, the BTUI editor
This commit is contained in:
parent
de8f64e99f
commit
b9bad81b80
227
Python/bed.py
Executable file
227
Python/bed.py
Executable file
@ -0,0 +1,227 @@
|
||||
#!/usr/bin/env python3
|
||||
import sys
|
||||
from btui import open_btui
|
||||
from collections import namedtuple
|
||||
|
||||
Vec2 = namedtuple('Vec2', 'x y')
|
||||
|
||||
def clamp(x, low, high):
|
||||
if x < low: return low
|
||||
elif x > high: return high
|
||||
return x
|
||||
|
||||
class BedFile:
|
||||
def __init__(self, bt, lines, pos=Vec2(0,0), size=None):
|
||||
self.bt = bt
|
||||
self.lines = lines
|
||||
self.drawn = set()
|
||||
self.numlen = len(str(len(self.lines)))
|
||||
self.pos = pos
|
||||
self.size = size if size else Vec2(self.bt.width-pos.x, self.bt.height-pos.y)
|
||||
self.scroll = 0
|
||||
self._cursor = Vec2(0, 0)
|
||||
self.unsaved = False
|
||||
|
||||
@property
|
||||
def cursor(self):
|
||||
return self._cursor
|
||||
|
||||
@cursor.setter
|
||||
def cursor(self, p):
|
||||
x, y = p.x, p.y
|
||||
y = clamp(y, 0, len(self.lines)-1)
|
||||
x = clamp(x, 0, len(self.lines[y]))
|
||||
self._cursor = Vec2(x, y)
|
||||
|
||||
def update_term_cursor(self):
|
||||
self.bt.move(self.pos.x + self.numlen + 1 + self.cursor.x,
|
||||
self.pos.y + self.cursor.y - self.scroll)
|
||||
|
||||
def set_scroll(self, newscroll):
|
||||
newscroll = clamp(newscroll, 0, len(self.lines)-1 - (self.size.y-1))
|
||||
if newscroll == self.scroll: return
|
||||
delta = newscroll - self.scroll
|
||||
self.bt.scroll(self.pos.y, self.pos.y + self.size.y - 1, delta)
|
||||
self.scroll = newscroll
|
||||
self.drawn &= set(range(self.scroll, self.scroll + (self.size.y-1)+1))
|
||||
if self.cursor.y < newscroll:
|
||||
self.cursor = Vec2(self.cursor.x, newscroll)
|
||||
elif self.cursor.y > newscroll + (self.size.y-1):
|
||||
self.cursor = Vec2(self.cursor.x, newscroll + (self.size.y-1))
|
||||
|
||||
def handle_input(self, key, mx=None, my=None):
|
||||
if key == 'Space': key = ' '
|
||||
if key == 'Tab': key = ' '
|
||||
|
||||
if key == 'Left':
|
||||
self.cursor = Vec2(self.cursor.x - 1, self.cursor.y)
|
||||
elif key == 'Right':
|
||||
self.cursor = Vec2(self.cursor.x + 1, self.cursor.y)
|
||||
elif key == 'Up':
|
||||
self.cursor = Vec2(self.cursor.x, self.cursor.y - 1)
|
||||
if self.cursor.y < self.scroll:
|
||||
self.set_scroll(self.cursor.y)
|
||||
elif key == 'Down':
|
||||
self.cursor = Vec2(self.cursor.x, self.cursor.y + 1)
|
||||
newscroll = clamp(self.scroll, self.cursor.y - (self.size.y-1), self.cursor.y)
|
||||
if self.cursor.y > self.scroll + (self.size.y-1):
|
||||
self.set_scroll(self.cursor.y - (self.size.y-1))
|
||||
elif key == 'Page Down' or key == 'Ctrl-d':
|
||||
self.set_scroll(self.scroll + self.bt.height // 2)
|
||||
elif key == 'Page Up' or key == 'Ctrl-u':
|
||||
self.set_scroll(self.scroll - self.bt.height // 2)
|
||||
elif key == 'Delete':
|
||||
line = self.lines[self.cursor.y]
|
||||
i = self.cursor.x
|
||||
if i == len(line) and self.cursor.y < len(self.lines)-1:
|
||||
line.extend(self.lines.pop(self.cursor.y+1))
|
||||
self.drawn &= {d for d in self.drawn if d < self.cursor.y or d >= len(self.lines)+1}
|
||||
self.unsaved = True
|
||||
elif i < len(line)-1:
|
||||
del line[i:i+1]
|
||||
self.drawn.discard(self.cursor.y)
|
||||
self.unsaved = True
|
||||
elif key == 'Backspace':
|
||||
line = self.lines[self.cursor.y]
|
||||
i = self.cursor.x
|
||||
if i == 0 and self.cursor.y > 0:
|
||||
self.lines.pop(self.cursor.y)
|
||||
prevline = self.lines[self.cursor.y - 1]
|
||||
self.cursor = Vec2(len(prevline), self.cursor.y-1)
|
||||
prevline.extend(line)
|
||||
self.drawn &= {d for d in self.drawn if d < self.cursor.y or d >= len(self.lines)+1}
|
||||
self.unsaved = True
|
||||
elif i > 0:
|
||||
del line[i-1:i]
|
||||
self.drawn.discard(self.cursor.y)
|
||||
self.cursor = Vec2(self.cursor.x - 1, self.cursor.y)
|
||||
self.unsaved = True
|
||||
elif key == 'Enter':
|
||||
line = self.lines[self.cursor.y]
|
||||
before, after = line[:self.cursor.x], line[self.cursor.x:]
|
||||
self.lines[self.cursor.y] = before
|
||||
self.lines.insert(self.cursor.y+1, after)
|
||||
self.drawn &= {d for d in self.drawn if d < self.cursor.y or d >= len(self.lines)}
|
||||
self.cursor = Vec2(0, self.cursor.y + 1)
|
||||
self.unsaved = True
|
||||
elif key and (len(key) == 1 or key == ' '):
|
||||
line = self.lines[self.cursor.y]
|
||||
i = self.cursor.x
|
||||
line[i:i] = bytearray(key, 'utf8')
|
||||
self.cursor = Vec2(self.cursor.x + len(key), self.cursor.y)
|
||||
self.drawn.discard(self.cursor.y)
|
||||
self.unsaved = True
|
||||
|
||||
def render(self):
|
||||
self.bt.hide_cursor()
|
||||
for i in range(self.size.y):
|
||||
lineno = self.scroll + i
|
||||
if lineno in self.drawn: continue
|
||||
y = self.pos.y + i
|
||||
x = self.pos.x
|
||||
self.bt.move(x, y)
|
||||
if lineno >= len(self.lines):
|
||||
self.bt.clear('line')
|
||||
self.drawn.add(lineno)
|
||||
continue
|
||||
with self.bt.attributes("faint"):
|
||||
self.bt.write(("{:>"+str(self.numlen)+"}").format(lineno + 1))
|
||||
x += self.numlen + 1
|
||||
self.bt.move(x, y)
|
||||
self.bt.write_bytes(bytes(self.lines[lineno]))
|
||||
self.bt.clear('right')
|
||||
self.drawn.add(lineno)
|
||||
|
||||
class BED:
|
||||
def __init__(self, bt, filename):
|
||||
self.bt = bt
|
||||
self.filename = filename
|
||||
try:
|
||||
lines = [bytearray(line.rstrip(b'\n')) for line in open(filename, 'rb').readlines()]
|
||||
except FileNotFoundError:
|
||||
lines = [bytearray(b'')]
|
||||
self.file = BedFile(bt, lines, Vec2(0, 1), Vec2(bt.width, bt.height-2))
|
||||
self.drawn = set()
|
||||
self.renders = 0
|
||||
|
||||
def edit(self):
|
||||
self.bt.set_cursor('blinking bar')
|
||||
self.render()
|
||||
key, mx, my = self.bt.getkey()
|
||||
while True:
|
||||
if key == 'Ctrl-c':
|
||||
sys.exit(1)
|
||||
elif key == 'Ctrl-q':
|
||||
if self.file.unsaved:
|
||||
self.confirm_save()
|
||||
break
|
||||
elif key == 'Ctrl-s':
|
||||
self.save()
|
||||
else:
|
||||
self.file.handle_input(key, mx, my)
|
||||
self.render()
|
||||
key, mx, my = self.bt.getkey()
|
||||
|
||||
def confirm_save(self):
|
||||
self.bt.move(0, self.bt.height)
|
||||
with self.bt.attributes('bold'):
|
||||
self.bt.write(f"Do you want to save {self.filename}? [y/n]")
|
||||
key, mx, my = self.bt.getkey()
|
||||
if key in ('y', 'Y'):
|
||||
self.save()
|
||||
return True
|
||||
elif key in ('n', 'N', 'Ctrl-c', 'Escape', 'Ctrl-q'):
|
||||
return False
|
||||
|
||||
def save(self):
|
||||
with open(self.filename, 'wb') as f:
|
||||
for line in self.file.lines:
|
||||
f.write(line)
|
||||
f.write(b'\n')
|
||||
self.file.unsaved = False
|
||||
|
||||
def render(self):
|
||||
self.renders += 1
|
||||
|
||||
k = 'title-modified' if self.file.unsaved else 'title'
|
||||
if k not in self.drawn:
|
||||
self.drawn -= {'title', 'title-modified'}
|
||||
self.bt.move(0,0)
|
||||
with self.bt.attributes("bold"):
|
||||
self.bt.write(self.filename)
|
||||
if self.file.unsaved:
|
||||
with self.bt.attributes("dim"):
|
||||
self.bt.write(" (modified)")
|
||||
self.bt.clear('right')
|
||||
self.drawn.add(k)
|
||||
|
||||
self.file.render()
|
||||
|
||||
# Just redraw this every time:
|
||||
with self.bt.attributes("faint"):
|
||||
self.bt.move(0, self.bt.height-1)
|
||||
self.bt.write("Ctrl-Q to quit, Ctrl-S to save")
|
||||
self.bt.clear('right')
|
||||
s = f"Line {self.file.cursor.y}, Col {self.file.cursor.x}, {self.renders} redraws"
|
||||
self.bt.move(self.bt.width-len(s), self.bt.height-1)
|
||||
self.bt.write(s)
|
||||
|
||||
self.file.update_term_cursor()
|
||||
self.bt.show_cursor()
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
args = sys.argv[1:]
|
||||
debug = False
|
||||
if args and args[0] in ('-d', '--debug'):
|
||||
args.pop(0)
|
||||
debug = True
|
||||
|
||||
if not args:
|
||||
print("Usage: bed.py [-d|--debug] file")
|
||||
sys.exit(1)
|
||||
|
||||
filename = args[0]
|
||||
with open_btui(debug=debug) as bt:
|
||||
bed = BED(bt, filename)
|
||||
bed.edit()
|
||||
Loading…
Reference in New Issue
Block a user