Environment setting support.

Author: tasdomas

tmuxp/config.py

@@ -189,7 +189,13 @@ def expand(sconf, cwd=None, parent=None):
         sconf['session_name'] = expandshell(sconf['session_name'])
     if 'window_name' in sconf:
         sconf['window_name'] = expandshell(sconf['window_name'])
-
+    if 'environment' in sconf:
+        for key in sconf['environment']:
+            val = sconf['environment'][key]
+            val = expandshell(val)
+            if any(val.startswith(a) for a in ['.', './']):
+                val = os.path.normpath(os.path.join(cwd, val))
+            sconf['environment'][key] = val
     # Any config section, session, window, pane that can contain the
     # 'shell_command' value
     if 'start_directory' in sconf:

tmuxp/session.py

@@ -330,6 +330,27 @@ def set_option(self, option, value):
                 proc.stderr = proc.stderr[0]
             raise ValueError('tmux set-option stderr: %s' % proc.stderr)
 
+    def set_environment(self, name, value):
+        """Set environment ``$ tmux set-environment <name> <value>``.
+
+        todo: needs tests
+
+        :param name: the environment variable name. such as 'PATH'.
+        :type option: string
+        :param value: environment value.
+        :type value: string
+
+        """
+
+        proc = self.cmd(
+            'set-environment', name, value
+        )
+
+        if proc.stderr:
+            if isinstance(proc.stderr, list) and len(proc.stderr) == int(1):
+                proc.stderr = proc.stderr[0]
+            raise ValueError('tmux set-environment stderr: %s' % proc.stderr)
+
     def show_options(self, option=None, g=False):
         """Return a dict of options for the window.
 

tmuxp/testsuite/workspacebuilder.py

@@ -186,7 +186,7 @@ def test_focus_pane_index(self):
         self.assertNotEqual(w.get('window_name'), 'man')
 
         pane_path = '/usr'
-        for i in range(10):
+        for i in range(20):
             p = w.attached_pane()
             p.server._update_panes()
             if p.get('pane_current_path') == pane_path:

tmuxp/workspacebuilder.py

@@ -143,6 +143,9 @@ def build(self, session=None):
             except Exception as e:
                 self.session.kill_session()
                 raise(e)
+        if 'environment' in self.sconf:
+            for option, value in self.sconf['environment'].items():
+                self.session.set_environment(option, value)
 
         for w, wconf in self.iter_create_windows(session):
             assert(isinstance(w, Window))