98 lines
2.6 KiB
Python
98 lines
2.6 KiB
Python
import datetime
|
|
import json
|
|
import subprocess
|
|
from collections import defaultdict
|
|
|
|
from kitty.boss import get_boss
|
|
from kitty.fast_data_types import Screen, add_timer, get_options
|
|
from kitty.rgb import to_color
|
|
from kitty.tab_bar import (
|
|
DrawData,
|
|
ExtraData,
|
|
Formatter,
|
|
TabBarData,
|
|
as_rgb,
|
|
draw_attributed_string,
|
|
draw_tab_with_powerline,
|
|
)
|
|
|
|
timer_id = None
|
|
|
|
def draw_tab(
|
|
draw_data: DrawData,
|
|
screen: Screen,
|
|
tab: TabBarData,
|
|
before: int,
|
|
max_title_length: int,
|
|
index: int,
|
|
is_last: bool,
|
|
extra_data: ExtraData,
|
|
) -> int:
|
|
global timer_id
|
|
|
|
if timer_id is None:
|
|
timer_id = add_timer(_redraw_tab_bar, 2.0, True)
|
|
draw_tab_with_powerline(
|
|
draw_data, screen, tab, before, max_title_length, index, is_last, extra_data
|
|
)
|
|
if is_last: draw_right_status(draw_data, screen)
|
|
return screen.cursor.x
|
|
|
|
def draw_right_status(draw_data: DrawData, screen: Screen) -> None:
|
|
# The tabs may have left some formats enabled. Disable them now.
|
|
draw_attributed_string(Formatter.reset, screen)
|
|
tab_bg = as_rgb(int(draw_data.inactive_bg))
|
|
tab_fg = as_rgb(int(draw_data.inactive_fg))
|
|
default_bg = as_rgb(int(draw_data.default_bg))
|
|
|
|
cells = create_cells()
|
|
|
|
# Drop cells that wont fit
|
|
while True:
|
|
if not cells:
|
|
return
|
|
padding = screen.columns - screen.cursor.x - sum(len(" ".join([c.get("icon", ""), c["text"]])) + 2 for c in cells)
|
|
if padding >= 0:
|
|
break
|
|
cells = cells[1:]
|
|
|
|
if padding: screen.draw(" " * padding)
|
|
|
|
for c in cells:
|
|
screen.cursor.bg = default_bg
|
|
icon = c.get("icon")
|
|
if icon:
|
|
fg = to_color(c.get("color")) if c.get("color") else tab_fg
|
|
screen.cursor.fg = as_rgb(int(fg))
|
|
screen.draw(f" {icon}")
|
|
screen.cursor.fg = tab_fg
|
|
text = c["text"]
|
|
screen.draw(f" {text} ")
|
|
|
|
|
|
def create_cells():
|
|
cells = [
|
|
get_todo(),
|
|
get_date(),
|
|
get_time()
|
|
]
|
|
return [c for c in cells if c is not None]
|
|
|
|
def get_time():
|
|
now = datetime.datetime.now().strftime("%H:%M")
|
|
return { "icon": " ", "color": "#669bbc", "text": now }
|
|
|
|
def get_date():
|
|
today = datetime.date.today()
|
|
if today.weekday() < 5:
|
|
return { "icon": " ", "color": "#2a9d8f", "text": today.strftime("%b %e") }
|
|
else:
|
|
return { "icon": " ", "color": "#f2e8cf", "text": today.strftime("%b %e") }
|
|
|
|
def get_todo():
|
|
return None
|
|
|
|
def _redraw_tab_bar(timer_id):
|
|
for tm in get_boss().all_tab_managers:
|
|
tm.mark_tab_bar_dirty()
|