tauri-plugins-workspace/plugins/websocket/examples/tauri-app/src-tauri/src/main.rs

// Copyright 2019-2023 Tauri Programme within The Commons Conservancy
// SPDX-License-Identifier: Apache-2.0
// SPDX-License-Identifier: MIT

#![cfg_attr(
  all(not(debug_assertions), target_os = "windows"),
  windows_subsystem = "windows"
)]

use futures_util::StreamExt;
use tokio::net::{TcpListener, TcpStream};

async fn start_server() {
  let addr = "127.0.0.1:8080".to_string();

  // Create the event loop and TCP listener we'll accept connections on.
  let try_socket = TcpListener::bind(&addr).await;
  let listener = try_socket.expect("Failed to bind");

  while let Ok((stream, _)) = listener.accept().await {
    tokio::spawn(accept_connection(stream));
  }
}

async fn accept_connection(stream: TcpStream) {
  let ws_stream = tokio_tungstenite::accept_async(stream)
    .await
    .expect("Error during the websocket handshake occurred");

  let (write, read) = ws_stream.split();
  if let Err(e) = read.forward(write).await {
    eprintln!("Error: {}", e);
  }
}

fn main() {
  tauri::async_runtime::spawn(start_server());
  tauri::Builder::default()
    .plugin(tauri_plugin_websocket::init())
    .run(tauri::generate_context!())
    .expect("error while running tauri application");
}