Files
rootiest f2ddf78fe4 feat(save-session): implement session saving function
Allows saving the current session with a keypress

The session can then be restored via a right-click option in the kitty application
2024-08-02 09:02:47 -04:00

74 lines
1.9 KiB
Python
Executable File

#!/usr/bin/env python3
"""
this tool is used to convert Kitty session dump to Kitty session file which can be loaded by Kitty
"""
import json
import os
import sys
def env_to_str(env):
"""Convert an env list to a series of '--env key=value' parameters and return as a string."""
s = ""
for key in env:
s += f"--env {key}={env[key]} "
return s.strip()
def cmdline_to_str(cmdline):
"""Convert a cmdline list to a space separated string."""
s = ""
for e in cmdline:
s += f"{e} "
return s.strip()
def fg_proc_to_str(fg):
"""Convert a foreground_processes list to a space separated string."""
s = ""
fg = fg[0]
# s += f"--cwd {fg['cwd']} {cmdline_to_str(fg['cmdline'])}"
s += f"{cmdline_to_str(fg['cmdline'])}"
if s == "kitty @ ls":
return os.getenv("SHELL")
return s
def convert(session):
"""Convert a kitty session dict, into a kitty session file and output it."""
first = True
for os_window in session:
if first:
first = False
else:
print("\nnew_os_window\n")
for tab in os_window["tabs"]:
print("\n")
print(f"new_tab {tab['title']}")
# print('enabled_layouts *)
print(f"layout {tab['layout']}")
# This is a bit of a kludge to set cwd for the tab, as
# setting it in the launch command didn't work, for some reason?
if tab["windows"]:
print(f"cd {tab['windows'][0]['cwd']}")
for w in tab["windows"]:
print(f"title {w['title']}")
print(
f"launch {env_to_str(w['env'])} {fg_proc_to_str(w['foreground_processes'])}"
)
if w["is_focused"]:
print("focus")
if __name__ == "__main__":
stdin = sys.stdin.readlines()
session = json.loads("".join(stdin))
convert(session)