import SwiftWebSocketClient
import Combine

let url = URL(string: "wss://yourwebsocketserver.com")!
let client = WebSocketBuilder(url: url).build(implementation: .urlSession)

var cancellables = Set<AnyCancellable>()

// Subscribe to status updates
client.status
    .sink { status in
        print("Connection Status: \(status)")
        
        // When connected, send a message.
        if case .connected = status {
            Task {
                print("Sending 'Hello' to the server.")
                try? await client.send(message: .string("Hello from Swift!"))
            }
        }
    }
    .store(in: &cancellables)

// Subscribe to incoming messages
client.message
    .sink { message in
        switch message {
        case .string(let text):
            print("Received message: \(text)")
        case .data(let data):
            print("Received data: \(data.count) bytes")
        }
    }
    .store(in: &cancellables)

// Connect to the server
client.connect(autoListen: true)