[client] Add TLS support

1 files changed, 11 insertions, 8 deletions

diff --git a/cmd/client/main.go b/cmd/client/main.go
index bc9f4ff..15033ab 100644
--- a/cmd/client/main.go
+++ b/cmd/client/main.go
@@ -13,10 +13,8 @@ import (
 )
 
 var (
-	serverAddr string
-	user       string
-	conn       net.Conn
-	c          *client.Client
+	conn net.Conn
+	c    *client.Client
 )
 
 type Handler struct{}
@@ -35,13 +33,18 @@ func main() {
 	prompt.Commands["leave"] = leave
 	prompt.Commands["history"] = history
 
-	flag.StringVar(&serverAddr, "a", "localhost:9999", "Server address:port")
-	flag.StringVar(&user, "u", "user", "username")
+	var cfg client.Config
+
+	flag.StringVar(&cfg.Addr, "a", "localhost:9999", "Server address:port")
+	flag.StringVar(&cfg.User, "u", "user", "username")
+	flag.StringVar(&cfg.Pass, "p", "valid", "password")
+	flag.BoolVar(&cfg.Tls, "tls", false, "Enable TLS")
+	flag.BoolVar(&cfg.TlsInsecure, "tls-insecure", false, "Allow insecure TLS certificate")
 	flag.Parse()
 
-	log.Println("connecting to " + serverAddr + " as " + user)
+	log.Println("connecting to " + cfg.Addr + " as " + cfg.User)
 
-	c = client.NewClient(&Handler{}, serverAddr, user, "valid")
+	c = client.NewClient(&Handler{}, cfg)
 	go prompt.Read()
 
 	if err := c.Connect(); err != nil {