bundlewrap/node.py: add node_ssh_connect hook

bundlewrap/node.py

@@ -898,6 +898,12 @@ def log_function(msg):
             # has completed on the node before trying to reuse the
             # multiplexed connection.
             if self._ssh_first_conn_lock.acquire(False):
+
+                self.repo.hooks.node_ssh_connect(
+                    self.repo,
+                    self,
+                )
+
                 try:
                     with io.job(_("{}  establishing connection...").format(bold(self.name))):
                         operations.run(

bundlewrap/repo.py

@@ -53,6 +53,7 @@
     'lock_add',
     'lock_remove',
     'lock_show',
+    'node_ssh_connect',
     'node_apply_end',
     'node_apply_start',
     'node_run_end',

docs/content/repo/hooks.md

@@ -99,6 +99,16 @@ Called each time a `bw apply` command reaches a new item.
 
 ---
 
+**`node_ssh_connect(repo, node, **kwargs)`**
+
+Called before the first ssh connection attempt is made once per node.
+
+`repo` The current repository (instance of `bundlewrap.repo.Repository`).
+
+`node` The current node (instance of `bundlewrap.node.Node`).
+
+---
+
 **`item_apply_end(repo, node, item, duration=None, status_code=None, status_before=None, status_after=None, **kwargs)`**
 
 Called each time a `bw apply` command completes processing an item.