feat(builder): allow environments for windows and panes

This fixes #832.

Author: zappolowski

src/tmuxp/workspace/builder.py

@@ -352,6 +352,7 @@ def iter_create_windows(self, session, append=False):
                 attach=False,  # do not move to the new window
                 window_index=wconf.get("window_index", ""),
                 window_shell=ws,
+                environment=panes[0].get("environment", wconf.get("environment")),
             )
 
             if is_first_window_pass:  # if first window, use window 1
@@ -423,6 +424,7 @@ def get_pane_shell():
                     start_directory=get_pane_start_directory(),
                     shell=get_pane_shell(),
                     target=p.id,
+                    environment=pconf.get("environment", wconf.get("environment")),
                 )
 
             assert isinstance(p, Pane)

tests/fixtures/workspace/builder/environment_vars.yaml

@@ -1,9 +1,27 @@
 session_name: test env vars
 start_directory: '~'
 environment:
-  FOO: BAR
+  FOO: SESSION
   PATH: /tmp
+options:
+  default-shell: /bin/sh
 windows:
-- window_name: editor
+- window_name: no_overrides
   panes:
   - pane
+- window_name: window_overrides
+  environment:
+    FOO: WINDOW
+  panes:
+  - pane
+- window_name: pane_overrides
+  panes:
+  - environment:
+      FOO: PANE
+- window_name: both_overrides
+  environment:
+    FOO: WINDOW
+  panes:
+  - pane
+  - environment:
+      FOO: PANE

tests/workspace/test_builder.py

@@ -340,9 +340,32 @@ def test_environment_variables(session):
     builder = WorkspaceBuilder(sconf=workspace)
     builder.build(session)
 
-    assert session.getenv("FOO") == "BAR"
+    assert session.getenv("FOO") == "SESSION"
     assert session.getenv("PATH") == "/tmp"
 
+    no_overrides_win = session.windows[0]
+    pane = no_overrides_win.panes[0]
+    pane.send_keys("echo $FOO")
+    assert pane.capture_pane()[1] == "SESSION"
+
+    window_overrides_win = session.windows[1]
+    pane = window_overrides_win.panes[0]
+    pane.send_keys("echo $FOO")
+    assert pane.capture_pane()[1] == "WINDOW"
+
+    pane_overrides_win = session.windows[2]
+    pane = pane_overrides_win.panes[0]
+    pane.send_keys("echo $FOO")
+    assert pane.capture_pane()[1] == "PANE"
+
+    both_overrides_win = session.windows[3]
+    pane = both_overrides_win.panes[0]
+    pane.send_keys("echo $FOO")
+    assert pane.capture_pane()[1] == "WINDOW"
+    pane = both_overrides_win.panes[1]
+    pane.send_keys("echo $FOO")
+    assert pane.capture_pane()[1] == "PANE"
+
 
 def test_automatic_rename_option(session):
     """With option automatic-rename: on."""