#!/usr/bin/env python3
#
# Splits the current terminal into a tab layout, runs a command, then
# restores the original layout. Handy for opening images and videos
# "inside" a terminal.
#
# This effect is similar to dwm's window swallowing patch: https://www.youtube.com/watch?v=92uo5OBOKfY
#
# Stolen from https://github.com/aduros/dotfiles/blob/master/home/bin/i3-tabbed

from i3ipc import Connection
import subprocess
import sys

if len(sys.argv) < 2:
    print("Usage: %s <command...>" % sys.argv[0])
    sys.exit(1)

i3 = Connection()
orig = i3.get_tree().find_focused()

# If the layout was already tabbed or stacked, don't do anything
layout = orig.parent.layout
if layout == "splith":
    orig.command("split v")
    orig.command("layout tabbed")
elif layout == "splitv":
    orig.command("split h")
    orig.command("layout tabbed")

# Ensure we don't collide with exclusive app handling
i3.send_tick("inhibit-exclusive")

try:
    # Run the given command
    code = subprocess.run(sys.argv[1:]).returncode

finally:
    # Unsplit the container if it was previously split to restore the old layout
    if layout == "splith":
        orig.command("layout default")
        orig.command("move left")
    elif layout == "splitv":
        orig.command("layout default")
        orig.command("move up")

# Pass along the command's return code
sys.exit(code)