Calvin Park
80 lines
2.7 KiB
Python
Executable File
80 lines
2.7 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
# tsk/main.py
|
|
import sys
|
|
|
|
import pyray as rl
|
|
|
|
from openpilot.system.ui.lib.application import gui_app
|
|
from tsk.common.env import is_calvins_c3x
|
|
from tsk.reboot_menu.ui import RebootMenuUI
|
|
from tsk.tools_menu.ui import ToolsMenuUI
|
|
from tsk.ui.header import Header
|
|
from tsk.ui.layout import Theme
|
|
|
|
|
|
class TSKManager:
|
|
"""Manages the TSK Manager UI, switching between menus."""
|
|
|
|
def __init__(self):
|
|
"""Initializes the TSK Manager."""
|
|
self.current_menu = Theme.menu_tools
|
|
self.header = Header()
|
|
self.tools_menu = ToolsMenuUI()
|
|
self.reboot_menu = RebootMenuUI()
|
|
|
|
def render(self, rect: rl.Rectangle):
|
|
"""Renders the current menu within the given rectangle."""
|
|
|
|
new_menu = self.header.render(rect, self.current_menu)
|
|
if new_menu is not None:
|
|
self.current_menu = new_menu
|
|
|
|
menu_rect = rl.Rectangle(rect.x, rect.y + self.header._calculate_height(), rect.width,
|
|
rect.height - self.header._calculate_height())
|
|
|
|
if self.current_menu == Theme.menu_tools:
|
|
self.tools_menu.render(menu_rect)
|
|
elif self.current_menu == Theme.menu_reboot:
|
|
self.reboot_menu.render(menu_rect)
|
|
return True
|
|
|
|
|
|
def setup_environment():
|
|
"""Performs initial environment setup, such as enabling SSH."""
|
|
if is_calvins_c3x():
|
|
with open("/data/params/d/GithubUsername", "w") as f:
|
|
f.write("calvinpark")
|
|
with open("/data/params/d/GithubSshKeys", "w") as f:
|
|
f.write("ssh-rsa AAAAB3NzaC1yc2EAAAADAQABAAABgQD30Dz8yY3n1DchzsPbuuWMXMBtyeW2Yh5aOjrjLSvUBjqs9OoPrPfOMAPiaKqE6EfEcjV90He9A6q7OywTy5kTD6JsjjoULJKHiGbDdQlclXE2fO/wTnmxPO9yjdDJqiFrPsSGbT/4R78TVUUkEwD+6DcDGtJd7hHQ/GQCWn78kZ/UsZqcukGjhuwI98gOnIOmX3ui2W6/2NrP3IH7GJWnIvDIHafHYwnRkNU7WQ5zyiUw2GX65dTrXt0pDpX/nYp0qjwORf91DTZCg6fimdUo2WAmhYXnQb66IKESpNVfIVA8L0PRNkSepc3RARX0bPgqYGj6TLy9s87UT11mq/ASuIo9IVYWt6okYvloQcwrX6uxKsGutXouXDraxP648s1ErM6BC3tOOagay19cZdQl53k0CZbkIXODlpM/QaW7MdagH7PVzlGGIuHohDAe3M/ltJjRmRfdj89cCGusBlFB5RuLZpzYskp353NZ1qxhL086Mfyg0bBdDK+CGLJ7bY0=\n"
|
|
"ssh-ed25519 AAAAC3NzaC1lZDI1NTE5AAAAILXx7npi7/QYSOu2Z0Bhldtey4L2nxEyZKYQY/BIHdak")
|
|
with open("/data/params/d/SshEnabled", "w") as f:
|
|
f.write("1")
|
|
with open("/data/params/d/HasAcceptedTerms", "w") as f:
|
|
f.write("2")
|
|
with open("/data/params/d/CompletedTrainingVersion", "w") as f:
|
|
f.write("0.2.0")
|
|
|
|
|
|
def main():
|
|
"""Main function to initialize and run the TSK Manager."""
|
|
|
|
setup_environment()
|
|
|
|
gui_app.init_window("TSK Manager")
|
|
tskm = TSKManager()
|
|
|
|
while not rl.window_should_close():
|
|
rl.begin_drawing()
|
|
rl.clear_background(rl.BLACK)
|
|
|
|
tskm.render(rl.Rectangle(0, 0, gui_app.width, gui_app.height))
|
|
|
|
rl.end_drawing()
|
|
|
|
rl.close_window()
|
|
sys.exit(0) # Necessary for macOS
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|