add option to connect with custom TLS

This commit is contained in:
ice-cronus
2024-03-28 17:23:02 +03:00
committed by fiatjaf_
parent cff9af9aca
commit c0f1c4f510
2 changed files with 10 additions and 2 deletions

View File

@@ -4,6 +4,7 @@ import (
"bytes"
"compress/flate"
"context"
"crypto/tls"
"errors"
"fmt"
"io"
@@ -28,12 +29,13 @@ type Connection struct {
msgStateW *wsflate.MessageState
}
func NewConnection(ctx context.Context, url string, requestHeader http.Header) (*Connection, error) {
func NewConnection(ctx context.Context, url string, requestHeader http.Header, tlsConfig *tls.Config) (*Connection, error) {
dialer := ws.Dialer{
Header: ws.HandshakeHeaderHTTP(requestHeader),
Extensions: []httphead.Option{
wsflate.DefaultParameters.Option(),
},
TLSConfig: tlsConfig,
}
conn, _, hs, err := dialer.Dial(ctx, url)
if err != nil {