highperfocused/go-nostr
1
0
Fork 0
mirror of https://github.com/nbd-wtf/go-nostr.git synced 2025-06-05 04:31:04 +02:00
Code Issues Packages Projects Releases Wiki Activity

add option to connect with custom TLS

Browse Source
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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
connection.go
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 {

8
relay.go
View File

@ -3,6 +3,7 @@ package nostr
import (
"bytes"
"context"
"crypto/tls"
"fmt"
"log"
"net/http"
@ -119,6 +120,11 @@ func (r *Relay) IsConnected() bool { return r.connectionContext.Err() == nil }
// pass a custom context to the underlying relay connection, use NewRelay() and
// then Relay.Connect().
func (r *Relay) Connect(ctx context.Context) error {
return r.ConnectWithTLS(ctx, nil)
}
// ConnectWithTLS tries to establish a secured websocket connection to r.URL using customized tls.Config (CA's, etc).
func (r *Relay) ConnectWithTLS(ctx context.Context, tlsConfig *tls.Config) error {
if r.connectionContext == nil || r.Subscriptions == nil {
return fmt.Errorf("relay must be initialized with a call to NewRelay()")
}
@ -134,7 +140,7 @@ func (r *Relay) Connect(ctx context.Context) error {
defer cancel()
}
conn, err := NewConnection(ctx, r.URL, r.RequestHeader)
conn, err := NewConnection(ctx, r.URL, r.RequestHeader, tlsConfig)
if err != nil {
return fmt.Errorf("error opening websocket to '%s': %w", r.URL, err)
}