Ssh tunnel configuration for dsn aliases (dbcli#1524)

* Allow per dsn_alias ssh tunnel selection

* update AUTHORS & changelog.rst

---------

Co-authored-by: Charbel Jacquin <[email protected]>

Author: charbeljc

AUTHORS

@@ -143,6 +143,7 @@ Contributors:
     * Doug Harris (dougharris)
     * Jay Knight (jay-knight)
     * fbdb
+    * Charbel Jacquin (charbeljc)
 
 Creator:
 --------

changelog.rst

@@ -8,6 +8,7 @@ Features:
     * Provide `init-command` in the config file
     * Support dsn specific init-command in the config file
 * Add suggestion when setting the search_path
+* Allow per dsn_alias ssh tunnel selection
 
 Internal:
 ---------

pgcli/main.py

@@ -290,6 +290,7 @@ def __init__(
 
         self.prompt_app = None
 
+        self.dsn_ssh_tunnel_config = c.get("dsn ssh tunnels")
         self.ssh_tunnel_config = c.get("ssh tunnels")
         self.ssh_tunnel_url = ssh_tunnel_url
         self.ssh_tunnel = None
@@ -643,6 +644,12 @@ def should_ask_for_password(exc):
             if "port" in parsed_dsn:
                 port = parsed_dsn["port"]
 
+        if self.dsn_alias and self.dsn_ssh_tunnel_config and not self.ssh_tunnel_url:
+            for dsn_regex, tunnel_url in self.dsn_ssh_tunnel_config.items():
+                if re.search(dsn_regex, self.dsn_alias):
+                    self.ssh_tunnel_url = tunnel_url
+                    break
+
         if self.ssh_tunnel_config and not self.ssh_tunnel_url:
             for db_host_regex, tunnel_url in self.ssh_tunnel_config.items():
                 if re.search(db_host_regex, host):
@@ -1540,8 +1547,8 @@ def cli(
                 fg="red",
             )
             sys.exit(1)
-        pgcli.connect_uri(dsn_config)
         pgcli.dsn_alias = dsn
+        pgcli.connect_uri(dsn_config)
     elif "://" in database:
         pgcli.connect_uri(database)
     elif "=" in database and service is None:

pgcli/pgclirc

@@ -258,3 +258,12 @@ float = ""
 [column_date_formats]
 # use strftime format, e.g.
 # created = "%Y-%m-%d"
+
+# Per host ssh tunnel configuration
+[ssh tunnels]
+# ^example.*\.host$ = myuser:[email protected]:4000
+# .*\.net = another.tunnel.com
+
+# Per dsn_alias ssh tunnel configuration
+[dsn ssh tunnels]
+# ^example_dsn$  = myuser:[email protected]:4000