xevion/go-ha
1
0
Fork 0
mirror of https://github.com/Xevion/go-ha.git synced 2025-12-08 04:07:18 -06:00
Code Issues Packages Projects Releases Wiki Activity

feat: support secure connections

Browse Source
This commit is contained in:
Dominik Siebel
2024-02-22 11:28:05 +01:00
parent a9aad16cf3
commit 8b436ce4ee
3 changed files with 56 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
internal/websocket/websocket.go
View File

@@ -50,14 +50,24 @@ func ReadMessage(conn *websocket.Conn, ctx context.Context) ([]byte, error) {
}
func SetupConnection(ip, port, authToken string) (*websocket.Conn, context.Context, context.CancelFunc, error) {
uri := fmt.Sprintf("ws:///%s:%s/api/websocket", ip, port)
return ConnectionFromUri(uri, authToken)
}
func SetupSecureConnection(ip, port, authToken string) (*websocket.Conn, context.Context, context.CancelFunc, error) {
uri := fmt.Sprintf("wss://%s:%s/api/websocket", ip, port)
return ConnectionFromUri(uri, authToken)
}
func ConnectionFromUri(uri, authToken string) (*websocket.Conn, context.Context, context.CancelFunc, error) {
ctx, ctxCancel := context.WithTimeout(context.Background(), time.Second*3)
// Init websocket connection
dialer := websocket.DefaultDialer
conn, _, err := dialer.DialContext(ctx, fmt.Sprintf("ws://%s:%s/api/websocket", ip, port), nil)
conn, _, err := dialer.DialContext(ctx, uri, nil)
if err != nil {
ctxCancel()
slog.Error("Failed to connect to websocket at ws://%s:%s/api/websocket. Check IP address and port\n", ip, port)
slog.Error("Failed to connect to websocket. Check URI\n", "uri", uri)
return nil, nil, nil, err
}