implementation "com.wannaverse:kmp-web-sockets:<version>"

class AppViewModel : ViewModel() {
    private var wsManager: WebSocketManager? = null

    var isLoading = mutableStateOf(false)
    var connected = mutableStateOf(false)
    val websocketURL = mutableStateOf("wss://ws.ifelse.io")
    val messages = mutableStateListOf<String>()

    fun connectToServer() = viewModelScope.launch {
        isLoading.value = true

        if (wsManager == null) {
            wsManager = WebSocketManager(websocketURL.value)
            wsManager?.incomingMessages?.onEach { message ->
                messages.add("Server: $message")
            }?.launchIn(viewModelScope)
        }

        wsManager?.connect()

        isLoading.value = false
        connected.value = true
    }

    fun sendMessage(message: String) {
        messages.add("You: $message")
        wsManager?.send(message)
    }

    fun disconnect() {
        connected.value = false
        wsManager?.disconnect()
    }

    override fun onCleared() {
        connected.value = false
        super.onCleared()
        wsManager?.clear()
    }
}