fix(data-warehouse): add username option to sshtunnel key pair (#23030)

add username option to sshtunnel
PostHog · Jun 17, 2024 · aa01904 · aa01904
1 parent cf99128
commit aa01904
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/frontend/src/scenes/data-warehouse/new/sourceWizardLogic.tsx b/frontend/src/scenes/data-warehouse/new/sourceWizardLogic.tsx
@@ -167,6 +167,13 @@ export const SOURCE_DETAILS: Record<ExternalDataSourceType, SourceConfig> = {
                                 label: 'Key pair',
                                 value: 'keypair',
                                 fields: [
+                                    {
+                                        name: 'username',
+                                        label: 'Tunnel username',
+                                        type: 'text',
+                                        required: false,
+                                        placeholder: 'User1',
+                                    },
                                     {
                                         name: 'private_key',
                                         label: 'Tunnel private key',

diff --git a/posthog/warehouse/models/ssh_tunnel.py b/posthog/warehouse/models/ssh_tunnel.py
@@ -119,6 +119,7 @@ def get_tunnel(self, remote_host: str, remote_port: int) -> SSHTunnelForwarder:
         else:
             return SSHTunnelForwarder(
                 (self.host, int(self.port)),
+                ssh_username=self.username,
                 ssh_pkey=self.parse_private_key(),
                 ssh_private_key_password=self.passphrase,
                 remote_bind_address=(remote_host, remote_port),