mirror of
https://github.com/kovidgoyal/kitty.git
synced 2024-11-14 01:20:31 +03:00
72 lines
1.9 KiB
Python
Executable File
72 lines
1.9 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
# License: GPL v3 Copyright: 2017, Kovid Goyal <kovid at kovidgoyal.net>
|
|
|
|
import os
|
|
import sys
|
|
import zlib
|
|
from base64 import standard_b64encode
|
|
from contextlib import suppress
|
|
|
|
write = getattr(sys.stdout, 'buffer', sys.stdout).write
|
|
|
|
|
|
def clear_screen():
|
|
write(b'\033[2J')
|
|
|
|
|
|
def move_cursor(x, y):
|
|
write(f'\033[{y};{x}H'.encode('ascii'))
|
|
|
|
|
|
def write_gr_cmd(cmd, payload):
|
|
cmd = ','.join(f'{k}={v}' for k, v in cmd.items())
|
|
w = write
|
|
w(b'\033_G'), w(cmd.encode('ascii')), w(b';'), w(payload), w(b'\033\\')
|
|
sys.stdout.flush()
|
|
|
|
|
|
def display(data, width, height, x, y, z, ncols=0, nrows=0):
|
|
move_cursor(x, y)
|
|
cmd = {'a': 'T', 's': width, 'v': height, 'c': ncols, 'r': nrows, 'S': len(data), 'z': z}
|
|
data = zlib.compress(data)
|
|
cmd['o'] = 'z'
|
|
data = standard_b64encode(data)
|
|
while data:
|
|
chunk, data = data[:4096], data[4096:]
|
|
m = 1 if data else 0
|
|
cmd['m'] = m
|
|
write_gr_cmd(cmd, chunk)
|
|
cmd.clear()
|
|
|
|
|
|
def display_png_file(path):
|
|
cmd = {'a': 'T', 't': 'f', 'f': '100'}
|
|
path = os.path.abspath(path)
|
|
if not isinstance(path, bytes):
|
|
path = path.encode(sys.getfilesystemencoding() or 'utf-8')
|
|
data = standard_b64encode(path)
|
|
write_gr_cmd(cmd, data)
|
|
|
|
|
|
def main():
|
|
from kitty.constants import logo_png_file
|
|
photo = sys.argv[-1]
|
|
if not photo.lower().endswith('.png'):
|
|
raise SystemExit('Must specify a PNG file to display')
|
|
clear_screen()
|
|
display(b'\xdd\xdd\xdd\xff', 1, 1, 0, 0, -10, 40, 20)
|
|
with open(logo_png_file.replace('.png', '.rgba'), 'rb') as f:
|
|
display(f.read(), 256, 256, 0, 5, -9)
|
|
display(b'\0\0\0\xaa', 1, 1, 0, 7, -8, 40, 3)
|
|
move_cursor(5, 8)
|
|
print('kitty is \033[3m\033[32mawesome\033[m!')
|
|
move_cursor(0, 21)
|
|
print('Photo...')
|
|
display_png_file(photo)
|
|
with suppress(EOFError, KeyboardInterrupt):
|
|
input()
|
|
|
|
|
|
if __name__ == '__main__':
|
|
main()
|