!squash test

Author: tony

src/libtmux/neo.py

@@ -6,7 +6,13 @@
 from typing import overload
 
 from libtmux import exc
-from libtmux.common import PaneDict, handle_option_error, tmux_cmd
+from libtmux.common import (
+    PaneDict,
+    handle_option_error,
+    has_version,
+    session_check_name,
+    tmux_cmd,
+)
 from libtmux.formats import FORMAT_SEPARATOR
 
 if t.TYPE_CHECKING:
@@ -359,10 +365,6 @@ def select_window(self, target_window: t.Union[str, int]) -> "_Window":
     def attached_window(self) -> "_Window":
         """
         Return active :class:`Window` object.
-
-        Returns
-        -------
-        :class:`Window`
         """
         active_windows = []
         for window in self.windows:
@@ -380,6 +382,173 @@ def attached_window(self) -> "_Window":
         if len(self._windows) == 0:
             raise exc.LibTmuxException("No Windows")
 
+    def attach_session(self) -> "_Session":
+        """Return ``$ tmux attach-session`` aka alias: ``$ tmux attach``."""
+        proc = self.cmd("attach-session", "-t%s" % self.session_id)
+
+        if proc.stderr:
+            raise exc.LibTmuxException(proc.stderr)
+
+        return self
+
+    def kill_session(self) -> None:
+        """``$ tmux kill-session``."""
+
+        proc = self.cmd("kill-session", "-t%s" % self.session_id)
+
+        if proc.stderr:
+            raise exc.LibTmuxException(proc.stderr)
+
+    def switch_client(self) -> "_Session":
+        """
+        Switch client to this session.
+
+        Raises
+        ------
+        :exc:`exc.LibTmuxException`
+        """
+        proc = self.cmd("switch-client", "-t%s" % self.session_id)
+
+        if proc.stderr:
+            raise exc.LibTmuxException(proc.stderr)
+
+        return self
+
+    def rename_session(self, new_name: str) -> "_Session":
+        """
+        Rename session and return new :class:`Session` object.
+
+        Parameters
+        ----------
+        new_name : str
+            new session name
+
+        Raises
+        ------
+        :exc:`exc.BadSessionName`
+        """
+        session_check_name(new_name)
+
+        proc = self.cmd("rename-session", new_name)
+
+        if proc.stderr:
+            if has_version("2.7") and "no current client" in proc.stderr:
+                """tmux 2.7 raises "no current client" warning on BSD systems.
+
+                Should be fixed next release:
+
+                - https://www.mail-archive.com/[email protected]/msg45186.html
+                - https://marc.info/?l=openbsd-cvs&m=152183263526828&w=2
+                """
+            else:
+                raise exc.LibTmuxException(proc.stderr)
+
+        return self
+
+    def new_window(
+        self,
+        window_name: t.Optional[str] = None,
+        start_directory: None = None,
+        attach: bool = True,
+        window_index: str = "",
+        window_shell: t.Optional[str] = None,
+    ) -> "_Window":
+        """
+        Return :class:`Window` from ``$ tmux new-window``.
+
+        By default, this will make the window active. For the new window
+        to be created and not set to current, pass in ``attach=False``.
+
+        Parameters
+        ----------
+        window_name : str, optional
+        start_directory : str, optional
+            working directory in which the new window is created.
+        attach : bool, optional
+            make new window the current window after creating it, default True.
+        window_index : str
+            create the new window at the given index position. Default is empty
+            string which will create the window in the next available position.
+        window_shell : str, optional
+            execute a command on starting the window.  The window will close
+            when the command exits.
+
+            .. note::
+                When this command exits the window will close.  This feature is
+                useful for long-running processes where the closing of the
+                window upon completion is desired.
+
+        Returns
+        -------
+        :class:`Window`
+        """
+        tmux_formats = ["#{window_id}" + FORMAT_SEPARATOR]
+
+        window_args: t.Tuple[str, ...] = tuple()
+
+        if not attach:
+            window_args += ("-d",)
+
+        window_args += ("-P",)
+
+        if start_directory:
+            # as of 2014-02-08 tmux 1.9-dev doesn't expand ~ in new-window -c.
+            start_directory = os.path.expanduser(start_directory)
+            window_args += ("-c%s" % start_directory,)
+
+        window_args += ('-F"%s"' % tmux_formats,)  # output
+        if window_name:
+            window_args += ("-n%s" % window_name,)
+
+        window_args += (
+            # empty string for window_index will use the first one available
+            "-t%s:%s"
+            % (self.session_id, window_index),
+        )
+
+        if window_shell:
+            window_args += (window_shell,)
+
+        cmd = self.cmd("new-window", *window_args)
+
+        if cmd.stderr:
+            raise exc.LibTmuxException(cmd.stderr)
+
+        window_output = cmd.stdout[0]
+
+        window_formatters = dict(
+            zip(["window_id"], window_output.split(FORMAT_SEPARATOR))
+        )
+
+        return _Window.from_window_id(
+            server=self.server, window_id=window_formatters["window_id"]
+        )
+
+    def kill_window(self, target_window: t.Optional[str] = None) -> None:
+        """Close a tmux window, and all panes inside it, ``$ tmux kill-window``
+
+        Kill the current window or the window at ``target-window``. removing it
+        from any sessions to which it is linked.
+
+        Parameters
+        ----------
+        target_window : str, optional
+            window to kill
+        """
+
+        if target_window:
+            if isinstance(target_window, int):
+                target = "-t%s:%d" % (self.name, target_window)
+            else:
+                target = "-t%s" % target_window
+
+        proc = self.cmd("kill-window", target)
+
+        if proc.stderr:
+            raise exc.LibTmuxException(proc.stderr)
+
+        self.server._update_windows()
+
     #
     # Redundant stuff we want to remove
     #