internal/{kahless|minipaas}: configure some values

2 files changed, 12 insertions, 10 deletions

diff --git a/internal/kahless/kahless.go b/internal/kahless/kahless.go
index 84bdd68..afcc184 100644
--- a/internal/kahless/kahless.go
+++ b/internal/kahless/kahless.go
@@ -2,6 +2,7 @@ package kahless
 
 import (
 	"bytes"
+	"flag"
 	"io"
 	"io/ioutil"
 	"log"
@@ -19,9 +20,9 @@ func getAgent() (agent.Agent, error) {
 	return agent.NewClient(agentConn), err
 }
 
-const (
-	greedoAddr = `kahless.cetacean.club:22`
-	greedoUser = `cadey`
+var (
+	greedoAddr = flag.String("kahless-addr", "kahless.cetacean.club:22", "address to use for kahless")
+	greedoUser = flag.String("kahless-user", "cadey", "username to use for kahless")
 )
 
 // Dial opens a SSH client to greedo.
@@ -31,8 +32,8 @@ func Dial() (*ssh.Client, error) {
 		return nil, err
 	}
 
-	client, err := ssh.Dial("tcp", greedoAddr, &ssh.ClientConfig{
-		User: greedoUser,
+	client, err := ssh.Dial("tcp", *greedoAddr, &ssh.ClientConfig{
+		User: *greedoUser,
 		Auth: []ssh.AuthMethod{
 			ssh.PublicKeysCallback(agent.Signers),
 		},
diff --git a/internal/minipaas/minipaas.go b/internal/minipaas/minipaas.go
index e08c8f0..a9dce93 100644
--- a/internal/minipaas/minipaas.go
+++ b/internal/minipaas/minipaas.go
@@ -1,6 +1,7 @@
 package minipaas
 
 import (
+	"flag"
 	"net"
 	"os"
 
@@ -13,9 +14,9 @@ func getAgent() (agent.Agent, error) {
 	return agent.NewClient(agentConn), err
 }
 
-const (
-	minipaasAddr = `minipaas.xeserv.us:22`
-	minipaasUser = `dokku`
+var (
+	minipaasAddr = flag.String("minipaas-addr", "minipaas.xeserv.us:22", "address of minipaas")
+	minipaasUser = flag.String("minipaas-user", "dokku", "username on minipaas")
 )
 
 // Dial opens a SSH client to minipaas as the dokku user.
@@ -25,8 +26,8 @@ func Dial() (*ssh.Client, error) {
 		return nil, err
 	}
 
-	client, err := ssh.Dial("tcp", minipaasAddr, &ssh.ClientConfig{
-		User: minipaasUser,
+	client, err := ssh.Dial("tcp", *minipaasAddr, &ssh.ClientConfig{
+		User: *minipaasUser,
 		Auth: []ssh.AuthMethod{
 			ssh.PublicKeysCallback(agent.Signers),
 		},