Add flag to enable compression

connection.go

@@ -18,15 +18,16 @@ type session struct {
 	errChan chan error
 }
 
-func connect(url, origin string, rlConf *readline.Config, allowInsecure bool) error {
+func connect(url, origin string, rlConf *readline.Config, allowInsecure, enableCompression bool) error {
 	headers := make(http.Header)
 	headers.Add("Origin", origin)
 
 	dialer := websocket.Dialer{
 		Proxy: http.ProxyFromEnvironment,
-		TLSClientConfig:&tls.Config{
+		TLSClientConfig: &tls.Config{
 			InsecureSkipVerify: allowInsecure,
 		},
+		EnableCompression: enableCompression,
 	}
 	ws, _, err := dialer.Dial(url, headers)
 	if err != nil {

main.go

@@ -17,7 +17,8 @@ const Version = "0.2.1"
 var options struct {
 	origin       string
 	printVersion bool
-	insecure bool
+	insecure     bool
+	compression  bool
 }
 
 func main() {
@@ -29,6 +30,7 @@ func main() {
 	rootCmd.Flags().StringVarP(&options.origin, "origin", "o", "", "websocket origin")
 	rootCmd.Flags().BoolVarP(&options.printVersion, "version", "v", false, "print version")
 	rootCmd.Flags().BoolVarP(&options.insecure, "insecure", "k", false, "skip ssl certificate check")
+	rootCmd.Flags().BoolVarP(&options.compression, "compression", "c", false, "enable compression")
 
 	rootCmd.Execute()
 }
@@ -72,7 +74,7 @@ func root(cmd *cobra.Command, args []string) {
 	err = connect(dest.String(), origin, &readline.Config{
 		Prompt:      "> ",
 		HistoryFile: historyFile,
-	}, options.insecure)
+	}, options.insecure, options.compression)
 	if err != nil {
 		fmt.Fprintln(os.Stderr, err)
 		if err != io.EOF && err != readline.ErrInterrupt {