Desktop And PTY

from nullspace import Machine

with Machine.create(template="desktop") as machine:
    width, height = machine.desktop.screen_size()
    display = machine.desktop.display_info()
    machine.desktop.type("hello")
    png = machine.desktop.screenshot()
    jpeg = machine.desktop.screenshot_compressed(format="jpeg", quality=70)
    print(width, height, display.virtual_screen.width, len(png), len(jpeg))

stream = machine.desktop.stream.start()
print(stream.viewer_url)
machine.desktop.set_clipboard_text("copied")
print(machine.desktop.get_clipboard_text())
machine.desktop.stream.stop()

machine.desktop.move(100, 100)
machine.desktop.left_click()
machine.desktop.right_click(200, 200)
machine.desktop.hotkey("ctrl+l")
machine.desktop.open("https://example.com")

for window in machine.desktop.list_windows():
    print(window.id, window.title)

recording = machine.desktop.start_recording(name="demo")
finished = machine.desktop.stop_recording(recording.id)
machine.desktop.download_recording(finished.id, local_path="./desktop.mp4")

session = machine.pty.create_session(cols=100, rows=30)
chunks = []
session.send_input("pwd\nexit\n")
session.wait(on_data=lambda chunk: chunks.append(chunk.decode("utf-8")), timeout=5)
print("".join(chunks))

session = machine.pty.create_session(cwd="/workspace")
session_id = session.session_id
session.disconnect()

reattached = machine.pty.connect(
    session_id=session_id,
    on_data=lambda chunk: print(chunk.decode("utf-8"), end=""),
)
reattached.send_input("echo reattached\n")

sessions = machine.pty.list_sessions()
session_id = sessions[0].session_id
info = machine.pty.get_session_info(session_id)
machine.pty.resize_session(session_id, 140, 40)
machine.pty.kill_session(session_id)