From 410ca88465f065f244f88c1d4089cb0fa4a45799 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Paul=20B=C3=BCtow?= <pbuetow@mimecast.com>
Date: Mon, 10 Feb 2020 14:06:03 +0000
Subject: initial scheduled jobs

---
 internal/clients/args.go       | 20 ++++++++++++--------
 internal/clients/baseclient.go |  3 ++-
 2 files changed, 14 insertions(+), 9 deletions(-)

(limited to 'internal/clients')

diff --git a/internal/clients/args.go b/internal/clients/args.go
index dea5a9e..f55ce90 100644
--- a/internal/clients/args.go
+++ b/internal/clients/args.go
@@ -2,16 +2,20 @@ package clients
 
 import (
 	"github.com/mimecast/dtail/internal/omode"
+
+	gossh "golang.org/x/crypto/ssh"
 )
 
 // Args is a helper struct to summarize common client arguments.
 type Args struct {
-	Mode              omode.Mode
-	ServersStr        string
-	UserName          string
-	What              string
-	Regex             string
-	TrustAllHosts     bool
-	Discovery         string
-	ConnectionsPerCPU int
+	Mode               omode.Mode
+	ServersStr         string
+	UserName           string
+	What               string
+	Regex              string
+	TrustAllHosts      bool
+	Discovery          string
+	ConnectionsPerCPU  int
+	SSHAuthMethods     []gossh.AuthMethod
+	SSHHostKeyCallback gossh.HostKeyCallback
 }
diff --git a/internal/clients/baseclient.go b/internal/clients/baseclient.go
index b1540ea..725910a 100644
--- a/internal/clients/baseclient.go
+++ b/internal/clients/baseclient.go
@@ -40,7 +40,8 @@ func (c *baseClient) init(maker maker) {
 	logger.Info("Initiating base client")
 
 	c.maker = maker
-	c.sshAuthMethods, c.hostKeyCallback = client.InitSSHAuthMethods(c.TrustAllHosts, c.throttleCh)
+	c.sshAuthMethods, c.hostKeyCallback = client.InitSSHAuthMethods(c.Args, c.throttleCh)
+
 	discoveryService := discovery.New(c.Discovery, c.ServersStr, discovery.Shuffle)
 
 	for _, server := range discoveryService.ServerList() {
-- 
cgit v1.2.3